kosame_repr 0.2.1

use std::fmt::Write;

use crate::{clause::*, part::TargetTable};

pub struct Update<'a> {
    target_table: TargetTable<'a>,
    set: Set<'a>,
    from: Option<From<'a>>,
    r#where: Option<Where<'a>>,
    returning: Option<Returning<'a>>,
}

impl<'a> Update<'a> {
    #[inline]
    pub const fn new(
        target_table: TargetTable<'a>,
        set: Set<'a>,
        from: Option<From<'a>>,
        r#where: Option<Where<'a>>,
        returning: Option<Returning<'a>>,
    ) -> Self {
        Self {
            target_table,
            set,
            from,
            r#where,
            returning,
        }
    }

    #[inline]
    pub const fn target_table(&self) -> &TargetTable<'a> {
        &self.target_table
    }

    #[inline]
    pub const fn set(&self) -> &Set<'a> {
        &self.set
    }

    #[inline]
    pub const fn from(&self) -> Option<&From<'a>> {
        self.from.as_ref()
    }

    #[inline]
    pub const fn r#where(&self) -> Option<&Where<'a>> {
        self.r#where.as_ref()
    }

    #[inline]
    pub const fn returning(&self) -> Option<&Returning<'a>> {
        self.returning.as_ref()
    }
}

impl kosame_sql::FmtSql for Update<'_> {
    fn fmt_sql<D>(&self, formatter: &mut kosame_sql::Formatter<D>) -> kosame_sql::Result
    where
        D: kosame_sql::Dialect,
    {
        formatter.write_str("update ")?;
        self.target_table.fmt_sql(formatter)?;
        self.set.fmt_sql(formatter)?;

        if let Some(from) = &self.from {
            from.fmt_sql(formatter)?;
        }
        if let Some(r#where) = &self.r#where {
            r#where.fmt_sql(formatter)?;
        }
        if let Some(returning) = &self.returning {
            returning.fmt_sql(formatter)?;
        }

        Ok(())
    }
}