1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use crate::{model_extensions::AsColumns, Context};
use quaint::ast::{Column, Table};
use query_structure::Model;

pub(crate) fn db_name_with_schema(model: &Model, ctx: &Context<'_>) -> Table<'static> {
    let schema_prefix = model
        .walker()
        .schema_name()
        .map(ToOwned::to_owned)
        .unwrap_or_else(|| ctx.schema_name().to_owned());
    let model_db_name = model.db_name().to_owned();
    (schema_prefix, model_db_name).into()
}

pub(crate) trait AsTable {
    fn as_table(&self, ctx: &Context<'_>) -> Table<'static>;
}

impl AsTable for Model {
    fn as_table(&self, ctx: &Context<'_>) -> Table<'static> {
        let table: Table<'static> = db_name_with_schema(self, ctx);

        let id_cols: Vec<Column<'static>> = self
            .primary_identifier()
            .as_scalar_fields()
            .expect("Primary identifier has non-scalar fields.")
            .as_columns(ctx)
            .collect();

        let table = table.add_unique_index(id_cols);

        self.unique_indexes().fold(table, |table, index| {
            let fields: Vec<_> = index
                .fields()
                .map(|f| query_structure::ScalarFieldRef::from((self.dm.clone(), f)))
                .collect();
            let index: Vec<Column<'static>> = fields.as_columns(ctx).collect();
            table.add_unique_index(index)
        })
    }
}