func IsBotUserAgent(userAgent string, allowLegitimate bool) bool

package main

import (
    "fmt"
    "github.com/woodchen-ink/go-web-utils/uautil"
)

func main() {
    // 检测常见机器人
    ua1 := "python-requests/2.28.1"
    if uautil.IsBotUserAgent(ua1, false) {
        fmt.Println("python-requests 是机器人")
    }

    // 检测正常浏览器
    ua2 := "Mozilla/5.0 (Windows NT 10.0) Chrome/91.0"
    if !uautil.IsBotUserAgent(ua2, false) {
        fmt.Println("Chrome 不是机器人")
    }

    // 检测搜索引擎（SEO 友好模式）
    ua3 := "Mozilla/5.0 (compatible; Googlebot/2.1)"
    if !uautil.IsBotUserAgent(ua3, true) {
        fmt.Println("Googlebot 在 allowLegitimate=true 时被允许")
    }
}

func checkUserAgents(userAgents []string) {
    for _, ua := range userAgents {
        isBot := uautil.IsBotUserAgent(ua, true)
        status := "正常"
        if isBot {
            status = "机器人"
        }
        fmt.Printf("[%s] %s\n", status, ua)
    }
}

func main() {
    userAgents := []string{
        "python-requests/2.28.1",
        "Mozilla/5.0 (Windows NT 10.0) Chrome/91.0",
        "curl/7.68.0",
        "Mozilla/5.0 (compatible; Googlebot/2.1)",
    }

    checkUserAgents(userAgents)
}

import (
    "bufio"
    "os"
    "strings"
)

// 分析访问日志中的机器人
func analyzeLogs(logFile string) error {
    file, err := os.Open(logFile)
    if err != nil {
        return err
    }
    defer file.Close()

    scanner := bufio.NewScanner(file)
    botCount := 0
    totalCount := 0

    for scanner.Scan() {
        line := scanner.Text()
        // 假设日志格式包含 User-Agent
        ua := extractUserAgent(line)

        totalCount++
        if uautil.IsBotUserAgent(ua, true) {
            botCount++
            fmt.Printf("发现机器人: %s\n", ua)
        }
    }

    fmt.Printf("总请求: %d, 机器人: %d (%.2f%%)\n",
        totalCount, botCount, float64(botCount)/float64(totalCount)*100)

    return scanner.Err()
}

func extractUserAgent(logLine string) string {
    // 实现从日志行中提取 User-Agent 的逻辑
    // 这里只是示例
    return ""
}

func handleUserAgent(ua string) string {
    if ua == "" {
        return "缺少 User-Agent"
    }

    if uautil.IsBotUserAgent(ua, true) {
        // 恶意机器人
        return "检测到恶意机器人，访问被拒绝"
    }

    // 正常访问（包括合法搜索引擎）
    return "欢迎访问"
}