|
|
|
@ -10,16 +10,24 @@ use purrchance::Purrchance;
|
|
|
|
|
use serde::Deserialize;
|
|
|
|
|
use serenity::{
|
|
|
|
|
async_trait,
|
|
|
|
|
model::{channel::Message, gateway::Ready, id::ChannelId, user::User},
|
|
|
|
|
model::{
|
|
|
|
|
channel::Message,
|
|
|
|
|
gateway::Ready,
|
|
|
|
|
id::{ChannelId, GuildId},
|
|
|
|
|
user::User,
|
|
|
|
|
},
|
|
|
|
|
prelude::*,
|
|
|
|
|
};
|
|
|
|
|
use std::fs::read_to_string;
|
|
|
|
|
use std::sync::Arc;
|
|
|
|
|
use std::sync::Once;
|
|
|
|
|
use std::time::Duration;
|
|
|
|
|
|
|
|
|
|
enum TextGenerator {
|
|
|
|
|
Perchance { grammar: purrchance::Grammar, symbol: purrchance::Symbol },
|
|
|
|
|
// Tracery(tracery::Grammar),
|
|
|
|
|
Perchance {
|
|
|
|
|
grammar: purrchance::Grammar,
|
|
|
|
|
symbol: purrchance::Symbol,
|
|
|
|
|
},
|
|
|
|
|
// Tracery(tracery::Grammar),
|
|
|
|
|
Markov(markov::Chain<String>),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -27,7 +35,7 @@ impl TextGenerator {
|
|
|
|
|
fn generate(&self) -> String {
|
|
|
|
|
match self {
|
|
|
|
|
TextGenerator::Perchance { grammar, symbol } => symbol.eval(&grammar).unwrap(),
|
|
|
|
|
// TextGenerator::Tracery(grammar) => grammar.flatten(),
|
|
|
|
|
// TextGenerator::Tracery(grammar) => grammar.flatten(),
|
|
|
|
|
TextGenerator::Markov(chain) => chain.generate_str(),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -37,7 +45,7 @@ impl TextGenerator {
|
|
|
|
|
#[serde(tag = "type", rename_all = "lowercase")]
|
|
|
|
|
enum GeneratorSpec {
|
|
|
|
|
Perchance { path: String, symbol: String },
|
|
|
|
|
// Tracery { path: String, rule: String },
|
|
|
|
|
// Tracery { path: String, rule: String },
|
|
|
|
|
Markov { path: String, order: usize },
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -45,17 +53,18 @@ impl GeneratorSpec {
|
|
|
|
|
fn load(&self) -> TextGenerator {
|
|
|
|
|
match self {
|
|
|
|
|
GeneratorSpec::Perchance { path: p, symbol: s } => {
|
|
|
|
|
let grammar = purrchance::parser::load_grammar(&read_to_string(p).unwrap()).unwrap();
|
|
|
|
|
let grammar =
|
|
|
|
|
purrchance::parser::load_grammar(&read_to_string(p).unwrap()).unwrap();
|
|
|
|
|
let symbol = purrchance::Symbol::NonTerminal(String::from(s));
|
|
|
|
|
TextGenerator::Perchance { grammar, symbol }
|
|
|
|
|
}
|
|
|
|
|
/*
|
|
|
|
|
/*
|
|
|
|
|
GeneratorSpec::Tracery { path, rule } => {
|
|
|
|
|
let mut grammar = tracery::from_json(read_to_string(path).unwrap()).unwrap();
|
|
|
|
|
grammar.default_rule(rule);
|
|
|
|
|
TextGenerator::Tracery(grammar)
|
|
|
|
|
}
|
|
|
|
|
*/
|
|
|
|
|
*/
|
|
|
|
|
GeneratorSpec::Markov { path, order } => {
|
|
|
|
|
let mut chain = markov::Chain::of_order(*order);
|
|
|
|
|
for line in read_to_string(path).unwrap().lines() {
|
|
|
|
@ -75,7 +84,6 @@ struct Config {
|
|
|
|
|
interval: u64,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
impl TypeMapKey for TextGenerator {
|
|
|
|
|
type Value = TextGenerator;
|
|
|
|
|
}
|
|
|
|
@ -92,18 +100,14 @@ impl TypeMapKey for ChanId {
|
|
|
|
|
type Value = ChannelId;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct LoopStatus;
|
|
|
|
|
|
|
|
|
|
impl TypeMapKey for LoopStatus {
|
|
|
|
|
type Value = bool;
|
|
|
|
|
struct Handler {
|
|
|
|
|
start_loop: Once,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct Handler;
|
|
|
|
|
|
|
|
|
|
async fn mentions_me(ctx: &Context, msg: &Message) -> bool {
|
|
|
|
|
let me = User::from(ctx.http.get_current_user().await.unwrap());
|
|
|
|
|
if msg.mentions_user(&me) {
|
|
|
|
|
return true
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
let guild = msg.guild(ctx).await.unwrap();
|
|
|
|
|
let member = guild.member(ctx, me).await.unwrap();
|
|
|
|
@ -111,12 +115,21 @@ async fn mentions_me(ctx: &Context, msg: &Message) -> bool {
|
|
|
|
|
msg.mention_roles.iter().any(|r| my_roles.contains(r))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn interval_loop(ctx: Arc<Context>) {
|
|
|
|
|
async fn interval_loop(ctx: Context) {
|
|
|
|
|
let channel_id = ctx.data.read().await.get::<ChanId>().unwrap().clone();
|
|
|
|
|
loop {
|
|
|
|
|
let sentence = ctx.data.read().await.get::<TextGenerator>().unwrap().generate();
|
|
|
|
|
let sentence = ctx
|
|
|
|
|
.data
|
|
|
|
|
.read()
|
|
|
|
|
.await
|
|
|
|
|
.get::<TextGenerator>()
|
|
|
|
|
.unwrap()
|
|
|
|
|
.generate();
|
|
|
|
|
channel_id.say(&ctx.http, sentence).await.unwrap();
|
|
|
|
|
tokio::time::delay_for(Duration::from_secs(*ctx.data.read().await.get::<Interval>().unwrap())).await;
|
|
|
|
|
tokio::time::sleep(Duration::from_secs(
|
|
|
|
|
*ctx.data.read().await.get::<Interval>().unwrap(),
|
|
|
|
|
))
|
|
|
|
|
.await;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -124,27 +137,25 @@ async fn interval_loop(ctx: Arc<Context>) {
|
|
|
|
|
impl EventHandler for Handler {
|
|
|
|
|
async fn message(&self, ctx: Context, msg: Message) {
|
|
|
|
|
if mentions_me(&ctx, &msg).await {
|
|
|
|
|
let sentence = ctx.data.read().await.get::<TextGenerator>().unwrap().generate();
|
|
|
|
|
let sentence = ctx
|
|
|
|
|
.data
|
|
|
|
|
.read()
|
|
|
|
|
.await
|
|
|
|
|
.get::<TextGenerator>()
|
|
|
|
|
.unwrap()
|
|
|
|
|
.generate();
|
|
|
|
|
msg.channel_id.say(&ctx.http, sentence).await.unwrap();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn ready(&self, ctx: Context, ready: Ready) {
|
|
|
|
|
async fn ready(&self, _ctx: Context, ready: Ready) {
|
|
|
|
|
eprintln!("{} is connected!", ready.user.name);
|
|
|
|
|
let ctx = Arc::new(ctx);
|
|
|
|
|
let ctx_clone = Arc::clone(&ctx);
|
|
|
|
|
if !ctx.data.read().await.get::<LoopStatus>().unwrap() {
|
|
|
|
|
let interval_loop = tokio::spawn(async move {interval_loop(ctx_clone).await});
|
|
|
|
|
{
|
|
|
|
|
let mut data = ctx.data.write().await;
|
|
|
|
|
data.insert::<LoopStatus>(true);
|
|
|
|
|
}
|
|
|
|
|
let _ = interval_loop.await;
|
|
|
|
|
{
|
|
|
|
|
let mut data = ctx.data.write().await;
|
|
|
|
|
data.insert::<LoopStatus>(false);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn cache_ready(&self, ctx: Context, _guilds: Vec<GuildId>) {
|
|
|
|
|
self.start_loop.call_once(move || {
|
|
|
|
|
tokio::spawn(async move { interval_loop(ctx).await });
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -154,17 +165,15 @@ async fn main() {
|
|
|
|
|
|
|
|
|
|
let generator = config.generator.load();
|
|
|
|
|
|
|
|
|
|
let mut client = Client::new(&config.token)
|
|
|
|
|
.event_handler(Handler)
|
|
|
|
|
let mut client = Client::builder(&config.token)
|
|
|
|
|
.event_handler(Handler {
|
|
|
|
|
start_loop: Once::new(),
|
|
|
|
|
})
|
|
|
|
|
.type_map_insert::<TextGenerator>(generator)
|
|
|
|
|
.type_map_insert::<Interval>(config.interval)
|
|
|
|
|
.type_map_insert::<ChanId>(config.channel_id)
|
|
|
|
|
.await
|
|
|
|
|
.expect("err creating client");
|
|
|
|
|
|
|
|
|
|
{
|
|
|
|
|
let mut data = client.data.write().await;
|
|
|
|
|
data.insert::<TextGenerator>(generator);
|
|
|
|
|
data.insert::<Interval>(config.interval);
|
|
|
|
|
data.insert::<ChanId>(config.channel_id);
|
|
|
|
|
data.insert::<LoopStatus>(false);
|
|
|
|
|
}
|
|
|
|
|
client.start().await.unwrap();
|
|
|
|
|
}
|
|
|
|
|