akulij/gongbotrs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use MongodbStorage

Browse Source
This commit is contained in:
Akulij 2025-04-26 00:28:19 +03:00
parent 7fe3a76052
commit a765dce328
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main.rs
View File

@ -1,18 +1,20 @@
pub mod admin; pub mod admin;
pub mod db; pub mod db;
pub mod mongodb_storage;
use std::time::Duration; use std::time::Duration;
use crate::admin::{admin_command_handler, AdminCommands}; use crate::admin::{admin_command_handler, AdminCommands};
use crate::admin::{secret_command_handler, SecretCommands}; use crate::admin::{secret_command_handler, SecretCommands};
use crate::db::{CallDB, DB}; use crate::db::{CallDB, DB};
use crate::mongodb_storage::MongodbStorage;
use chrono::{DateTime, Utc}; use chrono::{DateTime, Utc};
use chrono_tz::Asia; use chrono_tz::Asia;
use envconfig::Envconfig; use envconfig::Envconfig;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use teloxide::dispatching::dialogue::serializer::Json; use teloxide::dispatching::dialogue::serializer::Json;
use teloxide::dispatching::dialogue::{GetChatId, PostgresStorage}; use teloxide::dispatching::dialogue::GetChatId;
use teloxide::types::{ use teloxide::types::{
InlineKeyboardButton, InlineKeyboardMarkup, InputFile, InputMedia, MediaKind, MessageKind, InlineKeyboardButton, InlineKeyboardMarkup, InputFile, InputMedia, MediaKind, MessageKind,
ParseMode, ReplyMarkup, ParseMode, ReplyMarkup,
@ -23,7 +25,7 @@ use teloxide::{
utils::{command::BotCommands, render::RenderMessageTextHelper}, utils::{command::BotCommands, render::RenderMessageTextHelper},
}; };
type BotDialogue = Dialogue<State, PostgresStorage<Json>>; type BotDialogue = Dialogue<State, MongodbStorage<Json>>;
#[derive(Envconfig)] #[derive(Envconfig)]
struct Config { struct Config {
@ -74,7 +76,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
let bot = Bot::new(&config.bot_token); let bot = Bot::new(&config.bot_token);
let db = DB::new(&config.db_url).await; let db = DB::new(&config.db_url).await;
let db_url2 = config.db_url.clone(); let db_url2 = config.db_url.clone();
let state_mgr = PostgresStorage::open(&db_url2, 8, Json).await?; let state_mgr = MongodbStorage::open(&db_url2, "gongbot", Json).await?;
// TODO: delete this in production // TODO: delete this in production
let events: Vec<DateTime<Utc>> = vec!["2025-04-09T18:00:00+04:00", "2025-04-11T16:00:00+04:00"] let events: Vec<DateTime<Utc>> = vec!["2025-04-09T18:00:00+04:00", "2025-04-11T16:00:00+04:00"]
@ -105,7 +107,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
.await; .await;
user.is_admin user.is_admin
}) })
.enter_dialogue::<Message, PostgresStorage<Json>, State>() .enter_dialogue::<Message, MongodbStorage<Json>, State>()
.branch( .branch(
Update::filter_message() Update::filter_message()
.filter(|msg: Message| { .filter(|msg: Message| {