update(feat): query in codeblocks

Author: Rooyca

Diff for: README.md

@@ -72,6 +72,18 @@ Furthermore, QJSON extends its functionality to external JSON files. For example
 @data.json>store.books.0.author;
 ```
 
+It's also posible to query the file directly using the `#qj-query` flag:
+
+~~~markdown
+```qjson
+#qj-id: 24
+#qj-file: data.json
+#qj-show-json
+#qj-hide-id
+#qj-query: pageProps.heroData[win_rate >= 55 && role == Mage]
+```
+~~~
+
 ## 🏳️ Flags
 
 Query JSON supports various flags to enhance customization and functionality:
@@ -96,6 +108,28 @@ This flag allows you to display the JSON within the rendered output. By default
 
 If provided, this flag specifies the file path containing the JSON data. In its absence, the plugin scans for JSON data within the code block.
 
+#### `#qj-query` (optional)
+
+This flag allows you to query the JSON file directly from your codeblock. The query syntax must be inside brackets `[]`. The supported operators are:
+
+- Logical operators: `&&`, `||`
+- Comparison operators: `==`, `!=`, `>`, `>=`, `<`, `<=`
+
+**Example**:
+
+~~~markdown
+```qjson
+#qj-id: 24
+#qj-file: data.json
+#qj-show-json
+#qj-hide-id
+#qj-query: pageProps.heroData[win_rate >= 55 && role == Mage]
+```
+~~~
+
+> [!NOTE]
+> The `#qj-show-json` flag is mandatory when using the `#qj-query` flag.
+
 ## 🛠️ Contribution
 
 If you encounter any issues or have suggestions for improvement, please feel free to contribute to the project. Your feedback is invaluable in enhancing the plugin's functionality and user experience.

Diff for: manifest.json

@@ -1,7 +1,7 @@
 {
 	"id": "query-json",
 	"name": "Query JSON",
-	"version": "0.0.6",
+	"version": "0.0.7",
 	"minAppVersion": "0.15.0",
 	"description": "Read, query and work with JSON.",
 	"author": "rooyca",

Diff for: package.json

@@ -1,6 +1,6 @@
 {
 	"name": "query-json",
-	"version": "0.0.6",
+	"version": "0.0.7",
 	"description": "Read, query and work with JSON.",
 	"main": "main.js",
 	"scripts": {

Diff for: src/functions.ts

@@ -0,0 +1,105 @@
+export function parseQuery(query) {
+    const parts = query.split('.');
+    return parts.map(part => {
+        const match = part.match(/(\w+)\[(.*)\]/);
+        if (match) {
+            return {
+                type: 'filter',
+                field: match[1],
+                condition: match[2]
+            };
+        }
+        return {
+            type: 'field',
+            name: part
+        };
+    });
+}
+
+function parseCondition(condition) {
+    const logicalOperators = ['&&', '||'];
+    let conditions = [];
+    let operators = [];
+
+    let remainingCondition = condition;
+    logicalOperators.forEach(operator => {
+        if (remainingCondition.includes(operator)) {
+            const parts = remainingCondition.split(operator).map(s => s.trim());
+            conditions.push(parseSingleCondition(parts[0]));
+            operators.push(operator);
+            remainingCondition = parts[1];
+        }
+    });
+
+    if (conditions.length === 0) {
+        conditions.push(parseSingleCondition(remainingCondition));
+    } else {
+        conditions.push(parseSingleCondition(remainingCondition));
+    }
+
+    return { conditions, operators };
+}
+
+function parseSingleCondition(condition) {
+    const comparisonOperators = ['>=', '<=', '==', '>', '<', '!='];
+    for (let operator of comparisonOperators) {
+        if (condition.includes(operator)) {
+            const [key, value] = condition.split(operator).map(s => s.trim());
+            return { key, operator, value };
+        }
+    }
+    throw new Error(`Invalid condition: ${condition}`);
+}
+
+export function executeQuery(json, parsedQuery) {
+    let result = json;
+    let finalResult = result; // Initialize finalResult to the full dataset initially
+
+    parsedQuery.forEach(part => {
+        if (part.type === 'field') {
+            result = result[part.name];
+        } else if (part.type === 'filter') {
+            const { conditions, operators } = parseCondition(part.condition);
+            result = result[part.field];
+            finalResult = result.filter(item => {
+                return evaluateConditions(item, conditions, operators);
+            });
+        }
+    });
+
+    if (parsedQuery.length > 0 && parsedQuery[parsedQuery.length - 1].type === 'field') {
+        finalResult = finalResult.map(item => item[parsedQuery[parsedQuery.length - 1].name]);
+    }
+
+    return finalResult;
+}
+
+function evaluateConditions(item, conditions, operators) {
+    let result = evaluateCondition(item, conditions[0]);
+
+    for (let i = 0; i < operators.length; i++) {
+        const operator = operators[i];
+        const nextConditionResult = evaluateCondition(item, conditions[i + 1]);
+
+        if (operator === '&&') {
+            result = result && nextConditionResult;
+        } else if (operator === '||') {
+            result = result || nextConditionResult;
+        }
+    }
+
+    return result;
+}
+
+function evaluateCondition(item, condition) {
+    const { key, operator, value } = condition;
+    switch (operator) {
+        case '>=': return item[key] >= Number(value);
+        case '<=': return item[key] <= Number(value);
+        case '==': return item[key] == value;
+        case '>': return item[key] > Number(value);
+        case '<': return item[key] < Number(value);
+        case '!=': return item[key] != value;
+        default: return false;
+    }
+}

Diff for: src/main.ts

@@ -1,4 +1,5 @@
 import { App, Notice, Plugin } from 'obsidian';
+import { parseQuery, executeQuery} from './functions';
 
 export default class QJSON extends Plugin {
 
@@ -23,6 +24,13 @@ export default class QJSON extends Plugin {
 				return;
 			}
 
+			let query;
+
+			if (source.includes('#qj-query:')) {
+				query = source.match(/#qj-query: (.+)/)[1];
+				query = parseQuery(query);
+			}
+
 			let desc;
 
 			if (!source.includes('#qj-hide-id')) {
@@ -48,7 +56,13 @@ export default class QJSON extends Plugin {
 			}
 
 			const json = JSON.parse(source);
-			el.createEl('pre', {text: JSON.stringify(json, null, 2), cls: 'QJSON-'+id+' cdQjson '+showJson});
+
+			if (query) {
+				const result = executeQuery(json, query);
+				el.createEl('pre', {text: JSON.stringify(result, null, 2), cls: 'QJSON-'+id+' cdQjson '+showJson});
+			} else {
+				el.createEl('pre', {text: JSON.stringify(json, null, 2), cls: 'QJSON-'+id+' cdQjson '+showJson});
+			}
 
 			qjCount = document.querySelectorAll('.cdQjson').length;
 			statusBarItemEl.setText('QJSON: ' + qjCount);