Skip to content

Add filterPath option #20

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

Open
dobrynin wants to merge 2 commits into
base: master
Choose a base branch
from
+55 −8
Open

Add filterPath option #20

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
aa6cd49
Add leafsPath option
dobrynin Jun 20, 2019
def5565
Change leafPath to filterFunc
dobrynin Jun 21, 2019
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
63 changes: 55 additions & 8 deletions index.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,18 @@ program
.option('--destDirPath [value]', 'dir you want to store the generated queries')
.option('--depthLimit [value]', 'query depth you want to limit(The default is 100)')
.option('-C, --includeDeprecatedFields [value]', 'Flag to include deprecated fields (The default is to exclude)')
.option('-L, --filterPath [value]', 'path of a file containing a javascript function which filters fields')
.parse(process.argv);

const { schemaFilePath, destDirPath, depthLimit = 100, includeDeprecatedFields = false } = program;
const {
schemaFilePath, destDirPath, depthLimit = 100, includeDeprecatedFields = false, filterPath,
} = program;
const typeDef = fs.readFileSync(schemaFilePath);
const source = new Source(typeDef);
const gqlSchema = buildSchema(source);

const filterFunc = filterPath ? require(path.resolve(filterPath)) : () => true;

del.sync(destDirPath);
path.resolve(destDirPath).split(path.sep).reduce((before, cur) => {
const pathTmp = path.join(before, cur + path.sep);
Expand Down Expand Up @@ -85,8 +90,10 @@ const generateQuery = (
duplicateArgCounts = {},
crossReferenceKeyList = [], // [`${curParentName}To${curName}Key`]
curDepth = 1,
curParentSchema,
) => {
const field = gqlSchema.getType(curParentType).getFields()[curName];
const curParentTypeName = curParentType.inspect().replace(/[[\]!]/g, '');
const curTypeName = field.type.inspect().replace(/[[\]!]/g, '');
const curType = gqlSchema.getType(curTypeName);
let queryStr = '';
Expand All @@ -98,13 +105,30 @@ const generateQuery = (
crossReferenceKeyList.push(crossReferenceKey);
const childKeys = Object.keys(curType.getFields());
childQuery = childKeys
.filter(fieldName => {
/* Exclude deprecated fields */
.filter((fieldName) => {
const fieldSchema = gqlSchema.getType(curType).getFields()[fieldName];
return includeDeprecatedFields || !fieldSchema.isDeprecated;
const fieldTypeName = fieldSchema.type.inspect().replace(/[[\]!]/g, '');
const fieldType = gqlSchema.getType(fieldTypeName);
const include = filterFunc({
parentSchema: curParentSchema,
parentType: curParentType,
parentTypeName: curParentTypeName,
parentName: curParentName,
curSchema: field,
curType,
curTypeName,
curName,
curDepth,
fieldSchema,
fieldType,
fieldTypeName,
fieldName,
argumentsDict,
});
return include && (includeDeprecatedFields || !fieldSchema.isDeprecated);
})
.map(cur => generateQuery(cur, curType, curName, argumentsDict, duplicateArgCounts,
crossReferenceKeyList, curDepth + 1).queryStr)
crossReferenceKeyList, curDepth + 1, field).queryStr)
.filter(cur => cur)
.join('\n');
}
Expand Down Expand Up @@ -133,8 +157,30 @@ const generateQuery = (
const valueTypeName = types[i];
const valueType = gqlSchema.getType(valueTypeName);
const unionChildQuery = Object.keys(valueType.getFields())
.filter((fieldName) => {
const fieldSchema = valueType.getFields()[fieldName];
const fieldTypeName = fieldSchema.type.inspect().replace(/[[\]!]/g, '');
const fieldType = gqlSchema.getType(fieldTypeName);
const include = filterFunc({
parentSchema: curParentSchema,
parentType: curParentType,
parentTypeName: curParentTypeName,
parentName: curParentName,
curSchema: field,
curType,
curTypeName,
curName,
curDepth,
fieldSchema,
fieldType,
fieldTypeName,
fieldName,
argumentsDict,
});
return include;
})
.map(cur => generateQuery(cur, valueType, curName, argumentsDict, duplicateArgCounts,
crossReferenceKeyList, curDepth + 2).queryStr)
crossReferenceKeyList, curDepth + 2, field).queryStr)
.filter(cur => cur)
.join('\n');
queryStr += `${fragIndent}... on ${valueTypeName} {\n${unionChildQuery}\n${fragIndent}}\n`;
Expand Down Expand Up @@ -169,10 +215,11 @@ const generateFile = (obj, description) => {
const writeFolder = path.join(destDirPath, `./${outputFolderName}`);
fs.mkdirSync(writeFolder);
Object.keys(obj).forEach((type) => {
const field = gqlSchema.getType(description).getFields()[type];
const schema = gqlSchema.getType(description);
const field = schema.getFields()[type];
/* Only process non-deprecated queries/mutations: */
if (includeDeprecatedFields || !field.isDeprecated) {
const queryResult = generateQuery(type, description);
const queryResult = generateQuery(type, schema);
const varsToTypesStr = getVarsToTypesStr(queryResult.argumentsDict);
let query = queryResult.queryStr;
query = `${description.toLowerCase()} ${type}${varsToTypesStr ? `(${varsToTypesStr})` : ''}{\n${query}\n}`;
Expand Down