From 4595dcb7ed1b9b38f7e2374c0787ca7633030d5f Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Mon, 16 Oct 2023 06:56:42 +0800
Subject: [PATCH] feat: finish adding chat role to user function

---
 api/handler/chat_role_handler.go   |  27 ++++++++
 api/main.go                        |   1 +
 api/utils/common.go                |   2 +-
 web/src/components/LoginDialog.vue | 108 +++++++++++++++++++++++++++++
 web/src/views/ChatApps.vue         |  52 +++++++++++---
 5 files changed, 178 insertions(+), 12 deletions(-)
 create mode 100644 web/src/components/LoginDialog.vue

diff --git a/api/handler/chat_role_handler.go b/api/handler/chat_role_handler.go
index 8e685497..c4b76bd4 100644
--- a/api/handler/chat_role_handler.go
+++ b/api/handler/chat_role_handler.go
@@ -2,6 +2,7 @@ package handler
 
 import (
 	"chatplus/core"
+	"chatplus/core/types"
 	"chatplus/store/model"
 	"chatplus/store/vo"
 	"chatplus/utils"
@@ -76,3 +77,29 @@ func (h *ChatRoleHandler) List(c *gin.Context) {
 	}
 	resp.SUCCESS(c, roleVos)
 }
+
+// AddRole 为用户添加角色
+func (h *ChatRoleHandler) AddRole(c *gin.Context) {
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.NotAuth(c)
+		return
+	}
+
+	var data struct {
+		Keys []string `json:"keys"`
+	}
+	if err = c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	res := h.db.Model(&model.User{}).Where("id = ?", user.Id).UpdateColumn("chat_roles_json", utils.JsonEncode(data.Keys))
+	if res.Error != nil {
+		logger.Error("添加应用失败：", err)
+		resp.ERROR(c, "更新数据库失败！")
+		return
+	}
+
+	resp.SUCCESS(c)
+}
diff --git a/api/main.go b/api/main.go
index 925130d5..94cfeba9 100644
--- a/api/main.go
+++ b/api/main.go
@@ -185,6 +185,7 @@ func main() {
 		fx.Invoke(func(s *core.AppServer, h *handler.ChatRoleHandler) {
 			group := s.Engine.Group("/api/role/")
 			group.GET("list", h.List)
+			group.POST("add", h.AddRole)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *handler.UserHandler) {
 			group := s.Engine.Group("/api/user/")
diff --git a/api/utils/common.go b/api/utils/common.go
index 01fdd32c..5147d470 100644
--- a/api/utils/common.go
+++ b/api/utils/common.go
@@ -36,7 +36,7 @@ func CopyObject(src interface{}, dst interface{}) error {
 				pType := reflect.New(value.Type())
 				v2 := pType.Interface()
 				err := json.Unmarshal([]byte(v.String()), &v2)
-				if err == nil {
+				if err == nil && v2 != nil {
 					value.Set(reflect.ValueOf(v2).Elem())
 				}
 				// map, struct, slice to string
diff --git a/web/src/components/LoginDialog.vue b/web/src/components/LoginDialog.vue
new file mode 100644
index 00000000..1effc4c5
--- /dev/null
+++ b/web/src/components/LoginDialog.vue
@@ -0,0 +1,108 @@
+<template>
+  <el-dialog
+      class="login-dialog"
+      v-model="showDialog"
+      :close-on-click-modal="true"
+      :show-close="true"
+      :before-close="close"
+      :width="400"
+      title="用户登录"
+  >
+    <div class="form">
+      <el-form label-width="65px">
+        <el-form-item>
+          <template #label>
+            <div class="label">
+              <el-icon>
+                <User/>
+              </el-icon>
+              <span>账号</span>
+            </div>
+          </template>
+          <template #default>
+            <el-input v-model="username" placeholder="手机号码"/>
+          </template>
+        </el-form-item>
+        <el-form-item>
+          <template #label>
+            <div class="label">
+              <el-icon>
+                <Lock/>
+              </el-icon>
+              <span>密码</span>
+            </div>
+          </template>
+          <template #default>
+            <el-input v-model="password" type="password" placeholder="密码"/>
+          </template>
+        </el-form-item>
+
+        <div class="login-btn">
+          <el-button type="primary" @click="submit" round>登录</el-button>
+        </div>
+      </el-form>
+    </div>
+  </el-dialog>
+</template>
+
+<script setup>
+import {computed, ref} from "vue"
+import {httpPost} from "@/utils/http";
+import {ElMessage} from "element-plus";
+import {setUserToken} from "@/store/session";
+import {validateMobile} from "@/utils/validate";
+import {Lock, User} from "@element-plus/icons-vue";
+
+// eslint-disable-next-line no-undef
+const props = defineProps({
+  show: Boolean,
+});
+const showDialog = computed(() => {
+  return props.show
+})
+const username = ref("")
+const password = ref("")
+// eslint-disable-next-line no-undef
+const emits = defineEmits(['hide']);
+const submit = function () {
+  if (!validateMobile(username.value)) {
+    return ElMessage.error('请输入合法的手机号');
+  }
+  if (password.value.trim() === '') {
+    return ElMessage.error('请输入密码');
+  }
+
+  httpPost('/api/user/login', {username: username.value.trim(), password: password.value.trim()}).then((res) => {
+    setUserToken(res.data)
+    ElMessage.success("登录成功！")
+    emits("hide")
+  }).catch((e) => {
+    ElMessage.error('登录失败，' + e.message)
+  })
+}
+const close = function () {
+  emits('hide', false);
+}
+</script>
+
+<style lang="stylus">
+.login-dialog {
+  border-radius 20px
+
+  .label {
+    .el-icon {
+      font-size 16px
+      margin-right 6px
+    }
+  }
+
+  .login-btn {
+    text-align center
+    padding-top 10px
+
+    .el-button {
+      width 50%
+    }
+  }
+}
+</style>
\ No newline at end of file
diff --git a/web/src/views/ChatApps.vue b/web/src/views/ChatApps.vue
index c663f4c3..10a47ba7 100644
--- a/web/src/views/ChatApps.vue
+++ b/web/src/views/ChatApps.vue
@@ -4,14 +4,21 @@
       AI 助手应用中心
     </div>
     <div class="inner" :style="{height: listBoxHeight + 'px'}">
-      <ItemList :items="list" v-if="list.length > 0" gap="20" width="250">
+      <ItemList :items="list" v-if="list.length > 0" :gap="20" :width="250">
         <template #default="scope">
           <div class="app-item" :style="{width: scope.width+'px'}">
             <el-image :src="scope.item.icon" fit="cover" :style="{height: scope.width+'px'}"/>
             <div class="title">
               <span class="name">{{ scope.item.name }}</span>
               <div class="opt">
-                <el-button size="small"
+
+                <el-button v-if="hasRole(scope.item.key)" size="small" type="danger">
+                  <el-icon>
+                    <Delete/>
+                  </el-icon>
+                  <span>移除应用</span>
+                </el-button>
+                <el-button v-else size="small"
                            style="--el-color-primary:#009999"
                            @click="addRole(scope.item)">
                   <el-icon>
@@ -26,34 +33,57 @@
         </template>
       </ItemList>
     </div>
+
+    <login-dialog :show="showLoginDialog" @hide="showLoginDialog = false"/>
   </div>
 </template>
 
 <script setup>
 import {onMounted, ref} from "vue"
 import {ElMessage} from "element-plus";
-import {httpGet} from "@/utils/http";
+import {httpGet, httpPost} from "@/utils/http";
 import ItemList from "@/components/ItemList.vue";
-import {Plus} from "@element-plus/icons-vue";
+import {Delete, Plus} from "@element-plus/icons-vue";
+import LoginDialog from "@/components/LoginDialog.vue";
+import {checkSession} from "@/action/session";
+import {arrayContains} from "@/utils/libs";
 
 const listBoxHeight = window.innerHeight - 97
 const list = ref([])
+const showLoginDialog = ref(false)
+const roles = ref([])
 onMounted(() => {
   httpGet("/api/role/list?all=true").then((res) => {
-    const data = res.data
-    for (let i = 0; i < data.length; i++) {
-      if (data[i].key === 'gpt') {
-        continue
-      }
-      list.value.push(data[i])
-    }
+    list.value = res.data
   }).catch(e => {
     ElMessage.error("获取应用失败：" + e.message)
   })
+
+  checkSession().then(user => {
+    roles.value = user.chat_roles
+  }).catch(() => {
+  })
 })
 
 const addRole = (row) => {
+  checkSession().then(() => {
+    const exists = arrayContains(roles.value, row.key, (v1, v2) => v1 === v2)
+    if (exists) {
+      return
+    }
+    roles.value.push(row.key)
+    httpPost("/api/role/add", {keys: roles.value}).then(() => {
+      ElMessage.success("添加应用成功！")
+    }).catch(e => {
+      ElMessage.error("添加应用失败：" + e.message)
+    })
+  }).catch(() => {
+    showLoginDialog.value = true
+  })
+}
 
+const hasRole = (roleKey) => {
+  return arrayContains(roles.value, roleKey, (v1, v2) => v1 === v2)
 }
 </script>