fscan/common/progress_manager.go

package common

import (
	"fmt"
	"os"
	"runtime"
	"strings"
	"sync"
	"sync/atomic"
	"time"

	"github.com/shadow1ng/fscan/common/i18n"
)

/*
ProgressManager.go - 固定底部进度条管理器

提供固定在终端底部的进度条显示，与正常输出内容分离。
使用终端控制码实现位置固定和内容保护。
*/

// ProgressManager 进度条管理器
type ProgressManager struct {
	mu              sync.RWMutex
	enabled         bool
	total           int64
	current         int64
	description     string
	startTime       time.Time
	isActive        bool
	terminalHeight  int
	reservedLines   int // 为进度条保留的行数
	lastContentLine int // 最后一行内容的位置

	// 输出缓冲相关
	outputMutex sync.Mutex

	// 活跃指示器相关
	spinnerIndex     int
	lastActivity     time.Time
	activityTicker   *time.Ticker
	stopActivityChan chan struct{}

	// 内存监控相关
	lastMemUpdate time.Time
	memStats      runtime.MemStats

	// 进度条更新控制（减少 Windows 终端的重复输出）
	lastRenderedPercent int
}

// =============================================================================
// ANSI终端控制码常量
// =============================================================================

const (
	// AnsiClearLine 光标和行控制 - 清除当前行并回到行首
	AnsiClearLine = "\033[2K\r"
	// AnsiMoveCursor 向上移动N行（格式化字符串）
	AnsiMoveCursor = "\033[%dA"

	// AnsiRed 颜色代码 - 红色文本
	AnsiRed = "\033[31m"
	// AnsiGreen 绿色文本
	AnsiGreen = "\033[32m"
	// AnsiYellow 黄色文本
	AnsiYellow = "\033[33m"
	// AnsiCyan 青色文本
	AnsiCyan = "\033[36m"
	// AnsiGray 灰色文本
	AnsiGray = "\033[90m"
	// AnsiReset 重置所有属性
	AnsiReset = "\033[0m"
)

var (
	globalProgressManager *ProgressManager
	progressMutex         sync.Mutex

	// 活跃指示器字符序列（旋转动画）
	spinnerChars = []string{"|", "/", "-", "\\"}

	// 活跃指示器更新间隔
	activityUpdateInterval = 500 * time.Millisecond
)

// GetProgressManager 获取全局进度条管理器
func GetProgressManager() *ProgressManager {
	progressMutex.Lock()
	defer progressMutex.Unlock()

	if globalProgressManager == nil {
		globalProgressManager = &ProgressManager{
			enabled:        true,
			reservedLines:  2, // 保留2行：进度条 + 空行
			terminalHeight: getTerminalHeight(),
		}
	}
	return globalProgressManager
}

// InitProgress 初始化进度条
func (pm *ProgressManager) InitProgress(total int64, description string) {
	fv := GetFlagVars()
	if fv.DisableProgress || fv.Silent {
		pm.enabled = false
		return
	}

	pm.mu.Lock()
	defer pm.mu.Unlock()

	pm.total = total
	pm.current = 0
	pm.description = description
	pm.startTime = time.Now()
	pm.isActive = true
	pm.enabled = true
	pm.lastActivity = time.Now()
	pm.spinnerIndex = 0
	pm.lastMemUpdate = time.Now().Add(-2 * time.Second) // 强制首次更新内存
	pm.lastRenderedPercent = -1                         // 强制首次渲染

	// 为进度条保留空间
	pm.setupProgressSpace()

	// 启动活跃指示器
	pm.startActivityIndicator()

	// 初始显示进度条
	pm.renderProgress()
}

// UpdateProgress 更新进度
func (pm *ProgressManager) UpdateProgress(increment int64) {
	if !pm.enabled || !pm.isActive {
		return
	}

	pm.mu.Lock()
	defer pm.mu.Unlock()

	pm.current += increment
	if pm.current > pm.total {
		pm.current = pm.total
	}

	// 更新活跃时间
	pm.lastActivity = time.Now()

	pm.renderProgress()
}

