Skip to content

Commit a1117b3

Browse files
committed
feat(default): add '' around enum default values
1 parent 798297f commit a1117b3

File tree

4 files changed

+12
-8
lines changed

4 files changed

+12
-8
lines changed

prisma/dbml/schema.dbml

+1-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ Table User {
66
name String
77
posts Post
88
profile Profile
9-
role Role [not null, note: 'user role']
9+
role Role [not null, default: 'USER', note: 'user role']
1010
}
1111

1212
Table Profile {

prisma/schema.prisma

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ model User {
2323
posts Post[]
2424
profile Profile?
2525
/// user role
26-
role Role
26+
role Role @default(USER)
2727
}
2828

2929
/// User profile

src/generator/enums.ts

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
import { DMMF } from "@prisma/generator-helper";
1+
import { DMMF } from '@prisma/generator-helper';
22

33
export function generateEnums(enums: DMMF.DatamodelEnum[]): string[] {
44
return enums.map(
55
(e) => `Enum ${e.name} {\n` + generateEnumValues(e.values) + '\n}'
66
);
7-
};
7+
}
88

99
const generateEnumValues = (values: DMMF.EnumValue[]): string => {
1010
return values.map((value) => ` ${value.name}`).join('\n');

src/generator/table.ts

+8-4
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
1-
import { DMMF } from "@prisma/generator-helper";
1+
import { DMMF } from '@prisma/generator-helper';
22

3-
export function generateTables(models: DMMF.Model[]): string[] {
3+
export function generateTables(models: DMMF.Model[]): string[] {
44
return models.map(
55
(model) =>
66
`Table ${model.name} {\n` +
77
generateFields(model.fields) +
88
generateTableDocumentation(model) +
99
'\n}'
1010
);
11-
};
11+
}
1212

1313
const generateTableDocumentation = (model: DMMF.Model): string => {
1414
const doc = (model as any).documentation;
@@ -51,7 +51,11 @@ const generateColumnDefinition = (field: DMMF.Field): string => {
5151
typeof field.default === 'boolean' ||
5252
typeof field.default === 'number'
5353
) {
54-
columnDefinition.push(`default: ${field.default}`);
54+
if (field.kind === 'enum') {
55+
columnDefinition.push(`default: '${field.default}'`);
56+
} else {
57+
columnDefinition.push(`default: ${field.default}`);
58+
}
5559
}
5660

5761
if ((field as any).documentation) {

0 commit comments

Comments
 (0)