// Copyright (c) Benjamin Kietzman (github.com/bkietz)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

#include <dbus/connection.hpp>
#include <dbus/endpoint.hpp>
#include <dbus/filter.hpp>
#include <dbus/match.hpp>
#include <dbus/message.hpp>
#include <functional>

#include <unistd.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>

TEST(AvahiTest, GetHostName) {
  dbus::endpoint test_daemon("org.freedesktop.Avahi", "/",
                             "org.freedesktop.Avahi.Server");
  boost::asio::io_service io;
  auto system_bus = std::make_shared<dbus::connection>(io, dbus::bus::system);

  dbus::message m = dbus::message::new_call(test_daemon, "GetHostName");

  system_bus->async_send(
      m, [&](const boost::system::error_code ec, dbus::message r) {

        std::string avahi_hostname;
        std::string hostname;

        // get hostname from a system call
        char c[1024];
        gethostname(c, 1024);
        hostname = c;

        r.unpack(avahi_hostname);

        // Get only the host name, not the fqdn
        auto unix_hostname = hostname.substr(0, hostname.find("."));
        EXPECT_EQ(unix_hostname, avahi_hostname);

        io.stop();
      });
  boost::asio::deadline_timer t(io, boost::posix_time::seconds(10));
  t.async_wait([&](const boost::system::error_code& /*e*/) {
    io.stop();
    FAIL() << "Callback was never called\n";
  });
  io.run();
}

TEST(AvahiTest, ServiceBrowser) {
  boost::asio::io_service io;
  auto system_bus = std::make_shared<dbus::connection>(io, dbus::bus::system);

  dbus::endpoint test_daemon("org.freedesktop.Avahi", "/",
                             "org.freedesktop.Avahi.Server");
  // create new service browser
  dbus::message m1 = dbus::message::new_call(test_daemon, "ServiceBrowserNew");
  m1.pack((int32_t)-1, (int32_t)-1, "_http._tcp", "local", (uint32_t)(0));

  dbus::message r = system_bus->send(m1);
  dbus::object_path browser_path;
  EXPECT_TRUE(r.unpack(browser_path));
  testing::Test::RecordProperty("browserPath", browser_path.value);

  dbus::match ma(system_bus, "type='signal',path='" + browser_path.value + "'");
  dbus::filter f(system_bus, [](dbus::message& m) {
    auto member = m.get_member();
    return member == "NameAcquired";
  });

  std::function<void(boost::system::error_code, dbus::message)> event_handler =
      [&](boost::system::error_code ec, dbus::message s) {
        testing::Test::RecordProperty("firstSignal", s.get_member());
        std::string a = s.get_member();
        std::string dude;
        s.unpack(dude);
        f.async_dispatch(event_handler);
        io.stop();
      };
  f.async_dispatch(event_handler);

  boost::asio::deadline_timer t(io, boost::posix_time::seconds(10));
  t.async_wait([&](const boost::system::error_code& /*e*/) {
    io.stop();
    FAIL() << "Callback was never called\n";
  });
  io.run();
}

TEST(BOOST_DBUS, ListServices) {
  boost::asio::io_service io;
  boost::asio::deadline_timer t(io, boost::posix_time::seconds(10));
  t.async_wait([&](const boost::system::error_code& /*e*/) {
    io.stop();
    FAIL() << "Callback was never called\n";
  });

  auto system_bus = std::make_shared<dbus::connection>(io, dbus::bus::system);

  dbus::endpoint test_daemon("org.freedesktop.DBus", "/",
                             "org.freedesktop.DBus");
  // create new service browser
  dbus::message m = dbus::message::new_call(test_daemon, "ListNames");
  system_bus->async_send(
      m, [&](const boost::system::error_code ec, dbus::message r) {
        io.stop();
        std::vector<std::string> services;
        r.unpack(services);
        // Test a couple things that should always be present.... adapt if
        // neccesary
        EXPECT_THAT(services, testing::Contains("org.freedesktop.DBus"));
        EXPECT_THAT(services, testing::Contains("org.freedesktop.Accounts"));

      });

  io.run();
}