// =============================================================================================
// 已删除的死代码（未使用）：SetProgress 设置当前进度
// =============================================================================================

// FinishProgress 完成进度条
func (pm *ProgressManager) FinishProgress() {
	if !pm.enabled || !pm.isActive {
		return
	}

	pm.mu.Lock()
	defer pm.mu.Unlock()

	pm.current = pm.total
	pm.renderProgress()

	// 停止活跃指示器
	pm.stopActivityIndicator()

	// 显示完成信息
	pm.showCompletionInfo()

	// 清理进度条区域，恢复正常输出
	pm.clearProgressArea()
	pm.isActive = false
}

// setupProgressSpace 设置进度条空间
func (pm *ProgressManager) setupProgressSpace() {
	// 简化设计：进度条在原地更新，不需要预留额外空间
	// 只是标记进度条开始的位置
	pm.lastContentLine = 0
}

// =============================================================================================
// 已删除的死代码（未使用）：moveToContentArea 和 moveToProgressLine 方法
// =============================================================================================

// renderProgress 渲染进度条（使用锁避免输出冲突）
func (pm *ProgressManager) renderProgress() {
	pm.outputMutex.Lock()
	defer pm.outputMutex.Unlock()

	pm.renderProgressUnsafe()
}

// generateProgressBar 生成进度条字符串
func (pm *ProgressManager) generateProgressBar() string {
	if pm.total == 0 {
		spinner := pm.getActivityIndicator()
		memInfo := pm.getMemoryInfo()

		// 获取TCP包统计（包含原HTTP请求）
		packetCount := GetGlobalState().GetPacketCount()
		tcpSuccess := GetGlobalState().GetTCPSuccessPacketCount()
		tcpFailed := GetGlobalState().GetTCPFailedPacketCount()
		udpCount := GetGlobalState().GetUDPPacketCount()

		packetInfo := ""
		if packetCount > 0 {
			// 构建简化的包统计信息：只显示TCP和UDP
			details := make([]string, 0, 2)
			if tcpSuccess > 0 || tcpFailed > 0 {
				details = append(details, fmt.Sprintf("TCP:%d✓%d✗", tcpSuccess, tcpFailed))
			}
			if udpCount > 0 {
				details = append(details, fmt.Sprintf("UDP:%d", udpCount))
			}

			if len(details) > 0 {
				packetInfo = fmt.Sprintf(" 发包:%d[%s]", packetCount, strings.Join(details, ","))
			} else {
				packetInfo = fmt.Sprintf(" 发包:%d", packetCount)
			}
		}

		return fmt.Sprintf("%s %s 等待中...%s %s", pm.description, spinner, packetInfo, memInfo)
	}

	percentage := float64(pm.current) / float64(pm.total) * 100
	elapsed := time.Since(pm.startTime)

	// 获取并发状态
	concurrencyStatus := GetConcurrencyMonitor().GetConcurrencyStatus()

	// 计算预估剩余时间
	var eta string
	if pm.current > 0 {
		totalTime := elapsed * time.Duration(pm.total) / time.Duration(pm.current)
		remaining := totalTime - elapsed
		if remaining > 0 {
			eta = fmt.Sprintf(" ETA:%s", formatDuration(remaining))
		}
	}

	// 计算速度
	speed := float64(pm.current) / elapsed.Seconds()
	speedStr := ""
	if speed > 0 {
		speedStr = fmt.Sprintf(" (%.1f/s)", speed)
	}

	// 生成进度条
	barWidth := 30
	filled := int(percentage * float64(barWidth) / 100)
	bar := ""

	if GetFlagVars().NoColor {
		// 无颜色版本
		bar = "[" +
			fmt.Sprintf("%s%s",
				string(make([]rune, filled)),
				string(make([]rune, barWidth-filled))) +
			"]"
		for i := 0; i < filled; i++ {
			bar = bar[:i+1] + "=" + bar[i+2:]
		}
		for i := filled; i < barWidth; i++ {
			bar = bar[:i+1] + "-" + bar[i+2:]
		}
	} else {
		// 彩色版本
		bar = "|"
		for i := 0; i < barWidth; i++ {
			if i < filled {
				bar += "#"
			} else {
				bar += "."
			}
		}
		bar += "|"
	}

	// 生成活跃指示器
	spinner := pm.getActivityIndicator()

	// 获取TCP包统计（包含原HTTP请求）
	packetCount := GetGlobalState().GetPacketCount()
	tcpSuccess := GetGlobalState().GetTCPSuccessPacketCount()
	tcpFailed := GetGlobalState().GetTCPFailedPacketCount()
	udpCount := GetGlobalState().GetUDPPacketCount()

	packetInfo := ""
	if packetCount > 0 {
		// 构建简化的包统计信息：只显示TCP和UDP
		details := make([]string, 0, 2)
		if tcpSuccess > 0 || tcpFailed > 0 {
			details = append(details, fmt.Sprintf("TCP:%d✓%d✗", tcpSuccess, tcpFailed))
		}
		if udpCount > 0 {
			details = append(details, fmt.Sprintf("UDP:%d", udpCount))
		}

		if len(details) > 0 {
			packetInfo = fmt.Sprintf(" 发包:%d[%s]", packetCount, strings.Join(details, ","))
		} else {
			packetInfo = fmt.Sprintf(" 发包:%d", packetCount)
		}
	}

	// 构建基础进度条
	baseProgress := fmt.Sprintf("%s %s %6.1f%% %s (%d/%d)%s%s%s",
		pm.description, spinner, percentage, bar, pm.current, pm.total, speedStr, eta, packetInfo)

	// 添加内存信息
	memInfo := pm.getMemoryInfo()

	// 添加并发状态
	if concurrencyStatus != "" {
		return fmt.Sprintf("%s [%s] %s", baseProgress, concurrencyStatus, memInfo)
	}

	return fmt.Sprintf("%s %s", baseProgress, memInfo)
}

