content/browser/web_package/web_bundle_source.cc

// Copyright 2019 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 "content/browser/web_package/web_bundle_source.h"

#include "base/files/file.h"
#include "base/memory/ptr_util.h"
#include "base/strings/string_util.h"
#include "build/build_config.h"
#include "net/base/filename_util.h"
#include "net/base/url_util.h"
#include "url/gurl.h"

#if BUILDFLAG(IS_ANDROID)
#include "base/android/content_uri_utils.h"
#endif

namespace content {

// static
std::unique_ptr<WebBundleSource> WebBundleSource::MaybeCreateFromTrustedFileUrl(
    const GURL& url) {
#if BUILDFLAG(IS_ANDROID)
  if (url.SchemeIs(url::kContentScheme)) {
    const base::FilePath file_path = base::FilePath(url.spec());
    return base::WrapUnique(
        new WebBundleSource(Type::kTrustedFile, file_path, url));
  }
#endif
  DCHECK(url.SchemeIsFile());
  base::FilePath file_path;
  if (!net::FileURLToFilePath(url, &file_path))
    return nullptr;
  return base::WrapUnique(
      new WebBundleSource(Type::kTrustedFile, file_path, url));
}

// static
std::unique_ptr<WebBundleSource> WebBundleSource::MaybeCreateFromFileUrl(
    const GURL& url) {
  base::FilePath file_path;
  if (url.SchemeIsFile()) {
    if (net::FileURLToFilePath(url, &file_path)) {
      return base::WrapUnique(new WebBundleSource(Type::kFile, file_path, url));
    }
  }
#if BUILDFLAG(IS_ANDROID)
  if (url.SchemeIs(url::kContentScheme)) {
    return base::WrapUnique(
        new WebBundleSource(Type::kFile, base::FilePath(url.spec()), url));
  }
#endif
  return nullptr;
}

// static
std::unique_ptr<WebBundleSource> WebBundleSource::MaybeCreateFromNetworkUrl(
    const GURL& url) {
  if (url.SchemeIs(url::kHttpsScheme) ||
      (url.SchemeIs(url::kHttpScheme) && net::IsLocalhost(url))) {
    return base::WrapUnique(
        new WebBundleSource(Type::kNetwork, base::FilePath(), url));
  }
  return nullptr;
}

std::unique_ptr<WebBundleSource> WebBundleSource::Clone() const {
  return base::WrapUnique(new WebBundleSource(type_, file_path_, url_));
}

std::unique_ptr<base::File> WebBundleSource::OpenFile() const {
  DCHECK(!file_path_.empty());
#if BUILDFLAG(IS_ANDROID)
  if (file_path_.IsContentUri()) {
    return std::make_unique<base::File>(
        base::OpenContentUriForRead(file_path_));
  }
#endif
  return std::make_unique<base::File>(
      file_path_, base::File::FLAG_OPEN | base::File::FLAG_READ);
}

bool WebBundleSource::IsPathRestrictionSatisfied(const GURL& url) const {
  DCHECK(is_network());
  return base::StartsWith(url.spec(), url_.GetWithoutFilename().spec(),
                          base::CompareCase::SENSITIVE);
}

WebBundleSource::WebBundleSource(Type type,
                                 const base::FilePath& file_path,
                                 const GURL& url)
    : type_(type), file_path_(file_path), url_(url) {}

}  // namespace content