use crate::alter_column::AlterColumnType;
use crate::common::{ColumnDefinition, FQName, Identifier};
use itertools::Itertools;
use std::fmt::{Display, Formatter};
#[derive(PartialEq, Debug, Clone)]
pub struct AlterType {
pub name: FQName,
pub operation: AlterTypeOperation,
}
impl Display for AlterType {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
write!(f, "ALTER TYPE {} {}", self.name, self.operation)
}
}
#[derive(PartialEq, Debug, Clone)]
pub enum AlterTypeOperation {
AlterColumnType(AlterColumnType),
Add(Vec<ColumnDefinition>),
Rename(Vec<(Identifier, Identifier)>),
}
impl Display for AlterTypeOperation {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
match self {
AlterTypeOperation::AlterColumnType(column_type) => write!(f, "{}", column_type),
AlterTypeOperation::Add(columns) => write!(
f,
"ADD {}",
columns.iter().map(|x| x.to_string()).join(", ")
),
AlterTypeOperation::Rename(pairs) => write!(
f,
"RENAME {}",
pairs
.iter()
.map(|(x, y)| format!("{} TO {}", x, y))
.join(" AND ")
),
}
}
}