// Copyright 2016 Etix Labs
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "dispatcher.h"

namespace etix {
namespace cameradar {

using namespace std::chrono_literals;

// The main loop of the binary
void
dispatcher::run() {
    if (not(*cache)->configure(std::make_shared<configuration>(conf))) {
        LOG_ERR_(
            "There was a problem with the cache manager, Cameradar can't work properly without "
            "cache management",
            "dispatcher");
        return;
    }
    std::thread worker(&dispatcher::do_stuff, this);
    using namespace std::chrono_literals;

    // Catch CTRL+C signal
    signal_handler::instance();

    // Wait for event or end
    while (signal_handler::instance().should_stop() not_eq stop_priority::stop &&
           current != task::finished) {
        std::this_thread::sleep_for(30ms);
    }

    if (doing_stuff()) {
        LOG_INFO_("Waiting for a task to terminate", "dispatcher");
        LOG_INFO_("Press CTRL+C again to force stop", "dispatcher");
    }

    // Waiting for task to cleanup / force stop command
    while ((signal_handler::instance().should_stop() not_eq stop_priority::force_stop) and
           doing_stuff()) {
        std::this_thread::sleep_for(30ms);
    }
    worker.join();
}

// This loop is used to add all the tasks specified in the command line
// And then run them successively
void
dispatcher::do_stuff() {
    if (opts.second.exist("-d")) {
        queue.push_back(new etix::cameradar::mapping(cache, conf, nmap_output));
        queue.push_back(new etix::cameradar::parsing(cache, conf, nmap_output));
    }
    if (opts.second.exist("-b")) {
        if (opts.second.exist("--gst-rtsp-server")) {
            queue.push_back(new etix::cameradar::path_attack(cache, conf, nmap_output));
            queue.push_back(new etix::cameradar::creds_attack(cache, conf, nmap_output));
        } else {
            queue.push_back(new etix::cameradar::creds_attack(cache, conf, nmap_output));
            queue.push_back(new etix::cameradar::path_attack(cache, conf, nmap_output));
        }
    }
    if (opts.second.exist("-t")) {
        queue.push_back(new etix::cameradar::thumbnail(cache, conf, nmap_output));
    }
    if (opts.second.exist("-g")) {
        queue.push_back(new etix::cameradar::stream_check(cache, conf, nmap_output));
    }
    if (!opts.second.exist("-d") && !opts.second.exist("-b") && !opts.second.exist("-t") &&
        !opts.second.exist("-g")) {
        queue.push_back(new etix::cameradar::mapping(cache, conf, nmap_output));
        queue.push_back(new etix::cameradar::parsing(cache, conf, nmap_output));
        if (opts.second.exist("--gst-rtsp-server")) {
            queue.push_back(new etix::cameradar::path_attack(cache, conf, nmap_output));
            queue.push_back(new etix::cameradar::creds_attack(cache, conf, nmap_output));
        } else {
            queue.push_back(new etix::cameradar::creds_attack(cache, conf, nmap_output));
            queue.push_back(new etix::cameradar::path_attack(cache, conf, nmap_output));
        }
        queue.push_back(new etix::cameradar::thumbnail(cache, conf, nmap_output));
        queue.push_back(new etix::cameradar::stream_check(cache, conf, nmap_output));
    }
    queue.push_back(new etix::cameradar::print(cache, conf, nmap_output));
    while (queue.size() > 0 && signal_handler::instance().should_stop() == stop_priority::running) {
        if (queue.front()->run())
            queue.pop_front();
        else {
            LOG_ERR_("An error occured in one of the tasks, Cameradar will now stop.",
                     "dispatcher");
            break;
        }
        std::this_thread::sleep_for(30ms);
    }
    this->current = task::finished;
}
}
}