summaryrefslogtreecommitdiffstats
path: root/src/models/warning.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/models/warning.rs')
-rw-r--r--src/models/warning.rs382
1 files changed, 0 insertions, 382 deletions
diff --git a/src/models/warning.rs b/src/models/warning.rs
deleted file mode 100644
index c420dd0..0000000
--- a/src/models/warning.rs
+++ /dev/null
@@ -1,382 +0,0 @@
-use crate::{config::CONFIG, db::get_client, errors::AppError};
-use chrono::{Local, NaiveDateTime};
-use serde::{Deserialize, Serialize};
-use sqlx::types::JsonValue;
-use sqlx::Row;
-use std::convert::From;
-
-/// Model for warnings.
-#[derive(Deserialize, Serialize, sqlx::FromRow)]
-pub struct Warning {
- pub id: i32,
- pub user_id: Option<i32>,
- pub model_id: Option<i32>,
- pub resolved_by: Option<i32>,
- pub note: String,
- pub admin_note: String,
- pub created: NaiveDateTime,
- pub updated: NaiveDateTime,
-}
-
-#[derive(Serialize, sqlx::FromRow)]
-pub struct WarningUser {
- pub id: i32,
- pub user_id: Option<i32>,
- pub model_id: Option<i32>,
- pub resolved_by: Option<i32>,
- pub note: String,
- pub admin_note: String,
- pub created: NaiveDateTime,
- pub updated: NaiveDateTime,
- user: Option<JsonValue>,
- resolved: Option<JsonValue>,
-}
-
-/// Impl conversion from `WarningUser` to `Warning`
-impl From<WarningUser> for Warning {
- fn from(item: WarningUser) -> Self {
- Self {
- id: item.id,
- user_id: item.user_id,
- model_id: item.model_id,
- resolved_by: item.resolved_by,
- note: item.note,
- admin_note: item.admin_note,
- created: item.created,
- updated: item.created,
- }
- }
-}
-
-/// Payload used to create a new warning
-#[derive(Deserialize)]
-pub struct WarningCreate {
- pub model_id: i32,
- pub note: String,
-}
-
-/// Payload used to edit a warning
-#[derive(Deserialize)]
-pub struct WarningEdit {
- pub admin_note: String,
-}
-
-/// Payload used for warning filtering
-#[derive(Deserialize)]
-pub struct WarningFilterPayload {
- pub model_id: Option<i32>,
- pub resolved_by: Option<i32>,
-}
-
-/// Struct used as argument for filtering by the backend
-#[derive(Debug)]
-pub struct WarningFilter {
- pub model_id: Option<i32>,
- pub resolved_by: Option<i32>,
- pub user_id: Option<i32>,
-}
-
-impl Warning {
- /// Create a warning means create an object which has an `user_id` (creator of the warning), a
- /// `model_id` (suspect model) and a `note`
- pub fn new(user_id: i32, model_id: i32, note: String) -> Self {
- let now = Local::now().naive_utc();
- Self {
- id: 0,
- user_id: Some(user_id),
- model_id: Some(model_id),
- resolved_by: None,
- note,
- admin_note: String::new(),
- created: now,
- updated: now,
- }
- }
-
- /// Delete a report
- pub async fn delete(warning_id: i32) -> Result<(), AppError> {
- let pool = unsafe { get_client() };
-
- sqlx::query(
- r#"
- DELETE FROM warnings WHERE id = $1
- "#,
- )
- .bind(warning_id)
- .execute(pool)
- .await?;
-
- Ok(())
- }
-
- /// List all warnings. A staffer can see all the warnings, a user cannot
- pub async fn list(page: i64, user_id: Option<i32>) -> Result<Vec<WarningUser>, AppError> {
- let pool = unsafe { get_client() };
- let query = r#"
- SELECT
- warnings.*,
- json_build_object('id', users.id, 'name', users.name, 'email', users.email, 'username', users.username, 'is_staff', users.is_staff, 'avatar', users.avatar) as user,
- coalesce(r.data, '{}'::json) as resolved
- FROM warnings
- JOIN users ON users.id = warnings.user_id
- LEFT JOIN (
- SELECT id, json_build_object('id', r.id, 'name', r.name, 'email', r.email, 'username', r.username, 'is_staff', r.is_staff, 'avatar', r.avatar) as data
- FROM users r
- ) r ON r.id = warnings.resolved_by
- "#;
-
- let rows: Vec<WarningUser> = match user_id {
- Some(id) => {
- sqlx::query_as(&format!(
- r#"{} WHERE user_id = $1 ORDER BY id DESC LIMIT $2 OFFSET $3"#,
- query
- ))
- .bind(id)
- .bind(CONFIG.page_limit)
- .bind(CONFIG.page_limit * page)
- .fetch_all(pool)
- .await?
- }
- None => {
- sqlx::query_as(&format!(r#"{} ORDER BY id DESC LIMIT $1 OFFSET $2"#, query))
- .bind(CONFIG.page_limit)
- .bind(CONFIG.page_limit * page)
- .fetch_all(pool)
- .await?
- }
- };
-
- Ok(rows)
- }
-
- /// Returns the warning with id = `warning_id`
- pub async fn find_by_id(warning_id: i32) -> Result<WarningUser, AppError> {
- let pool = unsafe { get_client() };
-
- let rec: WarningUser = sqlx::query_as(
- r#"
- SELECT
- warnings.*,
- json_build_object('id', users.id, 'name', users.name, 'email', users.email, 'username', users.username, 'is_staff', users.is_staff, 'avatar', users.avatar) as user,
- coalesce(r.data, '{}'::json) as resolved
- FROM warnings
- JOIN users ON users.id = warnings.user_id
- LEFT JOIN (
- SELECT id, json_build_object('id', r.id, 'name', r.name, 'email', r.email, 'username', r.username, 'is_staff', r.is_staff, 'avatar', r.avatar) as data
- FROM users r
- ) r ON r.id = warnings.resolved_by
- WHERE warnings.id = $1
- "#)
- .bind(warning_id)
- .fetch_one(pool)
- .await?;
-
- Ok(rec)
- }
-
- /// Return the number of warnings.
- pub async fn count(user_id: Option<i32>) -> Result<i64, AppError> {
- let pool = unsafe { get_client() };
-
- let cursor = match user_id {
- Some(id) => {
- sqlx::query(r#"SELECT COUNT(id) as count FROM warnings WHERE user_id = $1"#)
- .bind(id)
- .fetch_one(pool)
- .await?
- }
- None => {
- sqlx::query(r#"SELECT COUNT(id) as count FROM warnings"#)
- .fetch_one(pool)
- .await?
- }
- };
-
- let count: i64 = cursor.try_get(0).unwrap();
- Ok(count)
- }
-
- /// Create a new upload for model
- pub async fn create(warning: Warning) -> Result<Warning, AppError> {
- let pool = unsafe { get_client() };
-
- let rec: Warning = sqlx::query_as(
- r#"
- INSERT INTO warnings (user_id, model_id, resolved_by, note, admin_note, created, updated)
- VALUES ( $1, $2, $3, $4, $5, $6, $7)
- RETURNING *
- "#,
- )
- .bind(warning.user_id)
- .bind(warning.model_id)
- .bind(warning.resolved_by)
- .bind(warning.note)
- .bind(warning.admin_note)
- .bind(warning.created)
- .bind(warning.updated)
- .fetch_one(pool)
- .await?;
-
- Ok(rec)
- }
-
- /// Filter warnings. Pass a `WarningFilter` argument. You can filter only by model_id or (not
- /// both) resolved by
- pub async fn filter(page: i64, args: WarningFilter) -> Result<Vec<WarningUser>, AppError> {
- let pool = unsafe { get_client() };
-
- let mut query = r#"
- SELECT
- warnings.*,
- json_build_object('id', users.id, 'name', users.name, 'email', users.email, 'username', users.username, 'is_staff', users.is_staff, 'avatar', users.avatar) as user,
- coalesce(r.data, '{}'::json) as resolved
- FROM warnings
- JOIN users ON users.id = warnings.user_id
- LEFT JOIN (
- SELECT id, json_build_object('id', r.id, 'name', r.name, 'email', r.email, 'username', r.username, 'is_staff', r.is_staff, 'avatar', r.avatar) as data
- FROM users r
- ) r ON r.id = warnings.resolved_by
- "#.to_string();
-
- if args.model_id.is_some() {
- query += r#"WHERE model_id = $1"#;
- } else {
- match args.resolved_by {
- Some(_) => {
- query += r#" WHERE warnings.resolved_by = $1"#;
- }
- None => {
- query += r#" WHERE warnings.resolved_by IS NULL"#;
- }
- };
- }
-
- let rows: Vec<WarningUser> = match args.user_id {
- Some(id) => {
- let q = if args.model_id.is_some() {
- query = format!(
- r#"{} AND user_id = $2 ORDER BY id DESC LIMIT $3 OFFSET $4"#,
- query
- );
- sqlx::query_as(&query).bind(args.model_id.unwrap())
- } else if args.resolved_by.is_some() {
- query = format!(
- r#"{} AND user_id = $2 ORDER BY id DESC LIMIT $3 OFFSET $4"#,
- query
- );
- sqlx::query_as(&query).bind(args.resolved_by.unwrap())
- } else {
- query = format!(
- r#"{} AND user_id = $1 ORDER BY id DESC LIMIT $2 OFFSET $3"#,
- query
- );
- sqlx::query_as(&query)
- };
-
- q.bind(id)
- .bind(CONFIG.page_limit)
- .bind(CONFIG.page_limit * page)
- .fetch_all(pool)
- .await?
- }
- None => {
- let q = if args.model_id.is_some() {
- query = format!(r#"{} ORDER BY id DESC LIMIT $2 OFFSET $3"#, query);
- sqlx::query_as(&query).bind(args.model_id.unwrap())
- } else if args.resolved_by.is_some() {
- query = format!(r#"{} ORDER BY id DESC LIMIT $2 OFFSET $3"#, query);
- sqlx::query_as(&query).bind(args.resolved_by.unwrap())
- } else {
- query = format!(r#"{} ORDER BY id DESC LIMIT $1 OFFSET $2"#, query);
- sqlx::query_as(&query)
- };
-
- q.bind(CONFIG.page_limit)
- .bind(CONFIG.page_limit * page)
- .fetch_all(pool)
- .await?
- }
- };
-
- Ok(rows)
- }
-
- /// Return the number of filtered warnings.
- pub async fn count_by_model_id(args: WarningFilter) -> Result<i64, AppError> {
- let pool = unsafe { get_client() };
-
- let mut query = r#"
- SELECT COUNT(id) as count FROM warnings
- "#
- .to_string();
-
- if args.model_id.is_some() {
- query += r#" WHERE model_id = $1"#;
- } else {
- match args.resolved_by {
- Some(_) => {
- query += r#" WHERE warnings.resolved_by = $1"#;
- }
- None => {
- query += r#" WHERE warnings.resolved_by IS NULL"#;
- }
- };
- }
-
- let cursor = match args.user_id {
- Some(id) => {
- let q = if args.model_id.is_some() {
- query = format!(r#"{} AND user_id = $2"#, query);
- sqlx::query(&query).bind(args.model_id.unwrap())
- } else if args.resolved_by.is_some() {
- query = format!(r#"{} AND user_id = $2"#, query);
- sqlx::query(&query).bind(args.resolved_by.unwrap())
- } else {
- query = format!(r#"{} AND user_id = $1"#, query);
- sqlx::query(&query)
- };
-
- q.bind(id).fetch_one(pool).await?
- }
- None => {
- let q = if args.model_id.is_some() {
- sqlx::query(&query).bind(args.model_id.unwrap())
- } else if args.resolved_by.is_some() {
- sqlx::query(&query).bind(args.resolved_by.unwrap())
- } else {
- sqlx::query(&query)
- };
-
- q.fetch_one(pool).await?
- }
- };
-
- let count: i64 = cursor.try_get(0).unwrap();
- Ok(count)
- }
-
- /// Edit a warning
- pub async fn edit(&mut self, resolver: i32, payload: WarningEdit) -> Result<(), AppError> {
- let pool = unsafe { get_client() };
-
- let now = Local::now().naive_utc();
-
- sqlx::query(
- r#"
- UPDATE warnings SET admin_note = $1, resolved_by = $2, updated = $3 WHERE id = $4
- "#,
- )
- .bind(&payload.admin_note)
- .bind(resolver)
- .bind(now)
- .bind(self.id)
- .execute(pool)
- .await?;
-
- self.admin_note = payload.admin_note;
- self.resolved_by = Some(resolver);
- self.updated = now;
-
- Ok(())
- }
-}
generated by cgit v1.2.3-71-g8e6c (git 2.46.0) at 2026-01-15 11:45:51 +0000