chore: add env overrides for storage and database paths
This commit is contained in:
+5
-1
@@ -1,9 +1,13 @@
|
|||||||
const Database = require("better-sqlite3");
|
const Database = require("better-sqlite3");
|
||||||
|
const path = require("path");
|
||||||
|
|
||||||
|
const DATABASE_DIR = process.env.DATABASE_DIR || "/database";
|
||||||
|
const DATABASE_FILE = process.env.DATABASE_FILE || path.join(DATABASE_DIR, "users.db");
|
||||||
|
|
||||||
class MyDatabase {
|
class MyDatabase {
|
||||||
dbObj;
|
dbObj;
|
||||||
constructor() {
|
constructor() {
|
||||||
this.dbObj = new Database("/database/users.db");
|
this.dbObj = new Database(DATABASE_FILE);
|
||||||
this.dbObj.exec(`
|
this.dbObj.exec(`
|
||||||
CREATE TABLE IF NOT EXISTS users (
|
CREATE TABLE IF NOT EXISTS users (
|
||||||
id INTEGER PRIMARY KEY,
|
id INTEGER PRIMARY KEY,
|
||||||
|
|||||||
+5
-1
@@ -1,8 +1,12 @@
|
|||||||
const fs = require('fs');
|
const fs = require('fs');
|
||||||
const fsp = require('fs/promises');
|
const fsp = require('fs/promises');
|
||||||
|
const path = require('path');
|
||||||
const readline = require('readline');
|
const readline = require('readline');
|
||||||
const readLine = require("node:readline");
|
const readLine = require("node:readline");
|
||||||
const myDb = require('./myDatabase');
|
const myDb = require('./myDatabase');
|
||||||
|
|
||||||
|
const STORAGE_DIR = process.env.STORAGE_DIR || "/storage";
|
||||||
|
|
||||||
class Storage {
|
class Storage {
|
||||||
userId;
|
userId;
|
||||||
dbObj;
|
dbObj;
|
||||||
@@ -13,7 +17,7 @@ class Storage {
|
|||||||
}
|
}
|
||||||
|
|
||||||
get getDir() {
|
get getDir() {
|
||||||
return "/storage/" + Math.floor(this.userId / 100);
|
return path.join(STORAGE_DIR, String(Math.floor(this.userId / 100)));
|
||||||
}
|
}
|
||||||
get fileName() {
|
get fileName() {
|
||||||
return this.getDir + "/" + this.userId + ".jsonl";
|
return this.getDir + "/" + this.userId + ".jsonl";
|
||||||
|
|||||||
Reference in New Issue
Block a user