1use std::fmt;
2
3use surrealdb_types::{SqlFormat, ToSql, write_sql};
4
5#[derive(Clone, Debug, Default, Eq, PartialEq, PartialOrd, Hash)]
6#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
7pub enum Base {
8 #[default]
9 Root,
10 Ns,
11 Db,
12}
13
14impl fmt::Display for Base {
15 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
16 match self {
17 Self::Ns => f.write_str("NAMESPACE"),
18 Self::Db => f.write_str("DATABASE"),
19 Self::Root => f.write_str("ROOT"),
20 }
21 }
22}
23
24impl ToSql for Base {
25 fn fmt_sql(&self, f: &mut String, _fmt: SqlFormat) {
26 match self {
27 Self::Ns => write_sql!(f, sql_fmt, "NAMESPACE"),
28 Self::Db => write_sql!(f, sql_fmt, "DATABASE"),
29 Self::Root => write_sql!(f, sql_fmt, "ROOT"),
30 }
31 }
32}