use json::{JsonValue, object};
use crate::{Field};
pub struct Int {
pub require: bool,
pub field: String,
pub mode: String,
pub title: String,
pub def: i64,
pub length: i32,
pub dec: String,
pub show: bool,
pub describe: 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(),
show: true,
describe: "".to_string(),
}
}
}
impl Field for Int {
fn sql(&mut self, model: &str) -> String {
let sql = if self.require {
format!("`{}` int({}) not null", self.field, self.length)
} else {
format!("`{}` int({}) not null default {}", self.field, self.length, self.def)
};
match model {
"sqlite" => sql,
_ => format!("{} comment '{}|{}|{}|{}|{}'", sql.clone(), self.mode, self.require, self.title, self.length, 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("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.insert("show", JsonValue::from(self.show.clone())).unwrap();
field.insert("describe", JsonValue::from(self.describe.clone())).unwrap();
field
}
fn hide(&mut self) -> &mut Self {
self.show = false;
self
}
fn describe(&mut self, text: &str) -> &mut Self {
self.describe = text.to_string();
self
}
}
pub struct Switch {
pub require: bool,
pub field: String,
pub mode: String,
pub title: String,
pub def: u8,
pub show: bool,
pub describe: String,
}
impl Switch {
pub fn new(require: bool, field: &str, title: &str, default: bool) -> Self {
let def = if default {
1
} else {
0
};
Self {
require,
field: field.to_string(),
mode: "switch".to_string(),
title: title.to_string(),
def,
show: true,
describe: "".to_string(),
}
}
}
impl Field for Switch {
fn sql(&mut self, model: &str) -> String {
match model {
"sqlite" => {
if self.require {
format!("`{}` TINYINT(1) not null", self.field)
} else {
format!("`{}` TINYINT(1) not null default {}", self.field, self.def)
}
}
_ => {
let sql = if self.require {
format!("`{}` tinyint(1) not null", self.field)
} else {
format!("`{}` tinyint(1) not null default {}", self.field, self.def)
};
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.insert("show", JsonValue::from(self.show.clone())).unwrap();
field.insert("describe", JsonValue::from(self.describe.clone())).unwrap();
field
}
fn hide(&mut self) -> &mut Self {
self.show = false;
self
}
fn describe(&mut self, text: &str) -> &mut Self {
self.describe = text.to_string();
self
}
}