From 391bac0b45ef4e8d8c30f6afdf1414e97a7c4336 Mon Sep 17 00:00:00 2001
From: yupi <592789970@qq.com>
Date: Tue, 10 May 2022 19:08:12 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E4=BC=98=E5=8C=96=E5=8C=B9=E9=85=8D?=
 =?UTF-8?q?=E5=AD=90=E6=9F=A5=E8=AF=A2=E7=9A=84=E6=96=B9=E6=B3=95=EF=BC=8C?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E5=8C=B9=E9=85=8D=E5=A4=9A=E5=B1=82=E7=BA=A7?=
 =?UTF-8?q?=E6=8B=AC=E5=8F=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/App.vue            | 35 +++++++++++++++++++++++----
 src/generator/index.ts | 54 ++++++++++++++++++++++++++++++++++++++----
 2 files changed, 79 insertions(+), 10 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index f63c105..9c92ba4 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -7,12 +7,25 @@ import {format} from "sql-formatter";
 
 import editorWorker from 'monaco-editor/esm/vs/editor/editor.worker?worker'
 import jsonWorker from 'monaco-editor/esm/vs/language/json/json.worker?worker'
+import cssWorker from 'monaco-editor/esm/vs/language/css/css.worker?worker'
+import htmlWorker from 'monaco-editor/esm/vs/language/html/html.worker?worker'
+import tsWorker from 'monaco-editor/esm/vs/language/typescript/ts.worker?worker'
 
-self.MonacoEnvironment = {
-  getWorker(_, label) {
+// @ts-ignore
+(self as any).MonacoEnvironment = {
+  getWorker(_: any, label: any) {
     if (label === 'json') {
       return new jsonWorker()
     }
+    if (label === 'css' || label === 'scss' || label === 'less') {
+      return new cssWorker()
+    }
+    if (label === 'html' || label === 'handlebars' || label === 'razor') {
+      return new htmlWorker()
+    }
+    if (label === 'typescript' || label === 'javascript') {
+      return new tsWorker()
+    }
     return new editorWorker()
   }
 }
@@ -33,6 +46,10 @@ const getSQL = () => {
   }
 }
 
+const getInvokeTree = () => {
+
+}
+
 const initJSONValue = "{\n" +
     "    \"main\": {\n" +
     "        \"sql\": \"select * from @union_all_layer(分区 = 2021) where 分区 = #{分区}\",\n" +
@@ -82,8 +99,13 @@ onMounted(() => {
 
 <template>
   <h1>
-    SQL 生成器
-    <t-button size="large" theme="primary" @click="getSQL" style="float: right"> 生成 SQL</t-button>
+    SQL 生成器 - 用 JSON 来写 SQL
+    <div style="float: right">
+      <t-button size="large" theme="primary" @click="getSQL" > 生成 SQL</t-button>
+      <t-divider theme="vertical" />
+      <t-button size="large" theme="default" @click="getInvokeTree"> 查看调用树</t-button>
+    </div>
+    <t-tree :data="[]" activable hover />
   </h1>
   <t-row :gutter="24">
     <t-col :xs="12" :sm="6">
@@ -93,7 +115,10 @@ onMounted(() => {
       <div id="outputContainer" ref="outputContainer" style="height: 80vh; max-width: 100%"/>
     </t-col>
   </t-row>
-  <div>yupi：你能体会手写 1500 行 SQL、牵一发而动全身的恐惧么？</div>
+  <br/>
+  <div>
+    yupi：你能体会手写 1500 行 SQL、牵一发而动全身的恐惧么？
+  </div>
 </template>
 
 <style>
diff --git a/src/generator/index.ts b/src/generator/index.ts
index 5c1e0e7..28e1b72 100644
--- a/src/generator/index.ts
+++ b/src/generator/index.ts
@@ -51,10 +51,9 @@ function replaceSubSql(sql: string, context: InputJSON): string {
     if (!sql) {
         return "";
     }
-    const regExp = /@([\u4e00-\u9fa5_a-zA-Z0-9]+)\((.*?)\)/;
     let result = sql;
     result = String(result);
-    let regExpMatchArray = result.match(regExp);
+    let regExpMatchArray = matchSubQuery(result);
     // 依次替换
     while (regExpMatchArray && regExpMatchArray.length > 2) {
         // 找到结果
@@ -72,11 +71,12 @@ function replaceSubSql(sql: string, context: InputJSON): string {
             paramsStr = paramsStr.trim();
         }
         // e.g. ["a = b", "c = d"]
-        const singleParamsStrArray = paramsStr.split(',');
+        const singleParamsStrArray = paramsStr.split('|||');
         // string => object
         const params: Record<string, string> = {};
         for (const singleParamsStr of singleParamsStrArray) {
-            const keyValueArray = singleParamsStr.split('=');
+            // 必须分成 2 段
+            const keyValueArray = singleParamsStr.split('=', 2);
             if (keyValueArray.length < 2) {
                 continue;
             }
@@ -85,7 +85,51 @@ function replaceSubSql(sql: string, context: InputJSON): string {
         }
         const replacement = replaceParams(replacementNode, context, params);
         result = result.replaceAll(regExpMatchArray[0], replacement);
-        regExpMatchArray = result.match(regExp);
+        regExpMatchArray = matchSubQuery(result);
     }
     return result;
+}
+
+/**
+ * 匹配子查询
+ * @param str
+ */
+function matchSubQuery(str: string) {
+    if (!str) {
+        return null;
+    }
+    const regExp = /@([\u4e00-\u9fa5_a-zA-Z0-9]+)\((.*?)\)/;
+    let regExpMatchArray = str.match(regExp);
+    if (!regExpMatchArray || regExpMatchArray.index === undefined) {
+        return null;
+    }
+    // @ 开始位置
+    let startPos = regExpMatchArray.index;
+    // 左括号右侧
+    let leftParenthesisPos = startPos + regExpMatchArray[1].length + 2;
+    // 遍历游标
+    let currPos = leftParenthesisPos;
+    // 默认匹配结束位置，需要对此结果进行修正
+    let endPos = startPos + regExpMatchArray[0].length;
+    // 剩余待匹配左括号数量
+    let leftCount = 1;
+    while (currPos < str.length) {
+        const currentChar = str.charAt(currPos);
+        if (currentChar === '(') {
+            leftCount++;
+        } else if (currentChar === ')') {
+            leftCount--;
+        }
+        // 匹配结束
+        if (leftCount == 0) {
+            endPos = currPos + 1;
+            break;
+        }
+        currPos++;
+    }
+    return [
+        str.slice(startPos, endPos),
+        regExpMatchArray[1],
+        str.slice(leftParenthesisPos, endPos - 1)
+    ]
 }
\ No newline at end of file