From 801b9a3dc1b141378c812ace7a38aaa67dd48d11 Mon Sep 17 00:00:00 2001
From: Alice Gaudon <alice@gaudon.pro>
Date: Sun, 6 Sep 2020 15:38:54 +0200
Subject: [PATCH] Improve field disambiguation to allow raw values when
 starting with ",',`

---
 package.json            |  2 +-
 src/db/ModelQuery.ts    | 34 +++++++++++++++++++---------------
 test/ModelQuery.test.ts |  7 +++++++
 3 files changed, 27 insertions(+), 16 deletions(-)

diff --git a/package.json b/package.json
index 2c38121..7ee866d 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "wms-core",
-  "version": "0.22.0-rc.4",
+  "version": "0.22.0-rc.5",
   "description": "Node web application framework and toolbelt.",
   "repository": "https://gitlab.com/ArisuOngaku/wms-core",
   "author": "Alice Gaudon <alice@gaudon.pro>",
diff --git a/src/db/ModelQuery.ts b/src/db/ModelQuery.ts
index 9eb94ad..e619752 100644
--- a/src/db/ModelQuery.ts
+++ b/src/db/ModelQuery.ts
@@ -18,7 +18,7 @@ export default class ModelQuery<M extends Model> implements WhereFieldConsumer<M
         const fields = [];
         for (let key in data) {
             if (data.hasOwnProperty(key)) {
-                fields.push(new UpdateFieldValue(key, data[key], false));
+                fields.push(new UpdateFieldValue(inputToFieldOrValue(key, factory.table), data[key], false));
             }
         }
         return new ModelQuery(QueryType.UPDATE, factory, fields);
@@ -96,7 +96,7 @@ export default class ModelQuery<M extends Model> implements WhereFieldConsumer<M
     }
 
     public sortBy(field: string, direction: SortDirection = 'ASC', raw: boolean = false): this {
-        this._sortBy = raw ? field : field.split('.').map(v => v.startsWith('`') ? v : `\`${v}\``).join('.');
+        this._sortBy = raw ? field : inputToFieldOrValue(field);
         this._sortDirection = direction;
         return this;
     }
@@ -119,7 +119,7 @@ export default class ModelQuery<M extends Model> implements WhereFieldConsumer<M
 
         this._union = {
             query: query,
-            sortBy: raw ? sortBy : sortBy.split('.').map(v => v.startsWith('`') ? v : `\`${v}\``).join('.'),
+            sortBy: raw ? sortBy : inputToFieldOrValue(sortBy),
             direction: direction,
             limit: limit,
         };
@@ -138,18 +138,11 @@ export default class ModelQuery<M extends Model> implements WhereFieldConsumer<M
         if (this._pivot) this.fields.push(...this._pivot);
 
         // Prevent wildcard and fields from conflicting
-        let fieldArray = this.fields.map(f => {
-            let field = f.toString();
-
+        let fields = this.fields.map(f => {
+            const field = f.toString();
             if (field.startsWith('(')) return f; // Skip sub-queries
-
-            let parts = field.split('.');
-            if (parts.length === 1) parts = [this.table, field]; // Add table disambiguation
-
-            return parts.map(v => v.startsWith('`') || v === '*' ? v : `\`${v}\``).join('.');
-        });
-
-        let fields = fieldArray.join(',');
+            return inputToFieldOrValue(field, this.table);
+        }).join(',');
 
         let join = '';
         if (this._leftJoin) {
@@ -316,6 +309,17 @@ export default class ModelQuery<M extends Model> implements WhereFieldConsumer<M
     }
 }
 
+function inputToFieldOrValue(input: string, addTable?: string): string {
+    if (input.startsWith('`') || input.startsWith('"') || input.startsWith("'")) {
+        return input;
+    }
+
+    let parts = input.split('.');
+    if (addTable && parts.length === 1) parts = [addTable, input]; // Add table disambiguation
+
+    return parts.map(v => v === '*' ? v : `\`${v}\``).join('.');
+}
+
 export interface ModelQueryResult<M extends Model> extends Array<M> {
     originalData?: any[];
     pagination?: Pagination<M>;
@@ -357,7 +361,7 @@ class FieldValue {
     public toString(first: boolean = true): string {
         const valueStr = (this.raw || this.value === null || this.value instanceof ModelQuery) ? this.value :
             (Array.isArray(this.value) ? `(${'?'.repeat(this.value.length).split('').join(',')})` : '?');
-        let field = this.field.split('.').map(p => `\`${p}\``).join('.');
+        let field = inputToFieldOrValue(this.field);
         return `${first ? '' : ','}${field}${this.test}${valueStr}`;
     }
 
diff --git a/test/ModelQuery.test.ts b/test/ModelQuery.test.ts
index dbc5872..3cdd3d8 100644
--- a/test/ModelQuery.test.ts
+++ b/test/ModelQuery.test.ts
@@ -3,6 +3,13 @@ import ModelFactory from "../src/db/ModelFactory";
 import Model from "../src/db/Model";
 
 describe('Test ModelQuery', () => {
+    test('select', () => {
+        const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>, 'f1', '"Test" as f2')
+            .where('f4', 'v4');
+        expect(query.toString(true)).toBe('SELECT `model`.`f1`,"Test" as f2 FROM `model` WHERE `f4`=?');
+        expect(query.variables).toStrictEqual(['v4']);
+    });
+
     test('order by', () => {
         const query = ModelQuery.select({table: 'model'} as unknown as ModelFactory<Model>)
             .sortBy('model.f2', 'ASC');