#include "SoilMoistureIo.h"
#include "json.hpp"
#include "apps.h"
#include <fstream>

namespace trygvis {
namespace apps {

using namespace std;
using namespace trygvis::apps;
using namespace trygvis::soil_moisture;
namespace po = boost::program_options;

class sample_convert : public app {

public:
    sample_convert() : table_name(""), input_file(""), input_format(""),
                       output_file(""), output_format("") {
    }

    void add_options(po::options_description_easy_init &options) override {
        options
                ("help", "produce this help message")
                ("input", po::value<string>(&input_file)->default_value("-"))
                ("input-format", po::value<string>(&input_format)->default_value("csv"))
                ("output", po::value<string>(&output_file)->default_value("-"))
                ("output-format", po::value<string>(&output_format)->default_value("plain"));
    }

    void add_extra_options(po::options_description &all_options) override {
        po::options_description sql("SQL");
        sql.add_options()
                ("table-name", po::value<string>(&table_name));

        all_options.add(sql);
    };

    int main(app_execution &execution) override {
        auto desc = execution.desc;
        auto vm = execution.vm;

        shared_ptr<SampleOutputStream> output;

        istream *inputStream;
        if (input_file == "-") {
            inputStream = &cin;
        } else {
            inputStream = new ifstream(input_file);
            if (inputStream->fail()) {
                cerr << "Unable to open input file " << input_file << endl;
                return EXIT_FAILURE;
            }
        }

        ostream *outputStream;
        if (output_file == "-") {
            outputStream = &cout;
        } else {
            outputStream = new ofstream(output_file);
            if (outputStream->fail()) {
                cerr << "Unable to open output file " << output_file << endl;
                return EXIT_FAILURE;
            }
        }

        if (output_format == "plain") {
            output = make_shared<CsvSampleOutputStream>(*outputStream);
        } else if (output_format == "json") {
            output = make_shared<JsonSampleOutputStream>(*outputStream);
        } else if (output_format == "sql") {
            if (table_name.size() == 0) {
                cerr << "Missing option: table-name" << endl;
                return EXIT_FAILURE;
            }

            output = make_shared<SqlSampleOutputStream>(*outputStream, table_name);
        } else {
            cerr << "Unsupported output format: " << output_format << endl;
            return EXIT_FAILURE;
        }

        auto input = make_shared<CsvParser>(output);

        char data[100];
        while (!inputStream->eof()) {
            inputStream->get(data[0]);
            input->process(boost::asio::buffer(data, 1));
        }

        delete outputStream;

        return EXIT_SUCCESS;
    }

private:
    string input_file, input_format;
    string output_file, output_format;

    string table_name;
};

}
}

using namespace trygvis::apps;

int main(int argc, char *argv[]) {
    sample_convert app;
    return launch_app(argc, argv, app);
}