Mirror/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2025-12-12 14:37:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

if in mirror mode, import CSV files

Browse Source
This commit is contained in:
Michael C
2021-10-21 03:49:25 -04:00
parent 93536976d0
commit 0163a52e55
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/databases/databases.ts
View File

@@ -3,7 +3,7 @@ import { Sqlite } from "./Sqlite";
import { Mysql } from "./Mysql"; import { Mysql } from "./Mysql";
import { Postgres } from "./Postgres"; import { Postgres } from "./Postgres";
import { IDatabase } from "./IDatabase"; import { IDatabase } from "./IDatabase";
import { readdirSync } from "fs";
let db: IDatabase; let db: IDatabase;
let privateDB: IDatabase; let privateDB: IDatabase;
@@ -68,6 +68,14 @@ async function initDb(): Promise<void> {
// Attach private db to main db // Attach private db to main db
(db as Sqlite).attachDatabase(config.privateDB, "privateDB"); (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 { export {