redsigc.h

#ifndef REDSIGC_H
#define REDSIGC_H

#define REDSIGC_VERSION std::string("0.8")

#include <vector>
#include <string>
#include <cxxopts.hpp>
#include <redasm/redasm_api.h>
#include "patterngenerator.h"

struct SelectedOptions
{
    SelectedOptions(): flags(0) { }

    int flags;
    std::vector<std::string> defaultargs;
    std::string prefix, suffix, symbol, infolder, name;

    bool has(int flag) const { return flags & flag; }
    const std::string& input() const { return defaultargs.front(); }
    std::string output() const { return (defaultargs.front() == defaultargs.back()) ? (defaultargs.back() + ".json") : defaultargs.back(); }
};

class REDSigC
{
    private:
        enum { None = 0, JSONSourceOutput = 1, AutoPrefix = 2, Disassemble = 4, Folder = 8 };

    public:
        REDSigC();
        int run(int argc, char **argv);

    private:
        static std::string autoModuleName(std::string infile);
        PatternGenerator* getPatternGenerator(const std::list<std::string>& infiles);
        void getInputFiles(std::list<std::string>& infiles) const;
        bool disassemblePattern(PatternGenerator* patterngenerator);
        bool checkOptions(int argc, char **argv);

    private:
        template<typename T> static bool checkOption(const cxxopts::ParseResult& parseresult, const std::string& option, T* result = NULL);

    private:
        SelectedOptions m_options;
};

template<typename T> bool REDSigC::checkOption(const cxxopts::ParseResult& parseresult, const std::string& option, T *result)
{
    try
    {
        if(!parseresult.count(option))
            return false;

        T opt = parseresult[option].as<T>();

        if(result)
            *result = opt;
    }
    catch(const cxxopts::OptionException& e)
    {
        std::cout << e.what() << std::endl;
        return false;
    }

    return true;
}

#endif // REDSIGC_H