Spells endpoints

src/commands/oai.rs

@@ -12,7 +12,7 @@ use serenity::model::channel::Message;
 use serenity::model::prelude::{ChannelType, PermissionOverwrite, PermissionOverwriteType};
 use serenity::prelude::*;
 
-use crate::db::{connection, NewMessageDB, MessageDB};
+use crate::db::{connection, messages::{MessageDB, NewMessageDB}};
 
 pub struct OAI {
   pub client: reqwest::Client,

src/db/mod.rs

@@ -1,26 +1,25 @@
 use crate::error_handler::ServiceError;
 use diesel::{r2d2::ConnectionManager, PgConnection};
+use serde::{Deserialize, Serialize};
 use crate::diesel_migrations::MigrationHarness;
 use lazy_static::lazy_static;
 use log::{error, info};
 use r2d2;
 use std::env;
 
-mod backgrounds;
-mod bestiary;
-mod classes;
-mod conditions;
-mod feats;
-mod items;
-mod messages;
-mod options;
-mod races;
-mod spells;
-mod users;
+pub mod backgrounds;
+pub mod bestiary;
+pub mod classes;
+pub mod conditions;
+pub mod feats;
+pub mod items;
+pub mod messages;
+pub mod options;
+pub mod races;
+pub mod spells;
+pub mod users;
 pub mod schema;
 
-pub use messages::*;
-
 type Pool = r2d2::Pool<ConnectionManager<PgConnection>>;
 pub type DbConnection = r2d2::PooledConnection<ConnectionManager<PgConnection>>;
 
@@ -52,3 +51,21 @@ pub fn connection() -> Result<DbConnection, ServiceError> {
   POOL.get()
     .map_err(|e| ServiceError::new(500, format!("Failed getting db connection: {}", e)))
 }
+
+pub fn load_data() {
+  spells::load_data();
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct GetResponse<T> {
+  pub data: T,
+  #[serde(skip_serializing_if = "Option::is_none")]
+  pub metadata: Option<Metadata>
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct Metadata {
+  pub total: i32,
+  pub limit: i32,
+  pub page: i32
+}

src/db/spells/mod.rs

@@ -1,3 +1,43 @@
 mod model;
+mod routes;
 
 pub use model::*;
+pub use routes::init_routes;
+
+pub fn load_data() {
+  let root_path = std::env::current_dir().unwrap();
+  let mut data_path = std::path::PathBuf::from(root_path);
+  data_path.push("data/spells.json");
+  match data_path.to_str() {
+    Some(path) => {
+      log::debug!("Loading spells from {}", path);
+      match std::fs::read_to_string(data_path) {
+        Ok(data) => {
+          match serde_json::from_str::<serde_json::Value>(&data) {
+            Ok(json) => {
+              match serde_json::from_value::<Vec<Spell>>(json) {
+                Ok(spells) => {
+                  let count = QuerySpell::get_count().unwrap();
+                  if count >= spells.len() as i64 {
+                    log::warn!("Spell data is already loaded");
+                    return;
+                  }
+                  for spell in spells {
+                    match InsertSpell::insert(spell.into()) {
+                      Ok(_) => {},
+                      Err(err) => log::error!("Failed to insert spell: {}", err)
+                    }
+                  }
+                },
+                Err(err) => log::error!("Failed to parse spells data: {}", err)
+              }
+            },
+            Err(err) => log::error!("Failed to parse spells data to value: {}", err)
+          };
+        },
+        Err(err) => log::error!("Failed to read spells data: {}", err)
+      };
+    },
+    None => log::error!("Failed to find spells data directory")
+  }
+}

src/db/spells/model.rs

@@ -1,7 +1,7 @@
 use diesel::prelude::*;
 use serde::{Deserialize, Serialize};
 
-use crate::db::schema::spells;
+use crate::{db::schema::spells, error_handler::ServiceError};
 
 #[derive(Queryable, QueryableByName)]
 #[diesel(table_name = spells)]
@@ -24,6 +24,38 @@ pub struct QuerySpell {
   pub description: String
 }
 
+impl QuerySpell {
+  pub fn get_all(limit: i32, page: i32) -> Result<Vec<Self>, ServiceError> {
+    let mut conn = crate::db::connection()?;
+    let mut query = spells::table
+      .limit(limit as i64)
+      .into_boxed();
+    query = query.filter(spells::id.gt(std::cmp::max(0, page - 1) * limit));
+    let spells = query.load::<QuerySpell>(&mut conn)?;
+    Ok(spells)
+  }
+
+  pub fn get_by_id(id: i32) -> Result<Self, ServiceError> {
+    let mut conn = crate::db::connection()?;
+    let spell = spells::table
+      .filter(spells::id.eq(id))
+      .first::<QuerySpell>(&mut conn)?;
+    Ok(spell)
+  }
+
+  pub fn get_count() -> Result<i64, ServiceError> {
+    let mut conn = crate::db::connection()?;
+    let count = spells::table.count().get_result(&mut conn)?;
+    Ok(count)
+  }
+
+  pub fn delete(id: i32) -> Result<Self, ServiceError> {
+    let mut conn = crate::db::connection()?;
+    let spell = diesel::delete(spells::table.filter(spells::id.eq(id))).get_result(&mut conn)?;
+    Ok(spell)
+  }
+}
+
 #[derive(Insertable, AsChangeset)]
 #[diesel(table_name = spells)]
 pub struct InsertSpell {
@@ -44,6 +76,20 @@ pub struct InsertSpell {
   pub description: String
 }
 
+impl InsertSpell {
+  pub fn insert(spell: Self) -> Result<QuerySpell, ServiceError> {
+    let mut conn = crate::db::connection()?;
+    let spell = diesel::insert_into(spells::table).values(spell).get_result(&mut conn)?;
+    Ok(spell)
+  }
+
+  pub fn update(id: i32, spell: Self) -> Result<QuerySpell, ServiceError> {
+    let mut conn = crate::db::connection()?;
+    let spell = diesel::update(spells::table.filter(spells::id.eq(id))).set(spell).get_result(&mut conn)?;
+    Ok(spell)
+  }
+}
+
 #[derive(Serialize, Deserialize)]
 pub struct Spell {
   pub name: String,
@@ -56,6 +102,7 @@ pub struct Spell {
   pub duration: String,
   pub classes: Vec<String>,
   pub sources: Vec<String>,
+  pub tags: Vec<String>,
   pub description: String
 }
 
@@ -67,30 +114,8 @@ pub struct Components {
   pub materials_needed: Option<String>
 }
 
-impl Spell {
-  /// Convert spell to insertable struct
-  pub fn to_insert(self) -> InsertSpell {
-    return InsertSpell {
-      name: self.name,
-      school: self.school,
-      level: self.level,
-      ritual: self.ritual,
-      casting_time: self.casting_time,
-      range: self.range,
-      components_verbal: self.components.verbal,
-      components_somatic: self.components.somatic,
-      components_material: self.components.material,
-      components_materials_needed: self.components.materials_needed,
-      duration: self.duration,
-      classes: self.classes,
-      sources: self.sources,
-      tags: vec![],
-      description: self.description
-    }
-  }
-
-  /// Convert query struct to spell
-  pub fn from_query(query: QuerySpell) -> Self {
+impl From<QuerySpell> for Spell {
+  fn from(query: QuerySpell) -> Self {
     return Self {
       name: query.name,
       school: query.school,
@@ -107,14 +132,30 @@ impl Spell {
       duration: query.duration,
       classes: query.classes,
       sources: query.sources,
+      tags: query.tags,
       description: query.description
     }
   }
+}
 
-  /// Convert file to spell
-  pub fn from_file(file: String) -> Self {
-    let data = std::fs::read_to_string(file).unwrap();
-    let spell: Spell = serde_json::from_str(&data).unwrap();
-    return spell;
+impl Into<InsertSpell> for Spell {
+  fn into(self) -> InsertSpell {
+    return InsertSpell {
+      name: self.name,
+      school: self.school,
+      level: self.level,
+      ritual: self.ritual,
+      casting_time: self.casting_time,
+      range: self.range,
+      components_verbal: self.components.verbal,
+      components_somatic: self.components.somatic,
+      components_material: self.components.material,
+      components_materials_needed: self.components.materials_needed,
+      duration: self.duration,
+      classes: self.classes,
+      sources: self.sources,
+      tags: self.tags,
+      description: self.description
+    }
   }
 }

src/db/spells/routes.rs

@@ -0,0 +1,79 @@
+use actix_web::{get, post, put, delete, web, HttpResponse, HttpRequest, ResponseError};
+use serde::{Serialize, Deserialize};
+
+use crate::db::{spells::QuerySpell, GetResponse, Metadata};
+
+use super::{Spell, InsertSpell};
+
+#[derive(Serialize, Deserialize)]
+struct GetAllParams {
+  limit: Option<i32>,
+  page: Option<i32>,
+}
+
+#[get("/spells")]
+async fn get_all(req: HttpRequest) -> HttpResponse {
+  let params = web::Query::<GetAllParams>::from_query(req.query_string()).unwrap();
+  let limit = params.limit.unwrap_or(20);
+  let page = params.page.unwrap_or(1);
+  match web::block(move || QuerySpell::get_all(limit, page)).await.unwrap() {
+    Ok(spells) => {
+      let mut response: Vec<Spell> = Vec::new();
+      for spell in spells {
+        response.push(Spell::from(spell));
+      }
+      let total_count = QuerySpell::get_count().unwrap();
+      HttpResponse::Ok().json(GetResponse {
+        data: response,
+        metadata: Some(Metadata {
+          total: total_count as i32,
+          limit,
+          page
+        })
+      })
+    },
+    Err(err) => ResponseError::error_response(&err)
+  }
+}
+
+#[get("/spells/{id}")]
+async fn get_by_id(id: web::Path<i32>) -> HttpResponse {
+  match web::block(move || QuerySpell::get_by_id(id.into_inner())).await.unwrap() {
+    Ok(spell) => HttpResponse::Ok().json(GetResponse {
+      data: Spell::from(spell),
+      metadata: None
+    }),
+    Err(err) => ResponseError::error_response(&err)
+  }
+}
+
+#[post("/spells")]
+async fn create(spell: web::Json<Spell>) -> HttpResponse {
+  match InsertSpell::insert(spell.into_inner().into()) {
+    Ok(spell) => HttpResponse::Created().json(Spell::from(spell)),
+    Err(err) => ResponseError::error_response(&err)
+  }
+}
+
+#[put("/spells/{id}")]
+async fn update(id: web::Path<i32>, spell: web::Json<Spell>) -> HttpResponse {
+  match web::block(move || InsertSpell::update(id.into_inner(), spell.into_inner().into())).await.unwrap() {
+    Ok(spell) => HttpResponse::Ok().json(Spell::from(spell)),
+    Err(err) => ResponseError::error_response(&err)
+  }
+}
+
+#[delete("/spells/{id}")]
+async fn delete(id: web::Path<i32>) -> HttpResponse {
+  match web::block(move || QuerySpell::delete(id.into_inner())).await.unwrap() {
+    Ok(spell) => HttpResponse::Ok().json(Spell::from(spell)),
+    Err(err) => ResponseError::error_response(&err)
+  }
+}
+
+pub fn init_routes(config: &mut web::ServiceConfig) {
+  config.service(get_all);
+  config.service(get_by_id);
+  config.service(create);
+  config.service(delete);
+}

src/error_handler.rs

@@ -1,4 +1,6 @@
+use actix_web::{ResponseError, HttpResponse};
 use diesel::result::Error as DieselError;
+use reqwest::StatusCode;
 use serde::{Deserialize, Serialize};
 use std::fmt;
 
@@ -29,8 +31,27 @@ impl From<DieselError> for ServiceError {
       DieselError::DatabaseError(_, err) => ServiceError::new(409, err.message().to_string()),
       DieselError::NotFound => {
         ServiceError::new(404, "The record was not found".to_string())
-      }
+      },
+      DieselError::SerializationError(err) => {
+        ServiceError::new(422, err.to_string())
+      },
       err => ServiceError::new(500, format!("Unknown Diesel error: {}", err)),
     }
   }
 }
+
+impl ResponseError for ServiceError {
+  fn error_response(&self) -> HttpResponse {
+      let status_code = match StatusCode::from_u16(self.error_status_code) {
+          Ok(status_code) => status_code,
+          Err(_) => StatusCode::INTERNAL_SERVER_ERROR,
+      };
+
+      let error_message = match status_code.as_u16() < 500 {
+          true => self.error_message.clone(),
+          false => "Internal server error".to_string(),
+      };
+
+      HttpResponse::build(status_code).json(serde_json::json!({ "message": error_message }))
+  }
+}

src/main.rs

@@ -6,6 +6,7 @@ use std::collections::{HashSet, HashMap};
 use std::env;
 use std::sync::Arc;
 
+use actix_web::{HttpServer, App};
 use commands::audio::{create_response, AudioConfig, AudioConfigs};
 
 use dotenv::dotenv;
@@ -111,60 +112,86 @@ impl EventHandler for Handler {
   }
 }
 
-#[tokio::main]
-async fn main() {
+#[actix_web::main]
+async fn main() -> std::io::Result<()> {
   dotenv().ok();
   env_logger::init_from_env(env_logger::Env::default().filter_or("RUST_LOG", "warn,siren=info"));
-
-  let token: String = env::var("DISCORD_TOKEN").expect("Expected a token in the environment");
-  let intents: GatewayIntents = GatewayIntents::all();
-
-  let http: Http = Http::new(&token);
-  let (owners, _bot_id) = match http.get_current_application_info().await {
-    Ok(info) => {
-      let mut owners: HashSet<serenity::model::id::UserId> = HashSet::new();
-      if let Some(team) = info.team {
-        owners.insert(team.owner_user_id);
-      } else {
-        owners.insert(info.owner.id);
-      }
-      match http.get_current_user().await {
-        Ok(bot) => (owners, bot.id),
-        Err(why) => panic!("Could not access the bot id: {:?}", why)
-      }
-    },
-    Err(why) => panic!("Could not access application info: {:?}", why)
-  };
-
   db::init();
-  
-  let handler = match env::var("OPENAI_API_KEY") {
-    Ok(token) => {
-      info!("Loaded OpenAI token");
-      Handler {
-        oai: Some(commands::oai::OAI { client: reqwest::Client::new(), base_url: "https://api.openai.com/v1".to_string(), max_attempts: 5, token , max_context_questions: 15 })
-      }
-    }
+  db::load_data();
+
+  // setup_discord_bot();
+
+  let host = env::var("SERVICE_HOST").unwrap_or("localhost".to_string());
+  let port = env::var("SERVICE_PORT").unwrap_or("5000".to_string());
+
+  match HttpServer::new(|| {
+    App::new()
+      .configure(db::spells::init_routes)
+  })
+  .bind(format!("{}:{}", host, port)) {
+    Ok(b) => {
+      info!("Binding server to {}:{}", host, port);
+      b
+    },
     Err(err) => {
-      warn!("Could not load OpenAI token: {}", err);
-      Handler { oai: None }
+      error!("Could not bind server: {}", err);
+      return Err(err);
     }
-  };
-
-  let mut client = Client::builder(token, intents)
-    .event_handler(handler)
-    .framework(StandardFramework::new()
-    .configure(|c| c.owners(owners)))
-    .register_songbird()
-    .await
-    .expect("Error creating client");
-
-  {
-    let mut data = client.data.write().await;
-    data.insert::<AudioConfigs>(Arc::new(RwLock::new(HashMap::default())));
-  }
-
-  if let Err(why) = client.start_autosharded().await {
-    error!("An error occurred while running the client: {:?}", why);
   }
+  .run()
+  .await
+}
+
+fn setup_discord_bot() {
+  tokio::spawn(async {
+    let token: String = env::var("DISCORD_TOKEN").expect("Expected a token in the environment");
+    let intents: GatewayIntents = GatewayIntents::all();
+
+    let http: Http = Http::new(&token);
+    let (owners, _bot_id) = match http.get_current_application_info().await {
+      Ok(info) => {
+        let mut owners: HashSet<serenity::model::id::UserId> = HashSet::new();
+        if let Some(team) = info.team {
+          owners.insert(team.owner_user_id);
+        } else {
+          owners.insert(info.owner.id);
+        }
+        match http.get_current_user().await {
+          Ok(bot) => (owners, bot.id),
+          Err(why) => panic!("Could not access the bot id: {:?}", why)
+        }
+      },
+      Err(why) => panic!("Could not access application info: {:?}", why)
+    };
+    
+    let handler = match env::var("OPENAI_API_KEY") {
+      Ok(token) => {
+        info!("Loaded OpenAI token");
+        Handler {
+          oai: Some(commands::oai::OAI { client: reqwest::Client::new(), base_url: "https://api.openai.com/v1".to_string(), max_attempts: 5, token , max_context_questions: 15 })
+        }
+      }
+      Err(err) => {
+        warn!("Could not load OpenAI token: {}", err);
+        Handler { oai: None }
+      }
+    };
+
+    let mut client = Client::builder(token, intents)
+      .event_handler(handler)
+      .framework(StandardFramework::new()
+      .configure(|c| c.owners(owners)))
+      .register_songbird()
+      .await
+      .expect("Error creating client");
+
+    {
+      let mut data = client.data.write().await;
+      data.insert::<AudioConfigs>(Arc::new(RwLock::new(HashMap::default())));
+    }
+
+    if let Err(why) = client.start_autosharded().await {
+      error!("An error occurred while running the client: {:?}", why);
+    }
+  });
 }