Use cookie store

This lets us actually set the location of the store we want

Author: OverHash

ARCHITECTURE.md

@@ -11,6 +11,33 @@ The countdown API requires an API key for all requests, which can be found in th
 Running a Postgres instance with the following tables:
 
 ```
+CREATE TYPE supermarket AS ENUM ('Countdown', 'New World');
+
+countdown_stores
+----------------
+CREATE TABLE countdown_stores (
+	id INTEGER PRIMARY KEY
+	name VARCHAR(255) NOT NULL,
+)
+
+stores
+----------------
+CREATE TABLE stores (
+	id SERIAL PRIMARY KEY,
+	supermarket supermarket NOT NULL,
+	countdown_store_id INTEGER UNIQUE,
+
+	CONSTRAINT fk_countdown_id
+		FOREIGN KEY(countdown_store_id)
+			REFERENCES countdown_stores(id),
+
+	CONSTRAINT chk_supermarket_type
+		CHECK (
+			(supermarket = 'Countdown' AND countdown_store_id IS NOT NULL)
+			-- in the future, we add an OR to this to check new world store id
+		)
+)
+
 countdown_products
 ------------------
 CREATE TABLE countdown_products (
@@ -38,10 +65,14 @@ CREATE TABLE prices (
 	product_id INTEGER NOT NULL,
 	time TIMESTAMPTZ NOT NULL DEFAULT NOW(),
 	cost_in_cents INTEGER NOT NULL,
-	supermarket VARCHAR(255) NOT NULL,
+	store_id INTEGER NOT NULL,
 
 	CONSTRAINT fk_product
 		FOREIGN KEY(product_id)
 			REFERENCES products(id)
+
+	CONSTRAINT fk_store_id
+		FOREIGN KEY(store_id)
+			REFERENCES stores(id)
 )
 ```

Cargo.lock

@@ -14,11 +14,7 @@ name = "supermarket-tracker"
 path = "src/main.rs"
 
 [dependencies]
-reqwest = { version = "0.11.22", default-features = false, features = [
-	"gzip",
-	"json",
-	"native-tls",
-] }
+reqwest = { version = "0.11.22", default-features = false, features = ["gzip", "json", "native-tls", "cookies"] }
 serde = { version = "1.0.193", features = ["derive"] }
 serde_json = "1.0.108"
 tokio = { version = "1.34.0", features = ["full"] }

Cargo.toml

@@ -132,22 +132,14 @@ pub async fn get_products(
                 price,
                 sku,
                 ..
-            } => {
-                if &sku == "133211" {
-                    tracing::warn!(
-                        "Found {name} with price {} and barcode {barcode}",
-                        price.sale_price
-                    );
-                }
-                Some(Product {
-                    name,
-                    barcode,
-                    sku,
-                    // convert to cents from dollars
-                    #[allow(clippy::cast_possible_truncation)]
-                    per_unit_price: (price.sale_price * 100.0).round() as i32,
-                })
-            }
+            } => Some(Product {
+                name,
+                barcode,
+                sku,
+                // convert to cents from dollars
+                #[allow(clippy::cast_possible_truncation)]
+                per_unit_price: (price.sale_price * 100.0).round() as i32,
+            }),
             _ => None,
         })
         .collect::<HashSet<Product>>();

src/countdown/get_products.rs

@@ -39,6 +39,7 @@ pub async fn run(connection: PgPool, should_insert: bool) -> Result<(), Report<A
         reqwest::Client::builder()
             .user_agent(DEFAULT_USER_AGENT)
             .default_headers(default_headers)
+            .cookie_store(true)
             .build()
             .change_context(ApplicationError::HttpError)
     }?;