Upstream //ios/chrome/browser/translate

ios/chrome/browser/translate/before_translate_infobar_controller.mm

Index: ios/chrome/browser/translate/before_translate_infobar_controller.mm
diff --git a/ios/chrome/browser/translate/before_translate_infobar_controller.mm b/ios/chrome/browser/translate/before_translate_infobar_controller.mm
new file mode 100644
index 0000000000000000000000000000000000000000..92702350a49c75475bdab4df09a4f4f84e606a16
--- /dev/null
+++ b/ios/chrome/browser/translate/before_translate_infobar_controller.mm
@@ -0,0 +1,356 @@
+// Copyright 2013 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 "ios/chrome/browser/translate/before_translate_infobar_controller.h"
+
+#import <UIKit/UIKit.h>
+
+#include "base/logging.h"
+#include "base/strings/sys_string_conversions.h"
+#include "base/strings/utf_string_conversions.h"
+#include "components/translate/core/browser/translate_infobar_delegate.h"
+#include "grit/components_strings.h"
+#include "ios/chrome/browser/translate/translate_infobar_tags.h"
+#include "ios/public/provider/chrome/browser/chrome_browser_provider.h"
+#import "ios/public/provider/chrome/browser/ui/infobar_view_delegate.h"
+#import "ios/public/provider/chrome/browser/ui/infobar_view_protocol.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/gfx/image/image.h"
+
+namespace {
+
+CGFloat kNavigationBarHeight = 44;
+CGFloat kUIPickerHeight = 216;
+CGFloat kUIPickerFontSize = 26;
+NSTimeInterval kPickerAnimationDurationInSeconds = 0.2;
+
+}  // namespace
+
+// The class is a data source and delegate to the UIPickerView that contains the
+// language list.
+@interface LanguagePickerController
+    : UIViewController<UIPickerViewDataSource, UIPickerViewDelegate> {
+  __weak translate::TranslateInfoBarDelegate* translateInfoBarDelegate_;
+  NSInteger initialRow_;   // Displayed in bold font.
+  NSInteger disabledRow_;  // Grayed out.
+}
+@end
+
+@implementation LanguagePickerController
+
+- (instancetype)initWithDelegate:(translate::TranslateInfoBarDelegate*)
+                                     translateInfoBarDelegate
+                      initialRow:(NSInteger)initialRow
+                     disabledRow:(NSInteger)disabledRow {
+  if ((self = [super init])) {
+    translateInfoBarDelegate_ = translateInfoBarDelegate;
+    initialRow_ = initialRow;
+    disabledRow_ = disabledRow;
+  }
+  return self;
+}
+
+#pragma mark -
+#pragma mark UIPickerViewDataSource
+
+- (NSInteger)pickerView:(UIPickerView*)pickerView
+    numberOfRowsInComponent:(NSInteger)component {
+  NSUInteger numRows = translateInfoBarDelegate_->num_languages();
+  return numRows;
+}
+
+- (NSInteger)numberOfComponentsInPickerView:(UIPickerView*)pickerView {
+  return 1;
+}
+
+#pragma mark -
+#pragma mark UIPickerViewDelegate
+
+- (UIView*)pickerView:(UIPickerView*)pickerView
+           viewForRow:(NSInteger)row
+         forComponent:(NSInteger)component
+          reusingView:(UIView*)view {
+  DCHECK_EQ(0, component);
+  UILabel* label = [view isKindOfClass:[UILabel class]]
+                       ? (UILabel*)view
+                       : [[[UILabel alloc] init] autorelease];
+  [label setText:base::SysUTF16ToNSString(
+                     translateInfoBarDelegate_->language_name_at(row))];
+  [label setTextAlignment:NSTextAlignmentCenter];
+  UIFont* font = [UIFont systemFontOfSize:kUIPickerFontSize];
+  BOOL enabled = YES;
+  if (row == initialRow_)
+    font = [UIFont boldSystemFontOfSize:kUIPickerFontSize];
+  else if (row == disabledRow_)
+    enabled = NO;
+  [label setFont:font];
+  [label setEnabled:enabled];
+  return label;
+}
+
+@end
+
+@interface BeforeTranslateInfoBarController ()
+
+// Action for any of the user defined buttons.
+- (void)infoBarButtonDidPress:(id)sender;
+// Action for any of the user defined links.
+- (void)infobarLinkDidPress:(NSNumber*)tag;
+// Action for the language selection "Done" button.
+- (void)languageSelectionDone;
+// Dismisses the language selection view.
+- (void)dismissLanguageSelectionView;
+
+@end
+
+@implementation BeforeTranslateInfoBarController {
+  __weak translate::TranslateInfoBarDelegate* translateInfoBarDelegate_;
+  // A fullscreen view that catches all touch events and contains a UIPickerView
+  // and a UINavigationBar.
+  base::scoped_nsobject<UIView> languageSelectionView_;
+  // Stores whether the user is currently choosing in the UIPickerView the
+  // original language, or the target language.
+  NSUInteger languageSelectionType_;
+  // The language picker.
+  base::scoped_nsobject<UIPickerView> languagePicker_;
+  // Navigation bar associated with the picker with "Done" and "Cancel" buttons.
+  base::scoped_nsobject<UINavigationBar> navigationBar_;
+  // The controller of the languagePicker. Needs to be an ivar because
+  // |languagePicker_| does not retain it.
+  base::scoped_nsobject<LanguagePickerController> languagePickerController_;
+}
+
+#pragma mark -
+#pragma mark InfoBarControllerProtocol
+
+- (void)layoutForDelegate:(infobars::InfoBarDelegate*)delegate
+                    frame:(CGRect)frame {
+  translateInfoBarDelegate_ = delegate->AsTranslateInfoBarDelegate();
+  infobars::InfoBarDelegate* infoBarDelegate =
+      static_cast<infobars::InfoBarDelegate*>(translateInfoBarDelegate_);
+  DCHECK(!infoBarView_);
+  infoBarView_.reset([ios::GetChromeBrowserProvider()->CreateInfoBarView()
+      initWithFrame:frame
+           delegate:delegate_
+          isWarning:infoBarDelegate->GetInfoBarType() ==
+                    infobars::InfoBarDelegate::WARNING_TYPE]);
+  // Icon
+  gfx::Image icon = translateInfoBarDelegate_->GetIcon();
+  if (!icon.IsEmpty())
+    [infoBarView_ addLeftIcon:icon.ToUIImage()];
+
+  // Main text.
+  [self updateInfobarLabel];
+
+  // Close button.
+  [infoBarView_ addCloseButtonWithTag:TranslateInfoBarIos::BEFORE_DENY
+                               target:self
+                               action:@selector(infoBarButtonDidPress:)];
+  // Other buttons.
+  NSString* buttonAccept = l10n_util::GetNSString(IDS_TRANSLATE_INFOBAR_ACCEPT);
+  NSString* buttonDeny = l10n_util::GetNSString(IDS_TRANSLATE_INFOBAR_DENY);
+  [infoBarView_ addButton1:buttonAccept
+                      tag1:TranslateInfoBarIos::BEFORE_ACCEPT
+                   button2:buttonDeny
+                      tag2:TranslateInfoBarIos::BEFORE_DENY
+                    target:self
+                    action:@selector(infoBarButtonDidPress:)];
+}
+
+- (void)updateInfobarLabel {
+  NSString* originalLanguage =
+      base::SysUTF16ToNSString(translateInfoBarDelegate_->language_name_at(
+          translateInfoBarDelegate_->original_language_index()));
+  NSString* targetLanguage =
+      base::SysUTF16ToNSString(translateInfoBarDelegate_->language_name_at(
+          translateInfoBarDelegate_->target_language_index()));
+  base::string16 originalLanguageWithLink =
+      base::SysNSStringToUTF16([[infoBarView_ class]
+          stringAsLink:originalLanguage
+                   tag:TranslateInfoBarIos::BEFORE_SOURCE_LANGUAGE]);
+  base::string16 targetLanguageWithLink =
+      base::SysNSStringToUTF16([[infoBarView_ class]
+          stringAsLink:targetLanguage
+                   tag:TranslateInfoBarIos::BEFORE_TARGET_LANGUAGE]);
+  NSString* label =
+      l10n_util::GetNSStringF(IDS_TRANSLATE_INFOBAR_BEFORE_MESSAGE_IOS,
+                              originalLanguageWithLink, targetLanguageWithLink);
+  [infoBarView_ addLabel:label
+                  target:self
+                  action:@selector(infobarLinkDidPress:)];
+}
+
+- (void)languageSelectionDone {
+  size_t selectedRow = [languagePicker_ selectedRowInComponent:0];
+  if (languageSelectionType_ == TranslateInfoBarIos::BEFORE_SOURCE_LANGUAGE &&
+      selectedRow != translateInfoBarDelegate_->target_language_index()) {
+    translateInfoBarDelegate_->UpdateOriginalLanguageIndex(selectedRow);
+  }
+  if (languageSelectionType_ == TranslateInfoBarIos::BEFORE_TARGET_LANGUAGE &&
+      selectedRow != translateInfoBarDelegate_->original_language_index()) {
+    translateInfoBarDelegate_->UpdateTargetLanguageIndex(selectedRow);
+  }
+  // TODO(translate): Update the position of the infobar.

[sdefresne, 2015/01/08 09:22:33]

nit: is this still relevant? is there a tracking bug? if so, please update the
comment.

+  [self updateInfobarLabel];
+  [self dismissLanguageSelectionView];
+}
+
+- (void)dismissLanguageSelectionView {
+  DCHECK_EQ(languagePicker_ == nil, navigationBar_ == nil);
+  if (languagePicker_ == nil)
+    return;
+  // Sets the picker's delegate and data source to nil, because the
+  // |languagePickerController_| may be destroyed before the picker is hidden,
+  // and even though the interactions with the picker are disabled, it might
+  // still be turning and requesting data from the data source or calling the
+  // delegate.
+  [languagePicker_ setDataSource:nil];
+  [languagePicker_ setDelegate:nil];
+  languagePickerController_.reset();
+  // Animate the picker away.
+  CGRect languagePickerFrame = [languagePicker_ frame];
+  CGRect navigationBarFrame = [navigationBar_ frame];
+  const CGFloat animationHeight =
+      languagePickerFrame.size.height + navigationBarFrame.size.height;
+  languagePickerFrame.origin.y += animationHeight;
+  navigationBarFrame.origin.y += animationHeight;
+  UIView* blockLanguagePicker = languagePicker_.get();
+  UIView* blockNavigationBar = navigationBar_.get();
+  languagePicker_.reset();

[sdefresne, 2015/01/08 09:22:33]

I'm afraid that this may lead to blockLanguagePicker pointing to a dangling
object if languagePicker_.reset() remove the last reference to the object.
Shouldn't blockLanguagePicker and blockNavigationBar be scoped_nsobject<>?

[droger, 2015/01/08 11:57:33]

Right.
I think it's not a problem in practice because the views are displayed, but it
is nonetheless a good point.
I moved the reset() a bit further down.

+  navigationBar_.reset();
+  void (^animations)(void) = ^{
+    blockLanguagePicker.frame = languagePickerFrame;
+    blockNavigationBar.frame = navigationBarFrame;
+  };
+  void (^completion)(BOOL finished) = ^(BOOL finished) {
+    [languageSelectionView_ removeFromSuperview];

[sdefresne, 2015/01/08 09:22:33]

animations block goes to great length to avoid keeping a reference to self, but
this block does capture self. Shouldn't this be changed to also use a
blockLanguageSelectionView local variable?

[droger, 2015/01/08 11:57:33]

I would rather not change this one as part of upstreaming, because it's not
clear when languageSelectionView_.reset() should be called if it is no longer
called in the block.

+    languageSelectionView_.reset();
+  };
+  [UIView animateWithDuration:kPickerAnimationDurationInSeconds
+                   animations:animations
+                   completion:completion];
+}
+
+#pragma mark - Handling of User Events
+
+- (void)infoBarButtonDidPress:(id)sender {
+  // This press might have occurred after the user has already pressed a button,
+  // in which case the view has been detached from the delegate and this press
+  // should be ignored.
+  if (!delegate_) {
+    return;
+  }
+  if ([sender isKindOfClass:[UIButton class]]) {
+    NSUInteger buttonId = static_cast<UIButton*>(sender).tag;
+    DCHECK(buttonId == TranslateInfoBarIos::BEFORE_ACCEPT ||
+           buttonId == TranslateInfoBarIos::BEFORE_DENY);
+    delegate_->InfoBarButtonDidPress(buttonId);
+  }
+}
+
+- (void)infobarLinkDidPress:(NSNumber*)tag {
+  DCHECK([tag isKindOfClass:[NSNumber class]]);
+  languageSelectionType_ = [tag unsignedIntegerValue];
+  DCHECK(languageSelectionType_ ==
+             TranslateInfoBarIos::BEFORE_SOURCE_LANGUAGE ||
+         languageSelectionType_ == TranslateInfoBarIos::BEFORE_TARGET_LANGUAGE);
+  if (languagePicker_ != nil)
+    return;  // The UIPickerView is already up.
+
+  // Creates and adds the view containing the UIPickerView and the
+  // UINavigationBar.
+  UIView* parentView =
+      [[UIApplication sharedApplication] keyWindow].rootViewController.view;
+  // Convert the parent frame to handle device rotation.
+  CGRect parentFrame =
+      CGRectApplyAffineTransform([parentView frame], [parentView transform]);
+  const CGFloat totalPickerHeight = kUIPickerHeight + kNavigationBarHeight;
+  CGRect languageSelectionViewFrame =
+      CGRectMake(0, parentFrame.size.height - totalPickerHeight,
+                 parentFrame.size.width, totalPickerHeight);
+  languageSelectionView_.reset(
+      [[UIView alloc] initWithFrame:languageSelectionViewFrame]);
+  [languageSelectionView_
+      setAutoresizingMask:UIViewAutoresizingFlexibleWidth |
+                          UIViewAutoresizingFlexibleTopMargin];
+  [parentView addSubview:languageSelectionView_];
+
+  // Creates the navigation bar and its buttons.
+  CGRect finalPickerFrame = CGRectMake(
+      0, [languageSelectionView_ frame].size.height - kUIPickerHeight,
+      [languageSelectionView_ frame].size.width, kUIPickerHeight);
+  CGRect finalNavigationBarFrame = CGRectMake(
+      0, 0, [languageSelectionView_ frame].size.width, kNavigationBarHeight);
+  // The language picker animates from the bottom of the screen.
+  CGRect initialPickerFrame = finalPickerFrame;
+  initialPickerFrame.origin.y += totalPickerHeight;
+  CGRect initialNavigationBarFrame = finalNavigationBarFrame;
+  initialNavigationBarFrame.origin.y += totalPickerHeight;
+
+  navigationBar_.reset(
+      [[UINavigationBar alloc] initWithFrame:initialNavigationBarFrame]);
+  const UIViewAutoresizing resizingMask =
+      UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleTopMargin;
+  [navigationBar_ setAutoresizingMask:resizingMask];
+  base::scoped_nsobject<UIBarButtonItem> doneButton([[UIBarButtonItem alloc]
+      initWithBarButtonSystemItem:UIBarButtonSystemItemDone
+                           target:self
+                           action:@selector(languageSelectionDone)]);
+  base::scoped_nsobject<UIBarButtonItem> cancelButton([[UIBarButtonItem alloc]
+      initWithBarButtonSystemItem:UIBarButtonSystemItemCancel
+                           target:self
+                           action:@selector(dismissLanguageSelectionView)]);
+  base::scoped_nsobject<UINavigationItem> item(
+      [[UINavigationItem alloc] initWithTitle:nil]);
+  [item setRightBarButtonItem:doneButton];
+  [item setLeftBarButtonItem:cancelButton];
+  [item setHidesBackButton:YES];
+  [navigationBar_ pushNavigationItem:item animated:NO];
+
+  // Creates the PickerView and its controller.
+  NSInteger selectedRow;
+  NSInteger disabledRow;
+  if (languageSelectionType_ == TranslateInfoBarIos::BEFORE_SOURCE_LANGUAGE) {
+    selectedRow = translateInfoBarDelegate_->original_language_index();
+    disabledRow = translateInfoBarDelegate_->target_language_index();
+  } else {
+    selectedRow = translateInfoBarDelegate_->target_language_index();
+    disabledRow = translateInfoBarDelegate_->original_language_index();
+  }
+  languagePickerController_.reset([[LanguagePickerController alloc]
+      initWithDelegate:translateInfoBarDelegate_
+            initialRow:selectedRow
+           disabledRow:disabledRow]);
+  languagePicker_.reset(
+      [[UIPickerView alloc] initWithFrame:initialPickerFrame]);
+  [languagePicker_ setAutoresizingMask:resizingMask];
+  [languagePicker_ setShowsSelectionIndicator:YES];
+  [languagePicker_ setDataSource:languagePickerController_];
+  [languagePicker_ setDelegate:languagePickerController_];
+  [languagePicker_ setShowsSelectionIndicator:YES];
+  [languagePicker_ setBackgroundColor:[infoBarView_ backgroundColor]];
+  [languagePicker_ selectRow:selectedRow inComponent:0 animated:NO];
+
+  [UIView animateWithDuration:kPickerAnimationDurationInSeconds
+                   animations:^{
+                     languagePicker_.get().frame = finalPickerFrame;
+                     navigationBar_.get().frame = finalNavigationBarFrame;
+                   }];
+
+  // Add the subviews.
+  [languageSelectionView_ addSubview:languagePicker_];
+  [languageSelectionView_ addSubview:navigationBar_];
+}
+
+- (void)removeView {
+  [super removeView];
+  [self dismissLanguageSelectionView];
+}
+
+- (void)detachView {
+  [super detachView];
+  [self dismissLanguageSelectionView];
+}
+
+@end