-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
1,103 additions
and
47 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,24 @@ | ||
| const sessionDb = require("./utils/SessionDatabase"); | ||
| const express = require("express"); | ||
| const router = express.Router(); | ||
|
|
||
| router.get("/settings", async (_, res) => { | ||
| try { | ||
| const settings = await sessionDb.getSetting(); | ||
| res.send(settings); | ||
| } catch (err) { | ||
| return res.status(400).send({ error: err.message }); | ||
| } | ||
| }); | ||
|
|
||
| router.post("/settings", async (req, res) => { | ||
| try { | ||
| await sessionDb.setSetting(req.body); | ||
| res.send({ success: true }); | ||
| } catch (err) { | ||
| console.error(err); | ||
| return res.status(400).send({ error: err.message }); | ||
| } | ||
| }); | ||
|
|
||
| module.exports = router; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,135 @@ | ||
| const path = require("path"); | ||
| const fs = require("fs").promises; | ||
| const sqlite3 = require("sqlite3"); | ||
| const sqlite = require("sqlite"); | ||
| const constants = require("./Constants"); | ||
|
|
||
| const MODES = constants.MODES; | ||
| const DB_FILE_NAME = "explorer.db"; | ||
|
|
||
| class SessionDatabase { | ||
| constructor() { | ||
| this.isInitialized = false; | ||
| const dbPath = process.env.KUZU_PATH; | ||
| if (!dbPath) { | ||
| return; | ||
| } | ||
| this.dbPath = path.resolve(path.join(dbPath, DB_FILE_NAME)); | ||
| this.isReadOnly = !!( | ||
| process.env.MODE && process.env.MODE !== MODES.READ_WRITE | ||
| ); | ||
| this.initSqlite(); | ||
| } | ||
|
|
||
| initSqlite() { | ||
| if (this.isInitialized) { | ||
| return null; | ||
| } | ||
| if (this.sqlInitPromise) { | ||
| return this.sqlInitPromise; | ||
| } | ||
| this.sqlInitPromise = (async () => { | ||
| let isDbFileExists = false; | ||
| try { | ||
| await fs.access(this.dbPath, fs.constants.R_OK); | ||
| isDbFileExists = true; | ||
| } catch (err) { | ||
| // File does not exist | ||
| } | ||
| // Double check if the file is writable | ||
| if (isDbFileExists) { | ||
| try { | ||
| await fs.access(this.dbPath, fs.constants.W_OK); | ||
| } catch (err) { | ||
| this.isReadOnly = true; | ||
| } | ||
| } else { | ||
| try { | ||
| await fs.writeFile(this.dbPath, ""); | ||
| await fs.access(this.dbPath, fs.constants.W_OK); | ||
| await fs.unlink(this.dbPath); | ||
| } catch (err) { | ||
| this.isReadOnly = true; | ||
| } | ||
| if (this.isReadOnly) { | ||
| // In read-only mode, if the db file does not exist, we should not create it, | ||
| // but if it exists, we can still use it. | ||
| return; | ||
| } | ||
| try { | ||
| await new Promise((resolve, reject) => { | ||
| const newDb = new sqlite3.Database( | ||
| this.dbPath, | ||
| sqlite3.OPEN_READWRITE | sqlite3.OPEN_CREATE, | ||
| (err) => { | ||
| if (err) { | ||
| return reject(err); | ||
| } | ||
| newDb.close(); | ||
| resolve(); | ||
| } | ||
| ); | ||
| }); | ||
| } catch (err) { | ||
| return; | ||
| } | ||
| } | ||
| this.db = await sqlite.open({ | ||
| filename: this.dbPath, | ||
| driver: sqlite3.Database, | ||
| }); | ||
| if (!isDbFileExists) { | ||
| await this.createDbSchema(); | ||
| } | ||
| this.isInitialized = true; | ||
| delete this.sqlInitPromise; | ||
| })(); | ||
| return this.sqlInitPromise; | ||
| } | ||
|
|
||
| async createDbSchema() { | ||
| await this.db.exec(` | ||
| CREATE TABLE settings ( | ||
| key TEXT PRIMARY KEY, | ||
| value TEXT | ||
| ); | ||
| `); | ||
| await this.db.exec(` | ||
| CREATE TABLE history ( | ||
| uuid TEXT PRIMARY KEY, | ||
| isQueryGenerationMode BOOLEAN, | ||
| gptQuestion TEXT, | ||
| cypherQuery TEXT | ||
| );`); | ||
| } | ||
|
|
||
| async getSetting(key = "allSettings") { | ||
| await this.initSqlite(); | ||
| if (!this.isInitialized) { | ||
| return {}; | ||
| } | ||
| let settings = await this.db.get( | ||
| "SELECT * FROM settings WHERE key = ?", | ||
| key | ||
| ); | ||
| settings = settings ? settings.value : {}; | ||
| return settings; | ||
| } | ||
|
|
||
| async setSetting(value, key = "allSettings") { | ||
| await this.initSqlite(); | ||
| if (!this.isInitialized) { | ||
| return; | ||
| } | ||
| if (this.isReadOnly) { | ||
| return; | ||
| } | ||
| await this.db.run( | ||
| `INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)`, | ||
| key, | ||
| JSON.stringify(value) | ||
| ); | ||
| } | ||
| } | ||
|
|
||
| module.exports = new SessionDatabase(); |
Oops, something went wrong.