Files
sineQL/test/index.js
2021-03-31 00:54:34 +11:00

265 lines
6.5 KiB
JavaScript

//mock tools
const Op = {
eq: 'eq'
};
const books = {
findAll: async args => {
let arr = [
{ id: 1, title: 'The Wind in the Willows', published: '1908-06-15' },
{ id: 2, title: 'The Fart in the Fronds', published: '1908-06-15' }
];
const { attributes, where } = args;
//filter out non-matching elements
arr = arr.filter(element => {
if (Object.keys(where).length == 0) {
return true;
}
if (where.title && where.published) {
return element.title == where.title.eq && element.published == where.published.eq;
}
if (where.title) {
return element.title == where.title.eq;
}
if (where.published) {
return element.published == where.published.eq;
}
return false;
});
//filter out non-used attributes
arr = arr.map(element => {
//only they element keys that are in attributes
const keys = Object.keys(element).filter(key => attributes.includes(key));
//determine which fields to carry over
const ret = {};
keys.forEach(key => ret[key] = element[key]);
return ret;
});
return arr;
}
}
const authors = {
findAll: async args => {
let arr = [
{ id: 1, name: 'Kenneth Grahame', books: [1] },
{ id: 2, name: 'Frank', books: [1, 2] },
{ id: 3, name: 'Betty', books: [2] }
];
const { attributes, where } = args;
//filter out non-matching elements
arr = arr.filter(element => {
if (Object.keys(where).length == 0) {
return true;
}
if (where.name && where.books) {
return element.name == where.name.eq; //books is always true because they are never queried
}
if (where.name) {
return element.name == where.name.eq;
}
if (where.books) {
return true; //books is always true because they are never queried
}
return false;
});
//filter out non-used attributes
arr = arr.map(element => {
//only they element keys that are in attributes
const keys = Object.keys(element).filter(key => attributes.includes(key));
//determine which fields to carry over
const ret = {};
keys.forEach(key => ret[key] = element[key]);
return ret;
});
return arr;
}
}
//the handler functions return arrays for each type, containing every element that satisfies the queries
//the "query" argument contains the object built from the sineQL query
//the "graph" argument contains the typeGraph
//the task of the handler functions is to query the database, and return the correct results
/* possible values for "query" include:
{
typeName: 'Author',
name: { typeName: 'String', scalar: true, match: 'Kenneth Grahame' },
books: { typeName: 'Book', match: {
typeName: 'Book',
title: { typeName: 'String', scalar: true, match: 'The wind in the Willows' }
published: { typeName: 'Date', scalar: true }
}
}
*/
//depth-first search seems to be the best option
//Each query shouldn't know if it's a sub-query
const queryHandlers = {
//complex compound
Author: async (query, graph) => {
//DEBUGGING
// console.log('Author():', query);
//get the fields alone
const { typeName, ...fields } = query;
//hack the id into the fields list (if it's not there already)
fields['id'] = fields['id'] || { typeName: 'Integer', scalar: true };
//get the names of matched fields (fields to find)
const matchedNames = Object.keys(fields).filter(field => fields[field].match);
//short-circuit if querying everything
const where = {};
if (matchedNames.length > 0) {
//build the "where" object
matchedNames.forEach(mn => {
if (query[mn].match !== true) { //true means it's a compound type
where[mn] = { [Op.eq]: query[mn].match };
}
});
}
//these are field names
const scalars = Object.keys(fields).filter(field => graph[fields[field].typeName].scalar);
const nonScalars = Object.keys(fields).filter(field => !graph[fields[field].typeName].scalar);
let authorResults = await authors.findAll({
attributes: Object.keys(fields), //fields to find (keys)
where: where
}); //sequelize ORM model
const promiseArray = nonScalars.map(async nonScalar => {
//delegate to a deeper part of the tree
const nonScalarArray = await queryHandlers[fields[nonScalar].typeName](fields[nonScalar], graph);
//for each author, update this non-scalar field with the non-scalar's recursed value
authorResults.forEach(author => {
author[nonScalar] = nonScalarArray.filter(ns => author[nonScalar].includes(ns.id)); //using the hacked-in ID field (JOIN)
});
//prune the authors when matching, but their results are empty
authorResults = authorResults.filter(author => {
return !(fields[nonScalar].match && author[nonScalar].length == 0);
});
});
await Promise.all(promiseArray);
//finally, return the results
return authorResults;
},
//simple compound
Book: async (query, graph) => {
// console.log('Book():', query);
//get the fields alone
const { typeName, ...fields } = query;
//hack the id into the fields list (if it's not there already)
fields['id'] = fields['id'] || { typeName: 'Integer', scalar: true };
//get the names of matched fields
const matchedNames = Object.keys(fields).filter(field => fields[field].match);
//short-circuit if querying everything
const where = {};
if (matchedNames.length > 0) {
//build the "where" object
matchedNames.forEach(mn => {
if (query[mn].match !== true) {
where[mn] = { [Op.eq]: query[mn].match };
}
});
}
//return the result
return await books.findAll({
attributes: Object.keys(fields), //fields to find
where: where
}); //sequelize ORM model
}
};
//the matching schema
const schema = `
scalar Date
type Book {
String title
Date published
}
type Author {
String name
Book books
}
`;
//input tool
const readline = require('readline');
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout,
terminal: false
});
const question = (prompt, def = null) => {
return new Promise((resolve, reject) => {
rl.question(`${prompt}${def ? ` (${def})` : ''}> `, answer => {
//loop on required
if (def === null && !answer) {
return resolve(question(prompt, def));
}
return resolve(answer || def);
});
});
};
//the library to test
const sineQL = require('../source/index.js');
//run the function in debug mode (builds type graph)
const sine = sineQL(schema, { queryHandlers }, { debug: false });
//actually ask the question
(async () => {
while(true) {
const answer = await question('sineQL');
const [code, result] = await sine(answer);
//normal response
if (code == 200) {
console.dir(result, { depth: null });
}
}
})();