reifydb-engine 0.10.0

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

use arrow_array::RecordBatch;
use reifydb_catalog::catalog::namespace::NamespaceToCreate;
use reifydb_core::{
	interface::catalog::{change::CatalogTrackNamespaceChangeOperations, id::NamespaceId},
	value::batch::single_row,
};
use reifydb_rql::nodes::CreateRemoteNamespaceNode;
use reifydb_transaction::transaction::{Transaction, admin::AdminTransaction};
use reifydb_value::value::Value;

use crate::{Result, vm::services::Services};

pub(crate) fn create_remote_namespace(
	services: &Services,
	txn: &mut AdminTransaction,
	plan: CreateRemoteNamespaceNode,
) -> Result<RecordBatch> {
	let full_name: String = plan.segments.iter().map(|s| s.text()).collect::<Vec<_>>().join("::");

	let mut parent_id = NamespaceId::ROOT;
	for i in 0..plan.segments.len().saturating_sub(1) {
		let prefix: String = plan.segments[..=i].iter().map(|s| s.text()).collect::<Vec<_>>().join("::");
		if let Some(existing) =
			services.catalog.find_namespace_by_name(&mut Transaction::Admin(txn), &prefix)?
		{
			parent_id = existing.id();
		} else {
			let result = services.catalog.create_namespace(
				txn,
				NamespaceToCreate {
					namespace_fragment: Some(plan.segments[i].clone()),
					name: prefix,
					local_name: plan.segments[i].text().to_string(),
					parent_id,
					grpc: None,
					token: None,
				},
			)?;
			txn.track_namespace_created(result.clone())?;
			parent_id = result.id();
		}
	}

	if services.catalog.find_namespace_by_name(&mut Transaction::Admin(txn), &full_name)?.is_some()
		&& plan.if_not_exists
	{
		return single_row([("namespace", Value::Utf8(full_name)), ("created", Value::Boolean(false))]);
	}

	let grpc_text = plan.grpc.text().to_string();
	let token_text = plan.token.as_ref().map(|t| t.text().to_string());
	let result = services.catalog.create_namespace(
		txn,
		NamespaceToCreate {
			namespace_fragment: plan.segments.last().cloned(),
			name: full_name,
			local_name: plan.segments.last().unwrap().text().to_string(),
			parent_id,
			grpc: Some(grpc_text),
			token: token_text,
		},
	)?;
	txn.track_namespace_created(result.clone())?;

	single_row([("namespace", Value::Utf8(result.name().to_string())), ("created", Value::Boolean(true))])
}

#[cfg(test)]
pub mod tests {
	use reifydb_test_harness::engine::create_test_admin_transaction;
	use reifydb_transaction::transaction::query::QueryTransaction;
	use reifydb_value::{
		params::Params,
		value::{
			Value, column_view::ColumnView, frame::frame::Frame, identity::IdentityId,
			system_columns::user_columns,
		},
	};

	use crate::vm::{Admin, Query, executor::Executor};

	#[test]
	fn test_create_remote_namespace() {
		let instance = Executor::testing();
		let mut txn = create_test_admin_transaction();

		let r = instance.admin(
			&mut txn,
			Admin {
				rql: "CREATE REMOTE NAMESPACE remote_ns WITH { grpc: 'localhost:50051' }",
				params: Params::default(),
			},
		);
		if let Some(e) = r.error {
			panic!("{e:?}");
		}
		let frame = &r[0];

		assert_eq!(value_at(frame, 0), Value::Utf8("remote_ns".to_string()));
		assert_eq!(value_at(frame, 1), Value::Boolean(true));
	}

	#[test]
	fn test_create_remote_namespace_if_not_exists() {
		let instance = Executor::testing();
		let mut txn = create_test_admin_transaction();

		let r = instance.admin(
			&mut txn,
			Admin {
				rql: "CREATE REMOTE NAMESPACE remote_ns WITH { grpc: 'localhost:50051' }",
				params: Params::default(),
			},
		);
		if let Some(e) = r.error {
			panic!("{e:?}");
		}

		// IF NOT EXISTS must report created=false rather than fault or re-register the endpoint.
		let r = instance.admin(
			&mut txn,
			Admin {
				rql: "CREATE REMOTE NAMESPACE IF NOT EXISTS remote_ns WITH { grpc: 'localhost:50051' }",
				params: Params::default(),
			},
		);
		if let Some(e) = r.error {
			panic!("{e:?}");
		}
		let frame = &r[0];
		assert_eq!(value_at(frame, 0), Value::Utf8("remote_ns".to_string()));
		assert_eq!(value_at(frame, 1), Value::Boolean(false));
	}

	#[test]
	fn test_query_remote_namespace_without_registry_returns_empty() {
		let instance = Executor::testing();
		let mut txn = create_test_admin_transaction();
		let identity = IdentityId::root();

		let r = instance.admin(
			&mut txn,
			Admin {
				rql: "CREATE REMOTE NAMESPACE remote_ns WITH { grpc: 'http://localhost:50051' }",
				params: Params::default(),
			},
		);
		if let Some(e) = r.error {
			panic!("{e:?}");
		}
		txn.commit().unwrap();

		// A missing registry must degrade to empty rather than panic or dial the endpoint.
		let mut qt =
			QueryTransaction::new(txn.multi.clone().begin_query().unwrap(), txn.single.clone(), identity);
		let r = instance.query(
			&mut qt,
			Query {
				rql: "FROM remote_ns::some_table",
				params: Params::default(),
			},
		);
		if let Some(e) = r.error {
			panic!("{e:?}");
		}

		assert!(r.is_empty() || r.iter().all(|f| user_columns(&f.batch).next().is_none()));
	}

	#[test]
	fn test_create_remote_namespace_nested() {
		let instance = Executor::testing();
		let mut txn = create_test_admin_transaction();

		let r = instance.admin(
			&mut txn,
			Admin {
				rql: "CREATE REMOTE NAMESPACE blockchain::protocol WITH { grpc: '10.0.0.5:50051' }",
				params: Params::default(),
			},
		);
		if let Some(e) = r.error {
			panic!("{e:?}");
		}
		let frame = &r[0];

		assert_eq!(value_at(frame, 0), Value::Utf8("blockchain::protocol".to_string()));
		assert_eq!(value_at(frame, 1), Value::Boolean(true));
	}

	fn value_at(frame: &Frame, column: usize) -> Value {
		// Positional read: without it a reordered result column would still pass.
		ColumnView::try_from((frame.batch.column(column), frame.batch.schema_ref().field(column)))
			.unwrap()
			.get_value(0)
	}
}