code cleanup
This commit is contained in:
parent
b1b6573bcd
commit
b28e686228
3 changed files with 113 additions and 95 deletions
|
@ -7,7 +7,6 @@
|
|||
|
||||
#include <string>
|
||||
|
||||
#include "atom/browser/loader/layered_resource_handler.h"
|
||||
#include "base/macros.h"
|
||||
#include "content/public/browser/web_contents_observer.h"
|
||||
#include "content/public/browser/web_ui_controller.h"
|
||||
|
@ -15,21 +14,15 @@
|
|||
|
||||
namespace content {
|
||||
class BrowserContext;
|
||||
class ResourceContext;
|
||||
struct StreamInfo;
|
||||
}
|
||||
|
||||
namespace net {
|
||||
class HttpResponseHeaders;
|
||||
}
|
||||
|
||||
namespace atom {
|
||||
|
||||
class PdfViewerHandler;
|
||||
|
||||
class PdfViewerUI : public content::WebUIController,
|
||||
public content::WebContentsObserver,
|
||||
public LayeredResourceHandler::Delegate {
|
||||
public content::WebContentsObserver {
|
||||
public:
|
||||
PdfViewerUI(content::BrowserContext* browser_context,
|
||||
content::WebUI* web_ui,
|
||||
|
@ -41,14 +34,12 @@ class PdfViewerUI : public content::WebUIController,
|
|||
content::RenderFrameHost* render_frame_host) override;
|
||||
void RenderFrameCreated(content::RenderFrameHost* rfh) override;
|
||||
|
||||
// LayeredResourceHandler:
|
||||
void OnResponseStarted(content::ResourceResponse* response) override;
|
||||
|
||||
private:
|
||||
using StreamResponseCallback =
|
||||
base::OnceCallback<void(std::unique_ptr<content::StreamInfo>)>;
|
||||
class ResourceRequester;
|
||||
|
||||
void OnPdfStreamCreated(std::unique_ptr<content::StreamInfo> stream_info);
|
||||
void OnPdfStreamResponseStarted(
|
||||
scoped_refptr<net::HttpResponseHeaders> headers,
|
||||
const std::string& mime_type);
|
||||
void OnSaveURLAs(const GURL& url, const content::Referrer& referrer);
|
||||
|
||||
// Source URL from where the PDF originates.
|
||||
|
@ -56,6 +47,8 @@ class PdfViewerUI : public content::WebUIController,
|
|||
|
||||
PdfViewerHandler* pdf_handler_;
|
||||
|
||||
scoped_refptr<ResourceRequester> resource_requester_;
|
||||
|
||||
// Pdf Resource stream.
|
||||
std::unique_ptr<content::StreamInfo> stream_;
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue