57 lines
1.7 KiB
TypeScript
57 lines
1.7 KiB
TypeScript
import { Model, Sequelize } from "sequelize-typescript";
|
|
import config from "../config";
|
|
import { LogService } from "matrix-js-snippets";
|
|
import User from "./models/User";
|
|
import UserScalarToken from "./models/UserScalarToken";
|
|
import Upstream from "./models/Upstream";
|
|
import * as Promise from "bluebird";
|
|
import WidgetRecord from "./models/WidgetRecord";
|
|
import * as path from "path";
|
|
import * as Umzug from "umzug";
|
|
import AppService from "./models/AppService";
|
|
import AppServiceUser from "./models/AppServiceUser";
|
|
|
|
class _DimensionStore {
|
|
private sequelize: Sequelize;
|
|
|
|
constructor() {
|
|
this.sequelize = new Sequelize({
|
|
dialect: 'sqlite',
|
|
database: "dimension",
|
|
storage: config.database.file,
|
|
username: "",
|
|
password: "",
|
|
logging: i => LogService.verbose("DimensionStore [SQL]", i)
|
|
});
|
|
this.sequelize.addModels(<Array<typeof Model>>[
|
|
User,
|
|
UserScalarToken,
|
|
Upstream,
|
|
WidgetRecord,
|
|
AppService,
|
|
AppServiceUser,
|
|
]);
|
|
}
|
|
|
|
public updateSchema(): Promise<any> {
|
|
LogService.info("DimensionStore", "Updating schema...");
|
|
|
|
const migrator = new Umzug({
|
|
storage: "sequelize",
|
|
storageOptions: {sequelize: this.sequelize},
|
|
migrations: {
|
|
params: [this.sequelize.getQueryInterface()],
|
|
path: path.join(__dirname, "migrations"),
|
|
}
|
|
});
|
|
|
|
return migrator.up();
|
|
}
|
|
}
|
|
|
|
export const DimensionStore = new _DimensionStore();
|
|
|
|
export function resolveIfExists<T>(record: T): Promise<T> {
|
|
if (!record) return Promise.reject("Record not found");
|
|
return Promise.resolve(record);
|
|
} |