agregar links a supermercados

poder importar warcs especificos

.dockerignore

@@ -0,0 +1,7 @@
+data/warcs/
+data/carrefour/
+*/*.db*
+downloader/
+node_modules/
+*/node_modules/
+*/Containerfile

db-datos/.env

@@ -1 +0,0 @@
-DB_PATH=../scraper/sqlite.db

db-datos/drizzle.config.ts

@@ -7,7 +7,6 @@ export const DB_PATH = process.env.DB_PATH;
 export default {
   schema: "./schema.ts",
   out: "./drizzle",
-  driver: "better-sqlite",
   dbCredentials: {
     url: process.env.DB_PATH,
   },

db-datos/migrate.ts

@@ -1,12 +1,19 @@
 import Database from "bun:sqlite";
+import { join } from "node:path";
 import { drizzle } from "drizzle-orm/bun-sqlite";
 import { migrate } from "drizzle-orm/bun-sqlite/migrator";
 import * as schema from "./schema.js";
 import { DB_PATH } from "./drizzle.config.js";
 
-const sqlite = new Database(DB_PATH);
-const db = drizzle(sqlite, { schema });
+export function migrateDb() {
+  const sqlite = new Database(DB_PATH);
+  const db = drizzle(sqlite, { schema });
 
-migrate(db, { migrationsFolder: "./drizzle" });
+  migrate(db, { migrationsFolder: join(import.meta.dir, "drizzle") });
+  sqlite.run(`
+pragma journal_mode = WAL;
+PRAGMA synchronous = NORMAL;
+`);
 
-sqlite.close();
+  sqlite.close();
+}

db-datos/package.json

@@ -15,9 +15,7 @@
     "drizzle-orm": "^0.29.1"
   },
   "devDependencies": {
-    "better-sqlite3": "^9.2.2",
-    "drizzle-kit": "^0.20.7",
-    "tsx": "^4.7.0",
-    "@types/better-sqlite3": "^7.6.8"
+    "@types/bun": "^1.0.0",
+    "drizzle-kit": "^0.20.7"
   }
 }

scraper/.env

@@ -1 +0,0 @@
-DB_PATH=../scraper/sqlite.db

scraper/Containerfile

@@ -0,0 +1,27 @@
+FROM oven/bun:1-alpine AS base
+WORKDIR /usr/src/app
+
+FROM base AS builder
+ENV NODE_ENV=production
+COPY . .
+RUN bun install --frozen-lockfile \
+ && bun build scraper/cli.ts --target=bun --outfile=/tmp/cli.build.js \
+ && rm -rf node_modules/
+
+FROM base
+RUN apk add --no-cache wget zstd cronie tini
+RUN printf "#!/bin/sh\nexec bun /bin/scraper auto" > /etc/periodic/daily/scraper \
+ && chmod +x /etc/periodic/daily/scraper
+
+COPY --from=builder /tmp/cli.build.js /bin/scraper
+COPY --from=builder /usr/src/app/db-datos/drizzle /bin/drizzle
+COPY --from=builder /usr/src/app/data /listas
+WORKDIR /app
+
+VOLUME /db
+ENV NODE_ENV=production
+ENV DB_PATH=/db/db.db
+ENV LISTS_DIR=/listas/
+
+CMD ["tini", "/usr/sbin/crond", "-n"]
+# CMD ["bun", "/bin/scraper"]

scraper/auto.ts

@@ -10,28 +10,6 @@ import { S3Client } from "@aws-sdk/client-s3";
 import { Upload } from "@aws-sdk/lib-storage";
 import { BunFile } from "bun";
 
