Home | History | Annotate | Download | only in chromedriver 
      1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/test/chromedriver/keycode_text_conversion.h"
      6 
      7 #import <Carbon/Carbon.h>
      8 
      9 #include <cctype>
     10 
     11 #include "base/mac/scoped_cftyperef.h"
     12 #include "base/strings/utf_string_conversions.h"
     13 #include "chrome/test/chromedriver/chrome/ui_events.h"
     14 #include "ui/events/keycodes/keyboard_code_conversion_mac.h"
     15 
     16 bool ConvertKeyCodeToText(
     17     ui::KeyboardCode key_code, int modifiers, std::string* text,
     18     std::string* error_msg) {
     19   int mac_key_code = 0;
     20   {
     21     unichar character, unmodified_character;
     22     mac_key_code = ui::MacKeyCodeForWindowsKeyCode(
     23         key_code,
     24         0,
     25         &character,
     26         &unmodified_character);
     27   }
     28   *error_msg = std::string();
     29   if (mac_key_code < 0) {
     30     *text = std::string();
     31     return true;
     32   }
     33 
     34   int mac_modifiers = 0;
     35   if (modifiers & kShiftKeyModifierMask)
     36     mac_modifiers |= shiftKey;
     37   if (modifiers & kControlKeyModifierMask)
     38     mac_modifiers |= controlKey;
     39   if (modifiers & kAltKeyModifierMask)
     40     mac_modifiers |= optionKey;
     41   if (modifiers & kMetaKeyModifierMask)
     42     mac_modifiers |= cmdKey;
     43   // Convert EventRecord modifiers to format UCKeyTranslate accepts. See docs
     44   // on UCKeyTranslate for more info.
     45   UInt32 modifier_key_state = (mac_modifiers >> 8) & 0xFF;
     46 
     47   base::ScopedCFTypeRef<TISInputSourceRef> input_source_copy(
     48       TISCopyCurrentKeyboardLayoutInputSource());
     49   CFDataRef layout_data = static_cast<CFDataRef>(TISGetInputSourceProperty(
     50       input_source_copy, kTISPropertyUnicodeKeyLayoutData));
     51 
     52   UInt32 dead_key_state = 0;
     53   UniCharCount char_count = 0;
     54   UniChar character = 0;
     55   OSStatus status = UCKeyTranslate(
     56       reinterpret_cast<const UCKeyboardLayout*>(CFDataGetBytePtr(layout_data)),
     57       static_cast<UInt16>(mac_key_code),
     58       kUCKeyActionDown,
     59       modifier_key_state,
     60       LMGetKbdLast(),
     61       kUCKeyTranslateNoDeadKeysBit,
     62       &dead_key_state,
     63       1,
     64       &char_count,
     65       &character);
     66   if (status == noErr && char_count == 1 && !std::iscntrl(character)) {
     67     base::string16 text16;
     68     text16.push_back(character);
     69     *text = base::UTF16ToUTF8(text16);
     70     return true;
     71   }
     72   *text = std::string();
     73   return true;
     74 }
     75 
     76 bool ConvertCharToKeyCode(
     77     base::char16 key, ui::KeyboardCode* key_code, int *necessary_modifiers,
     78     std::string* error_msg) {
     79   base::string16 key_string;
     80   key_string.push_back(key);
     81   std::string key_string_utf8 = base::UTF16ToUTF8(key_string);
     82   bool found_code = false;
     83   *error_msg = std::string();
     84   // There doesn't seem to be a way to get a mac key code for a given unicode
     85   // character. So here we check every key code to see if it produces the
     86   // right character. We could cache the results and regenerate everytime the
     87   // language changes, but this brute force technique has negligble performance
     88   // effects (on my laptop it is a submillisecond difference).
     89   for (int i = 0; i < 256; ++i) {
     90     ui::KeyboardCode code = static_cast<ui::KeyboardCode>(i);
     91     // Skip the numpad keys.
     92     if (code >= ui::VKEY_NUMPAD0 && code <= ui::VKEY_DIVIDE)
     93       continue;
     94     std::string key_string;
     95     if (!ConvertKeyCodeToText(code, 0, &key_string, error_msg))
     96       return false;
     97     found_code = key_string_utf8 == key_string;
     98     std::string key_string_utf8_tmp;
     99     if (!ConvertKeyCodeToText(
    100         code, kShiftKeyModifierMask, &key_string_utf8_tmp, error_msg))
    101       return false;
    102     if (!found_code && key_string_utf8 == key_string_utf8_tmp) {
    103       *necessary_modifiers = kShiftKeyModifierMask;
    104       found_code = true;
    105     }
    106     if (found_code) {
    107       *key_code = code;
    108       break;
    109     }
    110   }
    111   return found_code;
    112 }
    113