ProgramOptions.cpp

/*
 * ProgramOptions.cpp
 * Copyright (C) 2016 Nemanja Mićović <nmicovic@outlook.com>
 *
 * Distributed under terms of the MIT license.
 */

#include "ProgramOptions.hpp"
#include "GlobalContainers.hpp"
#include "color.h"

#include <sstream>
#include <iomanip>

namespace opt = boost::program_options;

// =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
namespace vlang {
namespace util {
// =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-

bool ProgramOptions::is_init = false;

ProgramOptions& ProgramOptions::get() {
    static ProgramOptions instance;
    return instance;
}

const std::vector<std::string>& ProgramOptions::input() const {
    return m_vm["input-file"].as<std::vector<std::string>>();
}

std::string ProgramOptions::first_input_file() const {
    if (m_vm.count("input-file"))
        return m_vm["input-file"].as<std::vector<std::string>>()[0];
    else return "stdin";
}

bool ProgramOptions::contains_input_files() const {
    if (m_vm.count("input-file"))
        return true;
    else return false;
}

bool ProgramOptions::emit_source() const {
    return m_vm["emit-source"].as<bool>();
}

void ProgramOptions::set_input(opt::variables_map vm) {
    m_vm = vm;
}

const boost::program_options::variables_map& ProgramOptions::var_map() const {
    return m_vm;
}

std::string ProgramOptions::output_path() const {
    return m_vm["output"].as<std::string>();
}

std::string ProgramOptions::show_state() const {
    std::stringstream ss;
    std::string separator = std::string(BOLDRED) + DRAGON_SEPARATOR() + std::string(RESET);

    // Showing input
    ss << separator << std::endl << BOLDBLUE;

    ss << "Input: ";
    if (contains_input_files()) {
        ss << std::endl << BOLDWHITE;
        for (auto &a : input())
            ss << "* " << a << std::endl;
        ss << RESET;
    } else {
        ss << BOLDWHITE << "stdin" << RESET << std::endl;
    }

    // Showing output
    ss << BOLDBLUE << "Output: " << BOLDWHITE << output_path() << std::endl;

    ss << RESET << separator << std::endl;
    return ss.str();
}

bool ProgramOptions::syntax_highlight() const {
    return m_vm["color-dump"].as<bool>();
}

bool ProgramOptions::emit_llvm() const {
    return m_vm["emit-llvm"].as<bool>();
}

void ProgramOptions::init(int argc, char** argv) {
    if (ProgramOptions::get().is_init) {
        std::cerr << "Warning! Detected multiple init of ProgramOptions!" << std::endl;
        return;
    }

    opt::options_description desc("All options");
    desc.add_options()
        ("help", "produce help message")
        //("optimization", opt::value<std::string>()->default_value(""), "optimization level")
        ("input-file,i", opt::value<std::vector<std::string> >(), "input .vala file")
        ("output,o", opt::value<std::string>()->default_value("a.out"), " executable output path and name")
        ("emit-source,s", opt::value<bool>()->default_value(false), " shows the parsed source code")
        ("color-dump,C", opt::value<bool>()->default_value(false), " if code is shown, this option gives it syntax highlight")
        ("emit-llvm,l", opt::value<bool>()->default_value(true), " shows llvm ir on stdout")
    ;

    // Let's make any given unspecified argument as input file
    opt::positional_options_description po_desc;
    po_desc.add("input-file", -1);

    opt::variables_map vm;

    // TODO: Handle the cases when we are passed a dumb argument.
    // It seems LLVM has turned off exceptions which I need in order to acomplish this.
    try {
        opt::store(opt::command_line_parser(argc, argv).options(desc).positional(po_desc).run(), vm);
    } catch (...) {
        std::cout << desc << std::endl;
        exit(0);
    }

    opt::notify(vm);
    if (vm.count("help")) {
        std::cout << desc << std::endl;
        exit(0);
    }

    ProgramOptions::get().is_init = true;
    ProgramOptions::get().set_input(vm);
}

// =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
} // ;util
} // ;vlang
// =-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-