diff --git a/planetwars-server/src/db/maps.rs b/planetwars-server/src/db/maps.rs new file mode 100644 index 0000000..c9f50a0 --- /dev/null +++ b/planetwars-server/src/db/maps.rs @@ -0,0 +1,31 @@ +use diesel::{PgConnection, QueryDsl, QueryResult, RunQueryDsl}; + +use crate::schema::maps; + +#[derive(Insertable)] +#[table_name = "maps"] +pub struct NewMap<'a> { + pub name: &'a str, + pub file_path: &'a str, +} + +#[derive(Queryable, Clone, Debug)] +pub struct Map { + pub id: i32, + pub name: String, + pub file_path: String, +} + +pub fn create_map(new_map: NewMap, conn: &PgConnection) -> QueryResult { + diesel::insert_into(maps::table) + .values(new_map) + .get_result(conn) +} + +pub fn find_map(id: i32, conn: &PgConnection) -> QueryResult { + maps::table.find(id).get_result(conn) +} + +pub fn list_maps(conn: &PgConnection) -> QueryResult> { + maps::table.get_results(conn) +} \ No newline at end of file diff --git a/planetwars-server/src/db/mod.rs b/planetwars-server/src/db/mod.rs index 84ed2a6..f014cea 100644 --- a/planetwars-server/src/db/mod.rs +++ b/planetwars-server/src/db/mod.rs @@ -1,4 +1,5 @@ pub mod bots; +pub mod maps; pub mod matches; pub mod ratings; pub mod sessions;