// showCompletionInfo 显示完成信息
func (pm *ProgressManager) showCompletionInfo() {
	elapsed := time.Since(pm.startTime)

	// 换行并显示完成信息
	fmt.Print("\n")

	completionMsg := i18n.GetText("progress_scan_completed")
	if GetFlagVars().NoColor {
		fmt.Printf("[完成] %s %d/%d (耗时: %s)\n",
			completionMsg, pm.total, pm.total, formatDuration(elapsed))
	} else {
		fmt.Printf("%s[完成] %s %d/%d%s %s(耗时: %s)%s\n",
			AnsiGreen, completionMsg, pm.total, pm.total, AnsiReset,
			AnsiGray, formatDuration(elapsed), AnsiReset)
	}
}

// clearProgressArea 清理进度条区域
func (pm *ProgressManager) clearProgressArea() {
	// 简单清除当前行
	fmt.Print(AnsiClearLine)
}

// IsActive 检查进度条是否活跃
func (pm *ProgressManager) IsActive() bool {
	pm.mu.RLock()
	defer pm.mu.RUnlock()
	return pm.isActive && pm.enabled
}

// getTerminalHeight 获取终端高度
func getTerminalHeight() int {
	// 对于固定底部进度条，我们暂时禁用终端高度检测
	// 因为在不同终端环境中可能会有问题
	// 改为使用相对定位方式
	return 0 // 返回0表示使用简化模式
}

// formatDuration 格式化时间间隔
func formatDuration(d time.Duration) string {
	if d < time.Minute {
		return fmt.Sprintf("%.1fs", d.Seconds())
	}
	if d < time.Hour {
		return fmt.Sprintf("%.1fm", d.Minutes())
	}
	return fmt.Sprintf("%.1fh", d.Hours())
}

// InitProgressBar 初始化进度条（全局函数，方便其他模块调用）
func InitProgressBar(total int64, description string) {
	GetProgressManager().InitProgress(total, description)
}

// UpdateProgressBar 更新进度条
func UpdateProgressBar(increment int64) {
	GetProgressManager().UpdateProgress(increment)
}

