preciazo/sepa/db/schema.ts

import { max, relations, sql } from "drizzle-orm";
import {
  pgTable,
  integer,
  bigint,
  text,
  numeric,
  unique,
  serial,
  date,
  index,
  pgMaterializedView,
  pgView,
  timestamp,
  boolean,
} from "drizzle-orm/pg-core";

export const datasets = pgTable(
  "datasets",
  {
    id: serial("id").primaryKey().notNull(),
    name: text("name"),
    date: date("date"),
    id_comercio: integer("id_comercio"),
  },
  (table) => {
    return {
      datasets_name_key: unique("datasets_name_key").on(table.name),
    };
  }
);

export const precios = pgTable(
  "precios",
  {
    id_dataset: integer("id_dataset").references(() => datasets.id),
    id_comercio: integer("id_comercio"),
    id_bandera: integer("id_bandera"),
    id_sucursal: integer("id_sucursal"),
    // You can use { mode: "bigint" } if numbers are exceeding js number limitations
    id_producto: bigint("id_producto", { mode: "bigint" }),
    productos_ean: integer("productos_ean"),
    productos_descripcion: text("productos_descripcion"),
    productos_cantidad_presentacion: numeric(
      "productos_cantidad_presentacion",
      {
        precision: 10,
        scale: 2,
      }
    ),
    productos_unidad_medida_presentacion: text(
      "productos_unidad_medida_presentacion"
    ),
    productos_marca: text("productos_marca"),
    productos_precio_lista: numeric("productos_precio_lista", {
      precision: 10,
      scale: 2,
    }),
    productos_precio_referencia: numeric("productos_precio_referencia", {
      precision: 10,
      scale: 2,
    }),
    productos_cantidad_referencia: numeric("productos_cantidad_referencia", {
      precision: 10,
      scale: 2,
    }),
    productos_unidad_medida_referencia: text(
      "productos_unidad_medida_referencia"
    ),
    productos_precio_unitario_promo1: numeric(
      "productos_precio_unitario_promo1",
      {
        precision: 10,
        scale: 2,
      }
    ),
    productos_leyenda_promo1: text("productos_leyenda_promo1"),
    productos_precio_unitario_promo2: numeric(
      "productos_precio_unitario_promo2",
      {
        precision: 10,
        scale: 2,
      }
    ),
    productos_leyenda_promo2: text("productos_leyenda_promo2"),
  },
  (table) => {
    return {
      idx_precios_id_producto: index("idx_precios_id_producto").using(
        "btree",
        table.id_producto
      ),
      idx_precios_id_producto_id_dataset: index(
        "idx_precios_id_producto_id_dataset"
      ).using("btree", table.id_producto, table.id_dataset),
      idx_precios_id_producto_id_comercio_id_sucursal: index(
        "idx_precios_id_producto_id_comercio_id_sucursal"
      ).on(table.id_producto, table.id_comercio, table.id_sucursal),
    };
  }
);

export const sucursales = pgTable(
  "sucursales",
  {
    id_dataset: integer("id_dataset").references(() => datasets.id),
    id_comercio: integer("id_comercio"),
    id_bandera: integer("id_bandera"),
    id_sucursal: integer("id_sucursal"),
    sucursales_nombre: text("sucursales_nombre"),
    sucursales_tipo: text("sucursales_tipo"),
    sucursales_calle: text("sucursales_calle"),
    sucursales_numero: text("sucursales_numero"),
    sucursales_latitud: numeric("sucursales_latitud"),
    sucursales_longitud: numeric("sucursales_longitud"),
    sucursales_observaciones: text("sucursales_observaciones"),
    sucursales_barrio: text("sucursales_barrio"),
    sucursales_codigo_postal: text("sucursales_codigo_postal"),
    sucursales_localidad: text("sucursales_localidad"),
    sucursales_provincia: text("sucursales_provincia"),
    sucursales_lunes_horario_atencion: text(
      "sucursales_lunes_horario_atencion"
    ),
    sucursales_martes_horario_atencion: text(
      "sucursales_martes_horario_atencion"
    ),
    sucursales_miercoles_horario_atencion: text(
      "sucursales_miercoles_horario_atencion"
    ),
    sucursales_jueves_horario_atencion: text(
      "sucursales_jueves_horario_atencion"
    ),
    sucursales_viernes_horario_atencion: text(
      "sucursales_viernes_horario_atencion"
    ),
    sucursales_sabado_horario_atencion: text(
      "sucursales_sabado_horario_atencion"
    ),
    sucursales_domingo_horario_atencion: text(
      "sucursales_domingo_horario_atencion"
    ),
  },
  (table) => {
    return {
      sucursales_id_dataset_id_comercio_id_bandera_id_sucursal_key: unique(
        "sucursales_id_dataset_id_comercio_id_bandera_id_sucursal_key"
      ).on(
        table.id_dataset,
        table.id_comercio,
        table.id_bandera,
        table.id_sucursal
      ),
    };
  }
);

export const banderas = pgTable(
  "banderas",
  {
    id_dataset: integer("id_dataset").references(() => datasets.id),
    id_comercio: integer("id_comercio").notNull(),
    id_bandera: integer("id_bandera").notNull(),
    comercio_cuit: text("comercio_cuit").notNull(),
    comercio_razon_social: text("comercio_razon_social"),
    comercio_bandera_nombre: text("comercio_bandera_nombre"),
    comercio_bandera_url: text("comercio_bandera_url"),
    comercio_ultima_actualizacion: date("comercio_ultima_actualizacion"),
    comercio_version_sepa: text("comercio_version_sepa"),
  },
  (table) => {
    return {
      banderas_id_dataset: unique("banderas_id_dataset").on(table.id_dataset),
    };
  }
);

export const preciosRelations = relations(precios, ({ one }) => ({
  dataset: one(datasets, {
    fields: [precios.id_dataset],
    references: [datasets.id],
  }),
}));

export const datasetsRelations = relations(datasets, ({ many }) => ({
  precios: many(precios),
  sucursales: many(sucursales),
}));

export const sucursalesRelations = relations(sucursales, ({ one }) => ({
  dataset: one(datasets, {
    fields: [sucursales.id_dataset],
    references: [datasets.id],
  }),
  bandera: one(banderas, {
    fields: [sucursales.id_comercio, sucursales.id_bandera],
    references: [banderas.id_comercio, banderas.id_bandera],
  }),
}));

// para actualizar la tabla:
//  bun run scripts/refresh-descripciones-index.ts
export const productos_descripcion_index = pgTable(
  "productos_descripcion_index",
  {
    id_producto: bigint("id_producto", { mode: "bigint" }),
    productos_descripcion: text("productos_descripcion").unique(),
    productos_marca: text("productos_marca"),
  },
  (table) => ({
    // https://orm.drizzle.team/learn/guides/postgresql-full-text-search
    tableSearchIndex: index(
      "productos_descripcion_index_search_descripcion"
    ).using("gin", sql`to_tsvector('spanish', ${table.productos_descripcion})`),
  })
);

// véase scripts/refresh-scrapped-metadata.ts
export const productos_metadata_scrapped = pgTable(
  "productos_metadata_scrapped",
  {
    ean: bigint("ean", { mode: "bigint" }),
    fetchedAt: timestamp("fetched_at").notNull(),
    precioCentavos: integer("precio_centavos"),
    inStock: boolean("in_stock"),
    url: text("url").notNull(),
    name: text("name"),
    imageUrl: text("image_url"),
  },
  (table) => ({
    productos_metadata_scrapped_ean_idx: index(
      "productos_metadata_scrapped_ean_idx"
    ).on(table.ean),
  })
);