wecom_it_smart_desk/frontend-agent/src/views/Login.vue

<!-- =============================================================================
// IT智能服务台 — 坐席登录页 (v1.7, 2026-07-05)
// =============================================================================
// 说明: 简化版 - 默认显示二维码，可切换账号密码登录
// ============================================================================= -->

<template>
  <div class="login-page">
    <div class="login-card">
      <!-- 标题区 -->
      <div class="login-title">
        <h1>🛠️ IT智能服务台</h1>
        <p>坐席工作台</p>
      </div>

      <!-- 企微扫码登录（二维码） -->
      <div v-if="!showPasswordLogin" class="qr-login">
        <div class="qr-container">
          <div v-if="qrLoading" class="qr-loading">
            <el-icon class="is-loading"><Loading /></el-icon>
            <p>加载中...</p>
          </div>
          <div v-else-if="qrCode" class="qr-code">
            <img :src="qrCode" alt="企微扫码登录" />
          </div>
          <div v-else class="qr-error">
            <p>获取二维码失败</p>
            <el-button size="small" @click="fetchQrCode">重试</el-button>
          </div>
        </div>
        <p class="qr-hint">请使用企业微信扫码登录</p>

        <!-- 分隔线 -->
        <div class="divider">
          <span>其他登录方式</span>
        </div>

        <el-button
          size="large"
          class="password-login-btn"
          @click="showPasswordLogin = true"
        >
          <span>🔐</span>
          账号密码+OTP
        </el-button>
      </div>

      <!-- 账号密码+OTP 登录表单 -->
      <div v-else class="password-login">
        <el-button
          size="small"
          class="back-btn"
          @click="handleBackToQrCode"
        >
          ← 返回扫码登录
        </el-button>

        <el-form
          ref="formRef"
          :model="loginForm"
          :rules="rules"
          label-position="top"
          @submit.prevent="handleLogin"
        >
          <el-form-item label="账号" prop="userId">
            <el-input
              v-model="loginForm.userId"
              placeholder="请输入账号（如 sxn）"
              size="large"
              :prefix-icon="User"
              @keydown.enter="handleLogin"
            />
          </el-form-item>

          <el-form-item label="密码" prop="password">
            <el-input
              v-model="loginForm.password"
              type="password"
              placeholder="请输入密码"
              size="large"
              :prefix-icon="Lock"
              show-password
              @keydown.enter="handleLogin"
            />
          </el-form-item>

          <!-- OTP 输入区: 仅在 require_otp 时显示 -->
          <el-form-item v-if="requireOtp" label="OTP 验证码" prop="otpCode">
            <el-input
              v-model="loginForm.otpCode"
              placeholder="请输入 Google Authenticator 验证码"
              size="large"
              :prefix-icon="Key"
              maxlength="6"
              @keydown.enter="handleLogin"
            />
          </el-form-item>

          <el-form-item>
            <el-button
              type="primary"
              size="large"
              :loading="logging"
              :disabled="logging"
              @click="handleLogin"
              class="login-btn"
            >
              {{ logging ? '登录中...' : (requireOtp ? '验证 OTP' : '账号密码登录') }}
            </el-button>
          </el-form-item>
        </el-form>
      </div>

      <!-- 错误提示 -->
      <el-alert
        v-if="errorMsg"
        :title="errorMsg"
        type="error"
        show-icon
        :closable="true"
        @close="errorMsg = ''"
        style="margin-top: 16px"
      />

      <!-- 提示信息 -->
      <div class="login-hint">
        <p>登录即表示同意《IT智能服务台使用规范》</p>
      </div>
    </div>
  </div>
</template>

<script setup lang="ts">
import { ref, reactive, onMounted, onUnmounted } from 'vue'
import { useRouter } from 'vue-router'
import { ElMessage } from 'element-plus'
import type { FormInstance, FormRules } from 'element-plus'
import { User, Lock, Key, Loading } from '@element-plus/icons-vue'
import { useAgentStore } from '@/stores/agent'

const router = useRouter()
const agentStore = useAgentStore()
const formRef = ref<FormInstance>()

/** 是否显示账号密码登录 */
const showPasswordLogin = ref(false)

/** 企微二维码 */
const qrCode = ref('')
const qrLoading = ref(false)

/** 登录表单数据 */
const loginForm = reactive({
  userId: '',
  password: '',
  otpCode: '',
})

/** 是否需要 OTP 验证 */
const requireOtp = ref(false)

/** 登录中状态 */
const logging = ref(false)

/** 错误信息 */
const errorMsg = ref<string>('')

/** 轮询定时器 */
let pollTimer: ReturnType<typeof setInterval> | null = null
let currentTicket = ''

/** 表单校验规则 */
const rules: FormRules = {
  userId: [{ required: true, message: '请输入账号', trigger: 'blur' }],
  password: [{ required: true, message: '请输入密码', trigger: 'blur' }],
  otpCode: [
    { required: true, message: '请输入 OTP 验证码', trigger: 'blur' },
    { len: 6, message: '验证码为 6 位数字', trigger: 'blur' },
  ],
}

/**
 * 获取企微登录二维码
 */