// =============================================================================================
// 已删除的死代码（未使用）：SetProgressBar 全局函数
// =============================================================================================

// FinishProgressBar 完成进度条
func FinishProgressBar() {
	GetProgressManager().FinishProgress()
}

// IsProgressActive 检查进度条是否活跃
func IsProgressActive() bool {
	return GetProgressManager().IsActive()
}

// GetProgressPercent 获取当前进度百分比 (0-100)
func GetProgressPercent() float64 {
	return GetProgressManager().GetPercent()
}

// GetPercent 获取当前进度百分比
func (pm *ProgressManager) GetPercent() float64 {
	pm.mu.RLock()
	defer pm.mu.RUnlock()

	if !pm.isActive || pm.total == 0 {
		return 0
	}
	return float64(pm.current) / float64(pm.total) * 100
}

// =============================================================================
// 日志输出协调功能
// =============================================================================

// LogWithProgress 在进度条活跃时协调日志输出
func LogWithProgress(message string) {
	pm := GetProgressManager()
	if !pm.IsActive() {
		// 如果进度条不活跃，直接输出
		fmt.Println(message)
		return
	}

	pm.outputMutex.Lock()
	defer pm.outputMutex.Unlock()

	// 清除当前行（清除进度条）
	// Windows 通过 progress_manager_win.go 已启用 ANSI 支持
	fmt.Print(AnsiClearLine)

	// 输出日志消息
	fmt.Println(message)

	// 不重绘进度条，等待下次 UpdateProgress 自动绘制
}

// renderProgressUnsafe 不加锁的进度条渲染（内部使用）
func (pm *ProgressManager) renderProgressUnsafe() {
	if !pm.enabled || !pm.isActive {
		return
	}

	// 计算当前百分比（避免除零）
	currentPercent := 0
	if pm.total > 0 {
		currentPercent = int((pm.current * 100) / pm.total)
	}

	// 只在百分比变化时更新，减少不必要的渲染
	if currentPercent == pm.lastRenderedPercent && currentPercent < 100 {
		return
	}
	pm.lastRenderedPercent = currentPercent

	// 生成进度条内容
	progressBar := pm.generateProgressBar()

	// 移动到行首（Windows 已通过 progress_manager_win.go 启用 ANSI 支持）
	fmt.Print("\r")

	// 输出进度条（带颜色，如果启用）
	if GetFlagVars().NoColor {
		fmt.Print(progressBar)
	} else {
		fmt.Printf("%s%s%s", AnsiCyan, progressBar, AnsiReset)
	}

	// 刷新输出
	_ = os.Stdout.Sync()
}

// =============================================================================
// 活跃指示器相关方法
// =============================================================================

// startActivityIndicator 启动活跃指示器
func (pm *ProgressManager) startActivityIndicator() {
	// 防止重复启动
	if pm.activityTicker != nil {
		return
	}

	pm.activityTicker = time.NewTicker(activityUpdateInterval)
	pm.stopActivityChan = make(chan struct{})

	go func() {
		for {
			select {
			case <-pm.activityTicker.C:
				// 只有在活跃状态下才更新指示器
				if pm.isActive && pm.enabled {
					pm.mu.Lock()
					pm.spinnerIndex = (pm.spinnerIndex + 1) % len(spinnerChars)
					pm.mu.Unlock()

					// 只有在长时间没有进度更新时才重新渲染
					// 这样可以避免频繁更新时的性能问题
					if time.Since(pm.lastActivity) > 2*time.Second {
						pm.renderProgress()
					}
				}
			case <-pm.stopActivityChan:
				return
			}
		}
	}()
}

// stopActivityIndicator 停止活跃指示器
func (pm *ProgressManager) stopActivityIndicator() {
	if pm.activityTicker != nil {
		pm.activityTicker.Stop()
		pm.activityTicker = nil
	}

	if pm.stopActivityChan != nil {
		close(pm.stopActivityChan)
		pm.stopActivityChan = nil
	}
}

