"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Executor = void 0;
const job_manager_1 = require("./job-manager");
const axios_1 = require("axios");
class Executor {
    executorKey;
    scheduleCenterUrl;
    accessToken;
    executorUrl;
    jobHandlers;
    jobManager;
    constructor(executorKey, scheduleCenterUrl, accessToken, jobLogPath, jobHandlers) {
        this.executorKey = executorKey;
        this.scheduleCenterUrl = scheduleCenterUrl;
        this.accessToken = accessToken;
        this.jobHandlers = jobHandlers;
        this.jobManager = new job_manager_1.JobManager(jobLogPath);
    }
    applyMiddleware({ app, domain }) {
        this.executorUrl = domain;
        const express = require('express');
        var router = new express.Router();
        router.use(async (req, res, next) => {
            res.status(200);
            const token = req.headers && req.headers['xxl-job-access-token'];
            if (!!this.accessToken && this.accessToken !== token) {
                res.send({ code: 500, msg: 'access token incorrect' });
                return;
            }
            if (!req.body) {
                res.send({ code: 500, msg: 'body is null' });
                return;
            }
            next();
        });
        this.addJobRoutes(router);
        app.use(router);
    }
    addJobRoutes(router) {
        router.post(`/beat`, async (req, res, next) => {
            res.send(this.beat());
        });
        router.post(`/idleBeat`, async (req, res, next) => {
            res.send(this.idleBeat(req.body.jobId || -1));
        });
        router.post(`/run`, async (req, res, next) => {
            res.send(this.run(req.body || {}));
        });
        router.post(`/kill`, async (req, res, next) => {
            res.send(this.killJob(req.body.jobId || -1));
        });
        router.post(`/log`, async (req, res, next) => {
            const { logDateTim, logId, fromLineNum } = req.body || {};
            const data = await this.readLog(logDateTim, logId, fromLineNum);
            res.send(data);
        });
    }
    beat() {
        return { code: 200, msg: 'success' };
    }
    idleBeat(jobId) {
        return (this.jobManager.hasJob(jobId) ? { code: 500, msg: 'busy' } : { code: 200, msg: 'idle' });
    }
    run({ jobId, executorHandler: handlerName, executorParams: jobJsonParams, executorTimeout, logId, logDateTime }) {
        const jobHandler = this.jobHandlers.get(handlerName);
        if (!jobHandler) {
            return { code: 500, msg: `no matched jobHandler(${handlerName})` };
        }
        this.jobManager.runJob(jobId, jobJsonParams, logId, logDateTime, executorTimeout, handlerName, jobHandler, this.callback.bind(this));
        return { code: 200, msg: 'success' };
    }
    killJob(jobId) {
        return { code: 500, msg: `not yet support, jobId(${jobId})` };
    }
    async readLog(logDateTime, logId, fromLineNum) {
        return {
            code: 200, content: {
                "fromLineNum": 0,
                "toLineNum": 100,
                "logContent": "test",
                "isEnd": true
            }
        };
    }
    async registry() {
        return axios_1.default.post(`${this.scheduleCenterUrl}/api/registry`, {
            registryGroup: 'EXECUTOR',
            registryKey: this.executorKey,
            registryValue: this.executorUrl
        }, {
            headers: {
                'XXL-JOB-ACCESS-TOKEN': this.accessToken,
                'Content-Type': 'application/json'
            }
        }).then((response) => {
        }).catch(error => console.error('执行器注册失败:', error.message));
    }
    async registryRemove() {
        return axios_1.default.post(`${this.scheduleCenterUrl}/api/registryRemove`, {
            'registryGroup': 'EXECUTOR',
            'registryKey': this.executorKey,
            'registryValue': this.executorUrl
        }, {
            headers: {
                'XXL-JOB-ACCESS-TOKEN': this.accessToken,
                'Content-Type': 'application/json'
            }
        }).then((response) => console.log('执行器移除成功:', response.data)).catch(error => console.error('执行器移除失败:', error.message));
    }
    async callback({ error, logId, result, use_time }) {
        var handleMsg = JSON.stringify({ use_time, result, error: error && error.message });
        return axios_1.default.post(`${this.scheduleCenterUrl}/api/callback`, [{
                logId,
                logDateTim: Date.now(),
                handleCode: error ? 500 : 200,
                handleMsg
            }], {
            headers: {
                'XXL-JOB-ACCESS-TOKEN': this.accessToken,
                'Content-Type': 'application/json'
            }
        }).then((response) => response.data).catch((err) => {
            console.error(`callback error:${JSON.stringify(result)} ${err.message}`);
        });
    }
}
exports.Executor = Executor;
//# sourceMappingURL=executor.js.map