arreglar container scraper

.devcontainer/devcontainer.json

@@ -9,7 +9,8 @@
 		"ghcr.io/devcontainers/features/git-lfs:1": {},
 		"ghcr.io/devcontainers/features/node:1": {},
 		"ghcr.io/swift-server-community/swift-devcontainer-features/sqlite:1": {},
-		"ghcr.io/devcontainers/features/rust:1": {}
+		"ghcr.io/devcontainers/features/rust:1": {},
+		"ghcr.io/devcontainers/features/docker-in-docker:2": {}
 	},
 
 	// Use 'forwardPorts' to make a list of ports inside the container available locally.

.github/workflows/container.yml

@@ -52,3 +52,32 @@ jobs:
           push: true
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
+
+  build-and-push-scraper:
+    needs: check
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+      - name: Log in to the Container registry
+        uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - name: Extract metadata (tags, labels) for Docker
+        id: meta
+        uses: docker/metadata-action@9ec57ed1fcdbf14dcef7dfbe97b2010124a938b7
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}/scraper
+      - name: Build and push Docker image
+        uses: docker/build-push-action@f2a1d5e99d037542a71f64918e516c093c6f3fc4
+        with:
+          context: .
+          file: Dockerfile.scraper
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}

Dockerfile.scraper

@@ -0,0 +1,29 @@
+FROM cgr.dev/chainguard/wolfi-base AS base
+WORKDIR /usr/src/app
+RUN apk add --no-cache bun libgcc
+
+FROM base as build
+ENV NODE_ENV=production
+COPY . .
+RUN bun install --frozen-lockfile
+RUN bun build scraper/cli.ts --target=bun --outfile=/tmp/cli.build.js
+
+# nightly porque usamos tl con `simd` activado
+FROM base as rs-build
+RUN apk add --no-cache rust build-base sqlite-dev
+
+COPY scraper-rs/ .
+RUN cargo install --locked --path .
+
+FROM base
+RUN apk add --no-cache sqlite
+
+# Scraper
+COPY --from=build /tmp/cli.build.js /bin/scraper
+COPY --from=build /usr/src/app/db-datos/drizzle /bin/drizzle
+COPY --from=rs-build /root/.cargo/bin/scraper-rs /usr/local/bin/scraper-rs
+
+ENV NODE_ENV=production
+ENV DB_PATH=/db/db.db
+
+CMD ["bun", "/bin/scraper", "cron"]

scraper-rs/Cargo.lock

@@ -908,6 +908,28 @@ dependencies = [
  "proc-macro2",
 ]
 
+[[package]]
+name = "r2d2"
+version = "0.8.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "51de85fb3fb6524929c8a2eb85e6b6d363de4e8c48f9e2c2eac4944abc181c93"
+dependencies = [
+ "log",
+ "parking_lot 0.12.1",
+ "scheduled-thread-pool",
+]
+
+[[package]]
+name = "r2d2_sqlite"
+version = "0.23.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4dc290b669d30e20751e813517bbe13662d020419c5c8818ff10b6e8bb7777f6"
+dependencies = [
+ "r2d2",
+ "rusqlite",
+ "uuid",
+]
+
 [[package]]
 name = "rand"
 version = "0.7.3"
@@ -1111,6 +1133,15 @@ version = "1.0.16"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "f98d2aa92eebf49b69786be48e4477826b256916e84a57ff2a4f21923b48eb4c"
 
+[[package]]
+name = "scheduled-thread-pool"
+version = "0.2.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3cbc66816425a074528352f5789333ecff06ca41b36b0b0efdfbb29edc391a19"
+dependencies = [
+ "parking_lot 0.12.1",
+]
+
 [[package]]
 name = "scopeguard"
 version = "1.2.0"
