lezeh-db 0.0.1

use std::collections::{HashMap, HashSet};
use std::rc::Rc;

use itertools::Itertools;

use crate::psql::dto::FromSqlSink;
use crate::psql::dto::PsqlTable;
use crate::psql::dto::PsqlTableIdentity;
use crate::psql::dto::PsqlTableRow;
use lezeh_common::types::ResultAnyError;

pub struct TableInsertStatement<'a> {
  table: PsqlTable,
  columns: TableInsertRowColumns<'a>,
  row_values: Vec<TableInsertRowValues>,
}

impl<'a> std::fmt::Display for TableInsertStatement<'a> {
  fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
    // let template = ;

    return write!(
      f,
      indoc::indoc! {"
        ------------------------------------------------
        -- insert into table {}
        ------------------------------------------------
        insert into {} ({}) VALUES
          {};
        ---------------

      "},
      self.table.id,
      self.table.id,
      self.columns,
      self
        .row_values
        .iter()
        .map(|val| format!("{}", val))
        .collect::<Vec<String>>()
        .join(",\n"),
    );
  }
}

pub struct TableInsertRowColumns<'a> {
  column_names: Vec<&'a str>,
}

impl<'a> std::fmt::Display for TableInsertRowColumns<'a> {
  fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
    let column_string: String = self
      .column_names
      .iter()
      .map(|column_name| {
        return format!("\"{}\"", column_name);
      })
      .collect::<Vec<String>>()
      .join(", ");

    return write!(f, "{}", column_string);
  }
}

pub struct TableInsertRowValues {
  values: Vec<String>,
}

impl std::fmt::Display for TableInsertRowValues {
  fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
    return write!(f, "({})", self.values.join(", "));
  }
}

pub struct RelationInsert {}

impl RelationInsert {
  pub fn into_insert_statements(
    mut rows_by_level: HashMap<i32, HashSet<&Rc<PsqlTableRow>>>,
  ) -> ResultAnyError<Vec<String>> {
    let mut levels: Vec<i32> = rows_by_level.keys().cloned().collect();
    let mut insert_statement_map: HashMap<String, bool> = Default::default();

    levels.sort();

    let insert_statements: ResultAnyError<Vec<Vec<String>>> = levels
      .iter()
      .map(|level| {
        let rows: &mut HashSet<_> = rows_by_level.get_mut(level).unwrap();

        rows.retain(|row| {
          let row_key = format!("{}.{}", row.table.id, row.row_id_representation);
          let found = insert_statement_map.contains_key(&row_key);

          insert_statement_map.insert(row_key, true);

          // We don't want duplicate insert statement
          // TODO: Maybe we could use BTreeSet instead?
          return !found;
        });

        return RelationInsert::table_rows_into_insert_statement(rows);
      })
      .collect();

    return Ok(insert_statements?.into_iter().flatten().collect());
  }

  pub fn table_rows_into_insert_statement(
    rows: &HashSet<&Rc<PsqlTableRow>>,
  ) -> ResultAnyError<Vec<String>> {
    // Rows of the same table can be scattered through vec of psql table rows,
    // remember Vec<PsqlTableRows> meaning Vec<Vec<Row>> due to PsqlTableRows
    // contains `rows: Vec<Row>`. So here we're trying to group
    // scattered rows by table id
    let psql_table_by_id: HashMap<PsqlTableIdentity, PsqlTable> = rows
      .iter()
      .map(|row| (row.table.id.clone(), row.table.clone()))
      .collect();

    let psql_rows_by_table_id: HashMap<PsqlTableIdentity, Vec<&Rc<PsqlTableRow>>> = rows
      .iter()
      .map(|psql_table_row| (psql_table_row.table.id.clone(), psql_table_row.clone()))
      .into_group_map();

    let rows_by_table_id: HashMap<PsqlTableIdentity, Vec<&Rc<PsqlTableRow>>> =
      psql_rows_by_table_id
        .into_iter()
        .map(
          |(table_identity, row): (PsqlTableIdentity, Vec<&Rc<PsqlTableRow>>)| {
            return (table_identity, row);
          },
        )
        .collect();

    return rows_by_table_id
      .iter()
      .map(|(table_id, rows)| {
        return RelationInsert::table_row_into_insert_statement(
          psql_table_by_id.get(table_id).unwrap(),
          rows,
        );
      })
      .collect::<ResultAnyError<Vec<String>>>();
  }

  pub fn table_row_into_insert_statement(
    table: &PsqlTable,
    rows: &Vec<&Rc<PsqlTableRow>>,
  ) -> ResultAnyError<String> {
    let first_row: &PsqlTableRow = rows.get(0).unwrap();
    let table_insert_row_columns = TableInsertRowColumns {
      column_names: first_row.get_column_names(),
    };

    let row_values: Vec<TableInsertRowValues> = rows
      .iter()
      .map(|row| {
        let column_value_map: HashMap<&str, FromSqlSink> = row.get_column_value_map();

        // Use ordering on table insert row columns to preserve ordering
        return table_insert_row_columns
          .column_names
          .iter()
          .map(|column_name| {
            let from_sql_sink = column_value_map.get(column_name).unwrap();

            return from_sql_sink.to_string_for_statement();
          })
          .collect::<ResultAnyError<Vec<String>>>()
          .map(|values_in_string| {
            return TableInsertRowValues {
              values: values_in_string,
            };
          });
      })
      .collect::<ResultAnyError<Vec<TableInsertRowValues>>>()?;

    let table_insert_statement = TableInsertStatement {
      table: table.clone(),
      columns: table_insert_row_columns,
      row_values,
    };

    return Ok(format!("{}", table_insert_statement));
  }
}