surrealdb-core 3.3.1

A scalable, distributed, collaborative, document-graph database, for the realtime web
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use std::ops::Deref;

use anyhow::Result;

use crate::catalog::providers::DatabaseProvider;
use crate::ctx::FrozenContext;
use crate::dbs::Options;
use crate::expr::Base;
use crate::expr::statements::alter::AlterKind;
use crate::expr::statements::alter::module::AlterModuleStatement;
use crate::iam::{Action, ResourceKind};
use crate::val::Value;

#[instrument(level = "trace", name = "AlterModuleStatement::compute", skip_all)]
pub(crate) async fn alter_module_statement_compute(
	this: &AlterModuleStatement,
	ctx: &FrozenContext,
	opt: &Options,
) -> Result<Value> {
	ctx.is_allowed(opt, Action::Edit, ResourceKind::Module, Base::Db)?;
	let (_, _) = opt.ns_db()?;
	let (ns, db) = ctx.expect_ns_db_ids(opt).await?;
	let txn = ctx.tx();

	let storage_name = this.name.get_storage_name();
	let mut md = match txn.get_db_module(ns, db, &storage_name, None).await {
		Ok(v) => v.deref().clone(),
		Err(e) => {
			if this.if_exists {
				return Ok(Value::None);
			}
			return Err(e);
		}
	};

	match this.comment {
		AlterKind::Set(ref v) => md.comment = Some(v.clone()),
		AlterKind::Drop => md.comment = None,
		AlterKind::None => {}
	}

	if let Some(ref p) = this.permissions {
		md.permissions = p.clone();
	}

	txn.put_db_module(ns, db, &md).await?;
	txn.clear_cache();
	Ok(Value::None)
}