#include <wanda/command.hpp>
#include <wanda/commander.hpp>
#include <wanda/environment.hpp>
#include <wanda/logging.hpp>
#include <wanda/xdg.hpp>

#include <asio.hpp>
#include <lyra/lyra.hpp>
#include <spdlog/sinks/stdout_color_sinks.h>
#include <spdlog/spdlog.h>

#include <cstdlib>
#include <filesystem>
#include <iostream>
#include <memory>

struct cli
{
  std::string command{};
  bool help{};

  lyra::cli_parser parser{};

  auto parse(int argc, char const * const * argv, std::ostream & error)
  {
    parser |= lyra::arg{command, "command"}("The command to send to the deamon").required() |
              lyra::help(help);

    auto result = parser.parse({argc, argv});

    if (!result)
    {
      error << "Error while processing command line arguments: "
            << result.errorMessage()
            << '\n'
            << parser
            << '\n';
      return false;
    }

    return true;
  }
};

struct listener : wanda::commander::listener
{
  listener(::cli & cli, asio::io_service & service)
      : m_cli{cli}
      , m_service{service}
  {
  }

  void on_connected(wanda::commander & commander) override
  {
    if (m_cli.command == "change")
    {
      commander.send(wanda::make_change_command());
      m_service.post([&]{
        commander.stop();
      });
    }
  }

private:
  ::cli & m_cli;
  asio::io_service & m_service;
};

int main(int argc, char const * const * argv)
{
  auto cli = ::cli{};
  if (!cli.parse(argc, argv, std::cerr))
  {
    return EXIT_FAILURE;
  }
  else if (cli.help)
  {
    std::cout << cli.parser << '\n';
    return EXIT_SUCCESS;
  }

  wanda::initialize_logger(std::make_shared<spdlog::sinks::stderr_color_sink_st>());

  auto interface = wanda::xdg_path_for(wanda::xdg_directory::runtime_dir, wanda::environment{}) / ".wanda_interface";
  auto service = asio::io_service{};
  auto listener = ::listener{cli, service};

  auto commander = wanda::commander{service, interface, listener};

  wanda::get_logger()->info("trying to connect to wanda control interface on '{}'", interface.native());
  commander.start();

  service.run();
}