Fix couple hiccups, support *

This commit is contained in:
rijkvanzanten
2020-07-14 17:13:37 -04:00
parent 54a2b3b74d
commit dd6c7043ac
3 changed files with 28 additions and 74 deletions

View File

@@ -15,7 +15,7 @@ export default async function runAST(ast: AST, query = ast.query) {
for (const child of ast.children) {
if (child.type === 'field') {
if (columnsInCollection.includes(child.name)) {
if (columnsInCollection.includes(child.name) || child.name === '*') {
toplevelFields.push(child.name);
}
@@ -181,6 +181,10 @@ export default async function runAST(ast: AST, query = ast.query) {
const nestedCollectionKeys = nestedCollections.map(({ fieldKey }) => fieldKey);
if (toplevelFields.includes('*')) {
return results;
}
return results.map((result) =>
pick(result, uniq([...nestedCollectionKeys, ...toplevelFields]))
);