// Copyright (C) 2001 - 2012 Rob Caelers & Raymond Penners
// All rights reserved.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// 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/>.
//

#ifndef CORE_HH
#define CORE_HH

#include <cstdio>
#include <sys/stat.h>
#include <sys/types.h>
#include <cstddef>
#include <cstdlib>

#if defined(PLATFORM_OS_MACOS)
#  include "MacOSHelpers.hh"
#endif

#include <iostream>
#include <memory>
#include <string>
#include <map>

#include "Break.hh"
#include "IActivityMonitor.hh"
#include "core/ICore.hh"
#include "core/ICoreEventListener.hh"
#include "config/IConfiguratorListener.hh"
#include "Timer.hh"
#include "stats/IStatistics.hh"
#include "utils/Diagnostics.hh"
#include "utils/Signals.hh"
#include "CoreHooks.hh"
#include "LocalActivityMonitor.hh"

using namespace workrave;

class FakeActivityMonitor;
class IdleLogManager;
class BreakControl;
#if defined(HAVE_CORE_DBUS)
class LegacyRpcDBusServer;
#endif
#if defined(HAVE_GRPC) && !defined(HAVE_CORE_SHADOW)
class RpcCoreServer;
#endif

// @rpc(service="workrave.CoreService")
// @rpc.dbus(interface="org.workrave.CoreInterface")
class Core
  : public ICore
  , public workrave::config::IConfiguratorListener
{
public:
  Core();
  ~Core() override;

  static void set_configurator(workrave::config::IConfigurator::Ptr configurator);
  static Core *get_instance();
#if defined(HAVE_TESTS)
  static void reset_instance();
#endif

  // ICore
  // @rpc.signal(name="OperationModeChanged")
  boost::signals2::signal<void(workrave::OperationMode)> &signal_operation_mode_changed() override;
  // @rpc.signal(name="UsageModeChanged")
  boost::signals2::signal<void(workrave::UsageMode)> &signal_usage_mode_changed() override;

  Timer *get_timer(std::string name) const;
  Timer *get_timer(BreakId id) const;
  Break *get_break(BreakId id) override;
  Break *get_break(std::string name) override;
  IActivityMonitor::Ptr get_activity_monitor() const;
  // @rpc(name="IsActive")
  bool is_user_active() const override;
  // @rpc(name="IsTaking")
  bool is_taking() const override;
  std::string get_break_stage(BreakId id);

  workrave::stats::IStatistics *get_statistics() const override;
  void set_core_events_listener(ICoreEventListener *l) override;
  // @rpc(name="ForceBreak")
  void force_break(BreakId id, workrave::utils::Flags<BreakHint> break_hint) override;
  void time_changed() override;
  void set_powersave(bool down) override;

  int64_t get_time() const override;
  void post_event(CoreEvent event) override;

  // @rpc(name="GetActiveOperationMode")
  OperationMode get_active_operation_mode() override;
  // @rpc(name="GetOperationMode")
  OperationMode get_regular_operation_mode() override;
  // @rpc(name="IsOperationModeAnOverride")
  bool is_operation_mode_an_override() override;
  // @rpc(name="SetOperationMode")
  void set_operation_mode(OperationMode mode) override;
  // @rpc(name="SetOperationModeFor")
  void set_operation_mode_for(OperationMode mode, std::chrono::minutes duration) override;
  void set_operation_mode_override(OperationMode mode, const std::string &id) override;
  void remove_operation_mode_override(const std::string &id) override;

  // @rpc(name="GetUsageMode")
  UsageMode get_usage_mode() override;
  // @rpc(name="SetUsageMode")
  void set_usage_mode(UsageMode mode) override;

  void set_freeze_all_breaks(bool freeze);
  void set_insensitive_mode_all_breaks(InsensitiveMode mode);

  void stop_prelude(BreakId break_id);
  void do_force_break(BreakId id, workrave::utils::Flags<BreakHint> break_hint);

  void freeze();
  void defrost();

  void force_idle() override;
  void force_idle(BreakId break_id);

  ActivityState get_current_monitor_state() const;
  bool is_master() const;

  // @rpc(name="ReportActivity")
  void report_external_activity(std::string who, bool act) override;
  void is_timer_running(BreakId id, bool &value);
  void get_timer_elapsed(BreakId id, int *value);
  void get_timer_remaining(BreakId id, int *value);
  void get_timer_idle(BreakId id, int *value);
  void get_timer_overdue(BreakId id, int *value);

  void postpone_break(BreakId break_id);
  void skip_break(BreakId break_id);

  ICoreHooks::Ptr get_hooks() const override;

private:
#if !defined(NDEBUG)
  enum ScriptCommand
  {
    SCRIPT_START = 1,
  };
#endif

  void init(int argc, char **argv, IApp *application, const char *display_name) override;
  void init_breaks();
  void init_monitor(const char *display_name);
  void init_distribution_manager();
  void init_statistics();
#if defined(HAVE_CORE_DBUS)
  void init_rpc_dbus();
#endif
#if defined(HAVE_GRPC) && !defined(HAVE_CORE_SHADOW)
  void init_rpc();
  void update_rpc();
#endif

  void load_monitor_config();
  void config_changed_notify(const std::string &key) override;
  void heartbeat() override;
  void timer_action(BreakId id, TimerInfo info);
  void process_distribution();
  void process_state();
  bool process_timewarp();
  void process_timers();
  void start_break(BreakId break_id, BreakId resume_this_break = BREAK_ID_NONE);
  void stop_all_breaks();
  void daily_reset();
  void save_state() const;
  void load_state();
  void load_misc();
  void do_postpone_break(BreakId break_id);
  void do_skip_break(BreakId break_id);
  void do_stop_prelude(BreakId break_id);

  void set_insist_policy(InsistPolicy p) override;
  InsistPolicy get_insist_policy() const;

  void set_operation_mode_internal(OperationMode mode);
  void check_operation_mode_auto_reset();
  void update_active_operation_mode();
  void set_usage_mode_internal(UsageMode mode, bool persistent);

private:
  //! The one and only instance
  static Core *instance;

  //! Number of command line arguments passed to the program.
  int argc{};

  //! Command line arguments passed to the program.
  char **argv{};

  //! The time we last processed the timers.
  int64_t last_process_time{0};

  //! Are we the master node??
  TracedField<bool> master_node{"core.master_node", true};

  //! List of breaks.
  Break breaks[workrave::BREAK_ID_SIZEOF];

  //! The Configurator.
  static workrave::config::IConfigurator::Ptr configurator;

  //! The activity monitor
  IActivityMonitor::Ptr monitor;

  //! The activity monitor
  LocalActivityMonitor::Ptr local_monitor;

  //! GUI Widget factory.
  IApp *application{nullptr};

  //! The statistics collector.
  workrave::stats::IStatistics::Ptr statistics;

  //! Current operation mode.
  TracedField<OperationMode> operation_mode_active{"core.operation_mode_active", OperationMode::Normal};

  //! The same as operation_mode unless operation_mode is an override mode.
  TracedField<OperationMode> operation_mode_regular{"core.operation_mode_regular", OperationMode::Normal};

  //! Active operation mode overrides.
  std::map<std::string, OperationMode> operation_mode_overrides;

  //! Current usage mode.
  TracedField<UsageMode> usage_mode{"core.usage_mode", UsageMode::Normal};

  //! Where to send core events to?
  ICoreEventListener *core_event_listener{nullptr};

  //! Did the OS announce a powersave?
  TracedField<bool> powersave{"core.powersave", false};

  //! Time the OS announces a resume from powersave
  int64_t powersave_resume_time{0};

  //! What to do with activity during insisted break?
  TracedField<InsistPolicy> insist_policy{"core.insist_policy", InsistPolicy::Halt};

  //! Policy currently in effect.
  TracedField<InsistPolicy> active_insist_policy{"core.active_insist_policy", InsistPolicy::Invalid};

  //! Resumes this break if current break ends.
  TracedField<BreakId> resume_break{"core.resume_break", BREAK_ID_NONE};

  //! Current local monitor state.
  TracedField<ActivityState> local_state{"core.local_state", ACTIVITY_IDLE};

  //! Current overall monitor state.
  TracedField<ActivityState> monitor_state{"core.monitor_state", ACTIVITY_UNKNOWN};

  //! Hooks to alter the backend behaviour.
  CoreHooks::Ptr hooks;

  //! External activity
  std::map<std::string, int64_t> external_activity;

  //! Operation mode changed notification.
  boost::signals2::signal<void(workrave::OperationMode)> operation_mode_changed_signal;

  //! Usage mode changed notification.
  boost::signals2::signal<void(workrave::UsageMode)> usage_mode_changed_signal;

#if defined(HAVE_TESTS)
  friend class Test;
#endif

#if defined(HAVE_CORE_DBUS)
  // Declared last so registrations stop before the Core and Break instances
  // they reference are destroyed.
  std::unique_ptr<LegacyRpcDBusServer> rpc_dbus_server;
#endif
#if defined(HAVE_GRPC) && !defined(HAVE_CORE_SHADOW)
  std::string rpc_listen_address;
  std::unique_ptr<RpcCoreServer> rpc_server;
  workrave::utils::Trackable rpc_settings_tracker;
#endif
};

//! Returns the singleton Core instance.
inline Core *
Core::get_instance()
{
  if (instance == nullptr)
    {
      instance = new Core();
    }

  return instance;
}

#if defined(HAVE_TESTS)
inline void
Core::reset_instance()
{
  if (instance != nullptr)
    {
      delete instance;
      instance = nullptr;
    }
}
#endif

//!
inline ActivityState
Core::get_current_monitor_state() const
{
  return monitor_state;
}

//!
inline bool
Core::is_master() const
{
  return master_node;
}

//!
inline bool
Core::is_taking() const
{
  bool taking = false;
  for (int i = 0; i < BREAK_ID_SIZEOF; ++i)
    {
      if (breaks[i].is_taking())
        {
          taking = true;
        }
    }

  return taking;
}

#endif // CORE_HH
