Fix snapshotting on linux by creating a separate automation path for

chrome/browser/automation/automation_tab_helper.h

 // 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_AUTOMATION_AUTOMATION_TAB_HELPER_H_
 #define CHROME_BROWSER_AUTOMATION_AUTOMATION_TAB_HELPER_H_
 #pragma once
 
 #include <set>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/observer_list.h"
 #include "base/memory/weak_ptr.h"
 #include "content/browser/tab_contents/tab_contents_observer.h"
 
 class TabContents;
 class AutomationTabHelper;
 
+namespace gfx {
+class Size;

[Paweł Hajdan Jr., 2011/10/18 10:21:35]

nit: Is this needed?

[kkania, 2011/10/18 18:30:19]

Done.

+}
+
 namespace IPC {
 class Message;
 }
 
 // An observer API implemented by classes which are interested in various
 // tab events from AutomationTabHelper(s).
 class TabEventObserver {
  public:
   // |LOAD_START| and |LOAD_STOP| notifications may occur several times for a
   // sequence of loads that may appear as one complete navigation to a user.
@@ skipping 11 matching lines @@
   // Called when the tab that had no pending loads now has a new pending
   // load. |tab_contents| will always be valid.
   virtual void OnFirstPendingLoad(TabContents* tab_contents) { }
 
   // Called when the tab that had one or more pending loads now has no
   // pending loads. |tab_contents| will always be valid.
   //
   // This method will always be called if |OnFirstPendingLoad| was called.
   virtual void OnNoMorePendingLoads(TabContents* tab_contents) { }
 
+  // Called as a result of a tab being snapshotted.
+  virtual void OnSnapshotEntirePageACK(
+      bool success,
+      const std::vector<unsigned char>& png_data,
+      const std::string& error_msg) { }
+
  protected:
   TabEventObserver();
   virtual ~TabEventObserver();
 
   // On construction, this class does not observe any events. This method
   // sets us up to observe events from the given |AutomationTabHelper|.
   void StartObserving(AutomationTabHelper* tab_helper);
 
   // Stop observing events from the given |AutomationTabHelper|. This does not
   // need to be called before the helper dies, and it is ok if this object is
@@ skipping 16 matching lines @@
 class AutomationTabHelper
     : public TabContentsObserver,
       public base::SupportsWeakPtr<AutomationTabHelper> {
  public:
   explicit AutomationTabHelper(TabContents* tab_contents);
   virtual ~AutomationTabHelper();
 
   void AddObserver(TabEventObserver* observer);
   void RemoveObserver(TabEventObserver* observer);
 
+  // Snapshots the entire page without resizing.
+  void SnapshotEntirePage();
+
   // Returns true if the tab is loading or the tab is scheduled to load
   // immediately. Note that scheduled loads may be canceled.
   bool has_pending_loads() const;
 
  private:
   friend class AutomationTabHelperTest;
 
+  void OnSnapshotEntirePageACK(
+      bool success,
+      const std::vector<unsigned char>& png_data,
+      const std::string& error_msg);
+
   // TabContentsObserver implementation.
   virtual void DidStartLoading();
   virtual void DidStopLoading();
   virtual void RenderViewGone();
   virtual void TabContentsDestroyed(TabContents* tab_contents);
   virtual bool OnMessageReceived(const IPC::Message& message);
 
   void OnWillPerformClientRedirect(int64 frame_id, double delay_seconds);
   void OnDidCompleteOrCancelClientRedirect(int64 frame_id);
   void OnTabOrRenderViewDestroyed(TabContents* tab_contents);
 
   // True if the tab is currently loading. If a navigation is scheduled but not
   // yet loading, this will be false.
   bool is_loading_;
 
   // Set of all the frames (by frame ID) that are scheduled to perform a client
   // redirect.
   std::set<int64> pending_client_redirects_;
 
   // List of all the |TabEventObserver|s, which we broadcast events to.
   ObserverList<TabEventObserver> observers_;
 
   DISALLOW_COPY_AND_ASSIGN(AutomationTabHelper);
 };
 
 #endif  // CHROME_BROWSER_AUTOMATION_AUTOMATION_TAB_HELPER_H_