-if (
-  !process.env.S3_ACCESS_KEY_ID ||
-  !process.env.S3_SECRET_ACCESS_KEY ||
-  !process.env.S3_BUCKET_NAME
-)
-  throw new Error("missing s3 creds");
-if (!process.env.TELEGRAM_BOT_TOKEN)
-  console.warn("no hay TELEGRAM_BOT_TOKEN, no voy a loggear por allá");
-if (!process.env.TELEGRAM_BOT_CHAT_ID)
-  console.warn("no hay TELEGRAM_BOT_CHAT_ID, no voy a loggear por allá");
-const { S3_BUCKET_NAME, S3_ACCESS_KEY_ID, S3_SECRET_ACCESS_KEY } = process.env;
-
-// https://www.backblaze.com/docs/cloud-storage-use-the-aws-sdk-for-javascript-v3-with-backblaze-b2
-const s3 = new S3Client({
-  endpoint: "https://s3.us-west-004.backblazeb2.com",
-  region: "us-west-004",
-  credentials: {
-    accessKeyId: S3_ACCESS_KEY_ID,
-    secretAccessKey: S3_SECRET_ACCESS_KEY,
-  },
-});
-
 const supermercados: Supermercado[] = [
   Supermercado.Carrefour,
   Supermercado.Coto,
@@ -44,164 +22,230 @@ const compressionQueue = new PQueue({ concurrency: 1 });
 // hacemos una cola para el scrapeo para no tener varios writers a la BD y no sobrecargar la CPU
 const scrapQueue = new PQueue({ concurrency: 1 });
 
-supermercados.forEach(downloadList);
+export async function auto() {
+  const a = new Auto();
+  await Promise.all(supermercados.map((supr) => a.downloadList(supr)));
+}
+
+class Auto {
+  s3Config?: { s3: S3Client; bucketName: string };
+  telegramConfig?: { token: string; chatId: string };
+
+  constructor() {
+    if (
+      !process.env.S3_ACCESS_KEY_ID ||
+      !process.env.S3_SECRET_ACCESS_KEY ||
+      !process.env.S3_BUCKET_NAME
+    ) {
+      if (process.env.NODE_ENV === "development") {
+        console.warn("faltan creds de s3, no voy a subir a s3");
+      } else {
+        throw new Error("faltan creds de s3");
+      }
+    } else {
+      this.s3Config = {
+        // https://www.backblaze.com/docs/cloud-storage-use-the-aws-sdk-for-javascript-v3-with-backblaze-b2
+        s3: new S3Client({
+          endpoint: "https://s3.us-west-004.backblazeb2.com",
+          region: "us-west-004",
+          credentials: {
+            accessKeyId: process.env.S3_ACCESS_KEY_ID,
+            secretAccessKey: process.env.S3_SECRET_ACCESS_KEY,
+          },
+        }),
+        bucketName: process.env.S3_BUCKET_NAME,
+      };
+    }
+
+    if (!process.env.TELEGRAM_BOT_TOKEN)
+      console.warn("no hay TELEGRAM_BOT_TOKEN, no voy a loggear por allá");
+    else if (!process.env.TELEGRAM_BOT_CHAT_ID)
+      console.warn("no hay TELEGRAM_BOT_CHAT_ID, no voy a loggear por allá");
+    else
+      this.telegramConfig = {
+        token: process.env.TELEGRAM_BOT_TOKEN,
+        chatId: process.env.TELEGRAM_BOT_CHAT_ID,
+      };
+  }
+
+  async downloadList(supermercado: Supermercado) {
+    const listPath = resolve(
+      join(process.env.LISTS_DIR ?? "../data", `${supermercado}.txt`)
+    );
+    const date = new Date();
+    const ctxPath = await mkdtemp(join(tmpdir(), "preciazo-scraper-wget-"));
+    const zstdWarcName = `${supermercado}-${format(
+      date,
+      "yyyy-MM-dd-HH:mm"
+    )}.warc.zst`;
+    const zstdWarcPath = join(ctxPath, zstdWarcName);
+    const subproc = Bun.spawn({
+      cmd: [
+        "wget",
+        "--no-verbose",
+        "--tries=3",
+        "--delete-after",
+        "--input-file",
+        listPath,
+        `--warc-file=temp`,
+      ],
+      stderr: "ignore",
+      stdout: "ignore",
+      cwd: ctxPath,
+    });
+    const t0 = performance.now();
+    await subproc.exited;
+    this.inform(
+      `wget para ${zstdWarcName} tardó ${formatMs(performance.now() - t0)}`
+    );
+
+    const gzippedWarcPath = join(ctxPath, "temp.warc.gz");
+    if (!(await fileExists(gzippedWarcPath))) {
+      const err = this.report(`no encontré el ${gzippedWarcPath}`);
+      throw err;
+    }
+
+    await compressionQueue.add(() =>
+      this.recompress(gzippedWarcPath, zstdWarcPath)
+    );
+    if (!(await fileExists(zstdWarcPath))) {
+      const err = this.report(`no encontré el ${zstdWarcPath}`);
+      throw err;
+    }
+
+    this.scrapAndInform({ zstdWarcPath, zstdWarcName });
+
+    try {
+      await this.uploadToBucket({
+        fileName: zstdWarcName,
+        file: Bun.file(zstdWarcPath),
+      });
+    } catch (error) {
+      this.inform(`Falló subir ${zstdWarcName} a S3; ${error}`);
+      console.error(error);
+    }
+
+    // TODO: borrar archivos temporales
+  }
+
+  async scrapAndInform({
+    zstdWarcPath,
+    zstdWarcName,
+  }: {
+    zstdWarcPath: string;
+    zstdWarcName: string;
+  }) {
+    const res = await scrapQueue.add(async () => {
+      const t0 = performance.now();
+      const progress = await parseWarc(zstdWarcPath);
+      return { took: performance.now() - t0, progress };
+    });
+
+    if (res) {
+      const { took, progress } = res;
+      this.inform(
+        `Procesado ${zstdWarcName} (${progress.done} ok, ${
+          progress.errors.length
+        } errores) (tardó ${formatMs(took)})`
+      );
+    } else {
+      this.inform(`Algo falló en ${zstdWarcName}`);
+    }
+  }
+
+  /**
+   * toma un archivo gzippeado y lo recomprime con zstd.
+   * borra el archivo original.
+   */
+  recompress(inputPath: string, outputPath: string) {
+    // XXX: por alguna razón no funciona en Bun 1.0.20
+    // const decompressor = Bun.spawn({
+    //   cmd: ["gzip", "-dc", inputPath],
+    //   stderr: "inherit",
+    // });
+    // const compressor = Bun.spawn({
+    //   cmd: ["zstd", "-T0", "-15", "--long", "-o", outputPath],
+    //   stdin: decompressor.stdout,
+    //   // stderr: "inherit",
+    // });
+    // const errorCode = await compressor.exited;
+    // if (errorCode !== 0) {
+    //   const err = report(`zstd threw error code ${errorCode}`);
+    //   throw err;
+    // }
+
+    return new Promise((resolve, reject) => {
+      const decompressor = spawn("gzip", ["-dc", inputPath], {
+        stdio: [null, "pipe", null],
+      });
+      const compressor = spawn(
+        "zstd",
+        ["-T0", "-15", "--long", "-o", outputPath],
+        {
+          stdio: ["pipe", null, null],
+        }
+      );
+      // @ts-expect-error a los types de bun no le gusta????
+      decompressor.stdout.pipe(compressor.stdin);
+      compressor.on("close", (code) => {
+        if (code !== 0) {
+          const err = this.report(`zstd threw error code ${code}`);
+          reject(err);
+        }
+        resolve(void 0);
+      });
+    });
+  }
+
+  async uploadToBucket({
+    fileName,
+    file,
+  }: {
+    fileName: string;
+    file: BunFile;
+  }) {
+    if (!this.s3Config) {
+      this.inform(
+        `[s3] Se intentó subir ${fileName} pero no tenemos creds de S3`
+      );
+      return;
+    }
+    const parallelUploads3 = new Upload({
+      client: this.s3Config.s3,
+      params: {
+        Bucket: this.s3Config.bucketName,
+        Key: fileName,
+        Body: file,
+      },
+    });
+    await parallelUploads3.done();
+  }
+
+  inform(msg: string) {
+    this.sendTelegramMsg(msg);
+    console.info(msg);
+  }
+  report(msg: string) {
+    this.inform(msg);
+    const error = new Error(msg);
+
+    return error;
+  }
+
+  async sendTelegramMsg(text: string) {
+    if (!this.telegramConfig) return;
+    const url = new URL(
+      `https://api.telegram.org/bot${this.telegramConfig.token}/sendMessage`
+    );
+    url.searchParams.set("chat_id", this.telegramConfig.chatId);
+    url.searchParams.set("text", text);
+    await fetch(url);
+  }
+}
 // await recompress("sqlite.db.gz", "sqlite.db.zst");
 
-async function downloadList(supermercado: Supermercado) {
-  const listPath = resolve(
-    join(process.env.LISTS_DIR ?? "../data", `${supermercado}.txt`)
-  );
-  const date = new Date();
-  const ctxPath = await mkdtemp(join(tmpdir(), "preciazo-scraper-wget-"));
-  const zstdWarcName = `${supermercado}-${format(
-    date,
-    "yyyy-MM-dd-HH:mm"
-  )}.warc.zst`;
-  const zstdWarcPath = join(ctxPath, zstdWarcName);
-  const subproc = Bun.spawn({
-    cmd: [
-      "wget",
-      "--no-verbose",
-      "--tries=3",
-      "--delete-after",
-      "--input-file",
-      listPath,
-      `--warc-file=temp`,
-    ],
-    stderr: "ignore",
-    stdout: "ignore",
-    cwd: ctxPath,
-  });
-  const t0 = performance.now();
-  await subproc.exited;
-  inform(`wget para ${zstdWarcName} tardó ${formatMs(performance.now() - t0)}`);
-
-  const gzippedWarcPath = join(ctxPath, "temp.warc.gz");
-  if (!(await exists(gzippedWarcPath))) {
-    const err = report(`no encontré el ${gzippedWarcPath}`);
-    throw err;
-  }
-
-  await compressionQueue.add(() => recompress(gzippedWarcPath, zstdWarcPath));
-  if (!(await exists(zstdWarcPath))) {
-    const err = report(`no encontré el ${zstdWarcPath}`);
-    throw err;
-  }
-
-  scrapAndInform({ zstdWarcPath, zstdWarcName });
-
-  try {
-    await uploadToBucket({
-      fileName: zstdWarcName,
-      file: Bun.file(zstdWarcPath),
-    });
-  } catch (error) {
-    inform(`Falló subir ${zstdWarcName} a S3; ${error}`);
-    console.error(error);
-  }
-
-  // TODO: borrar archivos temporales
-}
-
-async function scrapAndInform({
-  zstdWarcPath,
-  zstdWarcName,
-}: {
-  zstdWarcPath: string;
-  zstdWarcName: string;
-}) {
-  const res = await scrapQueue.add(async () => {
-    const t0 = performance.now();
-    const progress = await parseWarc(zstdWarcPath);
-    return { took: performance.now() - t0, progress };
-  });
-
-  if (res) {
-    const { took, progress } = res;
-    inform(
-      `Procesado ${zstdWarcName} (${progress.done} ok, ${
-        progress.errors.length
-      } errores) (tardó ${formatMs(took)})`
-    );
-  } else {
-    inform(`Algo falló en ${zstdWarcName}`);
-  }
-}
-
-/**
- * toma un archivo gzippeado y lo recomprime con zstd.
- * borra el archivo original.
- */
-function recompress(inputPath: string, outputPath: string) {
-  // XXX: por alguna razón no funciona en Bun 1.0.20
-  // const decompressor = Bun.spawn({
-  //   cmd: ["gzip", "-dc", inputPath],
-  //   stderr: "inherit",
-  // });
-  // const compressor = Bun.spawn({
-  //   cmd: ["zstd", "-T0", "-15", "--long", "-o", outputPath],
-  //   stdin: decompressor.stdout,
-  //   // stderr: "inherit",
-  // });
-  // const errorCode = await compressor.exited;
-  // if (errorCode !== 0) {
-  //   const err = report(`zstd threw error code ${errorCode}`);
-  //   throw err;
-  // }
-
-  return new Promise((resolve, reject) => {
-    const decompressor = spawn("gzip", ["-dc", inputPath], {
-      stdio: [null, "pipe", null],
-    });
-    const compressor = spawn(
-      "zstd",
-      ["-T0", "-15", "--long", "-o", outputPath],
-      {
-        stdio: ["pipe", null, null],
-      }
-    );
-    // @ts-expect-error a los types de bun no le gusta????
-    decompressor.stdout.pipe(compressor.stdin);
-    compressor.on("close", (code) => {
-      if (code !== 0) {
-        const err = report(`zstd threw error code ${code}`);
-        reject(err);
-      }
-      resolve(void 0);
-    });
-  });
-}
-
-async function uploadToBucket({
-  fileName,
-  file,
-}: {
-  fileName: string;
-  file: BunFile;
-}) {
-  const parallelUploads3 = new Upload({
-    client: s3,
-    params: {
-      Bucket: S3_BUCKET_NAME,
-      Key: fileName,
-      Body: file,
-    },
-  });
-  await parallelUploads3.done();
-}
-
-function inform(msg: string) {
-  sendTelegramMsg(msg);
-  console.info(msg);
-}
-function report(msg: string) {
-  inform(msg);
-  const error = new Error(msg);
-
-  return error;
-}
-
-async function exists(path: string) {
+// no se llama exists porque bun tiene un bug en el que usa fs.exists por mas que exista una funcion llamada exists
+async function fileExists(path: string) {
   try {
     access(path);
     return true;
@@ -210,17 +254,6 @@ async function exists(path: string) {
   }
 }
 
-async function sendTelegramMsg(text: string) {
-  if (!process.env.TELEGRAM_BOT_TOKEN || !process.env.TELEGRAM_BOT_CHAT_ID)
-    return;
-  const url = new URL(
-    `https://api.telegram.org/bot${process.env.TELEGRAM_BOT_TOKEN}/sendMessage`
-  );
-  url.searchParams.set("chat_id", process.env.TELEGRAM_BOT_CHAT_ID);
-  url.searchParams.set("text", text);
-  await fetch(url);
-}
-
 function formatMs(ms: number) {
   return formatDuration(intervalToDuration({ start: 0, end: Math.round(ms) }));
 }

scraper/cli.ts

@@ -0,0 +1,19 @@
+import { auto } from "./auto.js";
+import { parseWarc } from "./scrap.js";
+
+if (process.argv[2] === "auto") {
+  await auto();
+} else if (process.argv[2] === "scrap") {
+  const warcPaths = process.argv.slice(3);
+  if (warcPaths.length > 0) {
+    for (const path of warcPaths) {
+      await parseWarc(path);
+    }
+  } else {
+    console.error("Especificá WARCs para scrapear.");
+    process.exit(1);
+  }
+} else {
+  console.error("Especificá una acción (tipo `auto` o `scrap`) para hacer.");
+  process.exit(1);
+}

scraper/package.json

@@ -5,7 +5,7 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "build:container": "podman build -t gitea.nulo.in/nulo/preciazo/scraper -f ./Containerfile .."
   },
   "keywords": [],
   "author": "",

scraper/scrap.ts

@@ -10,17 +10,16 @@ import { getCotoProduct } from "./parsers/coto.js";
 import { join } from "path";
 import { and, eq, sql } from "drizzle-orm";
 import { DB_PATH } from "db-datos/drizzle.config.js";
+import { migrateDb } from "db-datos/migrate.js";
 
 const DEBUG = false;
 const PARSER_VERSION = 2;
 
+migrateDb();
+
 const sqlite = new Database(DB_PATH);
 const db = drizzle(sqlite, { schema });
 
-sqlite.run(`
-pragma journal_mode = WAL;
-PRAGMA synchronous = NORMAL;
-`);
 const getPrevPrecio = db
   .select({ id: schema.precios.id })
   .from(schema.precios)
@@ -33,12 +32,6 @@ const getPrevPrecio = db
   .limit(1)
   .prepare();
 
-if (process.argv[1].endsWith("/scrap.ts")) {
-  for (const path of process.argv.slice(2)) {
-    await parseWarc(path);
-  }
-}
-
 export type Precio = typeof schema.precios.$inferInsert;
 export type Precioish = Omit<
   Precio,

sitio/Containerfile

@@ -7,4 +7,7 @@ EXPOSE 3000
 ENV PROTOCOL_HEADER=x-forwarded-proto
 ENV HOST_HEADER=x-forwarded-host
 
+VOLUME /db
+ENV DB_PATH=/db/db.db
+
 CMD ["bun", "run", "start"]

sitio/package.json

@@ -5,6 +5,7 @@
   "scripts": {
     "dev": "vite dev",
     "build": "vite build",
+    "build:container": "bun --bun vite build && podman build -t gitea.nulo.in/nulo/preciazo/sitio .",
     "preview": "vite preview",
     "check": "svelte-kit sync && svelte-check --tsconfig ./tsconfig.json",
     "check:watch": "svelte-kit sync && svelte-check --tsconfig ./tsconfig.json --watch",

sitio/src/routes/ean/[ean]/+page.svelte

@@ -1,28 +1,40 @@
 <script lang="ts">
+  import { Supermercado, hosts } from "db-datos/supermercado";
   import type { PageData } from "./$types";
   import Chart from "./Chart.svelte";
 
   export let data: PageData;
+
+  let urls: Map<Supermercado, string>;
+  $: urls = data.precios.toReversed().reduce((prev, curr) => {
+    const url = new URL(curr.url);
+    const supermercado = hosts[url.hostname];
+    prev.set(supermercado, curr.url);
+    return prev;
+  }, new Map<Supermercado, string>());
+
+  const classBySupermercado: { [supermercado in Supermercado]: string } = {
+    [Supermercado.Dia]: "bg-[#d52b1e] focus:ring-[#d52b1e]",
+    [Supermercado.Carrefour]: "bg-[#19549d] focus:ring-[#19549d]",
+    [Supermercado.Coto]: "bg-[#e20025] focus:ring-[#e20025]",
+  };
 </script>
 
 {#if data.meta}
   <h1 class="text-3xl font-bold">{data.meta.name}</h1>
   <img src={data.meta.imageUrl} class="max-h-48" />
+  <div class="flex gap-2">
+    {#each urls as [supermercado, url]}
+      <a
+        href={url}
+        rel="noreferrer noopener"
+        target="_blank"
+        class={`focus:shadow-outline inline-flex items-center justify-center rounded-md ${classBySupermercado[supermercado]} px-4 py-2 text-sm font-medium tracking-wide text-white transition-colors duration-200 hover:bg-opacity-80 focus:outline-none focus:ring-2 focus:ring-offset-2`}
+      >
+        Ver en {supermercado}
+      </a>
+    {/each}
+  </div>
 {/if}
 
-<ul>
-  {#each data.precios as precio}
-    <li>
-      {precio.url}
-      :
-      {#if precio.precioCentavos}
-        {precio.precioCentavos / 100}
-      {:else}
-        {precio.inStock}
-      {/if}
-      ({precio.fetchedAt})
-    </li>
-  {/each}
-</ul>
-
 <Chart precios={data.precios} />