use crate::Field;
use json::{object, JsonValue};
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:#}");
}
}