/*
 * Copyright © Canonical Ltd.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 or 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <mir/server.h>

#include <mir/emergency_cleanup.h>
#include <mir/fd.h>
#include <mir/frontend/connector.h>
#include <mir/frontend/wayland.h>
#include <mir/graphics/graphic_buffer_allocator.h>
#include <mir/graphics/display_sink.h>
#include <mir/input/composite_event_filter.h>
#include <mir/input/event_filter.h>
#include <mir/options/default_configuration.h>
#include <mir/default_server_configuration.h>
#include <mir/logging/logger.h>
#include <mir/log.h>
#include <mir/main_loop.h>
#include <mir/report_exception.h>
#include <mir/run_mir.h>

// TODO these are used to frig a stub renderer when running headless
#include <mir/renderer/renderer.h>
#include <mir/graphics/renderable.h>
#include <mir/renderer/renderer_factory.h>

#include "frontend_wayland/wayland_connector.h"
#include <iostream>
#include <optional>

namespace mo = mir::options;
namespace mi = mir::input;

namespace
{
struct TemporaryCompositeEventFilter : public mi::CompositeEventFilter
{
    bool handle(MirEvent const&) override { return false; }
    void append(std::weak_ptr<mi::EventFilter> const& filter) override
    {
        append_event_filters.push_back(filter);
    }
    void prepend(std::weak_ptr<mi::EventFilter> const& filter) override
    {
        prepend_event_filters.push_back(filter);
    }

    void move_filters(std::shared_ptr<mi::CompositeEventFilter> const& composite_event_filter)
    {
        for (auto const& filter : prepend_event_filters)
            composite_event_filter->prepend(filter);

        for (auto const& filter : append_event_filters)
            composite_event_filter->append(filter);

        append_event_filters.clear();
        prepend_event_filters.clear();
    }
    std::vector<std::weak_ptr<mi::EventFilter>> prepend_event_filters;
    std::vector<std::weak_ptr<mi::EventFilter>> append_event_filters;
};
}

#define FOREACH_WRAPPER(MACRO)\
    MACRO(cursor)\
    MACRO(cursor_observer_multiplexer)\
    MACRO(display_buffer_compositor_factory)\
    MACRO(display_configuration_policy)\
    MACRO(shell)\
    MACRO(surface_stack)

#define FOREACH_OVERRIDE(MACRO)\
    MACRO(compositor)\
    MACRO(cursor_images)\
    MACRO(display_buffer_compositor_factory)\
    MACRO(gl_config)\
    MACRO(input_dispatcher)\
    MACRO(input_targeter)\
    MACRO(logger)\
    MACRO(pointer_input_dispatcher)\
    MACRO(prompt_session_listener)\
    MACRO(prompt_session_manager)\
    MACRO(server_status_listener)\
    MACRO(session_authorizer)\
    MACRO(session_listener)\
    MACRO(shell)\
    MACRO(persistent_surface_store)\
    MACRO(renderer_factory)\
    MACRO(accessibility_manager)\
    MACRO(clock)

#define FOREACH_ACCESSOR(MACRO)\
    MACRO(the_compositor)\
    MACRO(the_compositor_report)\
    MACRO(the_cursor_observer_multiplexer)\
    MACRO(the_cursor)\
    MACRO(the_display)\
    MACRO(the_display_configuration_controller)\
    MACRO(the_focus_controller)\
    MACRO(the_gl_config)\
    MACRO(the_display_platforms)    \
    MACRO(the_rendering_platforms)  \
    MACRO(the_input_targeter)  \
    MACRO(the_logger)\
    MACRO(the_main_loop)\
    MACRO(the_output_filter)\
    MACRO(the_prompt_session_listener)\
    MACRO(the_session_authorizer)\
    MACRO(the_session_coordinator)\
    MACRO(the_session_listener)\
    MACRO(the_surface_factory)\
    MACRO(the_pointer_input_dispatcher)\
    MACRO(the_prompt_session_manager)\
    MACRO(the_shell)\
    MACRO(the_shell_display_layout)\
    MACRO(the_surface_stack)\
    MACRO(the_touch_visualizer)\
    MACRO(the_input_device_hub)\
    MACRO(the_persistent_surface_store)\
    MACRO(the_display_configuration_observer_registrar)\
    MACRO(the_seat_observer_registrar)\
    MACRO(the_session_lock)\
    MACRO(the_renderer_factory)\
    MACRO(the_decoration_strategy)\
    MACRO(the_input_device_registry)\
    MACRO(the_idle_handler)\
    MACRO(the_token_authority)\
    MACRO(the_accessibility_manager)\
    MACRO(the_display_configuration_observer)\
    MACRO(the_buffer_allocator)\
    MACRO(the_default_cursor_image)\
    MACRO(the_scene_report)\
    MACRO(the_scene)\
    MACRO(the_screen_shooter_factory)

#define MIR_SERVER_BUILDER(name)\
    std::function<std::invoke_result_t<decltype(&mir::DefaultServerConfiguration::the_##name),mir::DefaultServerConfiguration*>()> name##_builder;

#define MIR_SERVER_WRAPPER(name)\
    std::function<std::invoke_result_t<decltype(&mir::DefaultServerConfiguration::the_##name),mir::DefaultServerConfiguration*>\
        (std::invoke_result_t<decltype(&mir::DefaultServerConfiguration::the_##name),mir::DefaultServerConfiguration*> const&)> name##_wrapper;

struct mir::Server::Self
{
    bool exit_status{false};
    std::string config_file;
    std::shared_ptr<ServerConfiguration> server_config;

    std::function<void()> pre_init_callback{[]{}};
    std::function<void()> init_callback{[]{}};
    std::function<void()> stop_callback{[]{}};
    int argc{0};
    char const** argv{nullptr};
    std::function<void()> exception_handler{};
    Terminator terminator{};
    EmergencyCleanupHandler emergency_cleanup_handler;
    std::shared_ptr<TemporaryCompositeEventFilter> temporary_event_filter{
        std::make_shared<TemporaryCompositeEventFilter>()};

    std::function<void(int argc, char const* const* argv)> command_line_hander{};

    /// set a callback to introduce additional configuration options.
    /// this will be invoked by run() before server initialisation starts
    void set_add_configuration_options(
        std::function<void(options::DefaultConfiguration& config)> const& add_configuration_options);

    std::function<void(options::DefaultConfiguration& config)> add_configuration_options{
        [](options::DefaultConfiguration&){}};

    FOREACH_OVERRIDE(MIR_SERVER_BUILDER)

    shell::WindowManagerBuilder wmb;

    FOREACH_WRAPPER(MIR_SERVER_WRAPPER)
};

#undef MIR_SERVER_BUILDER
#undef MIR_SERVER_WRAPPER

#define MIR_SERVER_CONFIG_OVERRIDE(name)\
auto the_##name()\
-> decltype(mir::DefaultServerConfiguration::the_##name()) override\
{\
    if (self->name##_builder)\
    {\
        if (auto const result = name([this]{ return self->name##_builder(); }))\
            return result;\
    }\
\
    return mir::DefaultServerConfiguration::the_##name();\
}

#define MIR_SERVER_CONFIG_WRAP(name)\
auto wrap_##name(decltype(Self::name##_wrapper)::result_type const& wrapped)\
-> decltype(mir::DefaultServerConfiguration::wrap_##name({})) override\
{\
    if (self->name##_wrapper)\
        return name(\
            [&] { return self->name##_wrapper(wrapped); });\
\
    return mir::DefaultServerConfiguration::wrap_##name(wrapped);\
}

struct mir::Server::ServerConfiguration : mir::DefaultServerConfiguration
{
    ServerConfiguration(
        std::shared_ptr<options::Configuration> const& configuration_options,
        std::shared_ptr<Self> const& self) :
        DefaultServerConfiguration(configuration_options),
        self(self.get())
    {
    }

    std::function<void()> the_stop_callback() override
    {
        return self->stop_callback;
    }

    using mir::DefaultServerConfiguration::the_options;

    FOREACH_OVERRIDE(MIR_SERVER_CONFIG_OVERRIDE)

    FOREACH_WRAPPER(MIR_SERVER_CONFIG_WRAP)

    auto the_window_manager_builder() -> shell::WindowManagerBuilder override
    {
        if (self->wmb) return self->wmb;
        return mir::DefaultServerConfiguration::the_window_manager_builder();
    }

    Self* const self;
};

#undef MIR_SERVER_CONFIG_OVERRIDE
#undef MIR_SERVER_CONFIG_WRAP

namespace
{
std::shared_ptr<mo::DefaultConfiguration> configuration_options(
    int argc,
    char const** argv,
    std::function<void(int argc, char const* const* argv)> const& command_line_hander,
    std::string const& config_file)
{
    std::shared_ptr<mo::DefaultConfiguration> result;

    if (command_line_hander)
        result = std::make_shared<mo::DefaultConfiguration>(argc, argv, command_line_hander, config_file);
    else
        result = std::make_shared<mo::DefaultConfiguration>(argc, argv, config_file);

    return result;
}

template<typename ConfigPtr>
void verify_setting_allowed(ConfigPtr const& initialized)
{
    if (initialized)
       BOOST_THROW_EXCEPTION(std::logic_error("Cannot amend configuration after apply_settings() call"));
}

template<typename ConfigPtr>
void verify_accessing_allowed(ConfigPtr const& initialized)
{
    if (!initialized)
       BOOST_THROW_EXCEPTION(std::logic_error("Cannot use configuration before apply_settings() call"));
}
}

mir::Server::Server() :
    self(std::make_shared<Self>())
{
}

void mir::Server::Self::set_add_configuration_options(
    std::function<void(mo::DefaultConfiguration& config)> const& add_configuration_options)
{
    this->add_configuration_options = add_configuration_options;
}


void mir::Server::set_command_line(int argc, char const* argv[])
{
    verify_setting_allowed(self->server_config);
    self->argc = argc;
    self->argv = argv;
}

void mir::Server::add_pre_init_callback(std::function<void()> const& pre_init_callback)
{
    auto const& existing = self->pre_init_callback;

    auto const updated = [=]
        {
            existing();
            pre_init_callback();
        };

    self->pre_init_callback = updated;
}

void mir::Server::add_init_callback(std::function<void()> const& init_callback)
{
    auto const& existing = self->init_callback;

    auto const updated = [=]
        {
            existing();
            init_callback();
        };

    self->init_callback = updated;
}

void mir::Server::add_stop_callback(std::function<void()> const& stop_callback)
{
    auto const& existing = self->stop_callback;

    auto const updated = [=]
        {
            stop_callback();
            existing();
        };

    self->stop_callback = updated;
}

void mir::Server::set_command_line_handler(
    std::function<void(int argc, char const* const* argv)> const& command_line_hander)
{
    verify_setting_allowed(self->server_config);
    self->command_line_hander = command_line_hander;
}

void mir::Server::set_config_filename(std::string const& config_file)
{
    verify_setting_allowed(self->server_config);
    self->config_file = config_file;
}

auto mir::Server::get_options() const -> std::shared_ptr<options::Option>
{
    verify_accessing_allowed(self->server_config);
    return self->server_config->the_options();
}

void mir::Server::set_exception_handler(std::function<void()> const& exception_handler)
{
    self->exception_handler = exception_handler;
}

void mir::Server::set_terminator(Terminator const& terminator)
{
    self->terminator = terminator;
}

void mir::Server::add_emergency_cleanup(EmergencyCleanupHandler const& handler)
{
    if (auto const& existing = self->emergency_cleanup_handler)
    {
        self->emergency_cleanup_handler = [=]
            {
                existing();
                handler();
            };
    }
    else
    {
        self->emergency_cleanup_handler = handler;
    }
}

void mir::Server::apply_settings()
{
    if (self->server_config) return;

    auto const options = configuration_options(self->argc, self->argv, self->command_line_hander, self->config_file);
    self->add_configuration_options(*options);

    auto const config = std::make_shared<ServerConfiguration>(options, self);
    self->server_config = config;

    mir::logging::set_logger(config->the_logger());
}

void mir::Server::run()
{
    try
    {
        mir::log_info("Starting");
        mir::security_log(
            logging::Severity::warning,
            "sys_startup",
            "Mir is starting up");

        verify_accessing_allowed(self->server_config);

        auto const emergency_cleanup = self->server_config->the_emergency_cleanup();
        auto const composite_event_filter = self->server_config->the_composite_event_filter();

        // keep the default_reports alive while the server is running
        auto const default_reports = self->server_config->default_reports();

        self->temporary_event_filter->move_filters(composite_event_filter);

        if (self->emergency_cleanup_handler)
            emergency_cleanup->add(self->emergency_cleanup_handler);

        self->pre_init_callback();

        run_mir(
            *self->server_config,
            [&](DisplayServer&)
                { self->init_callback(); self->init_callback = []{}; },
            self->terminator);

        mir::security_log(
            logging::Severity::warning,
            "sys_shutdown",
            "Mir is shutting down");

        self->exit_status = true;
    }
    catch (...)
    {
        if (self->exception_handler)
            self->exception_handler();
        else
            mir::report_exception(std::cerr);
    }
    /*
     * The server_config *must* outlive exception processing as it holds references to
     * the loaded platforms, and those references are keeping the DSOs loaded.
     *
     * If exception has been thrown from a platform DSO then the DSO must not be unloaded
     * before exception processing has finished.
     *
     * This is safe to have outside a try {} catch {} as shared_ptr<>::reset() is specified
     * as noexcept.
     */
    self->server_config.reset();
}
auto mir::Server::supported_pixel_formats() const -> std::vector<MirPixelFormat>
{
    return self->server_config->the_buffer_allocator()->supported_pixel_formats();
}

