df-fields 0.1.16

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

/// 数字
///
/// * require 是否必填
/// * field 字段名
/// * mode 模式 string
/// * title 字段描述
/// * length 字段总长度(含小数位)
/// * def 默认值
/// * dec 符号
pub struct Int {
    pub require: bool,
    pub field: String,
    pub mode: String,
    pub title: String,
    pub def: i64,
    pub length: i32,
    pub dec: String,
}

impl Int {
    pub fn new(require: bool, field: &str, title: &str, length: i32, default: i64) -> Self {
        Self {
            require,
            field: field.to_string(),
            mode: "int".to_string(),
            title: title.to_string(),
            def: default,
            length,
            dec: "".to_string(),
        }
    }
}

impl Field for Int {
    fn sql(&mut self, model: &str) -> String {
        let mut sql = format!("{} int({})", self.field, self.length);
        if self.require {
            sql = format!("{} not null", sql.clone())
        };
        sql = format!("{} default {}", sql.clone(), self.def);
        match model {
            "sqlite" => sql,
            _ => format!("{} comment '{}|{}|{}|{}|{}|{}'", sql.clone(), self.mode, self.require, self.title, self.length, self.def, self.dec)
        }
    }
    fn field(&mut self) -> JsonValue {
        let mut field = object! {};
        field.insert("require", JsonValue::from(self.require.clone())).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.clone())).unwrap();
        field.insert("def", JsonValue::from(self.def.clone())).unwrap();
        field.insert("dec", JsonValue::from(self.dec.clone())).unwrap();
        field
    }
}


/// 自增主键
///
/// * field 字段名
/// * mode 模式
/// * title 字段描述
/// * length 字段总长度(含小数位)
/// * def 默认值
/// * dec 无效
pub struct AutoKey {
    pub require: bool,
    pub field: String,
    pub mode: String,
    pub title: String,
    pub length: i32,
}

impl AutoKey {
    pub fn new(require: bool, field: &str, title: &str) -> Self {
        Self {
            require,
            field: field.to_string(),
            mode: "autokey".to_string(),
            title: title.to_string(),
            length: 0,
        }
    }
}

impl Field for AutoKey {
    fn sql(&mut self, model: &str) -> String {
        let mut sql = "".to_string();
        //PRIMARY KEY
        match model {
            "sqlite" => {
                sql = format!("{}{} INTEGER PRIMARY KEY", sql.clone(), self.field);
                sql = format!("{} AUTOINCREMENT", sql.clone());
                return sql;
            }
            _ => {
                sql = format!("{}{} int({})", sql.clone(), self.field, self.length);
                sql = format!("{} not null AUTO_INCREMENT", sql.clone());
                format!("{} comment '{}|{}|{}|{}'", sql.clone(), self.mode, self.require, self.title, self.length)
            }
        }
    }
    fn field(&mut self) -> JsonValue {
        let mut field = object! {};
        field.insert("require", JsonValue::from(self.require.clone())).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.clone())).unwrap();
        field
    }
}


/// 开关
///
/// * field 字段名
/// * mode 模式 string
/// * title 字段描述
/// * default 默认值
/// * empty 是否可空
pub struct Switch {
    pub require: bool,
    pub field: String,
    pub mode: String,
    pub title: String,
    pub def: bool,
}

impl Switch {
    pub fn new(require: bool, field: &str, title: &str, default: bool) -> Self {
        Self {
            require,
            field: field.to_string(),
            mode: "switch".to_string(),
            title: title.to_string(),
            def: default,
        }
    }
}

impl Field for Switch {
    fn sql(&mut self, model: &str) -> String {
        let mut sql = format!("{} bool", self.field);
        if self.require {
            sql = format!("{} not null", sql.clone())
        };
        sql = format!("{} default {}", sql.clone(), self.def);
        match model {
            "sqlite" => sql,
            _ => format!("{} comment '{}|{}|{}|{}'", sql.clone(), self.mode, self.require, self.title, self.def)
        }
    }
    fn field(&mut self) -> JsonValue {
        let mut field = object! {};
        field.insert("require", JsonValue::from(self.require.clone())).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("def", JsonValue::from(self.def.clone())).unwrap();
        field
    }
}