zen-gitsync 2.0.5 → 2.0.7

package/src/ui/client/src/stores/gitLogStore.ts

@@ -1,7 +1,8 @@
 import { defineStore } from 'pinia'
-import { ref } from 'vue'
+import { ref, onMounted, onUnmounted } from 'vue'
 import { ElMessage } from 'element-plus'
 import { useGitStore } from './gitStore'
+import { io, Socket } from 'socket.io-client'
 
 // 定义Git操作间隔时间（毫秒）
 const GIT_OPERATION_DELAY = 300
@@ -10,6 +11,11 @@ export const useGitLogStore = defineStore('gitLog', () => {
   // 引用gitStore获取仓库状态
   const gitStore = useGitStore()
   
+  // 定义socket连接
+  let socket: Socket | null = null
+  // 自动更新状态开关
+  const autoUpdateEnabled = ref(true)
+  
   // 状态
   const log = ref<any[]>([])
   const status = ref<{ staged: string[], unstaged: string[], untracked: string[] }>({
@@ -28,6 +34,163 @@ export const useGitLogStore = defineStore('gitLog', () => {
   const isPushing = ref(false)
   const isResetting = ref(false)
   
+  // Socket.io连接处理
+  function initSocketConnection() {
+    // 如果已经有socket连接，先断开
+    if (socket) {
+      socket.disconnect()
+    }
+    
+    // 创建新连接
+    try {
+      // 修正Socket.IO连接URL，使用固定的后端服务器端口3000
+      const backendUrl = 'http://localhost:3000'
+      
+      console.log('尝试连接Socket.IO服务器:', backendUrl)
+      
+      socket = io(backendUrl, {
+        reconnectionDelayMax: 10000,
+        reconnection: true,
+        reconnectionAttempts: 10,
+        reconnectionDelay: 1000, // 初始重连延迟1秒
+        timeout: 20000, // 连接超时时间
+        autoConnect: true, // 自动连接
+        forceNew: true, // 强制创建新连接
+        transports: ['websocket', 'polling'] // 优先使用WebSocket
+      })
+      
+      // 检查socket是否成功创建
+      if (!socket) {
+        console.error('Socket.IO初始化失败: socket为null')
+        return
+      }
+      console.log('Socket.IO初始化成功，socket ID:', socket.id || '未连接')
+
+      // 监听连接事件
+      socket.on('connect', () => {
+        console.log('成功连接到Socket.IO服务器')
+        
+        // 如果自动更新开启，向服务器请求开始监控
+        if (autoUpdateEnabled.value && socket) {
+          socket.emit('start_monitoring')
+        }
+      })
+      
+      // 监听断开连接事件
+      socket.on('disconnect', (reason) => {
+        console.log('与Socket.IO服务器断开连接:', reason)
+      })
+      
+      // 监听Git状态更新事件
+      socket.on('git_status_update', (data) => {
+        if (!autoUpdateEnabled.value) return
+        
+        console.log('收到Git状态更新通知:', new Date().toLocaleTimeString())
+        
+        // 更新状态文本
+        if (data.status) {
+          statusText.value = data.status
+        }
+        
+        // 更新文件列表
+        if (data.porcelain !== undefined) {
+          parseStatusPorcelain(data.porcelain)
+        }
+        
+        // 根据文件列表判断是否需要更新日志
+        const hasChanges = fileList.value.length > 0
+        
+        // 如果有变化，更新状态
+        if (hasChanges) {
+          // 自动刷新日志，但不要显示消息通知
+          fetchLog(false)
+        }
+      })
+      
+      // 监听监控状态
+      socket.on('monitoring_status', (data) => {
+        console.log('文件监控状态:', data.active ? '已启动' : '已停止')
+      })
+      
+      // 添加额外的连接问题诊断
+      socket.on('connect_error', (error) => {
+        console.error('Socket连接错误:', error.message)
+      })
+      
+      socket.on('connect_timeout', () => {
+        console.error('Socket连接超时')
+      })
+      
+      socket.on('reconnect', (attemptNumber) => {
+        console.log(`Socket重连成功，尝试次数: ${attemptNumber}`)
+        
+        // 重连后检查自动更新状态
+        if (autoUpdateEnabled.value) {
+          console.log('重连后重新发送start_monitoring请求')
+          socket?.emit('start_monitoring')
+        }
+      })
+      
+      socket.on('reconnect_attempt', (attemptNumber) => {
+        console.log(`Socket尝试重连，第 ${attemptNumber} 次尝试`)
+      })
+      
+      socket.on('reconnect_error', (error) => {
+        console.error('Socket重连错误:', error.message)
+      })
+      
+      socket.on('reconnect_failed', () => {
+        console.error('Socket重连失败，已达到最大重试次数')
+      })
+      
+      // 手动尝试连接
+      if (socket && !socket.connected) {
+        console.log('Socket未连接，尝试手动连接...')
+        socket.connect()
+      }
+    } catch (error) {
+      console.error('Socket.IO连接初始化失败:', error)
+    }
+  }
+  
+  // 切换自动更新状态
+  function toggleAutoUpdate() {
+    // autoUpdateEnabled.value = !autoUpdateEnabled.value
+    console.log('toggleAutoUpdate调用，当前值:', autoUpdateEnabled.value)
+    
+    if (!socket) {
+      console.error('无法切换自动更新状态: socket连接不存在')
+      ElMessage.error('无法连接到服务器，自动更新可能不会生效')
+      
+      // 尝试重新初始化socket连接
+      console.log('尝试重新建立socket连接...')
+      initSocketConnection()
+      
+      // 即使socket可能不存在，也保存设置
+      localStorage.setItem('zen-gitsync-auto-update', autoUpdateEnabled.value.toString())
+      return
+    }
+    
+    try {
+      if (autoUpdateEnabled.value) {
+        console.log('发送start_monitoring命令...')
+        socket.emit('start_monitoring')
+        ElMessage.success('自动更新已启用')
+      } else {
+        console.log('发送stop_monitoring命令...')
+        socket.emit('stop_monitoring')
+        ElMessage.info('自动更新已禁用')
+      }
+      
+      // 保存设置到localStorage
+      localStorage.setItem('zen-gitsync-auto-update', autoUpdateEnabled.value.toString())
+      console.log('已保存自动更新设置到本地存储:', autoUpdateEnabled.value)
+    } catch (error) {
+      console.error('切换自动更新状态时出错:', error)
+      ElMessage.error(`切换自动更新失败: ${(error as Error).message}`)
+    }
+  }
+  
   // 解析 git status --porcelain 输出，提取文件及类型
   function parseStatusPorcelain(statusText: string) {
     if (statusText === undefined || statusText === '') {
@@ -44,12 +207,37 @@ export const useGitLogStore = defineStore('gitLog', () => {
       const match = line.match(/^([ MADRCU\?]{2})\s+(.+)$/)
       if (match) {
         let type = ''
-        const code = match[1].trim()
-        if (code === 'M' || code === 'MM' || code === 'AM' || code === 'RM') type = 'modified'
-        else if (code === 'A' || code === 'AA') type = 'added'
-        else if (code === 'D' || code === 'AD' || code === 'DA') type = 'deleted'
-        else if (code === '??') type = 'untracked'
-        else type = 'other'
+        const code = match[1]
+        const indexStatus = code.charAt(0)
+        const workTreeStatus = code.charAt(1)
+        
+        // 根据暂存区状态和工作区状态确定文件类型
+        if (indexStatus === 'A') {
+          // 已暂存的新文件
+          type = 'added'
+        } else if (indexStatus === 'M') {
+          // 已暂存的修改文件
+          type = 'added'
+        } else if (indexStatus === 'D') {
+          // 已暂存的删除文件
+          type = 'added'
+        } else if (indexStatus === 'R') {
+          // 已暂存的重命名文件
+          type = 'added'
+        } else if (indexStatus === ' ' && workTreeStatus === 'M') {
+          // 已修改未暂存的文件
+          type = 'modified'
+        } else if (indexStatus === ' ' && workTreeStatus === 'D') {
+          // 已删除未暂存的文件
+          type = 'deleted'
+        } else if (code === '??') {
+          // 未跟踪的文件
+          type = 'untracked'
+        } else {
+          // 其他情况
+          type = 'other'
+        }
+        
         files.push({ path: match[2], type })
       }
     }
@@ -57,7 +245,7 @@ export const useGitLogStore = defineStore('gitLog', () => {
   }
   
   // 获取提交日志
-  async function fetchLog() {
+  async function fetchLog(showMessage = true) {
     // 检查是否是Git仓库
     if (!gitStore.isGitRepo) {
       console.log('当前目录不是Git仓库，跳过加载提交历史')
@@ -67,18 +255,35 @@ export const useGitLogStore = defineStore('gitLog', () => {
     try {
       isLoadingLog.value = true
       console.log('开始加载提交历史...')
-      const response = await fetch('/api/log')
+      
+      // 增加时间戳参数避免缓存，确保获取最新数据
+      const timestamp = new Date().getTime()
+      const response = await fetch(`/api/log?page=1&_t=${timestamp}`)
       const data = await response.json()
-      if (data && Array.isArray(data)) {
-        log.value = data
+      
+      if (data && data.data && Array.isArray(data.data)) {
+        // 清空并更新日志数组
+        log.value = [...data.data]
+        console.log(`提交历史加载完成，共 ${log.value.length} 条记录`)
+      } else {
+        console.warn('API返回的提交历史格式不正确:', data)
+        log.value = [] // 如果数据格式不对，清空日志
+      }
+      
+      if (showMessage) {
+        ElMessage({
+          message: '提交历史已更新',
+          type: 'success'
+        })
       }
-      console.log(`提交历史加载完成，共 ${log.value.length} 条记录`)
     } catch (error) {
       console.error('获取提交历史失败:', error)
-      ElMessage({
-        message: `获取提交历史失败: ${(error as Error).message}`,
-        type: 'error'
-      })
+      if (showMessage) {
+        ElMessage({
+          message: `获取提交历史失败: ${(error as Error).message}`,
+          type: 'error'
+        })
+      }
     } finally {
       isLoadingLog.value = false
     }
@@ -121,6 +326,7 @@ export const useGitLogStore = defineStore('gitLog', () => {
   
   // 获取Git状态 (porcelain格式)
   async function fetchStatusPorcelain() {
+    console.log('开始获取Git状态(porcelain格式)...')
     // 检查是否是Git仓库
     if (!gitStore.isGitRepo) {
       console.log('当前目录不是Git仓库，跳过加载Git状态')
@@ -190,6 +396,100 @@ export const useGitLogStore = defineStore('gitLog', () => {
     }
   }
   
+  // 添加单个文件到暂存区
+  async function addFileToStage(filePath: string) {
+    // 检查是否是Git仓库
+    if (!gitStore.isGitRepo) {
+      ElMessage.warning('当前目录不是Git仓库')
+      return false
+    }
+    
+    try {
+      isAddingFiles.value = true
+      const response = await fetch('/api/add-file', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({ filePath })
+      })
+      
+      const result = await response.json()
+      if (result.success) {
+        ElMessage({
+          message: '文件已暂存',
+          type: 'success'
+        })
+        
+        // 刷新状态
+        fetchStatus()
+        
+        return true
+      } else {
+        ElMessage({
+          message: `暂存文件失败: ${result.error}`,
+          type: 'error'
+        })
+        return false
+      }
+    } catch (error) {
+      ElMessage({
+        message: `暂存文件失败: ${(error as Error).message}`,
+        type: 'error'
+      })
+      return false
+    } finally {
+      isAddingFiles.value = false
+    }
+  }
+  
+  // 取消暂存单个文件
+  async function unstageFile(filePath: string) {
+    // 检查是否是Git仓库
+    if (!gitStore.isGitRepo) {
+      ElMessage.warning('当前目录不是Git仓库')
+      return false
+    }
+    
+    try {
+      isResetting.value = true
+      const response = await fetch('/api/unstage-file', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({ filePath })
+      })
+      
+      const result = await response.json()
+      if (result.success) {
+        ElMessage({
+          message: '已取消暂存文件',
+          type: 'success'
+        })
+        
+        // 刷新状态
+        fetchStatus()
+        
+        return true
+      } else {
+        ElMessage({
+          message: `取消暂存失败: ${result.error}`,
+          type: 'error'
+        })
+        return false
+      }
+    } catch (error) {
+      ElMessage({
+        message: `取消暂存失败: ${(error as Error).message}`,
+        type: 'error'
+      })
+      return false
+    } finally {
+      isResetting.value = false
+    }
+  }
+  
   // 添加延时函数
   function delay(ms: number) {
     return new Promise(resolve => setTimeout(resolve, ms))
@@ -435,6 +735,25 @@ export const useGitLogStore = defineStore('gitLog', () => {
     }
   }
   
+  // 在组件挂载时初始化socket连接
+  onMounted(() => {
+    // 从localStorage加载自动更新设置
+    const savedAutoUpdate = localStorage.getItem('zen-gitsync-auto-update')
+    if (savedAutoUpdate !== null) {
+      autoUpdateEnabled.value = savedAutoUpdate === 'true'
+    }
+    
+    initSocketConnection()
+  })
+  
+  // 在组件卸载时断开socket连接
+  onUnmounted(() => {
+    if (socket) {
+      socket.disconnect()
+      socket = null
+    }
+  })
+  
   return {
     // 状态
     log,
@@ -447,6 +766,7 @@ export const useGitLogStore = defineStore('gitLog', () => {
     isResetting,
     isCommiting,
     isPushing,
+    autoUpdateEnabled,
     
     // 方法
     fetchLog,
@@ -454,11 +774,14 @@ export const useGitLogStore = defineStore('gitLog', () => {
     fetchStatusPorcelain,
     parseStatusPorcelain,
     addToStage,
+    addFileToStage,
+    unstageFile,
     commitChanges,
     pushToRemote,
     addAndCommit,
     addCommitAndPush,
     resetHead,
-    resetToRemote
+    resetToRemote,
+    toggleAutoUpdate
   }
 }) 

package/src/ui/client/src/stores/gitStore.ts

@@ -13,6 +13,12 @@ export const useGitStore = defineStore('git', () => {
   const isGitRepo = ref(false) // 当前目录是否是Git仓库
   const lastCheckedTime = ref(0) // 上次检查Git仓库状态的时间戳
   
+  // 添加分支状态相关变量
+  const branchAhead = ref(0) // 当前分支领先远程分支的提交数
+  const branchBehind = ref(0) // 当前分支落后远程分支的提交数
+  const hasUpstream = ref(false) // 当前分支是否有上游分支
+  const upstreamBranch = ref('') // 上游分支名称
+  
   // 添加重置方法
   function $reset() {
     currentBranch.value = ''
@@ -23,6 +29,34 @@ export const useGitStore = defineStore('git', () => {
     isCreatingBranch.value = false
     isGitRepo.value = false
     lastCheckedTime.value = 0
+    branchAhead.value = 0
+    branchBehind.value = 0
+    hasUpstream.value = false
+    upstreamBranch.value = ''
+  }
+
+  // 获取分支状态（领先/落后远程）
+  async function getBranchStatus() {
+    if (!isGitRepo.value) return;
+    
+    try {
+      const response = await fetch('/api/branch-status');
+      const data = await response.json();
+      
+      if (data) {
+        branchAhead.value = data.ahead || 0;
+        branchBehind.value = data.behind || 0;
+        hasUpstream.value = data.hasUpstream || false;
+        upstreamBranch.value = data.upstreamBranch || '';
+      }
+    } catch (error) {
+      console.error('获取分支状态失败:', error);
+      // 出错时重置状态
+      branchAhead.value = 0;
+      branchBehind.value = 0;
+      hasUpstream.value = false;
+      upstreamBranch.value = '';
+    }
   }
 
   // 检查当前目录是否是Git仓库
@@ -56,6 +90,8 @@ export const useGitStore = defineStore('git', () => {
       const data = await response.json()
       if (data.branch) {
         currentBranch.value = data.branch
+        // 获取分支状态信息
+        await getBranchStatus();
       }
     } catch (error) {
       console.error('获取分支信息失败:', error)
@@ -285,6 +321,10 @@ export const useGitStore = defineStore('git', () => {
     isCreatingBranch,
     isGitRepo,
     lastCheckedTime,
+    branchAhead,
+    branchBehind,
+    hasUpstream,
+    upstreamBranch,
     
     // 方法
     $reset,
@@ -296,6 +336,7 @@ export const useGitStore = defineStore('git', () => {
     createBranch,
     loadInitialData,
     clearUserConfig,
-    restoreUserConfig
+    restoreUserConfig,
+    getBranchStatus
   }
 })