import _ from "lodash"; import CONSTRAINTS from "../validation/constraints"; import ErrorTypes from "../utils/errorTypes"; import * as Resources from "../utils/resources"; import {getTasks, Task} from "../jobs/scheduler"; import {normalizeString} from "../utils/strings"; import {forConstraint} from "../validation/validator"; import {Request, Response} from "express"; import {Entity} from "../utils/resources"; const isValidId = forConstraint(CONSTRAINTS.id, false); interface ExternalTask { id: number, name: string, description: string, schedule: string, runningSince: number | null, lastRunStarted: number | null, lastRunDuration: number | null, state: string, enabled: boolean, } function toExternalTask(task: Task): ExternalTask { return { id: task.id, name: task.name, description: task.description, schedule: task.schedule, runningSince: task.runningSince && task.runningSince.unix(), lastRunStarted: task.lastRunStarted && task.lastRunStarted.unix(), lastRunDuration: task.lastRunDuration || null, state: task.state, enabled: task.enabled }; } async function withValidTaskId(req: Request): Promise { const id = normalizeString(Resources.getData(req).id); if (!isValidId(id)) { throw {data: 'Invalid task id.', type: ErrorTypes.badRequest}; } return id; } async function getTask(id: string): Promise { const tasks = getTasks(); const task = tasks[id]; if (!task) { throw {data: 'Task not found.', type: ErrorTypes.notFound}; } return task; } async function withTask(req: Request): Promise { const id = await withValidTaskId(req); return await getTask(id); } function setTaskEnabled(req: Request, res: Response, enable: boolean) { withTask(req) .then(task => { task.enabled = enable; Resources.success(res, toExternalTask(task)) }) .catch(err => Resources.error(res, err)) } async function doGetAll(req: Request): Promise<{total: number, pageTasks: Entity[]}> { const restParams = await Resources.getValidRestParams('list', null, req); const tasks = Resources.sort( _.values(getTasks()), ['id', 'name', 'schedule', 'state', 'runningSince', 'lastRunStarted'], restParams ); const filteredTasks = Resources.filter( tasks, ['id', 'name', 'schedule', 'state'], restParams ); const total = filteredTasks.length; const pageTasks = Resources.getPageEntities(filteredTasks, restParams); return { total, pageTasks, }; } export function getAll (req: Request, res: Response): void { doGetAll(req) .then(({total, pageTasks}) => { res.set('X-Total-Count', total.toString(10)); Resources.success(res, _.map(pageTasks, toExternalTask)); }) .catch(err => Resources.error(res, err)); } export function run (req: Request, res: Response): void { withTask(req) .then(task => { if (task.runningSince) { return Resources.error(res, {data: 'Task already running.', type: ErrorTypes.conflict}); } task.run(); Resources.success(res, toExternalTask(task)); }) .catch(err => Resources.error(res, err)); } export function enable (req: Request, res: Response): void { setTaskEnabled(req, res, true); } export function disable (req: Request, res: Response): void { setTaskEnabled(req, res, false); }