-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
drizzle.ts
105 lines (92 loc) · 2.94 KB
/
drizzle.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import path from "path";
import fs from "fs-extra";
import { type PackageJson } from "type-fest";
import { PKG_ROOT } from "~/consts.js";
import { type Installer } from "~/installers/index.js";
import { addPackageDependency } from "~/utils/addPackageDependency.js";
import { type AvailableDependencies } from "./dependencyVersionMap.js";
export const drizzleInstaller: Installer = ({
projectDir,
packages,
scopedAppName,
databaseProvider,
}) => {
const devPackages: AvailableDependencies[] = [
"drizzle-kit",
"eslint-plugin-drizzle",
];
if (databaseProvider === "planetscale") devPackages.push("mysql2");
if (databaseProvider === "postgres") devPackages.push("pg");
addPackageDependency({
projectDir,
dependencies: devPackages,
devMode: true,
});
addPackageDependency({
projectDir,
dependencies: [
"drizzle-orm",
(
{
planetscale: "@planetscale/database",
mysql: "mysql2",
postgres: "postgres",
sqlite: "@libsql/client",
} as const
)[databaseProvider],
],
devMode: false,
});
const extrasDir = path.join(PKG_ROOT, "template/extras");
const configFile = path.join(
extrasDir,
`config/drizzle-config-${
databaseProvider === "planetscale" ? "mysql" : databaseProvider
}.ts`
);
const configDest = path.join(projectDir, "drizzle.config.ts");
const schemaSrc = path.join(
extrasDir,
"src/server/db/schema-drizzle",
packages?.nextAuth.inUse
? `with-auth-${databaseProvider}.ts`
: `base-${databaseProvider}.ts`
);
const schemaDest = path.join(projectDir, "src/server/db/schema.ts");
// Replace placeholder table prefix with project name
let schemaContent = fs.readFileSync(schemaSrc, "utf-8");
schemaContent = schemaContent.replace(
"project1_${name}",
`${scopedAppName}_\${name}`
);
let configContent = fs.readFileSync(configFile, "utf-8");
configContent = configContent.replace("project1_*", `${scopedAppName}_*`);
const clientSrc = path.join(
extrasDir,
`src/server/db/index-drizzle/with-${databaseProvider}.ts`
);
const clientDest = path.join(projectDir, "src/server/db/index.ts");
// add db:push script to package.json
const packageJsonPath = path.join(projectDir, "package.json");
const packageJsonContent = fs.readJSONSync(packageJsonPath) as PackageJson;
packageJsonContent.scripts = {
...packageJsonContent.scripts,
"db:push": `drizzle-kit push:${
{
postgres: "pg",
sqlite: "sqlite",
mysql: "mysql",
planetscale: "mysql",
}[databaseProvider]
}`,
"db:studio": "drizzle-kit studio",
};
fs.copySync(configFile, configDest);
fs.mkdirSync(path.dirname(schemaDest), { recursive: true });
fs.writeFileSync(schemaDest, schemaContent);
fs.writeFileSync(configDest, configContent);
fs.copySync(clientSrc, clientDest);
fs.writeJSONSync(packageJsonPath, packageJsonContent, {
spaces: 2,
});
};