use revision::revisioned;
use serde::{Deserialize, Serialize};
use surrealdb_strand::Strand;
use surrealdb_types::{SqlFormat, ToSql};
use crate::catalog::Permission;
use crate::expr::statements::info::InfoStructure;
use crate::kvs::impl_kv_value_revisioned;
use crate::sql;
use crate::sql::statements::define::{DefineBucketStatement, DefineKind};
use crate::val::Value;
#[revisioned(revision = 1)]
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
pub struct BucketId(pub u32);
#[revisioned(revision = 1)]
#[derive(Clone, Debug, Default, Eq, PartialEq, Hash)]
pub struct BucketDefinition {
pub(crate) id: Option<BucketId>,
pub(crate) name: Strand,
pub(crate) backend: Option<Strand>,
pub(crate) permissions: Permission,
pub(crate) readonly: bool,
pub(crate) comment: Option<String>,
}
impl_kv_value_revisioned!(BucketDefinition);
impl BucketDefinition {
pub fn to_sql_definition(&self) -> DefineBucketStatement {
DefineBucketStatement {
kind: DefineKind::Default,
name: sql::Expr::Idiom(sql::Idiom::field(self.name.clone())),
backend: self.backend.clone().map(|v| sql::Expr::Literal(sql::Literal::String(v))),
permissions: self.permissions.clone().into(),
readonly: self.readonly,
comment: self
.comment
.clone()
.map(|v| sql::Expr::Literal(sql::Literal::String(v.into())))
.unwrap_or(sql::Expr::Literal(sql::Literal::None)),
}
}
}
impl InfoStructure for BucketDefinition {
fn structure(self) -> Value {
Value::from(map! {
"name" => self.name.into(),
"permissions" => self.permissions.structure(),
"backend", if let Some(backend) = self.backend => backend.into(),
"readonly" => self.readonly.into(),
"comment", if let Some(comment) = self.comment => comment.into(),
})
}
}
impl ToSql for BucketDefinition {
fn fmt_sql(&self, f: &mut String, fmt: SqlFormat) {
self.to_sql_definition().fmt_sql(f, fmt)
}
}