Add referral sites insertion to fimfic stats

PseudoBob-Delightus · Mar 11, 2024 · d0bb13e · d0bb13e
1 parent 485cb43
commit d0bb13e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/fimficstats/fimfic-stats.ts b/fimficstats/fimfic-stats.ts
@@ -199,6 +199,10 @@ async function mane() {
 			).run();
 		});
 
+		for (const site in referrals) {
+			db.query(sql.insert_referral_site(site)).run();
+		}
+
 		await sleep(start_time, Date.now(), request_interval);
 	}
 }

diff --git a/fimficstats/sql-patterns.ts b/fimficstats/sql-patterns.ts
@@ -196,9 +196,16 @@ export function insert_stats(
 
 export const referral_sites_table = `CREATE TABLE IF NOT EXISTS Referral_sites (
 	id                  integer     PRIMARY KEY,
-	site                string      NOT NULL
+	site                string      NOT NULL,
+
+	UNIQUE(site)
 )`;
 
+export function insert_referral_site(site: string) {
+	return `INSERT OR IGNORE INTO Referral_sites (site) 
+	VALUES ('${site}')`;
+}
+
 export const referrals_table = `CREATE TABLE IF NOT EXISTS Referrals (
 	story_id            integer     NOT NULL,
 	referral_site_id    integer     NOT NULL,