ios/web_view/shell/shell_translation_delegate.m

// Copyright 2014 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.

#import "ios/web_view/shell/shell_translation_delegate.h"

#import <UIKit/UIKit.h>

#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
#endif

@interface ShellTranslationDelegate ()
// Action Sheet to prompt user whether or not the page should be translated.
@property(nonatomic, strong) UIAlertController* beforeTranslateActionSheet;
@end

@implementation ShellTranslationDelegate

@synthesize beforeTranslateActionSheet = _beforeTranslateActionSheet;

- (void)dealloc {
  [_beforeTranslateActionSheet dismissViewControllerAnimated:YES
                                                  completion:nil];
}

#pragma mark - CWVTranslationDelegate methods

- (void)translationController:(CWVTranslationController*)controller
    canOfferTranslationFromLanguage:(CWVTranslationLanguage*)pageLanguage
                         toLanguage:(CWVTranslationLanguage*)userLanguage {
  __weak ShellTranslationDelegate* weakSelf = self;

  self.beforeTranslateActionSheet = [UIAlertController
      alertControllerWithTitle:nil
                       message:@"Pick Translate Action"
                preferredStyle:UIAlertControllerStyleActionSheet];
  UIAlertAction* cancelAction =
      [UIAlertAction actionWithTitle:@"Nope."
                               style:UIAlertActionStyleCancel
                             handler:^(UIAlertAction* action) {
                               weakSelf.beforeTranslateActionSheet = nil;
                             }];
  [_beforeTranslateActionSheet addAction:cancelAction];

  NSString* translateTitle = [NSString
      stringWithFormat:@"Translate to %@", userLanguage.localizedName];
  UIAlertAction* translateAction = [UIAlertAction
      actionWithTitle:translateTitle
                style:UIAlertActionStyleDefault
              handler:^(UIAlertAction* action) {
                weakSelf.beforeTranslateActionSheet = nil;
                if (!weakSelf) {
                  return;
                }
                CWVTranslationLanguage* source = pageLanguage;
                CWVTranslationLanguage* target = userLanguage;
                [controller translatePageFromLanguage:source
                                           toLanguage:target
                                        userInitiated:YES];
              }];
  [_beforeTranslateActionSheet addAction:translateAction];

  UIAlertAction* alwaysTranslateAction = [UIAlertAction
      actionWithTitle:@"Always Translate"
                style:UIAlertActionStyleDefault
              handler:^(UIAlertAction* action) {
                weakSelf.beforeTranslateActionSheet = nil;
                if (!weakSelf) {
                  return;
                }
                CWVTranslationPolicy* policy = [CWVTranslationPolicy
                    translationPolicyAutoTranslateToLanguage:userLanguage];
                [controller setTranslationPolicy:policy
                                 forPageLanguage:pageLanguage];
              }];
  [_beforeTranslateActionSheet addAction:alwaysTranslateAction];

  UIAlertAction* neverTranslateAction = [UIAlertAction
      actionWithTitle:@"Never Translate"
                style:UIAlertActionStyleDefault
              handler:^(UIAlertAction* action) {
                weakSelf.beforeTranslateActionSheet = nil;
                if (!weakSelf) {
                  return;
                }
                CWVTranslationPolicy* policy =
                    [CWVTranslationPolicy translationPolicyNever];
                [controller setTranslationPolicy:policy
                                 forPageLanguage:pageLanguage];
              }];
  [_beforeTranslateActionSheet addAction:neverTranslateAction];

  [[UIApplication sharedApplication].keyWindow.rootViewController
      presentViewController:_beforeTranslateActionSheet
                   animated:YES
                 completion:nil];
}

@end