br-fields 2.3.5

This is a shortcut tool related to database fields
Documentation
use crate::Field;
use json::{object, JsonValue};

/// 附件
///
/// * require 是否必填
/// * field 字段名
/// * mode 模式 file
/// * title 字段描述
/// * length 附件数量
/// * def 默认值
pub struct Files {
    pub require: bool,
    pub field: String,
    pub mode: String,
    pub title: String,
    pub length: i32,
    /// 加密
    pub encrypt: bool,
    /// 文件尺寸
    pub size: usize,
    /// 运行的文件类型
    pub option: Vec<String>,
    pub show: bool,
    pub describe: String,
    /// 是否上传
    pub upload: bool,
    pub def: String,
    pub example: JsonValue,
}

impl Files {
    pub fn new(require: bool, field: &str, title: &str) -> Self {
        Self {
            require,
            field: field.to_string(),
            mode: "file".to_string(),
            title: title.to_string(),
            length: 1,
            encrypt: false,
            size: 0,
            option: vec![],
            def: "".to_string(),
            show: true,
            describe: "".to_string(),
            example: JsonValue::Null,
            upload: false,
        }
    }

    pub fn encrypt(&mut self) -> &mut Self {
        self.encrypt = true;
        self
    }
    pub fn multiple(&mut self, number: i32) -> &mut Self {
        self.length = number;
        self
    }
    pub fn option(&mut self, option: Vec<&str>) -> &mut Self {
        self.option = option.iter().map(|c| c.to_string()).collect();
        self
    }
    pub fn size(&mut self, size: usize) -> &mut Self {
        self.size = if size == 0 { 1024 * 1024 } else { size };
        self
    }
    pub fn def(&mut self, def: &str) -> &mut Self {
        self.def = def.to_string();
        self
    }
    pub fn upload(&mut self, data: bool) -> &mut Self {
        self.upload = data;
        self
    }
}

impl Field for Files {
    fn sql(&mut self, model: &str) -> String {
        let not_null = if self.require { " not null" } else { "" };
        let length = self.length * 64 + self.length;
        match model {
            "sqlite" => format!(
                "`{}` varchar({}){} default '{}'",
                self.field, length, not_null, self.def
            ),
            "pgsql" => {
                let sql = format!(
                    r#""{}" varchar({}) default '{}'"#,
                    self.field, length, self.def
                );
                format!(
                    "{} --{}|{}|{}|{}|{}",
                    sql, self.title, self.mode, self.require, self.length, self.def
                )
            }
            _ => {
                let sql = format!(
                    "`{}` varchar({}){} default '{}'",
                    self.field, length, not_null, self.def
                );
                format!(
                    "{} comment '{}|{}|{}|{}|{}'",
                    sql.clone(),
                    self.title,
                    self.mode,
                    self.require,
                    self.length,
                    self.def
                )
            }
        }
    }
    fn hide(&mut self) -> &mut Self {
        self.show = false;
        self
    }
    fn describe(&mut self, text: &str) -> &mut Self {
        self.describe = text.to_string();
        self
    }

    fn field(&mut self) -> JsonValue {
        let mut field = object! {};
        field
            .insert("require", JsonValue::from(self.require))
            .unwrap();
        field
            .insert("field", JsonValue::from(self.field.clone()))
            .unwrap();
        field
            .insert("mode", JsonValue::from(self.mode.clone()))
            .unwrap();
        field
            .insert("title", JsonValue::from(self.title.clone()))
            .unwrap();
        field
            .insert("length", JsonValue::from(self.length))
            .unwrap();
        field
            .insert("encrypt", JsonValue::from(self.encrypt))
            .unwrap();
        field.insert("size", JsonValue::from(self.size)).unwrap();
        field
            .insert("option", JsonValue::from(self.option.clone()))
            .unwrap();
        field.insert("show", JsonValue::from(self.show)).unwrap();
        field
            .insert("describe", JsonValue::from(self.describe.clone()))
            .unwrap();
        field
            .insert("upload", JsonValue::from(self.upload))
            .unwrap();
        field
            .insert("def", JsonValue::from(self.def.clone()))
            .unwrap();
        field.insert("example", self.example.clone()).unwrap();
        field
    }

    fn swagger(&mut self) -> JsonValue {
        object! {
            "type": self.mode.clone(),
            "example": self.example.clone(),
        }
    }

    fn example(&mut self, data: JsonValue) -> &mut Self {
        self.example = data.clone();
        self
    }
}

#[cfg(test)]
mod test {
    use crate::Field;
    #[test]
    fn test() {
        let res = crate::files::Files::new(true, "str", "附件")
            .describe("1313")
            .upload(true)
            .field();
        println!("{res:#}");
    }
}