MISP/app/Controller/TasksController.php

<?php

App::uses('AppController', 'Controller');

class TasksController extends AppController
{
    public $components = array('RequestHandler', 'Session');

    public $paginate = array(
            'limit' => 20,
            'order' => array(
                    'Task.id' => 'desc'
            )
    );

    public function index()
    {
        if (!$this->_isSiteAdmin()) {
            throw new MethodNotAllowedException();
        }
        if (!Configure::read('MISP.background_jobs')) {
            throw new NotFoundException('Background jobs are not enabled on this instance.');
        }
        $this->__checkTasks();
        $this->recursive = 0;
        $tasks = $this->paginate();
        $this->set('list', $tasks);
        $this->set('time', time());
    }

    // checks if all the mandatory tasks exist, and if not, creates them
    // default tasks are:
    // 'cache_exports'
    private function __checkTasks()
    {
        $existingTasks = $this->Task->find('list', array('fields' => array('type')));
        foreach ($this->Task->tasks as $taskName => $taskData) {
            if (!in_array($taskName, $existingTasks)) {
                $this->Task->create();
                $this->Task->save($taskData);
            } else {
                $existingTask = $this->Task->find('first', array('recursive' => -1, 'conditions' => array('Task.type' => $taskName)));
                if ($taskData['description'] != $existingTask['Task']['description']) {
                    $existingTask['Task']['description'] = $taskData['description'];
                    $this->Task->save($existingTask);
                }
            }
        }
    }

    public function setTask()
    {
        if (!$this->_isSiteAdmin()) {
            throw new MethodNotAllowedException('You are not authorised to do that.');
        }
        $today = $this->_getTodaysTimestamp();
        if ($this->request->is('post') || $this->request->is('put')) {
            $tasks = $this->Task->find('all', array('fields' => array('id', 'timer', 'scheduled_time', 'type', 'next_execution_time')));
            foreach ($tasks as $k => $task) {
                if ($this->request->data['Task'][$task['Task']['id']]['timer'] !== $task['Task']['timer'] ||
                $this->request->data['Task'][$task['Task']['id']]['scheduled_time'] !== $task['Task']['scheduled_time'] ||
                $this->request->data['Task'][$task['Task']['id']]['next_execution_time'] !== date("Y-m-d", $task['Task']['next_execution_time'])) {
                    $this->request->data['Task'][$task['Task']['id']]['id'] = $task['Task']['id'];
                    if (isset($this->request->data['Task'][$task['Task']['id']]['next_execution_time'])) {
                        $temp = $this->request->data['Task'][$task['Task']['id']]['next_execution_time'];
                    } else {
                        $temp = date("Y-m-d", $task['Task']['next_execution_time']);
                    }
                    if (isset($this->request->data['Task'][$task['Task']['id']]['scheduled_time'])) {
                        $this->request->data['Task'][$task['Task']['id']]['next_execution_time'] = strtotime($temp . ' ' . $this->request->data['Task'][$task['Task']['id']]['scheduled_time']);
                    } else {
                        $this->request->data['Task'][$task['Task']['id']]['next_execution_time'] = strtotime($temp . ' ' . $task['Task']['scheduled_time']);
                    }
                    // schedule task
                    $this->_jobScheduler($task['Task']['type'], $this->request->data['Task'][$task['Task']['id']]['next_execution_time'], $task['Task']['id']);
                    $this->Task->save($this->request->data['Task'][$task['Task']['id']]);
                }
            }
            $this->Flash->success('Task edited');
            $this->redirect(array('action' => 'index'));
        }
    }

    private function _getTodaysTimestamp()
    {
        return strtotime(date("d/m/Y") . ' 00:00:00');
    }

    private function _jobScheduler($type, $timestamp, $id)
    {
        if ($type === 'cache_exports') {
            $this->_cacheScheduler($timestamp, $id);
        }
        if ($type === 'pull_all') {
            $this->_pullScheduler($timestamp, $id);
        }
        if ($type === 'push_all') {
            $this->_pushScheduler($timestamp, $id);
        }
        if ($type === 'cache_feeds') {
            $this->_feedScheduler($timestamp, $id, 1);
        }
        if ($type === 'fetch_feeds') {
            $this->_feedScheduler($timestamp, $id, 0);
        }
    }

    private function _cacheScheduler($timestamp, $id)
    {
        $process_id = CakeResque::enqueueAt(
                $timestamp,
                'cache',
                'EventShell',
                array('enqueueCaching', $timestamp),
                true
        );
        $this->Task->id = $id;
        $this->Task->saveField('process_id', $process_id);
    }

    private function _pushScheduler($timestamp, $id)
    {
        $process_id = CakeResque::enqueueAt(
                $timestamp,
                'default',
                'ServerShell',
                array('enqueuePush', $timestamp, $id, $this->Auth->user('id')),
                true
        );
        $this->Task->id = $id;
        $this->Task->saveField('process_id', $process_id);
    }

    private function _pullScheduler($timestamp, $id)
    {
        $process_id = CakeResque::enqueueAt(
                $timestamp,
                'default',
                'ServerShell',
                array('enqueuePull', $timestamp, $this->Auth->user('id'),  $id),
                true
        );
        $this->Task->id = $id;
        $this->Task->saveField('process_id', $process_id);
    }

    private function _feedScheduler($timestamp, $id, $type)
    {
        if ($type == 1) {
            $action = 'enqueueFeedCache';
        } else {
            $action = 'enqueueFeedFetch';
        }
        $process_id = CakeResque::enqueueAt(
                $timestamp,
                'default',
                'ServerShell',
                array($action, $timestamp, $this->Auth->user('id'),  $id),
                true
        );
        $this->Task->id = $id;
        $this->Task->saveField('process_id', $process_id);
    }
}