Add Custom Query Class

Author: Rayhan Arayilakath

index.js

@@ -10,5 +10,7 @@ module.exports = {
 	Board: replapi.Board,
 	Notifications: replapi.Notifications,
 	Misc: replapi.Misc,
-	Login: replapi.Login
+	Login: replapi.Login,
+	CustomDataQuery: replapi.CustomDataQuery,
+	CustomRecursiveQuery: replapi.CustomRecursiveQuery
 };

src/classes/Custom.js

@@ -0,0 +1,150 @@
+let headers = require('../utils/headers.js');
+let variables = require('../utils/variables.js');
+
+class CustomDataQuery {
+	constructor(queryName, customQuery, customVariables) {
+		this.queryName = queryName;
+		this.customQuery = customQuery;
+		this.customVariables = customVariables;
+	}
+
+	async getData() {
+		let queryName = this.queryName;
+		let customQuery = this.customQuery;
+		let customVariables = this.customVariables;
+		
+		let specialQueryVariables = {
+		  since: 'KarmaSince',
+		  count: 'Int',
+		  id: 'Int',
+		  limit: 'Int'
+		}
+		Object.freeze(specialQueryVariables);
+		
+		let queryVariables = Object.keys(customVariables);
+		let queryVariablesString = '';
+		for (let i = 0; i < queryVariables.length; i++) {
+		  let type;
+      if(specialQueryVariables.hasOwnProperty(queryVariables[i])) {
+	      type = String(specialQueryVariables[queryVariables[i]]);
+      } else {
+        type = String(typeof queryVariables[i]).split('');
+        type[0] = type[0].toUpperCase();
+	      type = type.join('')
+      }
+
+		  if(i !== queryVariables.length - 1) {
+		    queryVariablesString += `$${queryVariables[i]}: ${type}!, `
+		  } else {
+		    queryVariablesString += `$${queryVariables[i]}: ${type}!`
+		  }
+		}
+
+		let info = await variables
+			.fetch(variables.graphql, {
+				method: 'POST',
+				headers,
+				body: JSON.stringify({
+					query: `
+    			  query ${queryName}(${queryVariablesString}) {
+    				  ${customQuery}
+    				}`,
+					variables: JSON.stringify(customVariables)
+				})
+			})
+			.then(res => res.json());
+    
+		if (info.errors) {
+			throw new Error(`Custom Data Query ${this.queryName} returned an error: ${JSON.stringify(info.errors)}`);
+		} else {
+			return info.data;
+		}
+	}
+}
+
+class CustomRecursiveQuery {
+	constructor(queryName, customQuery, customVariables, treePath, customAfter, customCount) {
+		this.queryName = queryName;
+		this.customQuery = customQuery;
+		this.customVariables = customVariables;
+		this.treePath = treePath;
+		this.customAfter = customAfter;
+		this.customCount = customCount;
+	}
+
+	async getData() {
+		let queryName = this.queryName;
+		let customQuery = this.customQuery;
+		let customVariables = this.customVariables;
+		let treePath = this.treePath;
+		let customAfter = this.customAfter;
+		let customCount = this.customCount;
+		
+		let specialQueryVariables = {
+		  since: 'KarmaSince',
+		  count: 'Int',
+		  id: 'Int',
+		  limit: 'Int'
+		}
+		Object.freeze(specialQueryVariables);
+		
+		let queryVariables = Object.keys(customVariables);
+		let queryVariablesString = '';
+		for (let i = 0; i < queryVariables.length; i++) {
+		  let type;
+      if(specialQueryVariables.hasOwnProperty(queryVariables[i])) {
+	      type = String(specialQueryVariables[queryVariables[i]]);
+      } else {
+        type = String(typeof queryVariables[i]).split('');
+        type[0] = type[0].toUpperCase();
+	      type = type.join('')
+      }
+
+		  if(i !== queryVariables.length - 1) {
+		    queryVariablesString += `$${queryVariables[i]}: ${type}!, `
+		  } else {
+		    queryVariablesString += `$${queryVariables[i]}: ${type}!`
+		  }
+		}
+    
+		let output = [];
+
+	 	async function recurse(after) {
+			if (after === null) return;
+
+  		let info = await variables
+  			.fetch(variables.graphql, {
+  				method: 'POST',
+  				headers,
+  				body: JSON.stringify({
+  					query: `
+      			  query ${queryName}(${queryVariablesString}) {
+      				  ${customQuery}
+      				}`,
+  					variables: JSON.stringify(customVariables)
+  				})
+  			})
+  			.then(res => res.json());
+  		
+			if (info.errors) {
+				throw new Error(`Custom Recusive Query ${queryName} returned an error: ${JSON.stringify(info.errors)}`);
+			} else {
+				info.data[customQuery.trim().split('(')[0]][treePath].items.forEach(item => {
+					output.push(item);
+				});
+				if (output.length !== customCount) {
+					await recurse(info.data[customQuery.trim().split('(')[0]][treePath].pageInfo.nextCursor);
+				}
+			}
+
+		}
+
+		await recurse(this.customAfter);
+		return output;
+	}
+}
+
+module.exports = {
+	CustomDataQuery: CustomDataQuery,
+	CustomRecursiveQuery: CustomRecursiveQuery
+};

src/classes/User.js

@@ -41,7 +41,7 @@ class User {
 		let user = this.username;
 		let output = [];
 
-		async function recurse(after) {
+	 	async function recurse(after) {
 			if (after === null) return;
 
 			let info = await variables

src/index.js

@@ -8,6 +8,7 @@ const languages = require('./classes/Languages.js');
 const board = require('./classes/Board.js');
 const notifications = require('./classes/Notifications.js');
 const misc = require('./classes/Misc.js')
+const custom = require('./classes/Custom.js')
 
 if (!login) throw new Error('Login class not found');
 if (!user) throw new Error('User class not found');
@@ -19,6 +20,7 @@ if (!languages) throw new Error('Languages class not found');
 if (!board) throw new Error('Board class not found');
 if (!notifications) throw new Error('Notifications class not found');
 if (!misc) throw new Error('Miscellaneous class not found');
+if (!custom) throw new Error('Custom class not found');
 
 module.exports = {
   Login: login.Login,
@@ -30,5 +32,7 @@ module.exports = {
   Languages: languages.Languages,
   Board: board.Board,
   Notifications: notifications.Notifications,
-  Misc: misc.Misc
+  Misc: misc.Misc,
+  CustomDataQuery: custom.CustomDataQuery,
+  CustomRecursiveQuery: custom.CustomRecursiveQuery
 }