void mir::Server::stop()
{
    mir::log_info("Stopping");
    if (self->server_config)
        if (auto const main_loop = the_main_loop().get())
        {
            self->stop_callback();
            self->stop_callback = []{};
            main_loop->stop();
        }
}

bool mir::Server::exited_normally()
{
    return self->exit_status;
}

auto mir::Server::open_wayland_client_socket() -> Fd
{
    if (auto const config = self->server_config)
        return Fd{config->the_wayland_connector()->client_socket_fd()};

    BOOST_THROW_EXCEPTION(std::logic_error("Cannot open connection when not running"));
}

auto mir::Server::wayland_display() const -> std::optional<std::string>
{
    if (auto const config = self->server_config)
        return config->the_wayland_connector()->socket_name();

    BOOST_THROW_EXCEPTION(std::logic_error("Cannot open connection when not running"));
}

auto mir::Server::x11_display() const -> std::optional<std::string>
{
    if (auto const config = self->server_config)
        return config->the_xwayland_connector()->socket_name();

    BOOST_THROW_EXCEPTION(std::logic_error("Cannot open connection when not running"));
}

auto mir::Server::get_activation_token() const -> std::string
{
    if (auto const config = self->server_config)
    {
        return std::string(config->the_token_authority()->issue_token({}));
    }

    BOOST_THROW_EXCEPTION(std::logic_error("Cannot open connection when not running"));
}

