import fs from 'fs'
import { Tail } from 'tail'

const tail = new Tail('/opt/log/caddy/access.log')

// 指标(日榜, 周榜, 月榜, 总榜)
const 作品 = new Map()
const 游戏 = new Map()
const 截图 = new Map()
const 格式 = new Intl.DateTimeFormat('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })


var 日期 = 格式.format(new Date()).replace(/\//g, '-')
var 存档 = new Date()

if (fs.existsSync(`./data/${日期}.json`)) {
    console.log(日期, '启动时读入数据..')
    const file = fs.readFileSync(`./data/${日期}.json`, 'utf8')
    if (file) {
        const data = JSON.parse(file)
        Object.entries(data.作品).forEach(([key, value]) => 作品.set(key, value))
        Object.entries(data.游戏).forEach(([key, value]) => 游戏.set(key, value))
        Object.entries(data.截图).forEach(([key, value]) => 截图.set(key, value))
    }
}


console.log(日期, '开始收集日志..')


tail.on('line', (line) => {
    const item = JSON.parse(line)
    if (item.level !== 'debug') return
    if (item.msg !== 'upstream roundtrip') return

    if ((new Date()).getTime() - 存档.getTime() > 600000) {
        console.log('10 分钟存档一次..')
        fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({
            作品: Object.fromEntries(作品),
            游戏: Object.fromEntries(游戏),
            截图: Object.fromEntries(截图),
        }, null, 2))
        存档 = new Date()
    }

    const 日志日期 = 格式.format(new Date(item.ts * 1000)).replace(/\//g, '-')
    if (日期 !== 日志日期) {
        fs.promises.writeFile(`./data/${日期} 作品.json`, JSON.stringify(Object.fromEntries(作品), null, 2))
        fs.promises.writeFile(`./data/${日期} 游戏.json`, JSON.stringify(Object.fromEntries(游戏), null, 2))
        fs.promises.writeFile(`./data/${日期} 截图.json`, JSON.stringify(Object.fromEntries(截图), null, 2))
        日期 = 日志日期
    }

    if (item.request.uri.startsWith('/web/v1/article/get')) {
        const reg = /\/web\/v1\/article\/get\?id=(\d+)&userId=(\d+)/
        const [uri, id, user_id] = item.request.uri.match(reg) ?? []
        if (uri && id && user_id && item.request.headers.Referer) {
            if (item.request.headers.Referer[0].includes('/inspirationInfo/')) {
                游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1)
                return
            }
            if (item.request.headers.Referer[0].includes('/articleDetails/')) {
                作品.set(id, 作品.has(id) ? 作品.get(id) + 1 : 1)
                return
            }
            console.log(item.request.headers.Referer)
        }
        return
    }
    if (item.request.uri.startsWith('/api/images')) {
        if (item.request.uri.includes('page=')) {
            return
        }
        const reg = /\/api\/images\?similar=(\d+)/
        const [uri, id] = item.request.uri.match(reg) ?? []
        if (uri && id) {
            截图.set(id, 截图.has(id) ? 截图.get(id) + 1 : 1)
        }
        return
    }
})

tail.on('end', () => {
    console.log('没有更多内容，结束读取')
    const sorted = new Map([...游戏.entries()].sort((a, b) => b[1] - a[1]))
    const top10 = Array.from(sorted).slice(0, 10)
    console.log(top10)
})

tail.on('error', (error) => {
    console.error('错误:', error)
})

process.on('SIGINT', async () => {
    tail.unwatch()

    console.log('退出前储存数据..')
    await fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({
        作品: Object.fromEntries(作品),
        游戏: Object.fromEntries(游戏),
        截图: Object.fromEntries(截图),
    }, null, 2))

    process.exit()
})