Network connectivity errors handling (incl. CP) for enrollment flow

chrome/browser/ui/webui/chromeos/login/enrollment_screen_handler.h

 // Copyright (c) 2012 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_CHROMEOS_LOGIN_ENROLLMENT_SCREEN_HANDLER_H_
 #define CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_ENROLLMENT_SCREEN_HANDLER_H_
 
 #include <string>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/callback_forward.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/scoped_vector.h"
 #include "chrome/browser/browsing_data/browsing_data_remover.h"
 #include "chrome/browser/chromeos/login/enrollment/enrollment_screen_actor.h"
+#include "chrome/browser/chromeos/login/ui/webui_login_view.h"
 #include "chrome/browser/ui/webui/chromeos/login/base_screen_handler.h"
+#include "chrome/browser/ui/webui/chromeos/login/network_state_informer.h"
 
 namespace policy {
 class PolicyOAuth2TokenFetcher;
 }
 
 namespace chromeos {
 
 class AuthenticatedUserEmailRetriever;
 
 // WebUIMessageHandler implementation which handles events occurring on the
 // page, such as the user pressing the signin button.
 class EnrollmentScreenHandler
     : public BaseScreenHandler,
       public EnrollmentScreenActor,
-      public BrowsingDataRemover::Observer {
+      public BrowsingDataRemover::Observer,
+      public NetworkStateInformer::NetworkStateInformerObserver,
+      public WebUILoginView::FrameObserver {
  public:
-  EnrollmentScreenHandler();
+  EnrollmentScreenHandler(
+      const scoped_refptr<NetworkStateInformer>& network_state_informer,
+      ErrorScreenActor* error_screen_actor);
   virtual ~EnrollmentScreenHandler();
 
   // Implements WebUIMessageHandler:
   virtual void RegisterMessages() OVERRIDE;
 
   // Implements EnrollmentScreenActor:
   virtual void SetParameters(Controller* controller,
                              EnrollmentMode enrollment_mode,
                              const std::string& management_domain) OVERRIDE;
   virtual void PrepareToShow() OVERRIDE;
   virtual void Show() OVERRIDE;
   virtual void Hide() OVERRIDE;
   virtual void FetchOAuthToken() OVERRIDE;
   virtual void ResetAuth(const base::Closure& callback) OVERRIDE;
   virtual void ShowSigninScreen() OVERRIDE;
   virtual void ShowEnrollmentSpinnerScreen() OVERRIDE;
   virtual void ShowLoginSpinnerScreen() OVERRIDE;
   virtual void ShowAuthError(const GoogleServiceAuthError& error) OVERRIDE;
   virtual void ShowEnrollmentStatus(policy::EnrollmentStatus status) OVERRIDE;
   virtual void ShowUIError(UIError error_code) OVERRIDE;
 
   // Implements BaseScreenHandler:
   virtual void Initialize() OVERRIDE;
   virtual void DeclareLocalizedValues(LocalizedValuesBuilder* builder) OVERRIDE;
 
   // Implements BrowsingDataRemover::Observer:
   virtual void OnBrowsingDataRemoverDone() OVERRIDE;
 
+  // Implements NetworkStateInformer::NetworkStateInformerObserver
+  virtual void UpdateState(ErrorScreenActor::ErrorReason reason) OVERRIDE;
+
+  // Implements WebUILoginView::FrameObserver
+  virtual void OnFrameError(const std::string& frame_unique_name) OVERRIDE;
+
  private:
   // Handlers for WebUI messages.
   void HandleRetrieveAuthenticatedUserEmail(double attempt_token);
   void HandleClose(const std::string& reason);
   void HandleCompleteLogin(const std::string& user);
   void HandleRetry();
+  void HandleFrameLoadingCompleted(int status);
 
+  void SetupAndShowOfflineMessage(NetworkStateInformer::State state,
+                                  ErrorScreenActor::ErrorReason reason);
+  void HideOfflineMessage(NetworkStateInformer::State state,
+                          ErrorScreenActor::ErrorReason reason);
+
+  net::Error frame_error() const { return frame_error_; }
   // Shows a given enrollment step.
   void ShowStep(const char* step);
 
   // Display the given i18n resource as error message.
   void ShowError(int message_id, bool retry);
 
   // Display the given string as error message.
   void ShowErrorMessage(const std::string& message, bool retry);
 
   // Display the given i18n string as a progress message.
   void ShowWorking(int message_id);
 
   // Handles completion of the OAuth2 token fetch attempt.
   void OnTokenFetched(const std::string& token,
                       const GoogleServiceAuthError& error);
 
   // Shows the screen.
   void DoShow();
 
+  // Returns current visible screen.
+  OobeUI::Screen GetCurrentScreen() const;
+
+  // Returns true if current visible screen is the enrollment sign-in page.
+  bool IsOnEnrollmentScreen() const;
+
+  // Returns true if current visible screen is the error screen over
+  // enrollment sign-in page.
+  bool IsEnrollmentScreenHiddenByError() const;
+
   // Keeps the controller for this actor.
   Controller* controller_;
 
   bool show_on_init_;
 
   // The enrollment mode.
   EnrollmentMode enrollment_mode_;
 
   // The management domain, if applicable.
   std::string management_domain_;
 
   // Whether an enrollment attempt has failed.
   bool enrollment_failed_once_;
 
   // This intentionally lives here and not in the controller, since it needs to
   // execute requests in the context of the profile that displays the webui.
   scoped_ptr<policy::PolicyOAuth2TokenFetcher> oauth_fetcher_;
 
   // The browsing data remover instance currently active, if any.
   BrowsingDataRemover* browsing_data_remover_;
 
   // The callbacks to invoke after browsing data has been cleared.
   std::vector<base::Closure> auth_reset_callbacks_;
 
   // Helper that retrieves the authenticated user's e-mail address.
   scoped_ptr<AuthenticatedUserEmailRetriever> email_retriever_;
 
+  // Latest enrollment frame error.
+  net::Error frame_error_;
+
+  // Network state informer used to keep signin screen up.
+  scoped_refptr<NetworkStateInformer> network_state_informer_;
+
+  ErrorScreenActor* error_screen_actor_;
+
+  base::WeakPtrFactory<EnrollmentScreenHandler> weak_ptr_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(EnrollmentScreenHandler);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_ENROLLMENT_SCREEN_HANDLER_H_