import { fastify } from 'fastify'
import { mongoose } from 'mongoose'
import * as Hooks from './hooks.js'
import * as AuthenticateRoutings from './routes/authenticate.js'
import * as SessionServerRoutings from './routes/session.js'
import * as AdvancedRoutings from './routes/advanced.js'
import * as APIRoutings from './routes/api.js'
import { config } from './config.js'
import { readFileSync } from 'fs'
import { Scenes, session, Telegraf } from 'telegraf'
import { allScenes, registerAllPlayerCommands } from './telegram/player-commands.js';
import { Player } from './models/player.js'
import fastifySwagger from '@fastify/swagger'

export const server = fastify({
    logger: {
        prettyPrint: true,
        // level: 'error'
    }
})

export const telegraf = new Telegraf(config.telegram.token)

export const setup = async () => {
    server.log.info("老色批世界树 > 初始化中...")

    await mongoose.connect(config.database.url)
    const publicKey = readFileSync(config.signing.public).toString()
    const privateKey = readFileSync(config.signing.private).toString()

    server.decorate('keys', { publicKey, privateKey })

    config.custom.preHooks(server)
    server.addHook('preHandler', Hooks.headerValidation)
    server.register(fastifySwagger, {
        routePrefix: '/docs',
        swagger: {
            title: "lsp-yggdrasil 接口文档",
            version: "1.0.0",
            tags: [
                { name: "Authserver", description: "Yggdrasil Authserver 协议定义的接口"},
                { name: "Sessionserver", description: "Yggdrasil Authserver 协议定义的接口"},
                { name: "api", description: "Yggdrasil Authserver 协议定义的接口"},
                { name: "webapi", description: "web前端接口"},
            ]
        },
        exposeRoute: true,
    })

    config.custom.preRouting(server)
    
    // Authserver routings
    server.route(AuthenticateRoutings.authenticate)
    server.route(AuthenticateRoutings.refresh)
    server.route(AuthenticateRoutings.validate)
    server.route(AuthenticateRoutings.invalidate)
    server.route(AuthenticateRoutings.signout)

    server.route(SessionServerRoutings.join)
    server.route(SessionServerRoutings.hasJoined)
    server.route(SessionServerRoutings.profile)

    server.route(AdvancedRoutings.meta)
    server.route(AdvancedRoutings.status)

    server.route(APIRoutings.profiles)

    config.custom.postRouting(server)

    if(process.env["UNIT_TEST"] || process.env["DEVEL_FIRST_RUN"]) {
    // Create a test player
        await new Player({
            username: 'test',
            password: '8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92',
            email: 'i@lama.icu',
            uuid: '098f6bcd-4621-3373-8ade-4e832627b4f6',
            texture: {
                skin: 'assets.lama.icu/textures/skin/steve.png',
                cape: 'assets.lama.icu/textures/cape/default.png'
            },
            registerDate: Date.now(),
            permissions: [{ node: 'login', allowed: true, duration: 0, eternal: true, startDate: Date.now(), highPriority: false }],
            telegramBind: {
                username: 'test',
                verified: true,
            }
        }).save() 
    }

    const stage = new Scenes.Stage([ ...allScenes ])
    telegraf.use(session())
    telegraf.use(stage.middleware())

    registerAllPlayerCommands()
}

const launch = async () => { 
    process.on('SIGINT', shutdown)
    process.on('SIGTERM', shutdown)

    telegraf.launch()
    
    await server.listen(config.server.port, config.server.url)

    server.log.info("老色批世界树 > 基于 fastify 的高性能 HTTP 服务器已启动")
}

export const shutdown = async () => {
    await server.close()
    server.log.info("老色批世界树 > HTTP 服务器已关闭")
    try {
        telegraf.stop()
        server.log.info("老色批世界树 > Telegram Bot 已关闭")
    } catch(err) {
        server.log.info("老色批世界树 > Telegram Bot 未运行，已跳过")
    }
    mongoose.disconnect()
    server.log.info("老色批世界树 > 数据库连接已断开，服务器已关闭")
}

(async () => {
    if(!process.env["UNIT_TEST"]) {
        console.log(`
 ================================================================
    __   _____ ______  __                __                _ __
   / /  / ___// __ \\ \\/ /___ _____ _____/ /________ ______(_) /
  / /   \\__ \\/ /_/ /\\  / __ \`/ __ \`/ __  / ___/ __ \`/ ___/ / / 
 / /______/ / ____/ / / /_/ / /_/ / /_/ / /  / /_/ (__  ) / /  
/_____/____/_/     /_/\\__, /\\__, /\\__,_/_/   \\__,_/____/_/_/   
                     /____//____/                              
================================================================\n`)

        if(typeof PROGRAM_PRODUCTION === 'undefined') {
            console.warn("⚠ 警告: 您运行的不是正式版本，可能会不稳定，仅限开发环境运行！\n")
        }
        await setup()
        await launch()
    }
})()