Mirrors/notesnook
1
0
Fork 0
mirror of https://github.com/streetwriters/notesnook.git synced 2025-12-23 23:19:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

core: separate db setup & init

Browse Source
This commit is contained in:
Abdullah Atta
2024-03-29 15:38:36 +05:00
parent 8fb8c54157
commit a15978bf6a
1 changed files with 48 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
packages/core/src/database/index.ts
View File

@@ -255,8 +255,10 @@ const DataMappers: Partial<Record<ItemType, (row: any) => void>> = {
}
};
async function init(db: Kysely<RawDatabaseSchema>, options: SQLiteOptions) {
try {
async function setupDatabase(
db: Kysely<RawDatabaseSchema>,
options: SQLiteOptions
) {
if (options.password)
await sql`PRAGMA key = ${sql.ref(options.password)}`.execute(db);
await sql`PRAGMA journal_mode = ${sql.raw(
@@ -288,7 +290,10 @@ async function init(db: Kysely<RawDatabaseSchema>, options: SQLiteOptions) {
await sql`PRAGMA locking_mode = ${sql.raw(options.lockingMode)}`.execute(
db
);
}
export async function initializeDatabase(db: Kysely<RawDatabaseSchema>) {
try {
const migrator = new Migrator({
db,
provider: new NNMigrationProvider()
@@ -330,11 +335,19 @@ export type SQLiteOptions = {
export async function createDatabase(name: string, options: SQLiteOptions) {
const db = new Kysely<RawDatabaseSchema>({
dialect: options.dialect(name, async () => {
await init(db, options);
await db.connection().execute(async (db) => {
await setupDatabase(db, options);
await initializeDatabase(db);
});
}),
plugins: [new SqliteBooleanPlugin()]
});
if (!options.skipInitialization) return await init(db, options);
if (!options.skipInitialization)
await db.connection().execute(async (db) => {
await setupDatabase(db, options);
await initializeDatabase(db);
});
return db;
}