joke_bot on Nostr: use nostr_bot::{ get_reply, keypair_from_secret, wrap_state, Command, Event, ...

use nostr_bot::{
get_reply, keypair_from_secret, wrap_state, Command, Event, EventNonSigned, FunctorType};

struct Info {
started_at: std::time::SystemTime,
jokes_told: u64,
}

type State = nostr_bot::State<Info>;

async fn fetch_joke(category: &str) -> Result<String, reqwest::Error> {
let url = format!("https://v2.jokeapi.dev/joke/{}?format=txt";, category);
let response = reqwest::get(url).await?;

response.text().await
}

async fn joke(event: Event, state: State) -> EventNonSigned {
let args = event.content.split_whitespace().collect::<Vec<_>>();
let category = if args.len() > 1 {
match args[1] {
"any" | "programming" | "misc" | "dark" | "pun" | "spooky" | "christmas" => args[1],
_ => {
return get_reply(
event,
String::from(
"Unknown category, run !help command to see available categories.",
),
)
}
}
} else {
"any"
};

let response = match fetch_joke(category).await {
Ok(joke_text) => {
state.lock().await.jokes_told += 1;
joke_text
}
Err(e) => format!("I was unable to get the joke: {}", e),
};

get_reply(event, response)
}

async fn stats(event: Event, state: State) -> EventNonSigned {
let state = state.lock().await;
let uptime_seconds = std::time::SystemTime::now()
.duration_since(state.started_at)
.unwrap()
.as_secs();
get_reply(
event,
format!(
"I'm up {} and I told {} jokes since my last crash.",
compound_duration::format_dhms(uptime_seconds),
state.jokes_told
),
)
}

async fn code(event: Event, _state: State) -> EventNonSigned {
get_reply(event, String::from(std::include_str!("main.rs")))
}

async fn repo(event: Event, _state: State) -> EventNonSigned {
get_reply(event, String::from("https://github.com/slaninas/joke_bot";))
}

#[tokio::main]
async fn main() {
nostr_bot::init_logger();

let relays = vec![
"wss://nostr-pub.wellorder.net",
"wss://relay.damus.io",
"wss://relay.nostr.info",
];

let mut secret = std::fs::read_to_string("secret").unwrap();
secret.pop(); // Remove newline

let keypair = keypair_from_secret(&secret);

let pic_url = "https://media.istockphoto.com/vectors/yellow-emoticons-and-emojis-vector-id1179177852?k=20&m=1179177852&s=612x612&w=0&h=5UrMXuV5x2WCI-d8twGI9gPZ6810wpgqyF9oy7X-r9M=";;

let state = wrap_state(Info {
started_at: std::time::SystemTime::now(),
jokes_told: 0,
});

let mut bot = nostr_bot::Bot::new(keypair, relays, state)
.name("joke_bot")
.about("Just joking around. Blame https://sv443.net/jokeapi/v2/ if you don't like a joke.")
.picture(pic_url)
.intro_message("Wasup, I'm a joke bot. Reply to me with !help.")
.command(Command::new("!joke", nostr_bot::wrap!(joke)).description("Tells a random joke. Category can be specified: any, programming, misc, dark, pun, spooky, christmas"))
.command(
Command::new("!stats", nostr_bot::wrap!(stats))
.description("Shows for how long I'm running and how many jokes I told."),
)
.command(Command::new("!code", nostr_bot::wrap!(code)).description("Spits out bot's source code."))
.command(Command::new("!repo", nostr_bot::wrap!(repo)).description("Gives link to bot's repository."))
.help();

bot.run().await;
}