// Copyright 2015 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.

#ifndef MOJO_PACKAGE_MANAGER_PACKAGE_MANAGER_IMPL_H_
#define MOJO_PACKAGE_MANAGER_PACKAGE_MANAGER_IMPL_H_

#include "base/files/file_path.h"
#include "mojo/fetcher/url_resolver.h"
#include "mojo/services/network/public/interfaces/network_service.mojom.h"
#include "mojo/services/network/public/interfaces/url_loader_factory.mojom.h"
#include "mojo/shell/package_manager.h"

namespace mojo {
namespace shell {
class Fetcher;
}
namespace package_manager {

// This is the default implementation of shell::PackageManager. It loads
// http/s urls off the network as well as providing special handling for mojo:
// and about: urls.
class PackageManagerImpl : public shell::PackageManager {
 public:
  // mojo: urls are only supported if |shell_file_root| is non-empty.
  explicit PackageManagerImpl(const base::FilePath& shell_file_root);
  ~PackageManagerImpl() override;

  // Register a content handler to handle content of |mime_type|.
  void RegisterContentHandler(const std::string& mime_type,
                              const GURL& content_handler_url);

  // Registers a package alias. When attempting to load |alias|, it will
  // instead redirect to |content_handler_package|, which is a content handler
  // which will be passed the |alias| as the URLResponse::url. Different values
  // of |alias| with the same |qualifier| that are in the same
  // |content_handler_package| will run in the same process in multi-process
  // mode.
  void RegisterApplicationPackageAlias(
      const GURL& alias,
      const GURL& content_handler_package,
      const std::string& qualifier);

 private:
  using ApplicationPackagedAlias = std::map<GURL, std::pair<GURL, std::string>>;
  using MimeTypeToURLMap = std::map<std::string, GURL>;

  // Overridden from shell::PackageManager:
  void SetApplicationManager(shell::ApplicationManager* manager) override;
  GURL ResolveURL(const GURL& url) override;
  void FetchRequest(
      URLRequestPtr request,
      const shell::Fetcher::FetchCallback& loader_callback) override;
  bool HandleWithContentHandler(shell::Fetcher* fetcher,
                                const GURL& unresolved_url,
                                base::TaskRunner* task_runner,
                                URLResponsePtr* new_response,
                                GURL* content_handler_url,
                                std::string* qualifier) override;


  shell::ApplicationManager* application_manager_;
  scoped_ptr<fetcher::URLResolver> url_resolver_;
  const bool disable_cache_;
  NetworkServicePtr network_service_;
  URLLoaderFactoryPtr url_loader_factory_;
  ApplicationPackagedAlias application_package_alias_;
  MimeTypeToURLMap mime_type_to_url_;

  DISALLOW_COPY_AND_ASSIGN(PackageManagerImpl);
};

}  // namespace package_manager
}  // namespace mojo

#endif  // MOJO_PACKAGE_MANAGER_PACKAGE_MANAGER_IMPL_H_