desubot/src/main.rs

use std::{env, process};

use futures::StreamExt;
use telegram_bot::*;
#[macro_use]
extern crate log;
use env_logger::Env;

mod commands;
mod db;
mod errors;
mod mystem;
mod utils;
mod handlers;

use mystem::MyStem;

#[tokio::main]
async fn main() -> Result<(), errors::Error> {
    env_logger::from_env(Env::default().default_filter_or("info")).init();
    let mut mystem = MyStem::new()?;
    match db::update_scheme() {
        Ok(_) => {}
        Err(e) => panic!("Database error: {:?}", e),
    }
    let token = match env::var("TELEGRAM_BOT_TOKEN") {
        Ok(token) => token,
        Err(_) => {
            error!("TELEGRAM_BOT_TOKEN not set");
            process::exit(0x0001);
        }
    };
    let api = Api::new(token.clone());
    let mut stream = api.stream();
    let me = api.send(GetMe).await?;
    info!("GetMe result: Username: {}, First Name: {}, ID {}", me.username.as_ref().unwrap(), me.first_name, me.id);
    while let Some(update) = stream.next().await {
        let update = update?;
        if let UpdateKind::Message(message) = update.kind {
            db::add_conf(message.clone()).await?;
            db::add_user(message.clone()).await?;
            handlers::handler(api.clone(), message, token.clone(), &mut mystem, me.clone()).await?;
        }
    }
    Ok(())
}