surrealdb-core 3.2.3

use anyhow::Result;
use reblessive::tree::Stk;

use crate::catalog::Index;
use crate::catalog::providers::{DatabaseProvider, TableProvider};
use crate::ctx::FrozenContext;
use crate::dbs::Options;
use crate::doc::CursorDoc;
use crate::err::Error;
use crate::expr::parameterize::expr_to_ident;
use crate::expr::{Base, Expr, Literal, Value};
use crate::iam::{Action, ResourceKind};

#[derive(Clone, Debug, Eq, PartialEq, Hash)]
pub(crate) struct RemoveAnalyzerStatement {
	pub name: Expr,
	pub if_exists: bool,
}

impl Default for RemoveAnalyzerStatement {
	fn default() -> Self {
		Self {
			name: Expr::Literal(Literal::None),
			if_exists: false,
		}
	}
}

impl RemoveAnalyzerStatement {
	pub(crate) async fn compute(
		&self,
		stk: &mut Stk,
		ctx: &FrozenContext,
		opt: &Options,
		doc: Option<&CursorDoc>,
	) -> Result<Value> {
		let (ns, db) = ctx.expect_ns_db_ids(opt).await?;
		// Allowed to run?
		ctx.is_allowed(opt, Action::Edit, ResourceKind::Analyzer, Base::Db)?;
		// Compute the name
		let name = expr_to_ident(stk, ctx, opt, doc, &self.name, "analyzer name").await?;
		// Get the transaction
		let txn = ctx.tx();
		// Get the definition
		let az = txn.get_db_analyzer(ns, db, &name, None).await;
		let az = match az {
			Ok(x) => x,
			Err(e) => {
				if self.if_exists && matches!(e.downcast_ref(), Some(Error::AzNotFound { .. })) {
					return Ok(Value::None);
				} else {
					return Err(e);
				}
			}
		};
		// Full-text indexes load their analyzer via `FullTextIndex::new`, which
		// fails with AzNotFound if the analyzer is gone — refuse removal while
		// any index still references it.
		for tb in txn.all_tb(ns, db, None).await?.iter() {
			for ix in txn.all_tb_indexes(ns, db, &tb.name, None).await?.iter() {
				if let Index::FullText(p) = &ix.index
					&& p.analyzer.as_str() == az.name.as_str()
				{
					return Err(anyhow::Error::new(Error::AzInUse {
						name: az.name.to_string(),
						table: tb.name.to_string(),
						index: ix.name.to_string(),
					}));
				}
			}
		}
		// Delete the definition
		let key = crate::key::database::az::new(ns, db, &az.name);
		txn.del(&key).await?;
		// Clear the cache
		txn.clear_cache();
		// Cleanup in-memory mappers if not used anymore
		let azs = txn.all_db_analyzers(ns, db, None).await?;
		ctx.get_index_stores().mappers().cleanup(&azs);
		// Ok all good
		Ok(Value::None)
	}
}