1
0
Fork 0

[javascript] began reworking token handling to match what is done with the haskell client

This commit is contained in:
Julien Dessaux 2023-09-02 01:06:11 +02:00
parent a50ea1be7c
commit 9a71163dda
Signed by: adyxax
GPG key ID: F92E51B86E07177E
5 changed files with 50 additions and 41 deletions

View file

@ -1,8 +1,7 @@
CREATE TABLE schema_version ( CREATE TABLE schema_version (
version INTEGER NOT NULL version INTEGER NOT NULL
); );
CREATE TABLE config ( CREATE TABLE tokens (
id INTEGER PRIMARY KEY, id INTEGER PRIMARY KEY,
key TEXT NOT NULL UNIQUE, data TEXT NOT NULL
value TEXT NOT NULL
); );

View file

@ -1,23 +0,0 @@
import db from './db.js';
const getTokenStatement = db.prepare(`SELECT value->>'token' as token from config where key = 'register_data';`);
const registerAgentStatement = db.prepare(`INSERT INTO config(key, value) VALUES ('register_data', json(?));`);
export function getToken() {
try {
return getTokenStatement.get().token;
} catch (err) {
console.log(err);
return null;
}
}
export function registerAgent(data) {
try {
registerAgentStatement.run(JSON.stringify(data));
return true;
} catch (err) {
console.log(err);
return false;
}
}

View file

@ -15,7 +15,8 @@ const db = new Database(
db.pragma('foreign_keys = ON'); db.pragma('foreign_keys = ON');
db.pragma('journal_mode = WAL'); db.pragma('journal_mode = WAL');
db.transaction(function migrate() { function init() {
db.transaction(function migrate() {
let version; let version;
try { try {
version = db.prepare('SELECT version FROM schema_version').get().version; version = db.prepare('SELECT version FROM schema_version').get().version;
@ -28,6 +29,22 @@ db.transaction(function migrate() {
version++; version++;
} }
db.exec(`DELETE FROM schema_version; INSERT INTO schema_version (version) VALUES (${version});`); db.exec(`DELETE FROM schema_version; INSERT INTO schema_version (version) VALUES (${version});`);
})(); })();
}
export function reset() {
const indices = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'index';`).all();
const tables = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'table';`).all();
const triggers = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'trigger';`).all();
const views = db.prepare(`SELECT name FROM sqlite_master WHERE type = 'view';`).all();
indices.forEach(elt => db.exec(`DROP INDEX ${elt.name};`));
tables.forEach(elt => db.exec(`DROP TABLE ${elt.name};`));
triggers.forEach(elt => db.exec(`DROP TRIGGER ${elt.name};`));
views.forEach(elt => db.exec(`DROP VIEW ${elt.name};`));
db.exec(`VACUUM;`);
init();
}
init();
export default db; export default db;

16
nodejs/database/tokens.js Normal file
View file

@ -0,0 +1,16 @@
import db from './db.js';
const addTokenStatement = db.prepare(`INSERT INTO tokens(data) VALUES (?);`);
const getTokenStatement = db.prepare(`SELECT data FROM tokens;`);
export function addToken(token) {
return addTokenStatement.run(token).lastInsertRowid;
}
export function getToken() {
const data = getTokenStatement.get();
if (data === undefined) {
return null;
}
return data.data;
}

View file

@ -1,7 +1,7 @@
import * as fs from 'fs'; import * as fs from 'fs';
import * as events from 'events'; import * as events from 'events';
import { getToken } from '../database/config.js'; import { getToken } from '../database/tokens.js';
import { PriorityQueue } from './priority_queue.js'; import { PriorityQueue } from './priority_queue.js';
// queue processor module variables // queue processor module variables