Skip to main content

surql_parser/upstream/sql/statements/define/
bucket.rs

1use super::DefineKind;
2use crate::upstream::fmt::CoverStmts;
3use crate::upstream::sql::{Expr, Literal, Permission};
4use surrealdb_types::{SqlFormat, ToSql, write_sql};
5#[derive(Clone, Debug, Eq, PartialEq)]
6#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
7pub struct DefineBucketStatement {
8	pub kind: DefineKind,
9	pub name: Expr,
10	pub backend: Option<Expr>,
11	pub permissions: Permission,
12	pub readonly: bool,
13	pub comment: Expr,
14}
15impl Default for DefineBucketStatement {
16	fn default() -> Self {
17		Self {
18			kind: DefineKind::Default,
19			name: Expr::Literal(Literal::None),
20			backend: None,
21			permissions: Permission::default(),
22			readonly: false,
23			comment: Expr::Literal(Literal::None),
24		}
25	}
26}
27impl ToSql for DefineBucketStatement {
28	fn fmt_sql(&self, f: &mut String, sql_fmt: SqlFormat) {
29		write_sql!(f, sql_fmt, "DEFINE BUCKET");
30		match self.kind {
31			DefineKind::Default => {}
32			DefineKind::Overwrite => write_sql!(f, sql_fmt, " OVERWRITE"),
33			DefineKind::IfNotExists => write_sql!(f, sql_fmt, " IF NOT EXISTS"),
34		}
35		write_sql!(f, sql_fmt, " {}", CoverStmts(&self.name));
36		if self.readonly {
37			write_sql!(f, sql_fmt, " READONLY");
38		}
39		if let Some(ref backend) = self.backend {
40			write_sql!(f, sql_fmt, " BACKEND {}", CoverStmts(backend));
41		}
42		write_sql!(f, sql_fmt, " PERMISSIONS {}", self.permissions);
43		if !matches!(self.comment, Expr::Literal(Literal::None)) {
44			write_sql!(f, sql_fmt, " COMMENT {}", CoverStmts(&self.comment));
45		}
46	}
47}