#ifndef SOIL_MOISTURE_IO_H
#define SOIL_MOISTURE_IO_H

#include <ostream>
#include <vector>
#include <map>
#include <memory>
#include <boost/asio/buffer.hpp>
#include <functional>

namespace trygvis {
namespace soil_moisture {

using namespace std;
using namespace boost::asio;

class Sample {
public:
    Sample() : entries() {
    }

    Sample(map<string, string> entries) : entries(entries) {
    }

    map<string, string>::iterator find(string &s) {
        return entries.find(s);
    }

    map<string, string>::iterator begin() {
        return entries.begin();
    }

    map<string, string>::iterator end() {
        return entries.end();
    }

    string &operator[](string key) {
        return entries[key];
    }

private:
    map<string, string> entries;
};

class SampleOutputStream {
public:
    virtual void write(Sample sample) = 0;
};

class CsvSampleOutputStream : public SampleOutputStream {
public:
    CsvSampleOutputStream(ostream &stream);

    CsvSampleOutputStream(ostream &stream, vector<string> fields);

    void write(Sample values);

private:
    void writeHeader();

    ostream &stream;
    bool headerWritten;
    bool filterFields;
    vector<string> fields;
};

class JsonSampleOutputStream : public SampleOutputStream {
public:
    JsonSampleOutputStream(ostream &stream);

    JsonSampleOutputStream(ostream &stream, vector<string> fields);

    void write(Sample values);

private:
    ostream &stream;
    bool filterFields;
    vector<string> fields;
};

class SqlSampleOutputStream : public SampleOutputStream {
public:
    SqlSampleOutputStream(ostream &stream);

    SqlSampleOutputStream(ostream &stream, vector<string> fields);

    void write(Sample values);

private:
    ostream &stream;
    bool filterFields;
    vector<string> fields;
};

class CsvParser {

public:
    CsvParser(shared_ptr<SampleOutputStream> output) : output(output), line(make_shared<vector<uint8_t>>()) {
    }

    void process(mutable_buffers_1 buffer);

private:
    void process_line(shared_ptr<vector<uint8_t>> packet);

    static const uint8_t packet_delimiter = '\n';
    shared_ptr<SampleOutputStream> output;
    shared_ptr<vector<uint8_t>> line;
};

}
}

#endif