// 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 "mash/init/init.h"

#include "base/bind.h"
#include "base/command_line.h"
#include "base/guid.h"
#include "mash/login/public/interfaces/login.mojom.h"
#include "mojo/shell/public/cpp/connection.h"
#include "mojo/shell/public/cpp/connector.h"

namespace mash {
namespace init {

Init::Init()
    : connector_(nullptr) {}
Init::~Init() {}

void Init::Initialize(mojo::Connector* connector,
                      const mojo::Identity& identity,
                      uint32_t id) {
  connector_ = connector;
  connector_->Connect("mojo:mus");
  StartTracing();
  StartResourceProvider();
  StartLogin();
}

bool Init::AcceptConnection(mojo::Connection* connection) {
  connection->AddInterface<mojom::Init>(this);
  return true;
}

void Init::StartService(const mojo::String& name,
                        const mojo::String& user_id) {
  if (user_services_.find(user_id) == user_services_.end()) {
    mojo::Connector::ConnectParams params(mojo::Identity(name, user_id));
    scoped_ptr<mojo::Connection> connection = connector_->Connect(&params);
    connection->SetConnectionLostClosure(
        base::Bind(&Init::UserServiceQuit, base::Unretained(this), user_id));
    user_services_[user_id] = std::move(connection);
  }
}

void Init::StopServicesForUser(const mojo::String& user_id) {
  auto it = user_services_.find(user_id);
  if (it != user_services_.end())
    user_services_.erase(it);
}

void Init::Create(mojo::Connection* connection, mojom::InitRequest request) {
  init_bindings_.AddBinding(this, std::move(request));
}

void Init::UserServiceQuit(const std::string& user_id) {
  auto it = user_services_.find(user_id);
  DCHECK(it != user_services_.end());
  user_services_.erase(it);
}

void Init::StartTracing() {
  connector_->Connect("mojo:tracing");
}

void Init::StartResourceProvider() {
  connector_->Connect("mojo:resource_provider");
}

void Init::StartLogin() {
  login_connection_ = connector_->Connect("mojo:login");
  login_connection_->AddInterface<mojom::Init>(this);
  mash::login::mojom::LoginPtr login;
  login_connection_->GetInterface(&login);
  login->ShowLoginUI();
}

}  // namespace init
}  // namespace main