Provide basic MongoDB connection

This commit is contained in:
Victor Mignot 2022-10-18 23:23:54 -04:00
parent 1d3905a3b2
commit e951b3012e
No known key found for this signature in database
GPG key ID: FFE4EF056FB5E0D0
3 changed files with 70 additions and 16 deletions

View file

@ -1,3 +1,8 @@
//! Module containing the Yorokobot client and used structs
use crate::errors::ClientsError;
use mongodb::{options::ClientOptions as MongoClientOptions, Client as MongoClient};
use serenity::{prelude::GatewayIntents, Client as DiscordClient}; use serenity::{prelude::GatewayIntents, Client as DiscordClient};
/// Yorokobot's client. /// Yorokobot's client.
@ -20,38 +25,75 @@ use serenity::{prelude::GatewayIntents, Client as DiscordClient};
/// ///
/// # } /// # }
/// ``` /// ```
///
pub struct Client { pub struct Client {
/// The Serenity Discord Client
discord_client: DiscordClient, discord_client: DiscordClient,
/// The MongoDB Client
mongodb_client: Option<MongoClient>,
/// MongoDB Client Options
mongodb_options: MongoClientOptions,
} }
/// Yorokobot connection credentials /// Yorokobot connection credentials
pub struct ClientCredentials<'a> { pub struct ClientCredentials<'a> {
/// Token for Discord API /// Token for Discord API
pub discord_token: &'a String, pub discord_token: &'a String,
/// MongoDB connection string.
pub mongo_uri: &'a String,
} }
impl<'a> Client { impl<'a> Client {
/// Create a Yorokobot client /// Create a Yorokobot client
pub async fn new(credentials: ClientCredentials<'a>) -> Client { pub async fn new(credentials: ClientCredentials<'a>) -> Result<Self, ClientsError> {
let discord_client = let discord_client = match DiscordClient::builder(
DiscordClient::builder(credentials.discord_token, GatewayIntents::empty()) credentials.discord_token,
.await GatewayIntents::empty(),
.expect("Could not create Discord Client"); )
.await
{
Ok(c) => c,
Err(e) => return Err(ClientsError::Discord(e)),
};
Client { discord_client } let mongodb_options = match MongoClientOptions::parse(credentials.mongo_uri).await {
Ok(o) => o,
Err(e) => return Err(ClientsError::Database(e)),
};
Ok(Client {
discord_client,
mongodb_options,
mongodb_client: None,
})
} }
/// Start connection to Discord API. /// Start connection to Discord API.
/// Wrap [`serenity::client::Client`] start method. /// Wrap [`serenity::client::Client`] start method.
pub async fn connect_discord(&mut self) { pub async fn connect_discord(&mut self) -> Result<(), ClientsError> {
if let Err(error) = self.discord_client.start().await { match self.discord_client.start().await {
println!("Could not connect to Discord: {:?}", error); Ok(_) => Ok(()),
Err(e) => return Err(ClientsError::Discord(e)),
} }
} }
/// Connect to the Mongo Database
pub fn connect_mongodb(&mut self) -> Result<(), ClientsError> {
self.mongodb_client = match MongoClient::with_options(self.mongodb_options.clone()) {
Ok(c) => Some(c),
Err(e) => return Err(ClientsError::Database(e)),
};
Ok(())
}
/// Connect client to the Mongo database then to the Discord API. /// Connect client to the Mongo database then to the Discord API.
pub async fn connect(&mut self) { pub async fn connect(&mut self) -> Result<(), ClientsError> {
self.connect_discord().await; self.connect_mongodb()?;
self.connect_discord().await?;
Ok(())
} }
} }

View file

@ -6,5 +6,6 @@
#![deny(missing_docs)] #![deny(missing_docs)]
#![deny(warnings)] #![deny(warnings)]
/// Module containing the Yorokobot client and used structs
pub mod client; pub mod client;
pub mod errors;

View file

@ -1,16 +1,27 @@
use std::env; use std::env;
use yorokobot::client::{Client, ClientCredentials}; use yorokobot::{
client::{Client, ClientCredentials},
errors::ClientsError,
};
#[tokio::main] #[tokio::main]
async fn main() { async fn main() {
let discord_token = env::var("DISCORD_TOKEN").expect("Cannot fetch Discord token"); let discord_token = env::var("DISCORD_TOKEN").expect("Cannot fetch Discord token");
let mongodb_uri = env::var("MONGODB_URI").expect("Cannot fetch Mongo URI");
let credentials = ClientCredentials { let credentials = ClientCredentials {
discord_token: &discord_token, discord_token: &discord_token,
mongo_uri: &mongodb_uri,
}; };
let mut client = Client::new(credentials).await; let mut client = Client::new(credentials)
.await
.expect("Could not create client");
client.connect_discord().await; client.connect().await.unwrap_or_else(|error| match error {
ClientsError::Database(e) => panic!("Could not connect to database: {:?}", e),
ClientsError::Discord(e) => panic!("Could not connect to Discord: {:?}", e),
});
} }