// getActivityIndicator 获取当前活跃指示器字符
func (pm *ProgressManager) getActivityIndicator() string {
	// 如果最近有活动（2秒内），显示静态指示器
	if time.Since(pm.lastActivity) <= 2*time.Second {
		return "●" // 实心圆表示活跃
	}

	// 如果长时间没有活动，显示旋转指示器表明程序仍在运行
	return spinnerChars[pm.spinnerIndex]
}

// getMemoryInfo 获取内存使用信息
func (pm *ProgressManager) getMemoryInfo() string {
	// 限制内存统计更新频率以提高性能（每秒最多一次）
	now := time.Now()
	if now.Sub(pm.lastMemUpdate) >= time.Second {
		runtime.ReadMemStats(&pm.memStats)
		pm.lastMemUpdate = now
	}

	// 获取当前使用的内存（以MB为单位）
	memUsedMB := float64(pm.memStats.Alloc) / 1024 / 1024

	// 根据内存使用量选择颜色
	var colorCode string
	if GetFlagVars().NoColor {
		return fmt.Sprintf("内存:%.1fMB", memUsedMB)
	}

	// 根据内存使用量设置颜色
	if memUsedMB < 50 {
		colorCode = AnsiGreen // 绿色 - 内存使用较低
	} else if memUsedMB < 100 {
		colorCode = AnsiYellow // 黄色 - 内存使用中等
	} else {
		colorCode = AnsiRed // 红色 - 内存使用较高
	}

	return fmt.Sprintf("%s内存:%.1fMB%s", colorCode, memUsedMB, AnsiReset)
}

// =============================================================================
// 并发监控器 (从 concurrency_monitor.go 合并)
// =============================================================================

/*
ConcurrencyMonitor - 并发监控器

监控两个层级的并发：
1. 主扫描器线程数 (-t 参数控制)
2. 插件内连接线程数 (-mt 参数控制)
*/

// ConcurrencyMonitor 并发监控器
type ConcurrencyMonitor struct {
	// 主扫描器层级
	activePluginTasks int64 // 当前活跃的插件任务数
	totalPluginTasks  int64 // 总插件任务数

	// 插件内连接层级已移除 - 原代码为死代码，无任何调用者
}

// 已移除 PluginConnectionInfo 结构体 - 原为死代码，无任何使用

var (
	globalConcurrencyMonitor *ConcurrencyMonitor
	concurrencyMutex         sync.Once
)

// GetConcurrencyMonitor 获取全局并发监控器
func GetConcurrencyMonitor() *ConcurrencyMonitor {
	concurrencyMutex.Do(func() {
		globalConcurrencyMonitor = &ConcurrencyMonitor{
			activePluginTasks: 0,
			totalPluginTasks:  0,
		}
	})
	return globalConcurrencyMonitor
}

// =============================================================================
// 主扫描器层级监控
// =============================================================================

// StartPluginTask 开始插件任务
func (m *ConcurrencyMonitor) StartPluginTask() {
	atomic.AddInt64(&m.activePluginTasks, 1)
	atomic.AddInt64(&m.totalPluginTasks, 1)
}

// FinishPluginTask 完成插件任务
func (m *ConcurrencyMonitor) FinishPluginTask() {
	atomic.AddInt64(&m.activePluginTasks, -1)
}

// GetPluginTaskStats 获取插件任务统计
func (m *ConcurrencyMonitor) GetPluginTaskStats() (active int64, total int64) {
	return atomic.LoadInt64(&m.activePluginTasks), atomic.LoadInt64(&m.totalPluginTasks)
}

// =============================================================================
// 已移除插件内连接层级监控 - 原为死代码，无任何调用者
// =============================================================================

// 已移除未使用的 Reset 方法

// GetConcurrencyStatus 获取并发状态字符串
func (m *ConcurrencyMonitor) GetConcurrencyStatus() string {
	activePlugins, _ := m.GetPluginTaskStats()

	if activePlugins == 0 {
		return ""
	}

	return fmt.Sprintf("%s:%d", i18n.GetText("concurrency_plugin"), activePlugins)
}

// 已移除未使用的 GetDetailedStatus 方法