import { Prisma, PrismaClient } from "@prisma/client"; import * as crypto from "crypto"; const configVariables: ConfigVariables = { internal: { jwtSecret: { type: "string", defaultValue: crypto.randomBytes(256).toString("base64"), locked: true, }, }, general: { appName: { type: "string", defaultValue: "Pingvin Share", secret: false, }, appUrl: { type: "string", defaultValue: "http://localhost:3000", secret: false, }, showHomePage: { type: "boolean", defaultValue: "true", secret: false, }, }, share: { allowRegistration: { type: "boolean", defaultValue: "true", secret: false, }, allowUnauthenticatedShares: { type: "boolean", defaultValue: "false", secret: false, }, maxSize: { type: "number", defaultValue: "1000000000", secret: false, }, zipCompressionLevel: { type: "number", defaultValue: "9", }, }, email: { enableShareEmailRecipients: { type: "boolean", defaultValue: "false", secret: false, }, shareRecipientsSubject: { type: "string", defaultValue: "Files shared with you", }, shareRecipientsMessage: { type: "text", defaultValue: "Hey!\n\n{creator} shared some files with you, view or download the files with this link: {shareUrl}\n\nThe share will expire {expires}.\n\nNote: {desc}\n\nShared securely with Pingvin Share 🐧", }, reverseShareSubject: { type: "string", defaultValue: "Reverse share link used", }, reverseShareMessage: { type: "text", defaultValue: "Hey!\n\nA share was just created with your reverse share link: {shareUrl}\n\nShared securely with Pingvin Share 🐧", }, resetPasswordSubject: { type: "string", defaultValue: "Pingvin Share password reset", }, resetPasswordMessage: { type: "text", defaultValue: "Hey!\n\nYou requested a password reset. Click this link to reset your password: {url}\nThe link expires in a hour.\n\nPingvin Share 🐧", }, inviteSubject: { type: "string", defaultValue: "Pingvin Share invite", }, inviteMessage: { type: "text", defaultValue: "Hey!\n\nYou were invited to Pingvin Share. Click this link to accept the invite: {url}\n\nYour password is: {password}\n\nPingvin Share 🐧", }, }, smtp: { enabled: { type: "boolean", defaultValue: "false", secret: false, }, host: { type: "string", defaultValue: "", }, port: { type: "number", defaultValue: "0", }, email: { type: "string", defaultValue: "", }, username: { type: "string", defaultValue: "", }, password: { type: "string", defaultValue: "", obscured: true, }, }, }; type ConfigVariables = { [category: string]: { [variable: string]: Omit< Prisma.ConfigCreateInput, "name" | "category" | "order" >; }; }; const prisma = new PrismaClient({ datasources: { db: { url: process.env.DATABASE_URL || "file:../data/pingvin-share.db?connection_limit=1", }, }, }); async function seedConfigVariables() { for (const [category, configVariablesOfCategory] of Object.entries( configVariables )) { let order = 0; for (const [name, properties] of Object.entries( configVariablesOfCategory )) { const existingConfigVariable = await prisma.config.findUnique({ where: { name_category: { name, category } }, }); // Create a new config variable if it doesn't exist if (!existingConfigVariable) { await prisma.config.create({ data: { order, name, ...properties, category, }, }); } order++; } } } async function migrateConfigVariables() { const existingConfigVariables = await prisma.config.findMany(); for (const existingConfigVariable of existingConfigVariables) { const configVariable = configVariables[existingConfigVariable.category]?.[ existingConfigVariable.name ]; if (!configVariable) { await prisma.config.delete({ where: { name_category: { name: existingConfigVariable.name, category: existingConfigVariable.category, }, }, }); // Update the config variable if the metadata changed } else if ( JSON.stringify({ ...configVariable, name: existingConfigVariable.name, category: existingConfigVariable.category, value: existingConfigVariable.value, }) != JSON.stringify(existingConfigVariable) ) { await prisma.config.update({ where: { name_category: { name: existingConfigVariable.name, category: existingConfigVariable.category, }, }, data: { ...configVariable, name: existingConfigVariable.name, category: existingConfigVariable.category, value: existingConfigVariable.value, }, }); } } } seedConfigVariables() .then(() => migrateConfigVariables()) .then(async () => { await prisma.$disconnect(); }) .catch(async (e) => { console.error(e); await prisma.$disconnect(); process.exit(1); });