Send screenshots back to the client for debugging

chrome/test/webdriver/session.cc

 // 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.
 
 #include "chrome/test/webdriver/session_manager.h"
 
 #include <vector>
 
+#include "base/base64.h"
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/message_loop_proxy.h"
 #include "base/process.h"
 #include "base/process_util.h"
 #include "base/scoped_ptr.h"
+#include "base/scoped_temp_dir.h"
 #include "base/stringprintf.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/test/test_timeouts.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/common/chrome_constants.h"
@@ skipping 25 matching lines @@
   bool success = false;
   if (thread_.Start()) {
     RunSessionTask(NewRunnableMethod(
         this,
         &Session::InitOnSessionThread,
         browser_dir,
         &success));
   } else {
     LOG(ERROR) << "Cannot start session thread";
   }
+
   if (!success)
     delete this;
+
   return success;
 }
 
 void Session::Terminate() {
   RunSessionTask(NewRunnableMethod(
       this,
       &Session::TerminateOnSessionThread));
   delete this;
 }
 
@@ skipping 268 matching lines @@
 bool Session::GetWindowIds(std::vector<int>* window_ids) {
   bool success = false;
   RunSessionTask(NewRunnableMethod(
       automation_.get(),
       &Automation::GetTabIds,
       window_ids,
       &success));
   return success;
 }
 
+bool Session::CaptureEntirePageAsPNG(const FilePath& path) {
+  bool success = false;
+  RunSessionTask(NewRunnableMethod(
+      automation_.get(),
+      &Automation::CaptureEntirePageAsPNG,
+      current_window_id_,
+      path,
+      &success));
+  return success;
+}
+
+bool Session::ScreenshotAsBase64(std::string* screenshot) {
+  ScopedTempDir screenshots_dir;
+
+  // Create a temp directory for screenshots.
+  if (!screenshots_dir.CreateUniqueTempDir()) {
+    LOG(ERROR) << "Could not make a temp directory for screenshots";
+    return false;
+  }
+
+  // Generate a temp file name for the screenshot

[kkania, 2011/02/24 22:05:02]

we no longer need a temp file name if we are using a temp directory, remove

[Joe, 2011/03/02 02:21:25]

Done.

+  scoped_ptr<char> file(tempnam(screenshots_dir.path().value().c_str(),
+                                "screencapture"));
+  if (NULL == file.get()) {
+    LOG(ERROR) << "Temp file name was not allocated";
+    return false;
+  }
+
+#ifdef OS_POSIX
+  FilePath path = FilePath(file.get());
+#elif OS_WIN
+  FilePath path = FilePath(ASCIIToWide(file.get()));
+#endif
+
+  // Capture the current tab as a PNG file.
+  if (!CaptureEntirePageAsPNG(path)) {
+    LOG(ERROR) << "Capturing tab page as PNG failed";
+    return false;
+  }
+
+  std::string raw_bytes;
+  if (!file_util::ReadFileToString(path, &raw_bytes)) {
+    LOG(ERROR) << "Opening PNG file failed";
+    return false;
+  }
+
+  // Make a base64 encoded string for the raw PNG data
+  if (!base::Base64Encode(raw_bytes, screenshot)) {
+    LOG(ERROR) << "Encoding the PNG to base64 format failed";
+    return false;
+  }
+
+  return true;
+}
+
 ErrorCode Session::SwitchToWindow(const std::string& name) {
   int switch_to_id = 0;
   int name_no = 0;
   if (base::StringToInt(name, &name_no)) {
     bool does_exist = false;
     RunSessionTask(NewRunnableMethod(
         automation_.get(),
         &Automation::DoesTabExist,
         name_no,
         &does_exist));
@@ skipping 180 matching lines @@
   if (result->GetAsString(&xpath)) {
     if (current_frame_xpath_.length())
       current_frame_xpath_ += "\n";
     current_frame_xpath_ += xpath;
     return kSuccess;
   }
   return kNoSuchFrame;
 }
 
 }  // namespace webdriver