create webui message handler

atom/browser/ui/webui/pdf_viewer_handler.h

@@ -0,0 +1,41 @@
+// Copyright (c) 2017 GitHub, Inc.
+// Use of this source code is governed by the MIT license that can be
+// found in the LICENSE file.
+
+#ifndef ATOM_BROWSER_UI_WEBUI_PDF_VIEWER_HANDLER_H_
+#define ATOM_BROWSER_UI_WEBUI_PDF_VIEWER_HANDLER_H_
+
+#include <string>
+
+#include "base/compiler_specific.h"
+#include "base/macros.h"
+#include "content/public/browser/web_ui_message_handler.h"
+
+namespace base {
+class ListValue;
+}
+
+namespace atom {
+
+class PdfViewerHandler : public content::WebUIMessageHandler {
+ public:
+  PdfViewerHandler(const std::string& stream_url,
+                   const std::string& original_url);
+  ~PdfViewerHandler() override;
+
+  // WebUIMessageHandler implementation.
+  void RegisterMessages() override;
+
+ private:
+  void Initialize(const base::ListValue* args);
+  void GetTabId(const base::ListValue* args);
+
+  std::string stream_url_;
+  std::string original_url_;
+
+  DISALLOW_COPY_AND_ASSIGN(PdfViewerHandler);
+};
+
+}  // namespace atom
+
+#endif  // ATOM_BROWSER_UI_WEBUI_PDF_VIEWER_HANDLER_H_