reifydb-catalog 0.4.11

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

use reifydb_core::{
	interface::catalog::id::{PrimaryKeyId, TableId},
	key::table::TableKey,
	return_internal_error,
};
use reifydb_transaction::transaction::admin::AdminTransaction;

use crate::{CatalogStore, Result, store::table::shape::table};

impl CatalogStore {
	/// Set the primary key ID for a table
	/// Returns an internal error if the table doesn't exist
	pub(crate) fn set_table_primary_key(
		txn: &mut AdminTransaction,
		table_id: TableId,
		primary_key_id: PrimaryKeyId,
	) -> Result<()> {
		let multi = match txn.get(&TableKey::encoded(table_id))? {
			Some(v) => v,
			None => return_internal_error!(format!(
				"Table with ID {} not found when setting primary key. This indicates a critical catalog inconsistency.",
				table_id.0
			)),
		};

		let mut updated_row = multi.row.clone();
		table::SHAPE.set_u64(&mut updated_row, table::PRIMARY_KEY, primary_key_id.0);

		txn.set(&TableKey::encoded(table_id), updated_row)?;

		Ok(())
	}
}

#[cfg(test)]
pub mod tests {
	use reifydb_core::interface::catalog::id::{PrimaryKeyId, TableId};
	use reifydb_engine::test_harness::create_test_admin_transaction;

	use crate::{CatalogStore, test_utils::ensure_test_table};

	#[test]
	fn test_set_table_primary_key() {
		let mut txn = create_test_admin_transaction();
		let table = ensure_test_table(&mut txn);

		// Set primary key
		CatalogStore::set_table_primary_key(&mut txn, table.id, PrimaryKeyId(42)).unwrap();

		// The test succeeds if no error is thrown.
		// In real usage, create_primary_key would create both the
		// PrimaryKey record and update the table, and
		// find_primary_key would find it.
	}

	#[test]
	fn test_set_table_primary_key_nonexistent() {
		let mut txn = create_test_admin_transaction();

		// Try to set primary key on non-existent table
		let result = CatalogStore::set_table_primary_key(&mut txn, TableId(999), PrimaryKeyId(1));

		assert!(result.is_err());
		let err = result.unwrap_err();
		assert!(err.to_string().contains("Table with ID 999 not found"));
		assert!(err.to_string().contains("critical catalog inconsistency"));
	}
}