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.