import fs from 'fs'
import { Tail } from 'tail'
import { updateDatabase } from './put_database.js'

const 作品 = new Map()
const 游戏 = new Map()
const 截图 = new Map()
const 格式 = new Intl.DateTimeFormat('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })
const articleRegex = /\/web\/v1\/article\/get\?id=(\d+)&userId=(\d+)/
const imagesRegex = /\/api\/images\?similar=(\d+)/


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

const 存档 = async (message = '存档中..') => {
    if (正在存档) return
    正在存档 = true
    console.log(message)
    await fs.promises.writeFile(`./data/${当前日期}.json`, JSON.stringify({
        作品: Object.fromEntries(作品),
        游戏: Object.fromEntries(游戏),
        截图: Object.fromEntries(截图),
    }, null, 2))
    正在存档 = false
}

export default function () {
    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(当前日期, '开始收集日志..')
    setInterval(() => {
	    存档('10 分钟存档一次..')
	    updateDatabase(Object.fromEntries(截图))
    }, 600000)

    const tail = new Tail('/opt/log/caddy/access.log')
    tail.on('line', async (line) => {
        const item = JSON.parse(line)
        if (item.level !== 'debug') return
        if (item.msg !== 'upstream roundtrip') return

        const 日志日期 = 格式.format(new Date(item.ts * 1000)).replace(/\//g, '-')
        if (当前日期 !== 日志日期) {
            await 存档('跨日期存档..')
            作品.clear()
            游戏.clear()
            截图.clear()
            当前日期 = 日志日期
        }

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

    tail.on('end', () => console.log('没有更多内容，结束读取'))
    tail.on('error', (error) => console.error('错误:', error))

    process.on('SIGINT', async () => {
        tail.unwatch()
        await 存档('退出前储存数据..')
        process.exit()
    })

}