From 0163a52e5599a2f729412ebbeeaecaa15a25faa4 Mon Sep 17 00:00:00 2001
From: Michael C <michael@mchang.name>
Date: Thu, 21 Oct 2021 03:49:25 -0400
Subject: [PATCH] if in mirror mode, import CSV files

---
 src/databases/databases.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/databases/databases.ts b/src/databases/databases.ts
index 96a5b5f..be15a32 100644
--- a/src/databases/databases.ts
+++ b/src/databases/databases.ts
@@ -3,7 +3,7 @@ import { Sqlite } from "./Sqlite";
 import { Mysql } from "./Mysql";
 import { Postgres } from "./Postgres";
 import { IDatabase } from "./IDatabase";
-
+import { readdirSync } from "fs";
 
 let db: IDatabase;
 let privateDB: IDatabase;
@@ -68,6 +68,14 @@ async function initDb(): Promise<void> {
         // Attach private db to main db
         (db as Sqlite).attachDatabase(config.privateDB, "privateDB");
     }
+
+    if (config.mode === "mirror") {
+        readdirSync("/mirror").forEach(async file => {
+            const fileName = file.slice(0,-4);
+            const filePath = `/mirror/${file}`;
+            await db.prepare("run", `COPY "${fileName}" FROM '${filePath}' WITH (FORMAT CSV, HEADER true);`);
+        });
+    }
 }
 
 export {