Add method for identifying different InfoBars

chrome/browser/ui/chrome_select_file_policy.cc

 // 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.
 
 #include "chrome/browser/ui/chrome_select_file_policy.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/logging.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/grit/generated_resources.h"
+#include "components/infobars/core/infobar_delegate.h"
 #include "components/infobars/core/simple_alert_infobar_delegate.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/gfx/vector_icons_public.h"
 
 ChromeSelectFilePolicy::ChromeSelectFilePolicy(
     content::WebContents* source_contents)
     : source_contents_(source_contents) {
 }
 
 ChromeSelectFilePolicy::~ChromeSelectFilePolicy() {}
 
 bool ChromeSelectFilePolicy::CanOpenSelectFileDialog() {
   return FileSelectDialogsAllowed();
 }
 
 void ChromeSelectFilePolicy::SelectFileDenied() {
   // Show the InfoBar saying that file-selection dialogs are disabled.
   if (source_contents_) {
     SimpleAlertInfoBarDelegate::Create(
         InfoBarService::FromWebContents(source_contents_),
+        infobars::InfoBarDelegate::CHROME_SELECT_FILE_POLICY,
         infobars::InfoBarDelegate::kNoIconID,
         gfx::VectorIconId::VECTOR_ICON_NONE,
         l10n_util::GetStringUTF16(IDS_FILE_SELECTION_DIALOG_INFOBAR), true);
   } else {
     LOG(WARNING) << "File-selection dialogs are disabled but no WebContents "
                  << "is given to display the InfoBar.";
   }
 }
 
 // static
 bool ChromeSelectFilePolicy::FileSelectDialogsAllowed() {
   DCHECK(g_browser_process);
 
   // local_state() can return NULL for tests.
   if (!g_browser_process->local_state())
     return false;
 
   return !g_browser_process->local_state()->FindPreference(
              prefs::kAllowFileSelectionDialogs) ||
          g_browser_process->local_state()->GetBoolean(
              prefs::kAllowFileSelectionDialogs);
 }