TEST(BOOST_DBUS, SingleSensorChanged) {
  boost::asio::io_service io;

  auto system_bus = std::make_shared<dbus::connection>(io, dbus::bus::system);

  dbus::match ma(system_bus,
                 "type='signal',path_namespace='/xyz/openbmc_project/sensors'");

  dbus::filter f(system_bus, [](dbus::message& m) {
    auto member = m.get_member();
    return member == "PropertiesChanged";
  });

  f.async_dispatch([&](boost::system::error_code ec, dbus::message s) {
    std::string object_name;
    EXPECT_EQ(s.get_path(),
              "/xyz/openbmc_project/sensors/temperature/LR_Brd_Temp");

    std::vector<std::pair<std::string, dbus::dbus_variant>> values;
    s.unpack(object_name, values);
    EXPECT_EQ(object_name, "xyz.openbmc_project.Sensor.Value");

    EXPECT_EQ(values.size(), 1);
    auto expected = std::pair<std::string, dbus::dbus_variant>("Value", 42);
    EXPECT_EQ(values[0], expected);

    io.stop();
  });

  dbus::endpoint test_endpoint(
      "org.freedesktop.Avahi",
      "/xyz/openbmc_project/sensors/temperature/LR_Brd_Temp",
      "org.freedesktop.DBus.Properties");

  auto signal_name = std::string("PropertiesChanged");
  auto m = dbus::message::new_signal(test_endpoint, signal_name);

  m.pack("xyz.openbmc_project.Sensor.Value");

  std::vector<std::pair<std::string, dbus::dbus_variant>> map2;

  map2.emplace_back("Value", 42);

  m.pack(map2);

  auto removed = std::vector<uint32_t>();
  m.pack(removed);
  system_bus->async_send(m,
                         [&](boost::system::error_code ec, dbus::message s) {});

  io.run();
}

TEST(BOOST_DBUS, MultipleSensorChanged) {
  boost::asio::io_service io;
  auto system_bus = std::make_shared<dbus::connection>(io, dbus::bus::system);

  dbus::match ma(system_bus,
                 "type='signal',path_namespace='/xyz/openbmc_project/sensors'");
  dbus::filter f(system_bus, [](dbus::message& m) {
    auto member = m.get_member();
    return member == "PropertiesChanged";
  });

  int count = 0;
  std::function<void(boost::system::error_code, dbus::message)> callback = [&](
      boost::system::error_code ec, dbus::message s) {
    std::string object_name;
    EXPECT_EQ(s.get_path(),
              "/xyz/openbmc_project/sensors/temperature/LR_Brd_Temp");

    std::vector<std::pair<std::string, dbus::dbus_variant>> values;
    s.unpack(object_name, values);
    EXPECT_EQ(object_name, "xyz.openbmc_project.Sensor.Value");

    EXPECT_EQ(values.size(), 1);
    auto expected = std::pair<std::string, dbus::dbus_variant>("Value", 42);
    EXPECT_EQ(values[0], expected);
    count++;
    if (count == 2) {
      io.stop();
    } else {
      f.async_dispatch(callback);
    }
    s.unpack(object_name, values);

  };
  f.async_dispatch(callback);

  dbus::endpoint test_endpoint(
      "org.freedesktop.Avahi",
      "/xyz/openbmc_project/sensors/temperature/LR_Brd_Temp",
      "org.freedesktop.DBus.Properties");

  auto signal_name = std::string("PropertiesChanged");

  std::vector<std::pair<std::string, dbus::dbus_variant>> map2;

  map2.emplace_back("Value", 42);

  static auto removed = std::vector<uint32_t>();

  auto m = dbus::message::new_signal(test_endpoint, signal_name);
  m.pack("xyz.openbmc_project.Sensor.Value", map2, removed);

  system_bus->send(m, std::chrono::seconds(0));
  system_bus->send(m, std::chrono::seconds(0));

  io.run();
}

TEST(BOOST_DBUS, MethodCall) {
  boost::asio::io_service io;

  boost::asio::deadline_timer t(io, boost::posix_time::seconds(2));
  t.async_wait([&](const boost::system::error_code&) {
    io.stop();
    FAIL() << "Callback was never called\n";
  });

  auto bus = std::make_shared<dbus::connection>(io, dbus::bus::session);
  std::string requested_name = bus->get_unique_name();

  dbus::filter f(bus, [](dbus::message& m) {
    return (m.get_member() == "Get" &&
            m.get_interface() == "org.freedesktop.DBus.Properties" &&
            m.get_signature() == "ss");
  });

  std::function<void(boost::system::error_code, dbus::message)> method_handler =
      [&](boost::system::error_code ec, dbus::message s) {
        if (ec) {
          FAIL() << ec;
        } else {
          std::string intf_name, prop_name;
          s.unpack(intf_name, prop_name);

          EXPECT_EQ(intf_name, "xyz.openbmc_project.fwupdate1");
          EXPECT_EQ(prop_name, "State");

          // send a reply so dbus doesn't get angry?
          auto r = bus->reply(s);
          r.pack("IDLE");
          bus->async_send(
              r, [&](boost::system::error_code ec, dbus::message s) {});
          io.stop();
        }
      };
  f.async_dispatch(method_handler);

  dbus::endpoint test_endpoint(requested_name, "/xyz/openbmc_project/fwupdate1",
                               "org.freedesktop.DBus.Properties");

  auto method_name = std::string("Get");
  auto m = dbus::message::new_call(test_endpoint, method_name);

  m.pack("xyz.openbmc_project.fwupdate1", "State");
  bus->async_send(m, [&](boost::system::error_code ec, dbus::message s) {
    std::cerr << "received s: " << s << std::endl;
  });

  // system_bus->send(m, std::chrono::seconds(0));

  io.run();
}