void mir::Server::run_on_wayland_display(std::function<void(wl_display*)> const& functor)
{
    if (auto const config = self->server_config)
    {
        std::dynamic_pointer_cast<mir::frontend::WaylandConnector>(config->the_wayland_connector())
            ->run_on_wayland_display(functor);
    }
}

void mir::Server::add_wayland_extension(
    std::string const& name, std::function<std::shared_ptr<void>(
        wl_display*,
        std::function<void(std::function<void()>&& work)> const&)> builder)
{
    if (auto const config = self->server_config)
    {
        config->add_wayland_extension(name, builder);
    }
}

void mir::Server::set_wayland_extension_policy(
    std::string const& interface_name,
    std::function<bool(std::shared_ptr<scene::Session> const&, char const*)> const& policy)
{
    if (auto const config = self->server_config)
    {
        config->set_wayland_extension_policy(interface_name, policy);
    }
}

auto mir::Server::open_client_wayland(ConnectHandler const& connect_handler) -> int
{
    if (auto const config = self->server_config)
        return config->the_wayland_connector()->client_socket_fd(connect_handler);

    BOOST_THROW_EXCEPTION(std::logic_error("Cannot open connection when not running"));
}

#define MIR_SERVER_ACCESSOR(name)\
auto mir::Server::name() const -> decltype(self->server_config->name())\
{\
    verify_accessing_allowed(self->server_config);\
    return self->server_config->name();\
}