async function fetchQrCode(): Promise<void> {
  qrLoading.value = true
  errorMsg.value = ''

  try {
    const baseUrl = import.meta.env.VITE_API_BASE_URL || '/api'
    const response = await fetch(`${baseUrl}/auth_qrcode/create`, {
      method: 'POST',
      headers: { 'Accept': 'application/json' },
    })

    if (!response.ok) {
      throw new Error(`获取二维码失败: ${response.status}`)
    }

    const result = await response.json()

    if (result.code === 0 && (result.data?.qrcode_url || result.data?.qrcode_png_base64)) {
      // qrcode_png_base64 是纯 base64，需要添加 data:image 前缀才能被 img 标签渲染
      qrCode.value = result.data.qrcode_png_base64 
        ? `data:image/png;base64,${result.data.qrcode_png_base64}` 
        : result.data.qrcode_url
      currentTicket = result.data.ticket
      startPolling()
    } else {
      throw new Error(result.message || '获取二维码失败')
    }
  } catch (error) {
    console.error('获取企微二维码失败:', error)
    errorMsg.value = error instanceof Error ? error.message : '获取二维码失败'
  } finally {
    qrLoading.value = false
  }
}

/**
 * 轮询扫码状态
 */
async function pollQrCode(): Promise<void> {
  if (!currentTicket) return

  try {
    const baseUrl = import.meta.env.VITE_API_BASE_URL || '/api'
    const response = await fetch(`${baseUrl}/auth_qrcode/poll/${currentTicket}`, {
      method: 'GET',
      headers: { 'Accept': 'application/json' },
    })

    const result = await response.json()

    if (result.code === 0 && result.data) {
      const { status, token } = result.data

      if (status === 'confirmed' && token) {
        stopPolling()
        localStorage.setItem('agent_token', token)
        ElMessage.success('登录成功')
        router.push('/workspace')
      } else if (status === 'expired') {
        stopPolling()
        ElMessage.warning('二维码已过期，请重新获取')
        fetchQrCode()
      }
    }
  } catch (error) {
    console.error('轮询扫码状态失败:', error)
  }
}

/**
 * 启动轮询
 */
function startPolling(): void {
  stopPolling()
  pollTimer = setInterval(pollQrCode, 2000)
}

/**
 * 停止轮询
 */
function stopPolling(): void {
  if (pollTimer) {
    clearInterval(pollTimer)
    pollTimer = null
  }
}

/**
 * 返回扫码登录
 */
function handleBackToQrCode(): void {
  showPasswordLogin.value = false
  fetchQrCode()
}

/**
 * 处理登录
 */
async function handleLogin(): Promise<void> {
  const valid = await formRef.value?.validate().catch(() => false)
  if (!valid) return

  logging.value = true
  errorMsg.value = ''

  try {
    const result = await agentStore.login(
      loginForm.userId.trim(),
      loginForm.password,
      loginForm.otpCode.trim() || undefined
    )

    if (result && result.require_otp) {
      requireOtp.value = true
      loginForm.otpCode = ''
      ElMessage.info('请输入 OTP 验证码')
      logging.value = false
      return
    }

    ElMessage.success('登录成功')
    router.push('/workspace')
  } catch (error: unknown) {
    if (error instanceof Error && error.message === 'require_otp') {
      requireOtp.value = true
      loginForm.otpCode = ''
      ElMessage.info('请输入 OTP 验证码')
      logging.value = false
      return
    }
    const errMsg = error instanceof Error ? error.message : '登录失败，请重试'
    errorMsg.value = errMsg
  } finally {
    logging.value = false
  }
}

onMounted(() => {
  fetchQrCode()
})

onUnmounted(() => {
  stopPolling()
})
</script>

<style scoped>
.login-page {
  display: flex;
  align-items: center;
  justify-content: center;
  min-height: 100vh;
  background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
  padding: 24px;
}

.login-card {
  width: 100%;
  max-width: 400px;
  background: var(--bg-secondary, #ffffff);
  border-radius: 16px;
  padding: 40px 32px;
  box-shadow: 0 8px 32px rgba(0, 0, 0, 0.15);
}

.login-title {
  text-align: center;
  margin-bottom: 32px;
}

.login-title h1 {
  font-size: 24px;
  font-weight: 700;
  color: var(--text-primary, #303133);
  margin: 0 0 8px 0;
}

.login-title p {
  font-size: 14px;
  color: var(--text-tertiary, #909399);
  margin: 0;
}

.qr-login {
  display: flex;
  flex-direction: column;
  align-items: center;
}

.qr-container {
  width: 200px;
  height: 200px;
  display: flex;
  align-items: center;
  justify-content: center;
  background: #f5f5f5;
  border-radius: 12px;
  margin-bottom: 12px;
}

.qr-loading {
  display: flex;
  flex-direction: column;
  align-items: center;
  gap: 8px;
  color: #909399;
}

.qr-code img {
  width: 180px;
  height: 180px;
}

.qr-error {
  text-align: center;
  color: #f56c6c;
}

.qr-hint {
  font-size: 14px;
  color: #909399;
  margin: 0 0 24px 0;
}

.divider {
  display: flex;
  align-items: center;
  width: 100%;
  margin: 0 0 24px 0;
  color: #909399;
  font-size: 14px;
}

.divider::before,
.divider::after {
  content: '';
  flex: 1;
  height: 1px;
  background: #e4e7ed;
}

.divider span {
  padding: 0 16px;
}

.password-login-btn,
.login-btn {
  width: 100%;
}

.password-login {
  padding-top: 8px;
}

.back-btn {
  margin-bottom: 16px;
}

.login-hint {
  text-align: center;
  color: var(--text-placeholder, #c0c4cc);
  font-size: 12px;
  line-height: 1.6;
  border-top: 1px solid var(--border-color-lighter, #ebeef5);
  padding-top: 16px;
  margin-top: 24px;
}

.login-hint p {
  margin: 4px 0;
}
</style>