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

#ifndef MIDIS_CLIENT_TRACKER_H_
#define MIDIS_CLIENT_TRACKER_H_

#include <map>
#include <memory>

#include <base/files/file_path.h>
#include <base/files/scoped_file.h>
#include <base/memory/weak_ptr.h>
#include <gtest/gtest_prod.h>
#include <mojo/edk/embedder/process_delegate.h>

#include "midis/client.h"
#include "midis/device_tracker.h"
#include "mojo/midis.mojom.h"

namespace midis {

class ClientTracker : public mojo::edk::ProcessDelegate {
 public:
  ClientTracker();
  ~ClientTracker();
  void InitClientTracker();
  void SetDeviceTracker(DeviceTracker* ptr) { device_tracker_ = ptr; }
  size_t GetNumClientsForTesting() const { return clients_.size(); }
  void RemoveClient(uint32_t client_id);

  // Sets up the MidisHost Mojo interface using the FD passed in
  // via D-Bus. The net result of this function should be the creation
  // of a MidisHostImpl object which ClientTracker manages.
  void AcceptProxyConnection(base::ScopedFD fd);

  // mojo::edk::ProcessDelegate:
  void OnShutdownComplete() override;

  // Helper function to check whether a |midis_host_| object is
  // already associated with ClientTracker.
  bool IsProxyConnected();

  // Create a new Mojo Client and place it in the list of clients.
  void MakeMojoClient(arc::mojom::MidisServerRequest request,
                      arc::mojom::MidisClientPtr client_ptr);

 private:
  friend class ClientTrackerTest;
  FRIEND_TEST(ClientTrackerTest, AddClientPositive);
  // Helper function to set the base directory to be used for looking for the
  // Unix Domain socket path. Helpful for testing, where the we won't be allowed
  // to create directories in locations other than tempfs.
  void SetBaseDirForTesting(const base::FilePath& dir) { basedir_ = dir; }
  std::map<uint32_t, std::unique_ptr<Client>> clients_;
  base::ScopedFD server_fd_;
  int client_id_counter_;
  // ClientTracker and DeviceTracker both exist for the lifetime of the service.
  // As such, it is safe to maintain this pointer as a means to make updates and
  // derive information regarding devices.
  DeviceTracker* device_tracker_;
  base::FilePath basedir_;
  base::SequenceChecker sequence_checker_;
  std::unique_ptr<arc::mojom::MidisHost> midis_host_;

  base::WeakPtrFactory<ClientTracker> weak_factory_;
  DISALLOW_COPY_AND_ASSIGN(ClientTracker);
};

}  // namespace midis

#endif  // MIDIS_CLIENT_TRACKER_H_