-
Notifications
You must be signed in to change notification settings - Fork 38
/
SchemaGenerator.ts
86 lines (75 loc) · 2.7 KB
/
SchemaGenerator.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
import * as prettier from 'prettier';
import { Container } from 'typedi';
import { logger, Logger } from '../core/logger';
import {
generateEnumMapImports,
entityToCreateInput,
entityToCreateManyArgs,
entityToOrderByEnum,
entityToUpdateInput,
entityToUpdateInputArgs,
entityToWhereArgs,
entityToWhereInput,
entityToWhereUniqueInput
} from './TypeORMConverter';
import { getMetadataStorage, ModelMetadata } from '../metadata';
export class SchemaGenerator {
static logger: Logger = Container.has('warthog.logger')
? Container.get('warthog.logger')
: logger;
static generate(
// This will reference 'warthog in the deployed module, but we need to do a relative import in the examples library
warthogImportPath = 'warthog'
): string {
const metadata = getMetadataStorage();
let template = `
// This file has been auto-generated by Warthog. Do not update directly as it
// will be re-written. If you need to change this file, update models or add
// new TypeGraphQL objects
// @ts-ignore
import { GraphQLDateTime as DateTime } from 'graphql-iso-date';
import { GraphQLID as ID } from 'graphql';
// @ts-ignore
import { ArgsType, Field as TypeGraphQLField, Float, InputType as TypeGraphQLInputType, Int } from 'type-graphql';
// @ts-ignore
import { registerEnumType } from 'type-graphql';
// @ts-ignore eslint-disable-next-line @typescript-eslint/no-var-requires
const { GraphQLJSONObject } = require('graphql-type-json');
// @ts-ignore
import { BaseWhereInput, JsonObject, PaginationArgs } from '${warthogImportPath}';
${generateEnumMapImports().join('')}
`;
// console.log('metadata: ', metadata);
// logger.logObject(metadata.getModels());
Object.keys(metadata.getModels()).forEach((modelName: string) => {
const model: ModelMetadata = metadata.getModel(modelName);
template += `
${entityToOrderByEnum(model)}
${entityToWhereInput(model)}
${entityToWhereUniqueInput(model)}
${entityToCreateInput(model)}
${entityToUpdateInput(model)}
${entityToWhereArgs(model)}
${entityToCreateManyArgs(model)}
${entityToUpdateInputArgs(model)}
`;
});
return this.format(template);
}
static format(code: string, options: prettier.Options = {}) {
try {
// TODO: grab our prettier options (single quote, etc...)
return prettier.format(code, {
...options,
parser: 'typescript'
});
} catch (e) {
this.logger.error(
`There is a syntax error in generated code, unformatted code printed, error: ${JSON.stringify(
e
)}`
);
return code;
}
}
}