/* Copyright 2011-2012 (C) Daniel Richman. License: GNU GPL 3; see LICENSE. */

#ifndef HABITAT_UPLOADERTHREAD_H
#define HABITAT_UPLOADERTHREAD_H

#include <memory>
#include "jsoncpp.h"
#include "habitat/EZ.h"
#include "habitat/Uploader.h"

using namespace std;

/* Add some more EZ stuff. */
namespace habitat {

class NotInitialisedError : public runtime_error
{
public:
    NotInitialisedError() : runtime_error("habitat::NotInitialisedError") {};
};

class UploaderThread;

class UploaderAction
{
protected:
    void check(habitat::Uploader *u);

private:
    virtual void apply(UploaderThread &uthr) = 0;

    friend class UploaderThread;

public:
    virtual ~UploaderAction() {};
    virtual string describe() = 0;
};

class UploaderSettings : public UploaderAction
{
    const string callsign, couch_uri, couch_db;
    const int max_merge_attempts;

    UploaderSettings(const string &ca, const string &co_u,
                     const string &co_db, int mx)
        : callsign(ca), couch_uri(co_u), couch_db(co_db),
          max_merge_attempts(mx)
        {};
    ~UploaderSettings() {};

    void apply(UploaderThread &uthr);

    friend class UploaderThread;

public:
    string describe();
};

class UploaderReset: public UploaderAction
{
    UploaderReset() {};
    ~UploaderReset() {};

    void apply(UploaderThread &uthr);

    friend class UploaderThread;

public:
    string describe();
};

class UploaderPayloadTelemetry : public UploaderAction
{
    const string data;
    const Json::Value metadata;
    const int time_created;

    UploaderPayloadTelemetry(const string &da, const Json::Value &mda,
                             const int tc)
        : data(da), metadata(mda), time_created(tc) {};
    ~UploaderPayloadTelemetry() {};

    void apply(UploaderThread &uthr);

    friend class UploaderThread;

public:
    string describe();
};

class UploaderListenerTelemetry : public UploaderAction
{
    const Json::Value data;
    const int time_created;

    UploaderListenerTelemetry(const Json::Value &da, int tc)
        : data(da), time_created(tc) {};
    ~UploaderListenerTelemetry() {};

    void apply(UploaderThread &uthr);

    friend class UploaderThread;

public:
    string describe();
};

class UploaderListenerInfo : public UploaderAction
{
    const Json::Value data;
    const int time_created;

    UploaderListenerInfo(const Json::Value &da, int tc)
        : data(da), time_created(tc) {};
    ~UploaderListenerInfo() {};

    void apply(UploaderThread &uthr);

    friend class UploaderThread;

public:
    string describe();
};

class UploaderFlights : public UploaderAction
{
    void apply(UploaderThread &uthr);
    friend class UploaderThread;

public:
    string describe();
};

class UploaderPayloads : public UploaderAction
{
    void apply(UploaderThread &uthr);
    friend class UploaderThread;

public:
    string describe();
};

class UploaderShutdown : public UploaderAction
{
    void apply(UploaderThread &uthr);
    friend class UploaderThread;

public:
    string describe();
};

class UploaderThread : public EZ::SimpleThread
{
    EZ::Queue<UploaderAction *> queue;
    auto_ptr<habitat::Uploader> uploader;

    bool queued_shutdown;

    void queue_action(UploaderAction *ac);

    friend class UploaderAction;
    friend class UploaderSettings;
    friend class UploaderReset;
    friend class UploaderPayloadTelemetry;
    friend class UploaderListenerTelemetry;
    friend class UploaderListenerInfo;
    friend class UploaderFlights;
    friend class UploaderPayloads;

public:
    UploaderThread();
    virtual ~UploaderThread();

    void settings(const string &callsign,
                  const string &couch_uri="http://habitat.habhub.org",
                  const string &couch_db="habitat",
                  int max_merge_attempts=20);
    void reset();
    void payload_telemetry(const string &data,
                           const Json::Value &metadata=Json::Value::null,
                           int time_created=-1);
    void listener_telemetry(const Json::Value &data, int time_created=-1);
    void listener_information(const Json::Value &data, int time_created=-1);
    void flights();
    void payloads();
    void shutdown();

    void *run();
    void detach();

    virtual void log(const string &message) = 0;
    virtual void warning(const string &message);
    virtual void saved_id(const string &type, const string &id);
    virtual void initialised();
    virtual void reset_done();
    virtual void caught_exception(const NotInitialisedError &error);
    virtual void caught_exception(const runtime_error &error);
    virtual void caught_exception(const invalid_argument &error);
    virtual void got_flights(const vector<Json::Value> &flights);
    virtual void got_payloads(const vector<Json::Value> &payloads);
};

} /* namespace habitat */

#endif /* HABITAT_UPLOADERTHREAD_H */