package main

import (
    "context"
    "fmt"
    fs "github.com/Miuzarte/FlareSolverr-go"
    stb "github.com/Miuzarte/SoutuBot-go"
)

func main() {
    fsClient := fs.NewClient("http://127.0.0.1:8191/v1")
    client := stb.NewClient(fsClient)
    ctx := context.Background()

    var imgData []byte // raw png/jpg/webp
    resp, err := client.Search(ctx, imgData)
    if err != nil {
        panic(err)
    }

    // 获取之前的搜索结果
    const id = "2026011412270786"
    // resp, err := client.GetResult(ctx, id)

    fmt.Printf("上传的原图片: %s\n", resp.ImageUrl)
    fmt.Printf(
        "耗时 %.2fs\n%s\n找到了 %d 条相似的结果\n",
        resp.ExecutionTime,
        resp.SearchOption,
        len(resp.Data),
    )
    if len(resp.Data) == 0 {
        return
    }
    if resp.Data[0].Similarity < stb.MATCH_SIMILARITY_THRESHOLD {
        fmt.Println("最大匹配度低于45，结果可能不正确")
        fmt.Println("请自行判断，或更换严格模式/其他搜图引擎来搜索")
    }

    var lastI int
    for i, item := range resp.Data {
        if item.Similarity < stb.LOW_SIMILARITY_THRESHOLD {
            // 匹配度必定按从大到小排序
            if i > 3 {
                // 保留前三条低匹配度结果, 剩余的不展示
                break
            }
        }

        var hosts []string = item.Source.Hosts()
        // [0]: "https://nhentai.net"  / [1]: "https://nhentai.xxx"
        // [0]: "https://e-hentai.org" / [1]: "https://exhentai.org"
        // [0]: "https://panda.chaika.moe"

        fmt.Printf(
            "[%d] %s\n匹配度: %.2f%%\n语言: %s\n来源: %s\n",
            i, item.Title,
            item.Similarity,
            item.Language.Emoji(),
            item.Source,
        )
        fmt.Printf("详情页1: %s\n", hosts[0]+item.SubjectPath)
        if len(hosts) >= 2 {
            fmt.Printf("详情页2: %s\n", hosts[1]+item.SubjectPath)
        }
        fmt.Printf("图片页1: %s\n", hosts[0]+item.PagePath)
        if len(hosts) >= 2 {
            fmt.Printf("图片页2: %s\n", hosts[1]+item.PagePath)
        }

        lastI = i
    }

    if lastI+1 < len(resp.Data) {
        fmt.Printf("未显示剩余低匹配度结果(%d条)\n", len(resp.Data)-lastI-1),
    }
}