reifydb-rql 0.4.9

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

//! Maybe-qualified identifier types for AST
//! These types allow optional qualification as they come directly from user
//! input

use crate::{
	bump::BumpFragment,
	token::token::{Token, TokenKind},
};

/// Represents a source identifier that hasn't been resolved to a specific type yet
/// Used in AST parsing before we know whether it's a table, view, or ring buffer
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct UnresolvedShapeIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
	pub alias: Option<BumpFragment<'bump>>,
}

impl<'bump> UnresolvedShapeIdentifier<'bump> {
	pub fn new(namespace: Vec<BumpFragment<'bump>>, name: BumpFragment<'bump>) -> Self {
		Self {
			namespace,
			name,
			alias: None,
		}
	}

	pub fn with_alias(mut self, alias: BumpFragment<'bump>) -> Self {
		self.alias = Some(alias);
		self
	}

	pub fn effective_name(&self) -> &str {
		self.alias.as_ref().map(|a| a.text()).unwrap_or_else(|| self.name.text())
	}
}

/// An unqualified identifier that hasn't been parsed for qualification yet.
/// This is used in the AST for simple identifiers before they're resolved
/// to specific types (column, table, namespace, etc.)
#[derive(Debug, Clone, Copy, PartialEq)]
pub struct UnqualifiedIdentifier<'bump> {
	pub token: Token<'bump>,
}

impl<'bump> UnqualifiedIdentifier<'bump> {
	pub fn new(token: Token<'bump>) -> Self {
		Self {
			token,
		}
	}

	pub fn from_fragment(fragment: BumpFragment<'bump>) -> Self {
		Self {
			token: Token {
				kind: TokenKind::Identifier,
				fragment,
			},
		}
	}

	pub fn text(&self) -> &str {
		self.token.fragment.text()
	}

	pub fn fragment(&self) -> &BumpFragment<'bump> {
		&self.token.fragment
	}

	pub fn into_fragment(self) -> BumpFragment<'bump> {
		self.token.fragment
	}
}

/// Maybe-qualified namespace identifier - segments of a dot-separated path
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedNamespaceIdentifier<'bump> {
	pub segments: Vec<BumpFragment<'bump>>,
}

impl<'bump> MaybeQualifiedNamespaceIdentifier<'bump> {
	pub fn new(segments: Vec<BumpFragment<'bump>>) -> Self {
		Self {
			segments,
		}
	}
}

/// Maybe-qualified table identifier for tables - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedTableIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
	pub alias: Option<BumpFragment<'bump>>,
}

impl<'bump> MaybeQualifiedTableIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
			alias: None,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}

	pub fn with_alias(mut self, alias: BumpFragment<'bump>) -> Self {
		self.alias = Some(alias);
		self
	}
}

/// Maybe-qualified deferred view identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedDeferredViewIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
	pub alias: Option<BumpFragment<'bump>>,
}

impl<'bump> MaybeQualifiedDeferredViewIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
			alias: None,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}

	pub fn with_alias(mut self, alias: BumpFragment<'bump>) -> Self {
		self.alias = Some(alias);
		self
	}
}

/// Maybe-qualified transactional view identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedTransactionalViewIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
	pub alias: Option<BumpFragment<'bump>>,
}

impl<'bump> MaybeQualifiedTransactionalViewIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
			alias: None,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}

	pub fn with_alias(mut self, alias: BumpFragment<'bump>) -> Self {
		self.alias = Some(alias);
		self
	}
}

/// Maybe-qualified view identifier (generic) - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedViewIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
	pub alias: Option<BumpFragment<'bump>>,
}

impl<'bump> MaybeQualifiedViewIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
			alias: None,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}

	pub fn with_alias(mut self, alias: BumpFragment<'bump>) -> Self {
		self.alias = Some(alias);
		self
	}
}

/// Maybe-qualified ring buffer identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedRingBufferIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
	pub alias: Option<BumpFragment<'bump>>,
}

impl<'bump> MaybeQualifiedRingBufferIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
			alias: None,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}

	pub fn with_alias(mut self, alias: BumpFragment<'bump>) -> Self {
		self.alias = Some(alias);
		self
	}
}

/// Maybe-qualified dictionary identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedDictionaryIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedDictionaryIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified sum type identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedSumTypeIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedSumTypeIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified series identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedSeriesIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedSeriesIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified sequence identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedSequenceIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedSequenceIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified index identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedIndexIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub table: BumpFragment<'bump>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedIndexIdentifier<'bump> {
	pub fn new(table: BumpFragment<'bump>, name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			table,
			name,
		}
	}

	pub fn with_shape(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// How a maybe-qualified column is referenced
#[derive(Debug, Clone, PartialEq)]
pub enum MaybeQualifiedColumnShape<'bump> {
	/// Qualified by shape name (table/view) - namespace still optional
	Qualified {
		namespace: Vec<BumpFragment<'bump>>,
		name: BumpFragment<'bump>,
	},
	/// Qualified by alias
	Alias(BumpFragment<'bump>),
	/// Not qualified (needs resolution based on context)
	Unqualified,
}

/// Maybe-qualified column identifier - shape qualification is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedColumnIdentifier<'bump> {
	pub shape: MaybeQualifiedColumnShape<'bump>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedColumnIdentifier<'bump> {
	pub fn unqualified(name: BumpFragment<'bump>) -> Self {
		Self {
			shape: MaybeQualifiedColumnShape::Unqualified,
			name,
		}
	}

	pub fn with_shape(
		namespace: Vec<BumpFragment<'bump>>,
		shape: BumpFragment<'bump>,
		name: BumpFragment<'bump>,
	) -> Self {
		Self {
			shape: MaybeQualifiedColumnShape::Qualified {
				namespace,
				name: shape,
			},
			name,
		}
	}

	pub fn with_alias(alias: BumpFragment<'bump>, name: BumpFragment<'bump>) -> Self {
		Self {
			shape: MaybeQualifiedColumnShape::Alias(alias),
			name,
		}
	}
}

/// Maybe-qualified function identifier - namespaces can be partial
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedFunctionIdentifier<'bump> {
	/// Namespace chain (may be empty or partial)
	pub namespaces: Vec<BumpFragment<'bump>>,
	/// Function name
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedFunctionIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespaces: Vec::new(),
			name,
		}
	}

	pub fn with_namespaces(mut self, namespaces: Vec<BumpFragment<'bump>>) -> Self {
		self.namespaces = namespaces;
		self
	}
}

/// Maybe-qualified procedure identifier
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedProcedureIdentifier<'bump> {
	/// Namespace chain (may be empty or partial)
	pub namespace: Vec<BumpFragment<'bump>>,
	/// Procedure name
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedProcedureIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified test identifier
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedTestIdentifier<'bump> {
	/// Namespace chain (may be empty or partial)
	pub namespace: Vec<BumpFragment<'bump>>,
	/// Test name
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedTestIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Generic maybe-qualified identifier - namespace is optional.
/// Used for targets/sources in CREATE SOURCE/SINK where the entity type is not known at parse time.
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified source identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedSourceIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedSourceIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}

/// Maybe-qualified sink identifier - namespace is optional
#[derive(Debug, Clone, PartialEq)]
pub struct MaybeQualifiedSinkIdentifier<'bump> {
	pub namespace: Vec<BumpFragment<'bump>>,
	pub name: BumpFragment<'bump>,
}

impl<'bump> MaybeQualifiedSinkIdentifier<'bump> {
	pub fn new(name: BumpFragment<'bump>) -> Self {
		Self {
			namespace: Vec::new(),
			name,
		}
	}

	pub fn with_namespace(mut self, namespace: Vec<BumpFragment<'bump>>) -> Self {
		self.namespace = namespace;
		self
	}
}