116 lines
3.3 KiB
Rust
116 lines
3.3 KiB
Rust
use diesel::{insert_into, prelude::*, Insertable, PgConnection, Queryable};
|
|
use rb::errors::{RbError, RbOption, RbResult};
|
|
use serde::{Deserialize, Serialize};
|
|
use uuid::Uuid;
|
|
|
|
use crate::schema::{sections, sections::dsl::*};
|
|
|
|
/// A section inside the database.
|
|
#[derive(Queryable, Serialize)]
|
|
pub struct Section
|
|
{
|
|
pub id: Uuid,
|
|
pub title: String,
|
|
pub shortname: String,
|
|
pub description: String,
|
|
pub is_default: bool,
|
|
pub has_titles: bool,
|
|
pub is_private: bool,
|
|
pub is_archived: bool,
|
|
}
|
|
|
|
/// A new section to add. Any `Option` values will get replaced by their default value in the
|
|
/// database.
|
|
#[derive(Serialize, Deserialize, Insertable)]
|
|
#[table_name = "sections"]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct NewSection
|
|
{
|
|
pub title: String,
|
|
pub shortname: String,
|
|
pub description: Option<String>,
|
|
pub is_default: Option<bool>,
|
|
pub has_titles: Option<bool>,
|
|
pub is_private: Option<bool>,
|
|
pub is_archived: Option<bool>,
|
|
}
|
|
|
|
/// A patch to apply to a section.
|
|
#[derive(Deserialize, AsChangeset)]
|
|
#[table_name = "sections"]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct PatchSection
|
|
{
|
|
title: Option<String>,
|
|
shortname: Option<String>,
|
|
description: Option<String>,
|
|
is_default: Option<bool>,
|
|
has_titles: Option<bool>,
|
|
pub is_private: Option<bool>,
|
|
pub is_archived: Option<bool>,
|
|
}
|
|
|
|
/// Get an amount of sections from the database, given an offset & limit. The maximum value for
|
|
/// `limit_` is determined by `super::MAX_SECTIONS`.
|
|
pub fn get(conn: &PgConnection, offset_: u32, limit_: u32) -> RbResult<Vec<Section>>
|
|
{
|
|
Ok(sections
|
|
.offset(offset_.into())
|
|
.limit(std::cmp::min(limit_, super::MAX_SECTIONS).into())
|
|
.load(conn)
|
|
.map_err(|_| RbError::DbError("Couldn't query sections."))?)
|
|
}
|
|
|
|
/// Try to find a section given its shortname.
|
|
pub fn find_with_shortname(conn: &PgConnection, shortname_: &str) -> RbOption<Section>
|
|
{
|
|
match sections.filter(shortname.eq(shortname_)).first(conn) {
|
|
Ok(val) => Ok(Some(val)),
|
|
Err(diesel::NotFound) => Ok(None),
|
|
_ => Err(RbError::DbError("Couldn't find section.")),
|
|
}
|
|
}
|
|
|
|
/// Create a new section.
|
|
pub fn create(conn: &PgConnection, new_section: &NewSection) -> RbResult<Section>
|
|
{
|
|
Ok(insert_into(sections)
|
|
.values(new_section)
|
|
.get_result(conn)
|
|
.map_err(|_| RbError::DbError("Couldn't insert section."))?)
|
|
|
|
// TODO check for conflict?
|
|
}
|
|
|
|
/// Update a section given its ID.
|
|
pub fn update(conn: &PgConnection, post_id: &Uuid, patch_post: &PatchSection) -> RbResult<Section>
|
|
{
|
|
Ok(diesel::update(sections.filter(id.eq(post_id)))
|
|
.set(patch_post)
|
|
.get_result(conn)
|
|
.map_err(|_| RbError::DbError("Couldn't update section."))?)
|
|
}
|
|
|
|
// Update a section given its shortname.
|
|
pub fn update_with_shortname(
|
|
conn: &PgConnection,
|
|
shortname_: &str,
|
|
patch_section: &PatchSection,
|
|
) -> RbResult<Section>
|
|
{
|
|
Ok(diesel::update(sections.filter(shortname.eq(shortname_)))
|
|
.set(patch_section)
|
|
.get_result(conn)
|
|
.map_err(|_| RbError::DbError("Couldn't update section with shortname."))?)
|
|
}
|
|
|
|
/// Delete a section given its ID.
|
|
pub fn delete(conn: &PgConnection, post_id: &Uuid) -> RbResult<()>
|
|
{
|
|
diesel::delete(sections.filter(id.eq(post_id)))
|
|
.execute(conn)
|
|
.map_err(|_| RbError::DbError("Couldn't delete section."))?;
|
|
|
|
Ok(())
|
|
}
|