tibba-model 0.2.0

// Copyright 2025 Tree xie.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use super::Model;
use super::user::{ROLE_ADMIN, ROLE_SUPER_ADMIN};
use super::{
    Error, JsonSnafu, ModelListParams, Schema, SchemaAllowCreate, SchemaAllowEdit, SchemaType,
    SchemaView, SqlxSnafu, Status, format_datetime, new_schema_options, parse_primitive_datetime,
};
use http::header::{HeaderMap, HeaderName, HeaderValue};
use serde::de::DeserializeOwned;
use serde::{Deserialize, Serialize};
use snafu::ResultExt;
use sqlx::FromRow;
use sqlx::types::Json;
use sqlx::{Pool, Postgres, QueryBuilder};
use std::collections::HashMap;
use std::str::FromStr;
use time::{OffsetDateTime, PrimitiveDateTime};

type Result<T> = std::result::Result<T, Error>;

#[derive(FromRow)]
struct ConfigurationSchema {
    id: i64,
    status: i16,
    category: String,
    name: String,
    data: Json<serde_json::Value>,
    description: String,
    effective_start_time: PrimitiveDateTime,
    effective_end_time: PrimitiveDateTime,
    created: PrimitiveDateTime,
    modified: PrimitiveDateTime,
}

#[derive(Deserialize, Serialize)]
pub struct Configuration {
    pub id: i64,
    pub status: i16,
    pub category: String,
    pub name: String,
    pub data: HashMap<String, serde_json::Value>,
    pub description: String,
    pub effective_start_time: String,
    pub effective_end_time: String,
    pub created: String,
    pub modified: String,
}

impl From<ConfigurationSchema> for Configuration {
    fn from(schema: ConfigurationSchema) -> Self {
        Self {
            id: schema.id,
            status: schema.status,
            category: schema.category,
            name: schema.name,
            data: serde_json::from_value(schema.data.0).unwrap_or_default(),
            description: schema.description,
            effective_start_time: format_datetime(schema.effective_start_time),
            effective_end_time: format_datetime(schema.effective_end_time),
            created: format_datetime(schema.created),
            modified: format_datetime(schema.modified),
        }
    }
}

#[derive(Debug, Clone, Deserialize, Default)]
pub struct ConfigurationInsertParams {
    pub category: String,
    pub name: String,
    pub data: serde_json::Value,
    pub description: Option<String>,
    pub status: i16,
    pub effective_start_time: String,
    pub effective_end_time: String,
}

#[derive(Debug, Clone, Deserialize, Default)]
pub struct ConfigurationUpdateParams {
    pub data: Option<serde_json::Value>,
    pub description: Option<String>,
    pub status: Option<i16>,
    pub effective_start_time: Option<String>,
    pub effective_end_time: Option<String>,
}

#[derive(Debug, Clone, Deserialize, Default)]
pub struct AlarmConfig {
    pub category: String,
    pub url: String,
}

#[derive(Default)]
pub struct ConfigurationModel {}

impl Model for ConfigurationModel {
    type Output = Configuration;
    fn new() -> Self {
        Self::default()
    }
    async fn schema_view(&self, _pool: &Pool<Postgres>) -> SchemaView {
        SchemaView {
            schemas: vec![
                Schema::new_id(),
                Schema {
                    name: "name".to_string(),
                    category: SchemaType::String,
                    required: true,
                    read_only: true,
                    filterable: true,
                    fixed: true,
                    ..Default::default()
                },
                Schema {
                    name: "category".to_string(),
                    category: SchemaType::String,
                    required: true,
                    read_only: true,
                    filterable: true,
                    options: Some(new_schema_options(&[
                        "common",
                        "app",
                        "user",
                        "system",
                        "alarm",
                        "response_headers",
                    ])),
                    ..Default::default()
                },
                Schema::new_effective_start_time(),
                Schema::new_effective_end_time(),
                Schema {
                    name: "data".to_string(),
                    category: SchemaType::Json,
                    span: Some(2),
                    required: true,
                    popover: true,
                    ..Default::default()
                },
                Schema::new_status(),
                Schema {
                    name: "description".to_string(),
                    category: SchemaType::String,
                    ..Default::default()
                },
                Schema::new_created(),
                Schema::new_modified(),
            ],
            allow_edit: SchemaAllowEdit {
                owner: true,
                roles: vec![ROLE_SUPER_ADMIN.to_string(), ROLE_ADMIN.to_string()],
                ..Default::default()
            },
            allow_create: SchemaAllowCreate {
                roles: vec![ROLE_SUPER_ADMIN.to_string(), ROLE_ADMIN.to_string()],
                ..Default::default()
            },
        }
    }

