1 files changed, 39 insertions, 0 deletions
diff --git a/chrome/browser/ui/webui/screenshot_source.h b/chrome/browser/ui/webui/screenshot_source.h
new file mode 100644
index 0000000..e4f77a4
--- /dev/null
+++ b/chrome/browser/ui/webui/screenshot_source.h
@@ -0,0 +1,39 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef CHROME_BROWSER_UI_WEBUI_SCREENSHOT_SOURCE_H_
+#define CHROME_BROWSER_UI_WEBUI_SCREENSHOT_SOURCE_H_
+
+#include <string>
+#include <vector>
+
+#include "base/basictypes.h"
+#include "chrome/browser/ui/webui/chrome_url_data_manager.h"
+
+// ScreenshotSource is the data source that serves screenshots (saved
+// or current) to the bug report html ui
+class ScreenshotSource : public ChromeURLDataManager::DataSource {
+ public:
+  explicit ScreenshotSource(
+      std::vector<unsigned char>* current_screenshot);
+
+  // Called when the network layer has requested a resource underneath
+  // the path we registered.
+  virtual void StartDataRequest(const std::string& path,
+                                bool is_off_the_record,
+                                int request_id);
+
+  virtual std::string GetMimeType(const std::string&) const;
+
+  std::vector<unsigned char> GetScreenshot(const std::string& path);
+
+ private:
+  virtual ~ScreenshotSource();
+
+  std::vector<unsigned char> current_screenshot_;
+
+  DISALLOW_COPY_AND_ASSIGN(ScreenshotSource);
+};
+
+#endif  // CHROME_BROWSER_UI_WEBUI_SCREENSHOT_SOURCE_H_