reifydb-engine 0.4.11

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

use reifydb_catalog::error::{CatalogError, CatalogObjectKind};
use reifydb_core::value::column::columns::Columns;
use reifydb_rql::nodes::DropDictionaryNode;
use reifydb_transaction::transaction::{Transaction, admin::AdminTransaction};
use reifydb_type::value::Value;

use super::dependent::find_column_dependents;
use crate::{Result, vm::services::Services};

pub(crate) fn drop_dictionary(
	services: &Services,
	txn: &mut AdminTransaction,
	plan: DropDictionaryNode,
) -> Result<Columns> {
	let Some(dictionary_id) = plan.dictionary_id else {
		return Ok(Columns::single_row([
			("namespace", Value::Utf8(plan.namespace_name.text().to_string())),
			("dictionary", Value::Utf8(plan.dictionary_name.text().to_string())),
			("dropped", Value::Boolean(false)),
		]));
	};

	let def = services.catalog.get_dictionary(&mut Transaction::Admin(txn), dictionary_id)?;

	// Check for dependent columns across all entity types
	let columns = services.catalog.list_columns_all(&mut Transaction::Admin(txn))?;
	let dependents = find_column_dependents(&services.catalog, txn, &columns, |info| {
		(info.column.dictionary_id == Some(dictionary_id)).then(String::new)
	})?;
	if !dependents.is_empty() {
		let dependents_str = dependents.join(", ");
		return Err(CatalogError::InUse {
			kind: CatalogObjectKind::Dictionary,
			namespace: plan.namespace_name.text().to_string(),
			name: Some(plan.dictionary_name.text().to_string()),
			dependents: dependents_str,
			fragment: plan.dictionary_name.clone(),
		}
		.into());
	}

	services.catalog.drop_dictionary(txn, def)?;

	Ok(Columns::single_row([
		("namespace", Value::Utf8(plan.namespace_name.text().to_string())),
		("dictionary", Value::Utf8(plan.dictionary_name.text().to_string())),
		("dropped", Value::Boolean(true)),
	]))
}