[javascript] began reworking token handling to match what is done with the haskell client
This commit is contained in:
parent
a50ea1be7c
commit
9a71163dda
5 changed files with 50 additions and 41 deletions
|
@ -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
|
|
||||||
);
|
);
|
||||||
|
|
|
@ -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;
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -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
16
nodejs/database/tokens.js
Normal 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;
|
||||||
|
}
|
|
@ -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
|
||||||
|
|
Loading…
Add table
Reference in a new issue