Skip to content

fix: improve prisma reset command #568

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
exKAZUu merged 2 commits into from
Dec 16, 2025
Merged

fix: improve prisma reset command #568

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
5dd2511
fix: improve prisma reset command
exKAZUu Dec 16, 2025
83c0f9b
refactor: extract shared constants and simplify reset method in prism…
exKAZUu Dec 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 39 additions & 18 deletions packages/wb/src/scripts/prismaScripts.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,14 @@ import path from 'node:path';

import type { Project } from '../project.js';

const FILE_SCHEMA = 'file:';

const POSSIBLE_PRISMA_PATHS = [
{ schemaPath: path.join('prisma', 'schema.prisma'), dbPath: 'prisma' },
{ schemaPath: path.join('prisma', 'schema'), dbPath: path.join('prisma', 'schema') },
{ schemaPath: path.join('db', 'schema.prisma'), dbPath: 'db' },
];

/**
* A collection of scripts for executing Prisma commands.
* Note that `PRISMA` is replaced with `YARN prisma` or `YARN blitz prisma`
Expand Down Expand Up @@ -43,11 +51,16 @@ class PrismaScripts {

reset(project: Project, additionalOptions = ''): string {
// cf. https://www.prisma.io/docs/guides/database/seed-database#integrated-seeding-with-prisma-migrate
const steps: string[] = [];
const cleanupCommand = cleanUpSqliteDbIfNeeded(project);
if (cleanupCommand) steps.push(cleanupCommand);
const resetCommand = ['PRISMA migrate reset --force', additionalOptions].filter(Boolean).join(' ');
steps.push(resetCommand);
if (project.packageJson.dependencies?.blitz) {
// Blitz does not trigger seed automatically, so we need to run it manually.
return `PRISMA migrate reset --force ${additionalOptions} && ${this.seed(project)}`;
steps.push(this.seed(project));
}
return `PRISMA migrate reset --force ${additionalOptions}`;
return steps.filter(Boolean).join(' && ');
}

restore(project: Project, outputPath: string): string {
Expand All @@ -63,7 +76,6 @@ class PrismaScripts {
}

studio(project: Project, dbUrlOrPath?: string, additionalOptions = ''): string {
const FILE_SCHEMA = 'file:';
let prefix = '';
// Deal with Prisma issue: https://github.com/prisma/studio/issues/1273
if (dbUrlOrPath) {
Expand All @@ -75,21 +87,10 @@ class PrismaScripts {
prefix = `DATABASE_URL=${FILE_SCHEMA}${absolutePath} `;
}
} else if (project.env.DATABASE_URL?.startsWith(FILE_SCHEMA)) {
const POSSIBLE_PATHS = [
{ schemaPath: path.join('prisma', 'schema.prisma'), dbPath: 'prisma' },
{ schemaPath: path.join('prisma', 'schema'), dbPath: path.join('prisma', 'schema') },
{ schemaPath: path.join('db', 'schema.prisma'), dbPath: 'db' },
];
for (const { dbPath, schemaPath } of POSSIBLE_PATHS) {
if (fs.existsSync(path.resolve(project.dirPath, schemaPath))) {
const absolutePath = path.resolve(
project.dirPath,
dbPath,
project.env.DATABASE_URL.slice(FILE_SCHEMA.length)
);
prefix = `DATABASE_URL=${FILE_SCHEMA}${absolutePath} `;
break;
}
const baseDir = getPrismaBaseDir(project);
if (baseDir) {
const absolutePath = path.resolve(project.dirPath, baseDir, project.env.DATABASE_URL.slice(FILE_SCHEMA.length));
prefix = `DATABASE_URL=${FILE_SCHEMA}${absolutePath} `;
}
}
return `${prefix}PRISMA studio ${additionalOptions}`;
Expand All @@ -100,4 +101,24 @@ function getDatabaseDirPath(project: Project): string {
return project.packageJson.dependencies?.blitz ? 'db/mount' : 'prisma/mount';
}

function getPrismaBaseDir(project: Project): string | undefined {
return POSSIBLE_PRISMA_PATHS.find(({ schemaPath }) => fs.existsSync(path.resolve(project.dirPath, schemaPath)))
?.dbPath;
}

function cleanUpSqliteDbIfNeeded(project: Project): string | undefined {
const dbUrl = project.env.DATABASE_URL;
if (!dbUrl?.startsWith(FILE_SCHEMA)) return;

const rawDbPath = dbUrl.slice(FILE_SCHEMA.length).replace(/[?#].*$/, '');
if (!rawDbPath) return;

const baseDir = getPrismaBaseDir(project);
const absolutePath = path.isAbsolute(rawDbPath)
? rawDbPath
: path.resolve(project.dirPath, baseDir ?? '.', rawDbPath);

return `rm -f "${absolutePath}" "${absolutePath}-wal" "${absolutePath}-shm"`;
}

export const prismaScripts = new PrismaScripts();