71 lines
2.2 KiB
TypeScript
71 lines
2.2 KiB
TypeScript
import { defineNuxtModule } from '@nuxt/kit'
|
|
import { Database } from "bun:sqlite";
|
|
|
|
interface Schema
|
|
{
|
|
type: string;
|
|
name: string;
|
|
tbl_name: string;
|
|
sql: string;
|
|
}
|
|
interface Structure
|
|
{
|
|
cid: number;
|
|
name: string;
|
|
type: string;
|
|
[key: string]: number | string | null;
|
|
}
|
|
|
|
export default defineNuxtModule({
|
|
setup (options, nuxt) {
|
|
nuxt.hook('build:done', () => {
|
|
console.log("Building database");
|
|
|
|
const template = new Database(nuxt.options.runtimeConfig.templateFile);
|
|
const schema = template.query(`SELECT * FROM sqlite_schema WHERE type = 'table'`).all() as Schema[];
|
|
|
|
const db = new Database(nuxt.options.runtimeConfig.dbFile);
|
|
db.exec(`PRAGMA foreign_keys = 0`);
|
|
const oldSchema = db.query(`SELECT * FROM sqlite_schema WHERE type = 'table'`).all() as Schema[];
|
|
const structure = db.query(`SELECT * FROM pragma_table_info(?1)`);
|
|
|
|
(db.transaction((tables: Schema[], oldTables: Schema[]) => {
|
|
for(const table of tables)
|
|
{
|
|
const oldIdx = oldTables.findIndex(e => e.name === table.name);
|
|
|
|
if(table.name === 'sqlite_sequence')
|
|
{
|
|
oldTables.splice(oldIdx, 1);
|
|
continue;
|
|
}
|
|
|
|
const columns = structure.all(table.name) as Structure[];
|
|
|
|
if(oldIdx !== -1)
|
|
{
|
|
oldTables.splice(oldIdx, 1);
|
|
|
|
db.exec(`ALTER TABLE ${table.name} RENAME TO ${table.name}_old`);
|
|
db.exec(table.sql);
|
|
db.exec(`INSERT INTO ${table.name} (${columns.map(e => `"${e.name}"`).join(', ')}) SELECT * FROM ${table.name}_old`);
|
|
db.exec(`DROP TABLE ${table.name}_old`);
|
|
}
|
|
else
|
|
{
|
|
db.exec(table.sql);
|
|
}
|
|
}
|
|
for(const table of oldTables)
|
|
{
|
|
db.exec(`DROP TABLE ${table.name}`);
|
|
}
|
|
})).immediate(schema, oldSchema);
|
|
|
|
db.exec(`PRAGMA foreign_keys = 1`);
|
|
|
|
db.close();
|
|
template.close();
|
|
});
|
|
}
|
|
}) |