/* Ricochet - https://ricochet.im/
 * Copyright (C) 2014, John Brooks <john.brooks@dereferenced.net>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *    * Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *    * Redistributions in binary form must reproduce the above
 *      copyright notice, this list of conditions and the following disclaimer
 *      in the documentation and/or other materials provided with the
 *      distribution.
 *
 *    * Neither the names of the copyright owners nor the names of its
 *      contributors may be used to endorse or promote products derived from
 *      this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#pragma once

#include "CryptoKey.h"
#include "bytearray.h"

namespace Tor
{

// This class is used to receive synchroneous notifications from the hidden service.
// Each client should implement its own notification handling.

class HiddenServiceClient
{
public:
    virtual void hiddenServiceStatusChanged(int /* newStatus */, int /* oldStatus */) =0;
    virtual void hiddenServiceOnline() =0;
    virtual void hiddenServicePrivateKeyChanged() =0;
    virtual void hiddenServiceHostnameChanged() =0;
};

class HiddenService
{
    friend class TorControl;

public:
    struct Target
    {
        std::string targetAddress;
        uint16_t servicePort, targetPort;
    };

    enum Status
    {
        NotCreated = -1, /* Service has not been created yet */
        Offline = 0, /* Data exists, but service is not published */
        Online /* Published */
    };

    HiddenService(HiddenServiceClient *client);
    HiddenService(HiddenServiceClient *client, const std::string &dataPath);
    HiddenService(HiddenServiceClient *client, const CryptoKey &privateKey, const std::string &dataPath = std::string());

    Status status() const { return m_status; }

    const std::string& hostname() const { return m_hostname; }
    const std::string  serviceId() const { return m_service_id.toString(); }
    const std::string& dataPath() const { return m_dataPath; }

    CryptoKey privateKey() { return m_privateKey; }
    void setPrivateKey(const CryptoKey &privateKey);
    void setServiceId(const ByteArray &sid);

    const std::list<Target> &targets() const { return m_targets; }
    void addTarget(const Target &target);
    void addTarget(uint16_t servicePort, std::string targetAddress, uint16_t targetPort);

    void servicePublished();

private:
    std::string m_dataPath;
    std::list<Target> m_targets;
    std::string m_hostname;
    Status m_status;
    CryptoKey m_privateKey;
    ByteArray m_service_id;

    bool loadPrivateKey();
    void setStatus(Status newStatus);

    HiddenServiceClient *m_client;

    // make the object non copyable
    HiddenService(const HiddenService&) {}
    HiddenService& operator=(const HiddenService&) { return *this ; }
};

}