FOREACH_ACCESSOR(MIR_SERVER_ACCESSOR)

#undef MIR_SERVER_ACCESSOR

#define MIR_SERVER_OVERRIDE(name)\
void mir::Server::override_the_##name(decltype(Self::name##_builder) const& value)\
{\
    verify_setting_allowed(self->server_config);\
    self->name##_builder = value;\
}

FOREACH_OVERRIDE(MIR_SERVER_OVERRIDE)

#undef MIR_SERVER_OVERRIDE

void mir::Server::override_the_window_manager_builder(shell::WindowManagerBuilder const wmb)
{
    verify_setting_allowed(self->server_config);
    self->wmb = wmb;
}

#define MIR_SERVER_WRAP(name)\
void mir::Server::wrap_##name(decltype(Self::name##_wrapper) const& value)\
{\
    verify_setting_allowed(self->server_config);\
    self->name##_wrapper = value;\
}

FOREACH_WRAPPER(MIR_SERVER_WRAP)

#undef MIR_SERVER_WRAP

auto mir::Server::the_composite_event_filter() const -> decltype(self->server_config->the_composite_event_filter())
{
    if (self->server_config)
        return self->server_config->the_composite_event_filter();
    else
        return self->temporary_event_filter;
}

void mir::Server::add_configuration_option(
    std::string const& option,
    std::string const& description,
    int default_)
{
    verify_setting_allowed(self->server_config);
    namespace po = boost::program_options;

    auto const& existing = self->add_configuration_options;

    auto const option_adder = [=](options::DefaultConfiguration& config)
        {
            existing(config);

            config.add_options()
            (option.c_str(), po::value<int>()->default_value(default_), description.c_str());
        };

    self->set_add_configuration_options(option_adder);
}

