remoting/host/it2me_desktop_environment.h

// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef REMOTING_HOST_IT2ME_DESKTOP_ENVIRONMENT_H_
#define REMOTING_HOST_IT2ME_DESKTOP_ENVIRONMENT_H_

#include <memory>
#include <string>

#include "base/memory/scoped_refptr.h"
#include "base/memory/weak_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "remoting/host/base/desktop_environment_options.h"
#include "remoting/host/basic_desktop_environment.h"
#include "remoting/host/client_session_control.h"
#include "remoting/host/desktop_environment.h"
#include "remoting/host/desktop_interaction_strategy.h"

namespace remoting {

class HostWindow;
class LocalInputMonitor;

// Same as BasicDesktopEnvironment but also presents the Continue window to
// the local user.
class It2MeDesktopEnvironment : public BasicDesktopEnvironment {
 public:
  It2MeDesktopEnvironment(const It2MeDesktopEnvironment&) = delete;
  It2MeDesktopEnvironment& operator=(const It2MeDesktopEnvironment&) = delete;

  ~It2MeDesktopEnvironment() override;

  // Initializes the curtain mode if needed.
  // Returns `false` if the curtain mode failed to start for any reason.
  void InitializeCurtainMode(
      base::WeakPtr<ClientSessionControl> client_session_control);

  // BasicDesktopEnvironment implementation:
  std::string GetCapabilities() const override;

 protected:
  friend class It2MeDesktopEnvironmentFactory;
  It2MeDesktopEnvironment(
      scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
      scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
      std::unique_ptr<DesktopInteractionStrategy> interaction_strategy,
      base::WeakPtr<ClientSessionControl> client_session_control,
      const DesktopEnvironmentOptions& options);

 private:
  void InitializeCurtainModeIfNoUserLoggedIn(
      base::WeakPtr<ClientSessionControl> client_session_control,
      bool is_user_logged_in);

  // Presents the continue window to the local user.
  std::unique_ptr<HostWindow> continue_window_;

  // Presents the disconnect window to the local user.
  std::unique_ptr<HostWindow> disconnect_window_;

  // Notifies the client session about the local mouse movements.
  std::unique_ptr<LocalInputMonitor> local_input_monitor_;

  base::WeakPtrFactory<It2MeDesktopEnvironment> weak_ptr_factory_{this};
};

// Used to create |It2MeDesktopEnvironment| instances.
class It2MeDesktopEnvironmentFactory : public BasicDesktopEnvironmentFactory {
 public:
  It2MeDesktopEnvironmentFactory(
      scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner,
      scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner,
      std::unique_ptr<DesktopInteractionStrategyFactory>
          interaction_strategy_factory);

  It2MeDesktopEnvironmentFactory(const It2MeDesktopEnvironmentFactory&) =
      delete;
  It2MeDesktopEnvironmentFactory& operator=(
      const It2MeDesktopEnvironmentFactory&) = delete;

  ~It2MeDesktopEnvironmentFactory() override;

  // DesktopEnvironmentFactory interface.
  void Create(base::WeakPtr<ClientSessionControl> client_session_control,
              base::WeakPtr<ClientSessionEvents> client_session_events,
              const DesktopEnvironmentOptions& options,
              CreateCallback callback) override;
};

}  // namespace remoting

#endif  // REMOTING_HOST_IT2ME_DESKTOP_ENVIRONMENT_H_