// Copyright (c) 2021-present The Bitcoin Core developers

// Distributed under the MIT software license, see the accompanying

// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#include <btcsignals.h>

#include <interfaces/echo.h>

#include <interfaces/handler.h>

#include <memory>

#include <utility>

namespace common {

namespace {

class CleanupHandler : public interfaces::Handler

{

public:

    explicit CleanupHandler(std::function<void()> cleanup) : m_cleanup(std::move(cleanup)) {}

    ~CleanupHandler() override { if (!m_cleanup) return; m_cleanup(); m_cleanup = nullptr; }

    void disconnect() override { if (!m_cleanup) return; m_cleanup(); m_cleanup = nullptr; }

    std::function<void()> m_cleanup;

};

class SignalHandler : public interfaces::Handler

{

public:

    explicit SignalHandler(btcsignals::connection connection) : m_connection(std::move(connection)) {}

    void disconnect() override { m_connection.disconnect(); }

    btcsignals::scoped_connection m_connection;

};

class EchoImpl : public interfaces::Echo

{

public:

    std::string echo(const std::string& echo) override { return echo; }

};

} // namespace

} // namespace common

namespace interfaces {

std::unique_ptr<Handler> MakeCleanupHandler(std::function<void()> cleanup)

{

    return std::make_unique<common::CleanupHandler>(std::move(cleanup));

}

std::unique_ptr<Handler> MakeSignalHandler(btcsignals::connection connection)

{

    return std::make_unique<common::SignalHandler>(std::move(connection));

}

std::unique_ptr<Echo> MakeEcho() { return std::make_unique<common::EchoImpl>(); }

} // namespace interfaces