    fn push_filter_conditions<'args>(
        &self,
        qb: &mut QueryBuilder<'args, Postgres>,
        filters: &HashMap<String, String>,
    ) -> Result<()> {
        if let Some(category) = filters.get("category") {
            qb.push(" AND category = ");
            qb.push_bind(category.clone());
        }
        Ok(())
    }
    async fn insert(&self, pool: &Pool<Postgres>, data: serde_json::Value) -> Result<u64> {
        let params: ConfigurationInsertParams = serde_json::from_value(data).context(JsonSnafu)?;
        let row: (i64,) = sqlx::query_as(
            r#"
            INSERT INTO configurations (category, name, data, description, status, effective_start_time, effective_end_time) VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING id"#,
        )
        .bind(params.category)
        .bind(params.name)
        .bind(params.data)
        .bind(params.description)
        .bind(params.status)
        .bind(parse_primitive_datetime(&params.effective_start_time)?)
        .bind(parse_primitive_datetime(&params.effective_end_time)?)
        .fetch_one(pool)
        .await
        .context(SqlxSnafu)?;

        Ok(row.0 as u64)
    }

    async fn get_by_id(&self, pool: &Pool<Postgres>, id: u64) -> Result<Option<Self::Output>> {
        let result = sqlx::query_as::<_, ConfigurationSchema>(
            r#"SELECT * FROM configurations WHERE id = $1 AND deleted_at IS NULL"#,
        )
        .bind(id as i64)
        .fetch_optional(pool)
        .await
        .context(SqlxSnafu)?;

        Ok(result.map(|schema| schema.into()))
    }

    async fn delete_by_id(&self, pool: &Pool<Postgres>, id: u64) -> Result<()> {
        sqlx::query(
            r#"UPDATE configurations SET deleted_at = CURRENT_TIMESTAMP WHERE id = $1 AND deleted_at IS NULL"#,
        )
        .bind(id as i64)
        .execute(pool)
        .await
        .context(SqlxSnafu)?;

        Ok(())
    }

    async fn update_by_id(
        &self,
        pool: &Pool<Postgres>,
        id: u64,
        data: serde_json::Value,
    ) -> Result<()> {
        let params: ConfigurationUpdateParams = serde_json::from_value(data).context(JsonSnafu)?;
        let _ = sqlx::query(
            r#"UPDATE configurations SET data = COALESCE($1, data), description = COALESCE($2, description), status = COALESCE($3, status), effective_start_time = COALESCE($4, effective_start_time), effective_end_time = COALESCE($5, effective_end_time) WHERE id = $6 AND deleted_at IS NULL"#,
        )
        .bind(params.data)
        .bind(params.description)
        .bind(params.status)
        .bind(params.effective_start_time.as_deref().map(parse_primitive_datetime).transpose()?)
        .bind(params.effective_end_time.as_deref().map(parse_primitive_datetime).transpose()?)
        .bind(id as i64)
        .execute(pool)
        .await
        .context(SqlxSnafu)?;

        Ok(())
    }

    async fn count(&self, pool: &Pool<Postgres>, params: &ModelListParams) -> Result<i64> {
        let mut qb = QueryBuilder::new("SELECT COUNT(*) FROM configurations");
        self.push_conditions(&mut qb, params)?;
        let count = qb
            .build_query_scalar::<i64>()
            .fetch_one(pool)
            .await
            .context(SqlxSnafu)?;
        Ok(count)
    }

    async fn list(
        &self,
        pool: &Pool<Postgres>,
        params: &ModelListParams,
    ) -> Result<Vec<Self::Output>> {
        let mut qb = QueryBuilder::new("SELECT * FROM configurations");
        self.push_conditions(&mut qb, params)?;
        params.push_pagination(&mut qb);
        let configurations = qb
            .build_query_as::<ConfigurationSchema>()
            .fetch_all(pool)
            .await
            .context(SqlxSnafu)?;
        Ok(configurations.into_iter().map(|s| s.into()).collect())
    }
}

impl ConfigurationModel {
    pub async fn get_response_headers(
        &self,
        pool: &Pool<Postgres>,
        name: &str,
    ) -> Result<Option<HeaderMap>> {
        let now_utc = OffsetDateTime::now_utc();
        let now = PrimitiveDateTime::new(now_utc.date(), now_utc.time());
        let configurations = sqlx::query_as::<_, ConfigurationSchema>(
            r#"SELECT * FROM configurations
               WHERE category = 'response_headers'
               AND status = $1
               AND name = $2
               AND deleted_at IS NULL
               AND effective_start_time <= $3
               AND effective_end_time >= $4"#,
        )
        .bind(Status::Enabled as i16)
        .bind(name)
        .bind(now)
        .bind(now)
        .fetch_all(pool)
        .await
        .context(SqlxSnafu)?;

        let mut headers = HeaderMap::new();

        for configuration in configurations {
            let data = configuration.data;
            let Some(data) = data.as_object() else {
                continue;
            };
            for (key, value) in data.iter() {
                let Some(value_str) = value.as_str() else {
                    continue;
                };
                let Ok(header_value) = HeaderValue::from_str(value_str) else {
                    continue;
                };
                let Ok(header_name) = HeaderName::from_str(key) else {
                    continue;
                };
                headers.insert(header_name, header_value);
            }
        }
        Ok(Some(headers))
    }
    pub async fn get_config<T>(
        &self,
        pool: &Pool<Postgres>,
        category: &str,
        name: &str,
    ) -> Result<Option<T>>
    where
        T: DeserializeOwned,
    {
        let now_utc = OffsetDateTime::now_utc();
        let now = PrimitiveDateTime::new(now_utc.date(), now_utc.time());
        let configuration = sqlx::query_as::<_, ConfigurationSchema>(
            r#"SELECT * FROM configurations
               WHERE category = $1
               AND status = $2
               AND name = $3
               AND deleted_at IS NULL
               AND effective_start_time <= $4
               AND effective_end_time >= $5"#,
        )
        .bind(category)
        .bind(Status::Enabled as i16)
        .bind(name)
        .bind(now)
        .bind(now)
        .fetch_optional(pool)
        .await
        .context(SqlxSnafu)?;

        let Some(configuration) = configuration else {
            return Ok(None);
        };
        let data = configuration.data;
        let Some(data) = data.as_object() else {
            return Err(Error::NotFound);
        };
        let data: T =
            serde_json::from_value(serde_json::Value::Object(data.clone())).context(JsonSnafu)?;
        Ok(Some(data))
    }
}