-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathgenerateFields.js
More file actions
61 lines (56 loc) · 2.04 KB
/
generateFields.js
File metadata and controls
61 lines (56 loc) · 2.04 KB
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
const { format } = require('prettier');
const getBaseType = type => (type.ofType ? getBaseType(type.ofType) : type);
const generateFieldsAndValues = (maxDepth, fields) => {
const valueArray = [];
const generateFields = (maxDepth, fields, parents = []) =>
Object.values(fields)
.map(field => {
const nextFields = getBaseType(field.type)._fields;
if (!nextFields) {
const value =
[...parents, field.name].join('{') +
'}'.repeat(parents.length);
valueArray.push(`'${value}'`);
return `${field.name}: '${value}'`;
}
if (parents.length > maxDepth) {
return;
}
return `${field.name}: {
${generateFields(maxDepth, nextFields, [
...parents,
field.name,
])}
}`;
})
.filter(Boolean);
return [valueArray, generateFields(maxDepth, fields)];
};
module.exports = {
plugin: (schema, _documents, { maxDepth, baseTypes }) => {
const typeMap = schema.getTypeMap();
const final = Object.values(typeMap)
.filter(type => baseTypes.includes(type.name))
.map(type => {
const [values, generatedFields] = generateFieldsAndValues(
maxDepth,
type._fields,
);
return `export type ${type.name}Fields = {
${generatedFields}
};
export const ${type.name}Fields: ${type.name}Fields = {
${generatedFields}
};
export type ${type.name}Field = ${values.join('|')};
`;
})
.join('');
return format(final, {
parser: 'typescript',
trailingComma: 'all',
tabWidth: 4,
singleQuote: true,
});
},
};