From 5525d28f2d0484c96a2befb4f742af7acf29ea62 Mon Sep 17 00:00:00 2001
From: rd <1344903914@qq.com>
Date: Wed, 23 Jul 2025 17:02:24 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BF=AE=E6=94=B9=E7=99=BB=E5=BD=95?=
 =?UTF-8?q?=E9=A1=B5=E5=8A=A0=E5=85=A5=E4=BC=81=E4=B8=9A=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/join-modal.vue        |  8 ++++++--
 src/layouts/Basic.vue                | 13 +++++++++----
 src/router/guard/userLoginInfo.ts    |  3 ++-
 src/views/components/login/index.vue | 13 ++++++++++++-
 4 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/src/components/join-modal.vue b/src/components/join-modal.vue
index c05628b..0b32313 100644
--- a/src/components/join-modal.vue
+++ b/src/components/join-modal.vue
@@ -12,6 +12,7 @@ import { ref, onMounted } from 'vue';
 import { getQueryParam } from '@/utils/helper';
 
 import { getEnterpriseByInviteCode, joinEnterpriseByInviteCode } from '@/api/all';
+
 const enterprise = ref();
 const inviteCode = ref();
 
@@ -27,8 +28,11 @@ async function handleJoin() {
   await joinEnterpriseByInviteCode(inviteCode.value);
   AMessage.success('加入成功');
 }
-onMounted(() => {
-  getEnterprise();
+// onMounted(() => {
+//   getEnterprise();
+// });
+defineExpose({
+  getEnterprise,
 });
 </script>
 <style lang="scss">
diff --git a/src/layouts/Basic.vue b/src/layouts/Basic.vue
index 935161d..8f3f10d 100644
--- a/src/layouts/Basic.vue
+++ b/src/layouts/Basic.vue
@@ -1,13 +1,17 @@
-<script setup lang="ts">
+<script setup>
 import { useAppStore } from '@/stores';
 import { useResponsive } from '@/hooks';
 import JoinModal from '@/components/join-modal.vue';
 import { getQueryParam } from '@/utils/helper';
+import { useUserStore } from '@/stores';
+
 import { ref, onMounted, computed } from 'vue';
 import { useRoute } from 'vue-router';
 
 const joinEnterpriseVisible = ref(false);
+const joinModalRef = ref(null);
 const appStore = useAppStore();
+const userStore = useUserStore();
 const router = useRouter();
 const route = useRoute();
 
@@ -36,14 +40,15 @@ const paddingStyle = computed(() => {
 onMounted(() => {
   checkHasInviteCode();
 });
-const setCollapsed = (val: boolean) => {
+const setCollapsed = (val) => {
   appStore.updateSettings({ menuCollapse: val });
 };
 
 const checkHasInviteCode = () => {
   const inviteCode = getQueryParam('invite_code');
-  if (inviteCode) {
+  if (userStore.isLogin && inviteCode) {
     joinEnterpriseVisible.value = true;
+    joinModalRef.value?.getEnterprise?.();
   }
 };
 const drawerVisible = ref(false);
@@ -57,7 +62,7 @@ provide('toggleDrawerMenu', () => {
 
 <template>
   <a-layout :class="['layout', { mobile: appStore.hideMenu }]">
-    <JoinModal v-model:visible="joinEnterpriseVisible" />
+    <JoinModal v-model:visible="joinEnterpriseVisible" ref="joinModalRef" />
     <div v-if="navbar" class="layout-navbar">
       <base-navbar />
     </div>
diff --git a/src/router/guard/userLoginInfo.ts b/src/router/guard/userLoginInfo.ts
index 30e1169..cc3f891 100644
--- a/src/router/guard/userLoginInfo.ts
+++ b/src/router/guard/userLoginInfo.ts
@@ -18,9 +18,10 @@ export default function setupUserLoginInfoGuard(router: Router) {
     const routeName = to?.name as string;
     const requiresAuth = to?.meta?.requiresAuth || false;
     const requireLogin = to?.meta?.requireLogin || false;
+    const query = to?.query ?? {};
 
     if (requireLogin && !userStore.isLogin) {
-      goUserLogin();
+      goUserLogin(query);
       next();
       return;
     }
diff --git a/src/views/components/login/index.vue b/src/views/components/login/index.vue
index 9df7a08..238c240 100644
--- a/src/views/components/login/index.vue
+++ b/src/views/components/login/index.vue
@@ -125,13 +125,16 @@
 <script setup lang="ts">
 import PuzzleVerification from './components/PuzzleVerification.vue';
 import { fetchLoginCaptCha, fetchAuthorizationsCaptcha, fetchProfileInfo } from '@/api/all/login';
+import { joinEnterpriseByInviteCode } from '@/api/all';
 import { ref, reactive, onUnmounted, computed } from 'vue';
 import { useUserStore } from '@/stores';
 import { useEnterpriseStore } from '@/stores/modules/enterprise';
 import { handleUserLogin } from '@/utils/user';
 import router from '@/router';
+import { useRoute } from 'vue-router';
 
 const formRef = ref();
+const route = useRoute();
 const userStore = useUserStore();
 const enterpriseStore = useEnterpriseStore();
 const countdown = ref(0);
@@ -144,7 +147,6 @@ const submitting = ref(false);
 const hasCheck = ref(false);
 const mobileNumber = ref('');
 const selectedAccountIndex = ref(0);
-
 const accounts = ref([]);
 
 const loginForm = reactive({
@@ -299,6 +301,15 @@ const handleSubmit = async () => {
       // 处理登录成功逻辑
       AMessage.success(isLogin.value ? '登录成功' : '注册成功');
       userStore.setToken(data.access_token);
+
+      const { invite_code } = route.query;
+      if (invite_code) {
+        const { code } = await joinEnterpriseByInviteCode(invite_code as string);
+        if (code === 200) {
+          AMessage.success('加入企业成功');
+        }
+      }
+
       getProfileInfo();
     }
   } catch (error) {