#include <Tactility/service/ServiceRegistration.h>

#include <Tactility/Logger.h>
#include <Tactility/Mutex.h>
#include <Tactility/service/ServiceInstance.h>
#include <Tactility/service/ServiceManifest.h>

#include <string>

namespace tt::service {

static const auto LOGGER = Logger("ServiceRegistration");

typedef std::unordered_map<std::string, std::shared_ptr<const ServiceManifest>> ManifestMap;
typedef std::unordered_map<std::string, std::shared_ptr<ServiceInstance>> ServiceInstanceMap;

static ManifestMap service_manifest_map;
static ServiceInstanceMap service_instance_map;

static Mutex manifest_mutex;
static Mutex instance_mutex;

void addService(std::shared_ptr<const ServiceManifest> manifest, bool autoStart) {
    assert(manifest != nullptr);
    // We'll move the manifest pointer, but we'll need to id later
    const auto& id = manifest->id;

    LOGGER.info("Adding {}", id);

    manifest_mutex.lock();
    if (service_manifest_map[id] == nullptr) {
        service_manifest_map[id] = std::move(manifest);
    } else {
        LOGGER.error("Service id in use: {}", id);
    }
    manifest_mutex.unlock();

    if (autoStart) {
        startService(id);
    }
}

void addService(const ServiceManifest& manifest, bool autoStart) {
    addService(std::make_shared<const ServiceManifest>(manifest), autoStart);
}

std::shared_ptr<const ServiceManifest> findManifestById(const std::string& id) {
    manifest_mutex.lock();
    auto iterator = service_manifest_map.find(id);
    auto manifest = iterator != service_manifest_map.end() ? iterator->second : nullptr;
    manifest_mutex.unlock();
    return manifest;
}

static std::shared_ptr<ServiceInstance> findServiceInstanceById(const std::string& id) {
    manifest_mutex.lock();
    auto iterator = service_instance_map.find(id);
    auto service = iterator != service_instance_map.end() ? iterator->second : nullptr;
    manifest_mutex.unlock();
    return service;
}

// TODO: Return proper error/status instead of BOOL?
bool startService(const std::string& id) {
    LOGGER.info("Starting {}", id);
    auto manifest = findManifestById(id);
    if (manifest == nullptr) {
        LOGGER.error("manifest not found for service {}", id);
        return false;
    }

    auto service_instance = std::make_shared<ServiceInstance>(manifest);

    // Register first, so that a service can retrieve itself during onStart()
    instance_mutex.lock();
    service_instance_map[manifest->id] = service_instance;
    instance_mutex.unlock();

    service_instance->setState(State::Starting);
    if (service_instance->getService()->onStart(*service_instance)) {
        service_instance->setState(State::Started);
    } else {
        LOGGER.error("Starting {} failed", id);
        service_instance->setState(State::Stopped);
        instance_mutex.lock();
        service_instance_map.erase(manifest->id);
        instance_mutex.unlock();
    }

    LOGGER.info("Started {}", id);

    return true;
}

std::shared_ptr<ServiceContext> findServiceContextById(const std::string& id) {
    return findServiceInstanceById(id);
}

std::shared_ptr<Service> findServiceById(const std::string& id) {
    auto instance = findServiceInstanceById(id);
    return instance != nullptr ? instance->getService() : nullptr;
}

bool stopService(const std::string& id) {
    LOGGER.info("Stopping {}", id);
    auto service_instance = findServiceInstanceById(id);
    if (service_instance == nullptr) {
        LOGGER.warn("Service not running: {}", id);
        return false;
    }

    service_instance->setState(State::Stopping);
    service_instance->getService()->onStop(*service_instance);
    service_instance->setState(State::Stopped);

    instance_mutex.lock();
    service_instance_map.erase(id);
    instance_mutex.unlock();

    if (service_instance.use_count() > 1) {
        LOGGER.warn("Possible memory leak: service {} still has {} references", service_instance->getManifest().id, service_instance.use_count() - 1);
    }

    LOGGER.info("Stopped {}", id);

    return true;
}

State getState(const std::string& id) {
    auto service_instance = findServiceInstanceById(id);
    if (service_instance == nullptr) {
        return State::Stopped;
    }
    return service_instance->getState();
}

} // namespace