handle reload api call from pdf ui

diff --git a/atom/browser/ui/webui/pdf_viewer_handler.cc b/atom/browser/ui/webui/pdf_viewer_handler.cc
index 6c86173..7b8f00b 100644 (file)
--- 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 a061cb8..33e2157 100644 (file)
--- 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.