53 lines
1.2 KiB
JavaScript
53 lines
1.2 KiB
JavaScript
const sqlite3 = require('sqlite3')
|
|
const {open} = require('sqlite')
|
|
const config = require('./../config');
|
|
|
|
async function validateDatabase(db, logger) {
|
|
try {
|
|
db = await open({
|
|
filename: config.databaseFile,
|
|
driver: sqlite3.Database,
|
|
mode: sqlite3.OPEN_READWRITE
|
|
});
|
|
await db.get('SELECT * FROM accounts LIMIT 1')
|
|
await db.get('SELECT * FROM devices LIMIT 1')
|
|
await db.get('SELECT * FROM drives LIMIT 1')
|
|
await db.get('SELECT * FROM drive_segments LIMIT 1')
|
|
|
|
} catch (exception) {
|
|
logger.error(exception);
|
|
process.exit();
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
module.exports = async (logger) => {
|
|
let db;
|
|
|
|
try {
|
|
db = await open({
|
|
filename: config.databaseFile,
|
|
driver: sqlite3.Database,
|
|
mode: sqlite3.OPEN_READWRITE
|
|
});
|
|
|
|
} catch (exception) {
|
|
logger.error(exception);
|
|
process.exit();
|
|
}
|
|
|
|
// I'm not sure we _really_ need to wait for this, since it'll exit the application if it's invalid anyway.
|
|
|
|
await validateDatabase(db, logger);
|
|
|
|
|
|
return {
|
|
db,
|
|
models: {
|
|
drivesModel: require('./drives')(db),
|
|
users: require('./users')(db),
|
|
}
|
|
}
|
|
} |