void mir::Server::add_configuration_option(
    std::string const& option,
    std::string const& description,
    double default_)
{
    verify_setting_allowed(self->server_config);
    namespace po = boost::program_options;

    auto const& existing = self->add_configuration_options;

    auto const option_adder = [=](options::DefaultConfiguration& config)
        {
            existing(config);

            config.add_options()
            (option.c_str(), po::value<double>()->default_value(default_), description.c_str());
        };

    self->set_add_configuration_options(option_adder);
}

void mir::Server::add_configuration_option(
    std::string const& option,
    std::string const& description,
    std::string const& default_)
{
    verify_setting_allowed(self->server_config);
    namespace po = boost::program_options;

    auto const& existing = self->add_configuration_options;

    auto const option_adder = [=](options::DefaultConfiguration& config)
        {
            existing(config);

            config.add_options()
            (option.c_str(), po::value<std::string>()->default_value(default_), description.c_str());
        };

    self->set_add_configuration_options(option_adder);
}

void mir::Server::add_configuration_option(
    std::string const& option,
    std::string const& description,
    char const* default_value)
{
    add_configuration_option(option, description, std::string{default_value});
}

void mir::Server::add_configuration_option(
    std::string const& option,
    std::string const& description,
    bool default_)
{
    verify_setting_allowed(self->server_config);
    namespace po = boost::program_options;

    auto const& existing = self->add_configuration_options;

    auto const option_adder = [=](options::DefaultConfiguration& config)
        {
            existing(config);

            config.add_options()
            (option.c_str(), po::value<decltype(default_)>()->default_value(default_), description.c_str());
        };

    self->set_add_configuration_options(option_adder);
}

void mir::Server::add_configuration_option(
    std::string const& option,
    std::string const& description,
    OptionType type)
{
    verify_setting_allowed(self->server_config);
    namespace po = boost::program_options;

    auto const& existing = self->add_configuration_options;

    switch (type)
    {
    case OptionType::null:
        {
            auto const option_adder = [=](options::DefaultConfiguration& config)
                {
                    existing(config);

                    config.add_options()
                    (option.c_str(), description.c_str());
                };

            self->set_add_configuration_options(option_adder);
        }
        break;

    case OptionType::integer:
        {
            auto const option_adder = [=](options::DefaultConfiguration& config)
                {
                    existing(config);

                    config.add_options()
                    (option.c_str(), po::value<int>(), description.c_str());
                };

            self->set_add_configuration_options(option_adder);
        }
        break;

    case OptionType::string:
        {
            auto const option_adder = [=](options::DefaultConfiguration& config)
                {
                    existing(config);

                    config.add_options()
                    (option.c_str(), po::value<std::string>(), description.c_str());
                };

            self->set_add_configuration_options(option_adder);
        }
        break;

    case OptionType::boolean:
        {
            auto const option_adder = [=](options::DefaultConfiguration& config)
                {
                    existing(config);

                    config.add_options()
                    (option.c_str(), po::value<bool>(), description.c_str());
                };

            self->set_add_configuration_options(option_adder);
        }
        break;

    case OptionType::strings:
        {
            auto const option_adder = [=](options::DefaultConfiguration& config)
                {
                    existing(config);

                    config.add_options()
                    (option.c_str(), po::value<std::vector<std::string>>(), description.c_str());
                };

            self->set_add_configuration_options(option_adder);
        }
        break;

        case OptionType::real:
        {
            auto const option_adder = [=](options::DefaultConfiguration& config)
            {
                existing(config);

                config.add_options()
                        (option.c_str(), po::value<double>(), description.c_str());
            };

            self->set_add_configuration_options(option_adder);
        }
        break;
    }
}


void mir::Server::set_the_decoration_strategy(std::shared_ptr<DecorationStrategy> strategy)
{
    self->server_config->set_the_decoration_strategy(strategy);
}

void mir::Server::for_each_output_binding(wayland::Client* client, graphics::DisplayConfigurationOutputId output,
    std::function<void(wl_resource* output)> const& callback)
{
    if (auto const wc = dynamic_pointer_cast<frontend::WaylandTools>(self->server_config->the_wayland_connector()))
    {
        wc ->for_each_output_binding(client, output, callback);
    }
}
