mirror of
https://github.com/GiriNeko/YesPlayMusic.git
synced 2025-12-16 05:08:04 +00:00
102 lines
3.2 KiB
TypeScript
102 lines
3.2 KiB
TypeScript
import { app } from 'electron'
|
|
import path from 'path'
|
|
import { PrismaClient } from '../prisma/client'
|
|
import { isDev, isWindows } from './env'
|
|
import log from './log'
|
|
import { createFileIfNotExist, dirname, isFileExist } from './utils'
|
|
import fs from 'fs'
|
|
import { dialog } from 'electron'
|
|
|
|
export const dbPath = path.join(app.getPath('userData'), 'r3play.db')
|
|
export const dbUrl = 'file:' + (isWindows ? '' : '//') + dbPath
|
|
log.info('[prisma] dbUrl', dbUrl)
|
|
|
|
const extraResourcesPath = app.getAppPath().replace('app.asar', '') // impacted by extraResources setting in electron-builder.yml
|
|
function getPlatformName(): string {
|
|
const isDarwin = process.platform === 'darwin'
|
|
if (isDarwin && process.arch === 'arm64') {
|
|
return process.platform + 'Arm64'
|
|
}
|
|
|
|
return process.platform
|
|
}
|
|
const platformName = getPlatformName()
|
|
export const platformToExecutables: any = {
|
|
win32: {
|
|
migrationEngine: 'node_modules/@prisma/engines/migration-engine-windows.exe',
|
|
queryEngine: 'node_modules/@prisma/engines/query_engine-windows.dll.node',
|
|
},
|
|
linux: {
|
|
migrationEngine: 'node_modules/@prisma/engines/migration-engine-debian-openssl-1.1.x',
|
|
queryEngine: 'node_modules/@prisma/engines/libquery_engine-debian-openssl-1.1.x.so.node',
|
|
},
|
|
darwin: {
|
|
migrationEngine: 'node_modules/@prisma/engines/migration-engine-darwin',
|
|
queryEngine: 'node_modules/@prisma/engines/libquery_engine-darwin.dylib.node',
|
|
},
|
|
darwinArm64: {
|
|
migrationEngine: 'node_modules/@prisma/engines/migration-engine-darwin-arm64',
|
|
queryEngine: 'node_modules/@prisma/engines/libquery_engine-darwin-arm64.dylib.node',
|
|
},
|
|
}
|
|
export const queryEnginePath = path.join(
|
|
extraResourcesPath,
|
|
platformToExecutables[platformName].queryEngine
|
|
)
|
|
|
|
log.info('[prisma] dbUrl', dbUrl)
|
|
|
|
// Hacky, but putting this here because otherwise at query time the Prisma client
|
|
// gives an error "Environment variable not found: DATABASE_URL" despite us passing
|
|
// the dbUrl into the prisma client constructor in datasources.db.url
|
|
process.env.DATABASE_URL = dbUrl
|
|
|
|
createFileIfNotExist(dbPath)
|
|
|
|
// @ts-expect-error
|
|
let prisma: PrismaClient = null
|
|
try {
|
|
prisma = new PrismaClient({
|
|
log: isDev ? ['info', 'warn', 'error'] : ['error'],
|
|
datasources: {
|
|
db: {
|
|
url: dbUrl,
|
|
},
|
|
},
|
|
// see https://github.com/prisma/prisma/discussions/5200
|
|
// @ts-expect-error internal prop
|
|
// __internal: {
|
|
// engine: {
|
|
// binaryPath: queryEnginePath,
|
|
// },
|
|
// },
|
|
})
|
|
log.info('[prisma] prisma initialized')
|
|
} catch (e) {
|
|
log.error('[prisma] failed to init prisma', e)
|
|
dialog.showErrorBox('Failed to init prisma', String(e))
|
|
app.exit()
|
|
}
|
|
|
|
export const initDatabase = async () => {
|
|
try {
|
|
const initSQLFile = fs
|
|
.readFileSync(path.join(dirname, 'migrations/init.sql'), 'utf-8')
|
|
.toString()
|
|
const tables = initSQLFile.split(';')
|
|
await Promise.all(
|
|
tables.map(sql => {
|
|
if (!sql.trim()) return
|
|
return prisma.$executeRawUnsafe(sql.trim()).catch(() => {
|
|
log.error('[prisma] failed to execute init sql >>> ', sql.trim())
|
|
})
|
|
})
|
|
)
|
|
} catch (e) {
|
|
dialog.showErrorBox('Failed to init prisma database', String(e))
|
|
app.exit()
|
|
}
|
|
log.info('[prisma] database initialized')
|
|
}
|
|
|
|
export default prisma
|