kosame_dsl 0.3.0

use proc_macro2::TokenStream;
use quote::{ToTokens, quote};
use syn::{
    Ident,
    parse::{Parse, ParseStream},
};

use crate::{
    parse_option::ParseOption,
    part::{Alias, TablePath},
    path_ext::PathExt,
    pretty::{PrettyPrint, Printer},
    quote_option::QuoteOption,
    visit::Visit,
};

pub struct TargetTable {
    pub table: TablePath,
    pub alias: Option<Alias>,
}

impl TargetTable {
    #[must_use]
    pub fn name(&self) -> &Ident {
        self.alias.as_ref().map_or_else(
            || {
                &self
                    .table
                    .as_path()
                    .segments
                    .last()
                    .expect("path cannot be empty")
                    .ident
            },
            |alias| &alias.ident,
        )
    }
}

pub fn visit_target_table<'a>(
    visit: &mut (impl Visit<'a> + ?Sized),
    target_table: &'a TargetTable,
) {
    visit.visit_table_path(&target_table.table);
}

impl Parse for TargetTable {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        Ok(Self {
            table: input.parse()?,
            alias: input.call(Alias::parse_option)?,
        })
    }
}

impl ToTokens for TargetTable {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let table = self.table.as_path().to_call_site(1);
        let alias = QuoteOption::from(&self.alias);
        quote! {
            ::kosame::repr::part::TargetTable::new(#table::TABLE_NAME, #alias)
        }
        .to_tokens(tokens);
    }
}

impl PrettyPrint for TargetTable {
    fn pretty_print(&self, printer: &mut Printer<'_>) {
        self.table.pretty_print(printer);
        if let Some(alias) = &self.alias {
            " ".pretty_print(printer);
            alias.pretty_print(printer);
        }
    }
}