From 83f64e539c2dd74ef47fa4a7f5d9ad229c195f05 Mon Sep 17 00:00:00 2001
From: mayinrain <624899618@qq.com>
Date: Tue, 31 Dec 2024 11:37:59 +0800
Subject: [PATCH] =?UTF-8?q?upd:=20=E5=87=BD=E6=95=B0=E7=AE=A1=E7=90=86?=
 =?UTF-8?q?=E4=BA=A4=E4=BA=92=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 linkis-web/src/apps/URM/i18n/common/en.json   |  3 +-
 linkis-web/src/apps/URM/i18n/common/zh.json   |  3 +-
 .../functionManagement/addFunctionModal.vue   | 62 ++++++++++++++++++-
 3 files changed, 63 insertions(+), 5 deletions(-)

diff --git a/linkis-web/src/apps/URM/i18n/common/en.json b/linkis-web/src/apps/URM/i18n/common/en.json
index 5b4c887a00..89ab41bc82 100644
--- a/linkis-web/src/apps/URM/i18n/common/en.json
+++ b/linkis-web/src/apps/URM/i18n/common/en.json
@@ -89,7 +89,8 @@
         "XGHS": "Update Funtion",
         "XG": "Update",
         "QPZS": "View in Full Screen",
-        "QXQP": "Cancel Full Screen"
+        "QXQP": "Cancel Full Screen",
+        "XZHS": "Select Function"
       }
     }
   }
diff --git a/linkis-web/src/apps/URM/i18n/common/zh.json b/linkis-web/src/apps/URM/i18n/common/zh.json
index 2c6896869a..becc40b4fe 100644
--- a/linkis-web/src/apps/URM/i18n/common/zh.json
+++ b/linkis-web/src/apps/URM/i18n/common/zh.json
@@ -89,7 +89,8 @@
         "XGHS": "修改函数",
         "XG": "修改",
         "QPZS": "全屏展示",
-        "QXQP": "取消全屏"
+        "QXQP": "取消全屏",
+        "XZHS": "选择函数"
       }
     }
   }
diff --git a/linkis-web/src/apps/URM/module/functionManagement/addFunctionModal.vue b/linkis-web/src/apps/URM/module/functionManagement/addFunctionModal.vue
index ab25686202..51c2b64604 100644
--- a/linkis-web/src/apps/URM/module/functionManagement/addFunctionModal.vue
+++ b/linkis-web/src/apps/URM/module/functionManagement/addFunctionModal.vue
@@ -39,7 +39,7 @@
       <FormItem
         :label="$t('message.linkis.udf.functionName')"
         prop="name">
-        <Input v-model="setting.name" :disabled="model === 1" />
+        <Input v-model="setting.name" :disabled="model === 1 || !registerFunctionEditable" />
       </FormItem>
       <FormItem
         :label="$t('message.linkis.udf.functionType')"
@@ -95,6 +95,21 @@
           fs-type="share"
           @set-node="setNodePath"/>
       </FormItem>
+      <FormItem
+        v-if="registerFunctions.length > 0 && model !== 1"
+        :label="$t('message.linkis.udf.XZHS')"
+        prop="name">
+        <Select
+          v-model="setting.name"
+          @on-change="handleFuncChange"
+          filterable>
+          <Option
+            v-for="(item) in registerFunctions"
+            :label="item"
+            :value="item"
+            :key="item"/>
+        </Select>
+      </FormItem>
       <!-- <FormItem
         v-if="fnType === 0 && fnCategory.isCommon"
         :label="$t('message.linkis.udf.registerFormat')"
@@ -268,6 +283,7 @@ export default {
         isCommon: false,
         isCustom: true,
       },
+      registerFunctions: [],
       TYPELIB: {
         jar: 0,
         Spark: {
@@ -308,6 +324,8 @@ export default {
         directory: ''
       },
       isShareLoading: false,
+      
+      registerFunctionEditable: true,
       rules: {
         name: [
           { required: true, message: this.$t('message.linkis.udf.QSRMC'), trigger: 'blur' },
@@ -514,6 +532,11 @@ export default {
     getHeight() {
       this.resize(window.innerHeight);
     },
+    handleFuncChange(val) {
+      if(this.model !== 1) {
+        this.setting.name = val;
+      }
+    },
     open(node) {
       this.resetData();
       const titleMap = [this.$t('message.linkis.udf.addFunction'), this.$t('message.linkis.udf.updateFunction')];
@@ -531,7 +554,10 @@ export default {
       this.title = titleMap[this.model];
       this.btnLabel = btnLabelMap[this.model];
     },
-
+    async getRegisterFunction(path) {
+      const res = await api.fetch('/udf/get-register-functions', { path }, 'get');
+      return res.functions
+    },
     init() {
       let { name, shared, description, path, udfName, directory, udfType, registerFormat, load, useFormat } = this.node;
       let fnType = 'Spark'
@@ -556,6 +582,18 @@ export default {
       });
       this.$nextTick(() => {
         this.$set(this.setting, this.getTypes(), path);
+        // try {
+        //   this.registerFunctions = await this.getRegisterFunction(path);
+        //   if(this.registerFunctions.length !== 0) {
+        //     this.registerFunctionEditable = false;
+        //   } else {
+        //     this.registerFunctionEditable = true;
+        //   }
+        // } catch (err) {
+        //   window.console.error(err);
+        //   this.registerFunctions = [];
+        //   this.registerFunctionEditable = true;
+        // }
       });
     },
 
@@ -659,8 +697,24 @@ export default {
       this.modalHeight = h - 380 + 'px';
     },
 
-    setNodePath(node) {
+    async setNodePath(node) {
       this.$set(this.setting, this.getTypes(), node.path);
+      if(this.model !== 1) {
+        try {
+          this.registerFunctions = await this.getRegisterFunction(node.path);
+          if(this.registerFunctions.length !== 0) {
+            this.registerFunctionEditable = false;
+            this.setting.name = '';
+          } else {
+            this.registerFunctionEditable = true;
+          }
+        } catch (err) {
+          window.console.error(err);
+          this.registerFunctions = [];
+          this.registerFunctionEditable = true;
+        }
+      }
+      
     },
 
     getTypes() {
@@ -697,6 +751,8 @@ export default {
       };
       this.showScalaRF = '';
       this.showScalaUF = '';
+      this.registerFunctions = [];
+      this.registerFunctionEditable = true;
     },
 
     getDir() {