95 lines
3.1 KiB
C++
95 lines
3.1 KiB
C++
|
// Copyright (c) 2017 GitHub, Inc.
|
||
|
|
// Use of this source code is governed by the MIT license that can be
|
||
|
|
// found in the LICENSE file.
|
||
|
|
|
||
|
|
#include "atom/browser/ui/webui/pdf_viewer_ui.h"
|
||
|
|
|
||
|
|
#include <map>
|
||
|
|
|
||
|
|
#include "atom/browser/ui/webui/pdf_viewer_handler.h"
|
||
|
|
#include "content/public/browser/render_view_host.h"
|
||
|
|
#include "content/public/browser/url_data_source.h"
|
||
|
|
#include "content/public/common/bindings_policy.h"
|
||
|
|
#include "grit/pdf_viewer_resources_map.h"
|
||
|
|
#include "net/base/mime_util.h"
|
||
|
|
#include "ui/base/resource/resource_bundle.h"
|
||
|
|
|
||
|
|
namespace atom {
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
std::string PathWithoutParams(const std::string& path) {
|
||
|
|
return GURL(std::string("chrome://pdf-viewer/") + path).path().substr(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
class BundledDataSource : public content::URLDataSource {
|
||
|
|
public:
|
||
|
|
BundledDataSource() {
|
||
|
|
for (size_t i = 0; i < kPdfViewerResourcesSize; ++i) {
|
||
|
|
base::FilePath resource_path =
|
||
|
|
base::FilePath().AppendASCII(kPdfViewerResources[i].name);
|
||
|
|
resource_path = resource_path.NormalizePathSeparators();
|
||
|
|
|
||
|
|
DCHECK(path_to_resource_id_.find(resource_path) ==
|
||
|
|
path_to_resource_id_.end());
|
||
|
|
path_to_resource_id_[resource_path] = kPdfViewerResources[i].value;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// content::URLDataSource implementation.
|
||
|
|
std::string GetSource() const override { return PdfViewerUI::kHost; }
|
||
|
|
|
||
|
|
void StartDataRequest(const std::string& path,
|
||
|
|
int render_process_id,
|
||
|
|
int render_frame_id,
|
||
|
|
const GotDataCallback& callback) override {
|
||
|
|
std::string filename = PathWithoutParams(path);
|
||
|
|
std::map<base::FilePath, int>::const_iterator entry =
|
||
|
|
path_to_resource_id_.find(base::FilePath(filename));
|
||
|
|
if (entry != path_to_resource_id_.end()) {
|
||
|
|
int resource_id = entry->second;
|
||
|
|
const ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
|
||
|
|
callback.Run(rb.LoadDataResourceBytes(resource_id));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string GetMimeType(const std::string& path) const override {
|
||
|
|
std::string filename = PathWithoutParams(path);
|
||
|
|
std::string mime_type;
|
||
|
|
net::GetMimeTypeFromFile(base::FilePath(filename), &mime_type);
|
||
|
|
return mime_type;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool ShouldAddContentSecurityPolicy() const override { return false; }
|
||
|
|
|
||
|
|
bool ShouldDenyXFrameOptions() const override { return false; }
|
||
|
|
|
||
|
|
bool ShouldServeMimeTypeAsContentTypeHeader() const override { return true; }
|
||
|
|
|
||
|
|
private:
|
||
|
|
~BundledDataSource() override {}
|
||
|
|
|
||
|
|
// A map from a resource path to the resource ID.
|
||
|
|
std::map<base::FilePath, int> path_to_resource_id_;
|
||
|
|
|
||
|
|
DISALLOW_COPY_AND_ASSIGN(BundledDataSource);
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
const char PdfViewerUI::kHost[] = "pdf-viewer";
|
||
|
|
|
||
|
|
PdfViewerUI::PdfViewerUI(content::BrowserContext* browser_context,
|
||
|
|
content::WebUI* web_ui,
|
||
|
|
const std::string& view_id)
|
||
|
|
: content::WebUIController(web_ui) {
|
||
|
|
web_ui->AddMessageHandler(new PdfViewerHandler(view_id));
|
||
|
|
content::URLDataSource::Add(browser_context, new BundledDataSource);
|
||
|
|
}
|
||
|
|
|
||
|
|
void PdfViewerUI::RenderViewCreated(content::RenderViewHost* rvh) {
|
||
|
|
rvh->AllowBindings(content::BINDINGS_POLICY_WEB_UI);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace atom
|