@@ -1126,6 +1157,8 @@ dependencies = [
  "async-channel",
  "clap",
  "nanoid",
+ "r2d2",
+ "r2d2_sqlite",
  "rand 0.8.5",
  "reqwest",
  "rusqlite",
@@ -1333,7 +1366,8 @@ checksum = "1f3ccbac311fea05f86f61904b462b55fb3df8837a366dfc601a0161d0532f20"
 [[package]]
 name = "tl"
 version = "0.7.7"
-source = "git+https://github.com/evertedsphere/tl?branch=patch-1#56711166588fa6c7729a08e5740dca2526436316"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d5e993a1c7c32fdf90a308cec4d457f507b2573acc909bd6e7a092321664fdb3"
 
 [[package]]
 name = "tokio"
@@ -1514,6 +1548,16 @@ version = "0.2.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "711b9620af191e0cdc7468a8d14e709c3dcdb115b36f838e601583af800a370a"
 
+[[package]]
+name = "uuid"
+version = "1.6.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5e395fcf16a7a3d8127ec99782007af141946b4795001f876d54fb0d55978560"
+dependencies = [
+ "getrandom 0.2.11",
+ "rand 0.8.5",
+]
+
 [[package]]
 name = "valuable"
 version = "0.1.0"

scraper-rs/Cargo.toml

@@ -11,6 +11,8 @@ anyhow = "1.0.79"
 async-channel = "2.1.1"
 clap = { version = "4.4.15", features = ["derive"] }
 nanoid = "0.4.0"
+r2d2 = "0.8.10"
+r2d2_sqlite = "0.23.0"
 rand = "0.8.5"
 # lol_html = "1.2.0"
 reqwest = { version = "0.11.23", default-features = false, features = [
@@ -25,7 +27,7 @@ serde = { version = "1.0.193", features = ["derive"] }
 serde_json = "1.0.109"
 simple-error = "0.3.0"
 thiserror = "1.0.56"
-tl = { git = "https://github.com/evertedsphere/tl", branch = "patch-1", features = ["simd"] }
+tl = "0.7.7"
 tokio = { version = "1.35.1", features = ["full"] }
 tracing = "0.1"
 tracing-subscriber = "0.3"

scraper-rs/src/main.rs

@@ -1,9 +1,10 @@
 use again::RetryPolicy;
-use async_channel::{Receiver, Sender};
+use async_channel::Receiver;
 use clap::Parser;
 use nanoid::nanoid;
-use reqwest::Url;
-use rusqlite::Connection;
+use r2d2::Pool;
+use r2d2_sqlite::SqliteConnectionManager;
+use reqwest::{StatusCode, Url};
 use simple_error::{bail, SimpleError};
 use std::{
     env::{self},
@@ -12,7 +13,6 @@ use std::{
     time::Duration,
 };
 use thiserror::Error;
-use tl::VDom;
 
 #[derive(Parser)] // requires `derive` feature
 enum Args {
@@ -47,34 +47,37 @@ async fn fetch_list_cli(links_list_path: String) -> anyhow::Result<()> {
         .map(|s| s.to_owned())
         .collect::<Vec<_>>();
 
-    let handle = {
-        let (sender, receiver) = async_channel::bounded::<String>(1);
-        let (res_sender, res_receiver) = async_channel::unbounded::<PrecioPoint>();
+    let (sender, receiver) = async_channel::bounded::<String>(1);
 
-        let mut handles = Vec::new();
-        for _ in 1..env::var("N_COROUTINES")
-            .map_or(Ok(128), |s| s.parse::<usize>())
-            .expect("N_COROUTINES no es un número")
-        {
+    let db_path = env::var("DB_PATH").unwrap_or("../scraper/sqlite.db".to_string());
+    let manager = SqliteConnectionManager::file(db_path);
+    let pool = Pool::new(manager).unwrap();
+
+    let n_coroutines = env::var("N_COROUTINES")
+        .map_or(Ok(128), |s| s.parse::<usize>())
+        .expect("N_COROUTINES no es un número");
+    let handles = (1..n_coroutines)
+        .map(|_| {
             let rx = receiver.clone();
-            let tx = res_sender.clone();
-            handles.push(tokio::spawn(worker(rx, tx)));
-        }
+            let pool = pool.clone();
+            tokio::spawn(worker(rx, pool))
+        })
+        .collect::<Vec<_>>();
 
-        let db_writer_handle = tokio::spawn(db_writer(res_receiver));
+    for link in links {
+        sender.send_blocking(link).unwrap();
+    }
+    sender.close();
 
-        for link in links {
-            sender.send_blocking(link).unwrap();
-        }
-        sender.close();
+    let mut counters = Counters::default();
+    for handle in handles {
+        let c = handle.await.unwrap();
+        counters.success += c.success;
+        counters.errored += c.errored;
+        counters.skipped += c.skipped;
+    }
 
-        for handle in handles {
-            handle.await.unwrap();
-        }
-
-        db_writer_handle
-    };
-    handle.await.unwrap();
+    println!("Finished: {:?}", counters);
     Ok(())
 }
 
@@ -82,19 +85,46 @@ fn build_client() -> reqwest::Client {
     reqwest::ClientBuilder::default().build().unwrap()
 }
 
-async fn worker(rx: Receiver<String>, tx: Sender<PrecioPoint>) {
+#[derive(Default, Debug)]
+struct Counters {
+    success: u64,
+    errored: u64,
+    skipped: u64,
+}
+
+async fn worker(rx: Receiver<String>, pool: Pool<SqliteConnectionManager>) -> Counters {
     let client = build_client();
+
+    let mut counters = Counters::default();
     while let Ok(url) = rx.recv().await {
         let res = fetch_and_parse(&client, url.clone()).await;
         match res {
-            Ok(ex) => {
-                tx.send(ex).await.unwrap();
+            Ok(res) => {
+                counters.success += 1;
+                pool.get().unwrap().execute("INSERT INTO precios(ean, fetched_at, precio_centavos, in_stock, url, warc_record_id, parser_version, name, image_url) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9);",rusqlite::params![
+                    res.ean,
+                    res.fetched_at,
+                    res.precio_centavos,
+                    res.in_stock,
+                    res.url,
+                    None::<String>,
+                    res.parser_version,
+                    res.name,
+                    res.image_url,
+                ]).unwrap();
             }
             Err(err) => {
+                match err.downcast_ref::<FetchError>() {
+                    Some(FetchError::HttpStatus(StatusCode::NOT_FOUND)) => counters.skipped += 1,
+                    _ => counters.errored += 1,
+                }
+
                 tracing::error!(error=%err, url=url);
             }
         }
     }
+
+    counters
 }
 
 #[derive(Debug, Error)]
@@ -191,17 +221,6 @@ async fn scrap_url(
     }
 }
 
-async fn db_writer(rx: Receiver<PrecioPoint>) {
-    // let conn = Connection::open("../scraper/sqlite.db").unwrap();
-    // let mut stmt = conn.prepare("SELECT id, name, data FROM person")?;
-    let mut n = 0;
-    while let Ok(res) = rx.recv().await {
-        n += 1;
-        println!("{}", n);
-        // println!("{:?}", res)
-    }
-}
-
 use std::time::{SystemTime, UNIX_EPOCH};
 
 mod sites;

scraper/auto.ts

@@ -4,7 +4,6 @@ import { join } from "node:path";
 import { Supermercado, hosts, supermercados } from "db-datos/supermercado.js";
 import PQueue from "p-queue";
 import { formatDuration, intervalToDuration } from "date-fns";
-import { downloadList } from "./scrap.js";
 import { db } from "db-datos/db.js";
 import { like } from "drizzle-orm";
 import { productoUrls } from "db-datos/schema.js";
@@ -12,13 +11,16 @@ import { scrapDiaProducts } from "../link-scrapers/dia.js";
 import { scrapCotoProducts } from "../link-scrapers/coto.js";
 import { scrapCarrefourProducts } from "../link-scrapers/carrefour.js";
 import { scrapJumboProducts } from "../link-scrapers/jumbo.js";
+import { readableStreamToText } from "bun";
 
 // hacemos una cola para el scrapeo para no tener varios writers a la BD y no sobrecargar la CPU
-const scrapQueue = new PQueue({ concurrency: 4 });
+const scrapQueue = new PQueue({ concurrency: 1 });
 
 export async function auto() {
   const a = new Auto();
-  await Promise.all(supermercados.map((supr) => a.downloadList(supr)));
+  await Promise.all(
+    supermercados.filter((x) => x === "Dia").map((supr) => a.downloadList(supr))
+  );
 }
 
 class Auto {
@@ -38,32 +40,35 @@ class Auto {
     this.inform("[auto] Empezando scrap");
   }
 
+  async scrapUrls(supermercado: Supermercado) {
+    const t0 = performance.now();
+    switch (supermercado) {
+      case "Dia":
+        await scrapDiaProducts();
+        break;
+      case "Coto":
+        await scrapCotoProducts();
+        break;
+      case "Carrefour":
+        await scrapCarrefourProducts();
+        break;
+      case "Jumbo":
+        await scrapJumboProducts();
+        break;
+    }
+    this.inform(
+      `[scrapUrls[${supermercado}]] Tardó ${formatMs(performance.now() - t0)}`
+    );
+  }
+
   async downloadList(supermercado: Supermercado) {
     const ctxPath = await mkdtemp(join(tmpdir(), "preciazo-scraper-download-"));
 
-    let listPath: string;
-    {
-      const t0 = performance.now();
-      switch (supermercado) {
-        case "Dia":
-          await scrapDiaProducts();
-          break;
-        case "Coto":
-          await scrapCotoProducts();
-          break;
-        case "Carrefour":
-          await scrapCarrefourProducts();
-          break;
-        case "Jumbo":
-          await scrapJumboProducts();
-          break;
-      }
-      this.inform(
-        `[scrapUrls[${supermercado}]] Tardó ${formatMs(performance.now() - t0)}`
-      );
-    }
+    await scrapQueue.add(async () => {
+      await this.scrapUrls(supermercado);
+    });
 
-    listPath = join(ctxPath, `lista-${supermercado}.txt`);
+    const listPath = join(ctxPath, `lista-${supermercado}.txt`);
     const host = Object.entries(hosts).find(
       ([host, supe]) => supe === supermercado
     )![0];
@@ -82,16 +87,25 @@ class Auto {
   async scrapAndInform({ listPath }: { listPath: string }) {
     const res = await scrapQueue.add(async () => {
       const t0 = performance.now();
-      const progress = await downloadList(listPath);
-      return { took: performance.now() - t0, progress };
+
+      const sub = Bun.spawn({
+        cmd: ["scraper-rs", "fetch-list", listPath],
+        stdio: ["ignore", "pipe", "inherit"],
+      });
+      const text = await readableStreamToText(sub.stdout);
+      const code = await sub.exited;
+      if (code !== 0) throw new Error(`scraper-rs threw ${code}`);
+
+      return { took: performance.now() - t0, text };
     });
 
     if (res) {
-      const { took, progress } = res;
+      const { took, text } = res;
       this.inform(
-        `Procesado ${listPath} (${progress.done} ok, ${
-          progress.skipped
-        } skipped, ${progress.errors.length} errores) (tardó ${formatMs(took)})`
+        `Procesado ${listPath} (${text}) (tardó ${formatMs(took)})`
+        //(${progress.done} ok, ${
+        //   progress.skipped
+        // } skipped, ${progress.errors.length} errores)
       );
     } else {
       this.inform(`Algo falló en ${listPath}`);

scraper/cli.ts

@@ -4,9 +4,14 @@ import { scrapDiaProducts } from "../link-scrapers/dia.js";
 import { scrapJumboProducts } from "../link-scrapers/jumbo.js";
 import { auto } from "./auto.js";
 import { downloadList, getProduct } from "./scrap.js";
+import Cron from "croner";
 
 if (process.argv[2] === "auto") {
   await auto();
+} else if (process.argv[2] === "cron") {
+  Cron("0 2 * * *", () => {
+    auto();
+  });
 } else if (process.argv[2] === "scrap-carrefour-links") {
   await scrapCarrefourProducts();
 } else if (process.argv[2] === "scrap-dia-links") {

scraper/package.json

@@ -13,6 +13,7 @@
   "dependencies": {
     "@aws-sdk/client-s3": "^3.478.0",
     "@aws-sdk/lib-storage": "^3.478.0",
+    "croner": "^8.0.0",
     "date-fns": "^3.0.6",
     "db-datos": "workspace:^",
     "drizzle-orm": "^0.29.1",

sitio/package.json

@@ -38,7 +38,6 @@
     "better-sqlite3": "^9.2.2",
     "chart.js": "^4.4.1",
     "chartjs-adapter-dayjs-4": "^1.0.4",
-    "croner": "^8.0.0",
     "dayjs": "^1.11.10",
     "drizzle-orm": "^0.29.1"
   }

sitio/src/hooks.server.ts

@@ -1,12 +0,0 @@
-import { spawn } from "child_process";
-import Cron from "croner";
-
-if (process.env.NODE_ENV === "production") {
-  const job = Cron("15 3 * * *", () => {
-    runScraper();
-  });
-}
-
-function runScraper() {
-  spawn("bun", ["/bin/scraper", "auto"], { stdio: "inherit" });
-}

sitio/src/routes/+page.svelte

@@ -30,6 +30,9 @@
     <li>
       <a href="/ean/7790895000430">Gaseosa Coca-Cola Sabor Original 1,5 Lts.</a>
     </li>
+    <li>
+      <a href="/ean/7792200000128">Bizcochos Agridulc 9 De Oro Paq 200 Grm</a>
+    </li>
   </ul>
 </section>