use crate::Field;
use json::{object, JsonValue};
pub struct Dict {
pub require: bool,
pub field: String,
pub mode: String,
pub title: String,
pub table: String,
pub fields: String,
pub api: String,
pub def: String,
pub show: bool,
pub describe: String,
pub multiple: bool,
pub multiple_count: i32,
pub example: JsonValue,
}
impl Dict {
pub fn new(
require: bool,
field: &str,
title: &str,
table: &str,
fields: &str,
api: &str,
) -> Self {
Self {
require,
field: field.to_string(),
mode: "dict".to_string(),
title: title.to_string(),
table: table.to_string(),
api: api.to_string(),
show: true,
fields: fields.to_string(),
describe: "".to_string(),
def: "".to_string(),
multiple: false,
multiple_count: 1,
example: JsonValue::Null,
}
}
pub fn multiple(&mut self, count: i32) -> &mut Dict {
self.multiple = count > 1;
self.multiple_count = count;
self
}
pub fn default(&mut self, def: &str) -> &mut Dict {
self.def = def.to_string();
self
}
}
impl Field for Dict {
fn sql(&mut self, model: &str) -> String {
let not_null = if self.require { " not null" } else { "" };
let length = if self.multiple {
64 * self.multiple_count + self.multiple_count
} else {
64
};
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.mode,
self.title,
self.require,
self.multiple,
self.multiple_count,
self.def
)
}
_ => {
let sql = format!(
"`{}` varchar({}){} default '{}'",
self.field, length, not_null, self.def
);
format!(
"{} comment '{}|{}|{}|{}|{}|{}'",
sql.clone(),
self.mode,
self.title,
self.require,
self.multiple,
self.multiple_count,
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("table", JsonValue::from(self.table.clone()))
.unwrap();
field
.insert("api", JsonValue::from(self.api.clone()))
.unwrap();
field
.insert("fields", JsonValue::from(self.fields.clone()))
.unwrap();
field
.insert("def", JsonValue::from(self.def.clone()))
.unwrap();
field
.insert("multiple", JsonValue::from(self.multiple))
.unwrap();
field
.insert("multiple_count", JsonValue::from(self.multiple_count))
.unwrap();
field.insert("show", JsonValue::from(self.show)).unwrap();
field
.insert("describe", JsonValue::from(self.describe.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
}
}