compilescript/src/main.cpp

/*  This file is part of compilescript.
 *  Copyright © 2018, 2019 tastytea <tastytea@tastytea.de>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, version 3.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "version.hpp"
#include "xdgcfg.hpp"
#include <libconfig.h++>
#include <basedir.h>
#include <unistd.h>
#include <chrono>
#include <cstdlib>
#include <exception>
#include <experimental/filesystem>
#include <fstream>
#include <iostream>
#include <regex>
#include <string>
#include <system_error>
#include <vector>

namespace fs = std::experimental::filesystem;
using std::cout;
using std::cerr;
using std::endl;
using std::chrono::system_clock;
using std::chrono::hours;
using std::chrono::duration_cast;
using std::string;
using std::vector;

class Compilescript
{
private:
    string _compiler;
    fs::path _cache_dir;
    int _clean_after_hours;

public:
    Compilescript();

    //! Read settings and set member variables.
    void read_settings();
    //! Remove files older than `_clean_after_hours` from the cache.
    void cleanup();
    //! Compile `filename` if necessary and return filename of binary.
    string compile(const string &filename);
    //! Run binary. Pass original `argv`, the 1st entry is ignored.
    void run(const string &filename, char *argv[]);
    //! Print version, copyright and license.
    void print_version();
    //! Set compiler command.
    void set_compiler(const string &command);
};

Compilescript::Compilescript()
    : _compiler("g++ -x c++")
    , _clean_after_hours(30 * 24)
{}


void Compilescript::read_settings()
{
    bool need_save = false;
    xdgcfg config("compilescript.cfg");
    config.read();
    libconfig::Setting &cfg = config.get_cfg().getRoot();

    if (cfg.exists("compiler"))
    {
        _compiler = cfg["compiler"].c_str();
    }
    else
    {
        cfg.add("compiler", libconfig::Setting::TypeString) = _compiler;
        need_save = true;
    }

    if (cfg.exists("clean_after_hours"))
    {
        cfg.lookupValue("clean_after_hours", _clean_after_hours);
    }
    else
    {
        cfg.add("clean_after_hours",
                libconfig::Setting::TypeInt) = _clean_after_hours;
        need_save = true;
    }

    if (cfg.exists("cache_dir"))
    {
        _cache_dir = cfg["cache_dir"].c_str();
    }
    else
    {
        xdgHandle xdg;
        xdgInitHandle(&xdg);
        _cache_dir = xdgCacheHome(&xdg);
        _cache_dir /= "compilescript";
        xdgWipeHandle(&xdg);
    }
    if (!fs::is_directory(_cache_dir))
    {
        fs::create_directories(_cache_dir);
    }

    if (need_save)
    {
        config.write();
    }
}

void Compilescript::cleanup()
{
    for (const auto &entry : fs::recursive_directory_iterator(_cache_dir))
    {
        if (!fs::is_regular_file(entry))
        {
            continue;
        }

        const auto diff = system_clock::now() - fs::last_write_time(entry);
        if (duration_cast<hours>(diff).count() > _clean_after_hours)
        {
            fs::path current_path = entry.path();
            std::error_code e;

            while (fs::remove(current_path, e))
            {
                current_path = current_path.parent_path();
                if (current_path == _cache_dir)
                {
                    break;
                }
            }
        }
    }
}

string Compilescript::compile(const string &filename)
{
    const fs::path original = fs::canonical(filename);
    const fs::path source = _cache_dir / original;
    const fs::path binary = (source.string() + ".bin");
    string compiler_arguments;
    fs::create_directories(binary.parent_path());

    if (!fs::exists(binary) ||
        fs::last_write_time(original) > fs::last_write_time(binary))
    {
        std::ifstream in(original);
        if (in.is_open())
        {
            std::ofstream out(source);
            if (out.is_open())
            {
                string buf;

                std::getline(in, buf);
                if (!(buf.substr(0, 2) == "#!"))
                {
                    out << buf << endl;
                }

                std::getline(in, buf);
                const std::regex re("^(//|#|;) ?compilescript:");
                std::smatch match;
                if (std::regex_search(buf, match, re))
                {
                    compiler_arguments = match.suffix();
                }
                else
                {
                    out << buf << endl;
                }

                while (!in.eof())
                {
                    std::getline(in, buf);
                    out << buf << endl;
                }

                in.close();
                out.close();
            }
            else
            {
                throw std::runtime_error("Could not open file: "
                                         + source.string());
            }
        }
        else
        {
            throw std::runtime_error("Could not open file: "
                                     + original.string());
        }

        const string command = _compiler + " " + source.string() + " "
            + compiler_arguments + " -o " + binary.string();
        int ret = std::system(command.c_str()); // NOLINT(cert-env33-c)
        if (ret != 0)
        {
            throw std::runtime_error("Compilation failed.");
        }
    }

    return binary.string();
}

void Compilescript::run(const string &filename, char *argv[])
{
    execvp(filename.c_str(), argv);
}

void Compilescript::print_version()
{
    cout << "compilescript " << global::version << '\n' <<
        "Copyright (C) 2018, 2019 tastytea <tastytea@tastytea.de>\n"
        "License GPLv3: GNU GPL version 3 "
        "<https://www.gnu.org/licenses/gpl-3.0.html>.\n"
        "This program comes with ABSOLUTELY NO WARRANTY. This is free software,"
        "\nand you are welcome to redistribute it under certain conditions.\n";
}

void Compilescript::set_compiler(const string &command)
{
    _compiler = command;
}

int main(int argc, char *argv[])
{
    const vector<string> args(argv, argv + argc);

    try
    {
        Compilescript App;
        App.read_settings();

        if (args.size() <= 1)
        {
            cerr << "usage: " << args[0]
                 << " [file|--cleanup|--version|--compiler command] "
                 << "[arguments]\n";
            return 1;
        }
        if (args[1] == "--cleanup")
        {
            App.cleanup();
            return 0;
        }
        if (args[1] == "--version")
        {
            App.print_version();
            return 0;
        }
        if (args[1] == "--compiler")
        {
            if (args.size() <= 3)
            {
                cerr << "Error: You need to specify a command.\n";
                return 1;
            }
            App.set_compiler(args[2]);
            const string binary = App.compile(args[3]);
            // We know that argv[3] exists.
            // NOLINTNEXTLINE(cppcoreguidelines-pro-bounds-pointer-arithmetic)
            App.run(binary, &argv[3]);
            return 0;
        }

        const string binary = App.compile(args[1]);
        // We know that argv[1] exists.
        // NOLINTNEXTLINE(cppcoreguidelines-pro-bounds-pointer-arithmetic)
        App.run(binary, &argv[1]);
    }
    catch (const std::exception &e)
    {
        cerr << "Error: " << e.what() << endl;
        return 1;
    }

    return 0;
}