From 528853efbf6243f994df8a13678f6bf1be2e1fa8 Mon Sep 17 00:00:00 2001
From: deepak1556 <hop2deep@gmail.com>
Date: Tue, 14 Feb 2017 06:55:52 +0530
Subject: [PATCH] handle reload api call from pdf ui

---
 atom/browser/ui/webui/pdf_viewer_handler.cc | 7 +++++++
 atom/browser/ui/webui/pdf_viewer_handler.h  | 1 +
 2 files changed, 8 insertions(+)

diff --git a/atom/browser/ui/webui/pdf_viewer_handler.cc b/atom/browser/ui/webui/pdf_viewer_handler.cc
index 6c86173043ac..7b8f00b95795 100644
--- a/atom/browser/ui/webui/pdf_viewer_handler.cc
+++ b/atom/browser/ui/webui/pdf_viewer_handler.cc
@@ -62,6 +62,8 @@ void PdfViewerHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback(
       "getStrings",
       base::Bind(&PdfViewerHandler::GetStrings, base::Unretained(this)));
+  web_ui()->RegisterMessageCallback(
+      "reload", base::Bind(&PdfViewerHandler::Reload, base::Unretained(this)));
 }
 
 void PdfViewerHandler::OnJavascriptAllowed() {
@@ -158,6 +160,11 @@ void PdfViewerHandler::GetStrings(const base::ListValue* args) {
   ResolveJavascriptCallback(*callback_id, *result);
 }
 
+void PdfViewerHandler::Reload(const base::ListValue* args) {
+  CHECK_EQ(0U, args->GetSize());
+  web_ui()->GetWebContents()->ReloadFocusedFrame(false);
+}
+
 void PdfViewerHandler::OnZoomLevelChanged(
     const content::HostZoomMap::ZoomLevelChange& change) {
   // TODO(deepak1556): This will work only if zoom level is changed through host
diff --git a/atom/browser/ui/webui/pdf_viewer_handler.h b/atom/browser/ui/webui/pdf_viewer_handler.h
index a061cb8b7848..33e2157d9785 100644
--- a/atom/browser/ui/webui/pdf_viewer_handler.h
+++ b/atom/browser/ui/webui/pdf_viewer_handler.h
@@ -37,6 +37,7 @@ class PdfViewerHandler : public content::WebUIMessageHandler {
   void GetDefaultZoom(const base::ListValue* args);
   void GetInitialZoom(const base::ListValue* args);
   void GetStrings(const base::ListValue* args);
+  void Reload(const base::ListValue* args);
   void OnZoomLevelChanged(const content::HostZoomMap::ZoomLevelChange& change);
 
   // Keeps track of events related to zooming.