user_shell_client.cc

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

#include "services/user/user_shell_client.h"

#include "base/bind.h"
#include "base/memory/ptr_util.h"
#include "base/memory/weak_ptr.h"
#include "components/filesystem/lock_table.h"
#include "components/leveldb/leveldb_service_impl.h"
#include "services/shell/public/cpp/connection.h"
#include "services/user/user_id_map.h"
#include "services/user/user_service.h"

namespace user_service {

class UserShellClient::UserServiceObjects
    : public base::SupportsWeakPtr<UserServiceObjects> {
 public:
  // Created on the main thread.
  UserServiceObjects(base::FilePath user_dir) : user_dir_(user_dir) {}

  // Destroyed on the |user_service_runner_|.
  ~UserServiceObjects() {}

  // Called on the |user_service_runner_|.
  void OnUserServiceRequest(shell::Connection* connection,
                            mojom::UserServiceRequest request) {
    if (!lock_table_)
      lock_table_ = new filesystem::LockTable;
    user_service_bindings_.AddBinding(new UserService(user_dir_, lock_table_),
                                      std::move(request));
  }

 private:
  mojo::BindingSet<mojom::UserService> user_service_bindings_;
  scoped_refptr<filesystem::LockTable> lock_table_;
  base::FilePath user_dir_;

  DISALLOW_COPY_AND_ASSIGN(UserServiceObjects);
};

class UserShellClient::LevelDBServiceObjects
    : public base::SupportsWeakPtr<LevelDBServiceObjects> {
 public:
  // Created on the main thread.
  LevelDBServiceObjects(scoped_refptr<base::SingleThreadTaskRunner> task_runner)
      : task_runner_(std::move(task_runner)) {}

  // Destroyed on the |leveldb_service_runner_|.
  ~LevelDBServiceObjects() {}

  // Called on the |leveldb_service_runner_|.
  void OnLevelDBServiceRequest(shell::Connection* connection,
                               leveldb::mojom::LevelDBServiceRequest request) {
    if (!leveldb_service_)
      leveldb_service_.reset(new leveldb::LevelDBServiceImpl(task_runner_));
    leveldb_bindings_.AddBinding(leveldb_service_.get(), std::move(request));
  }

 private:
  scoped_refptr<base::SingleThreadTaskRunner> task_runner_;

  // Variables that are only accessible on the |leveldb_service_runner_| thread.
  std::unique_ptr<leveldb::mojom::LevelDBService> leveldb_service_;
  mojo::BindingSet<leveldb::mojom::LevelDBService> leveldb_bindings_;

  DISALLOW_COPY_AND_ASSIGN(LevelDBServiceObjects);
};

std::unique_ptr<shell::ShellClient> CreateUserShellClient(
    scoped_refptr<base::SingleThreadTaskRunner> user_service_runner,
    scoped_refptr<base::SingleThreadTaskRunner> leveldb_service_runner,
    const base::Closure& quit_closure) {
  return base::WrapUnique(new UserShellClient(
      std::move(user_service_runner), std::move(leveldb_service_runner)));
}

UserShellClient::UserShellClient(
    scoped_refptr<base::SingleThreadTaskRunner> user_service_runner,
    scoped_refptr<base::SingleThreadTaskRunner> leveldb_service_runner)
    : user_service_runner_(std::move(user_service_runner)),
      leveldb_service_runner_(std::move(leveldb_service_runner)) {}

UserShellClient::~UserShellClient() {
  user_service_runner_->DeleteSoon(FROM_HERE, user_objects_.release());
  leveldb_service_runner_->DeleteSoon(FROM_HERE, leveldb_objects_.release());
}

void UserShellClient::Initialize(shell::Connector* connector,
                                 const shell::Identity& identity,
                                 uint32_t id) {
  user_objects_.reset(new UserShellClient::UserServiceObjects(
      GetUserDirForUserId(identity.user_id())));
  leveldb_objects_.reset(
      new UserShellClient::LevelDBServiceObjects(leveldb_service_runner_));
}

bool UserShellClient::AcceptConnection(shell::Connection* connection) {
  connection->AddInterface<leveldb::mojom::LevelDBService>(this);
  connection->AddInterface<mojom::UserService>(this);
  return true;
}

void UserShellClient::Create(shell::Connection* connection,
                             mojom::UserServiceRequest request) {
  user_service_runner_->PostTask(
      FROM_HERE,
      base::Bind(&UserShellClient::UserServiceObjects::OnUserServiceRequest,
                 user_objects_->AsWeakPtr(), connection,
                 base::Passed(&request)));
}

void UserShellClient::Create(shell::Connection* connection,
                             leveldb::mojom::LevelDBServiceRequest request) {
  leveldb_service_runner_->PostTask(
      FROM_HERE,
      base::Bind(
          &UserShellClient::LevelDBServiceObjects::OnLevelDBServiceRequest,
          leveldb_objects_->AsWeakPtr(), connection,
          base::Passed(&request)));
}

}  // namespace user_service