1 files changed, 108 insertions, 0 deletions
diff --git a/chrome/browser/translate/translate_infobars_delegates.h b/chrome/browser/translate/translate_infobars_delegates.h
new file mode 100644
index 0000000..80bada0
--- /dev/null
+++ b/chrome/browser/translate/translate_infobars_delegates.h
@@ -0,0 +1,108 @@
+// Copyright (c) 2010 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_TRANSLATE_TRANSLATE_INFOBARS_DELEGATES_H_
+#define CHROME_BROWSER_TRANSLATE_TRANSLATE_INFOBARS_DELEGATES_H_
+
+#include "chrome/browser/tab_contents/infobar_delegate.h"
+#include "chrome/browser/translate/translate_prefs.h"
+
+class SkBitmap;
+
+// An interface derived from InfoBarDelegate to form the base interface for
+// translate infobars.
+class TranslateInfoBarDelegate : public InfoBarDelegate {
+ public:
+  virtual void GetAvailableOriginalLanguages(
+      std::vector<std::string>* languages);
+  virtual void GetAvailableTargetLanguages(
+      std::vector<std::string>* languages);
+  virtual void ModifyOriginalLanguage(const std::string& original_language);
+  virtual void ModifyTargetLanguage(const std::string& target_language);
+  virtual void Translate();
+  virtual bool IsLanguageBlacklisted();
+  virtual void ToggleLanguageBlacklist();
+  virtual bool IsSiteBlacklisted();
+  virtual void ToggleSiteBlacklist();
+  virtual bool ShouldAlwaysTranslate();
+  virtual void ToggleAlwaysTranslate();
+
+  const std::string& original_language() const {
+    return original_language_;
+  }
+  const std::string& target_language() const {
+    return target_language_;
+  }
+  TabContents* tab_contents() const {
+    return tab_contents_;
+  }
+
+  // Overridden from InfoBarDelegate.
+  virtual Type GetInfoBarType() {
+    return PAGE_ACTION_TYPE;
+  }
+  virtual SkBitmap* GetIcon() const;
+  virtual bool EqualsDelegate(InfoBarDelegate* delegate) const;
+  virtual TranslateInfoBarDelegate* AsTranslateInfoBarDelegate() {
+    return this;
+  }
+  virtual void InfoBarClosed();
+
+ protected:
+  TranslateInfoBarDelegate(TabContents* contents, PrefService* user_prefs,
+      const std::string& original_language, const std::string& target_language);
+
+  TabContents* tab_contents_;  // Weak.
+  std::string original_language_;
+  std::string target_language_;
+  TranslatePrefs prefs_;
+
+  DISALLOW_COPY_AND_ASSIGN(TranslateInfoBarDelegate);
+};
+
+class BeforeTranslateInfoBarDelegate : public TranslateInfoBarDelegate {
+ public:
+  BeforeTranslateInfoBarDelegate(TabContents* contents, PrefService* user_prefs,
+      const GURL& url, const std::string& original_language,
+      const std::string& target_language);
+
+  // Overriden from TranslateInfoBarDelegate:
+  virtual bool IsLanguageBlacklisted();
+  virtual void ToggleLanguageBlacklist();
+  virtual bool IsSiteBlacklisted();
+  virtual void ToggleSiteBlacklist();
+
+  // Overridden from InfoBarDelegate:
+  virtual InfoBar* CreateInfoBar();
+
+ private:
+  std::string site_;
+  bool never_translate_language_;
+  bool never_translate_site_;
+
+  DISALLOW_COPY_AND_ASSIGN(BeforeTranslateInfoBarDelegate);
+};
+
+class AfterTranslateInfoBarDelegate : public TranslateInfoBarDelegate {
+ public:
+  AfterTranslateInfoBarDelegate(TabContents* contents, PrefService* user_prefs,
+      const std::string& original_language, const std::string& target_language);
+
+  // Overriden from TranslateInfoBar:
+  virtual void GetAvailableTargetLanguages(
+      std::vector<std::string>* languages);
+  virtual void ModifyTargetLanguage(const std::string& target_language);
+  virtual bool ShouldAlwaysTranslate();
+  virtual void ToggleAlwaysTranslate();
+
+  // Overridden from InfoBarDelegate:
+  virtual InfoBar* CreateInfoBar();
+
+ private:
+  bool always_translate_;
+
+  DISALLOW_COPY_AND_ASSIGN(AfterTranslateInfoBarDelegate);
+};
+
+#endif  // CHROME_BROWSER_TRANSLATE_TRANSLATE_INFOBARS_DELEGATES_H_