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/test_util.h"
      6 
      7 #include "base/logging.h"
      8 #include "base/strings/utf_string_conversions.h"
      9 
     10 RestoreKeyboardLayoutOnDestruct::RestoreKeyboardLayoutOnDestruct() {
     11 #if defined(OS_WIN)
     12   layout_ = GetKeyboardLayout(NULL);
     13 #elif defined(OS_MACOSX)
     14   layout_.reset(TISCopyCurrentKeyboardInputSource());
     15 #elif defined(OS_LINUX)
     16   NOTIMPLEMENTED();
     17 #endif
     18 }
     19 
     20 RestoreKeyboardLayoutOnDestruct::~RestoreKeyboardLayoutOnDestruct() {
     21 #if defined(OS_WIN)
     22   ActivateKeyboardLayout(layout_, 0);
     23 #elif defined(OS_MACOSX)
     24   TISSelectInputSource(layout_);
     25 #elif defined(OS_LINUX)
     26   NOTIMPLEMENTED();
     27 #endif
     28 }
     29 
     30 #if defined(OS_WIN)
     31 bool SwitchKeyboardLayout(const std::string& input_locale_identifier) {
     32   HKL layout = LoadKeyboardLayout(
     33       base::UTF8ToWide(input_locale_identifier).c_str(), 0);
     34   if (!layout)
     35     return false;
     36   return !!ActivateKeyboardLayout(layout, 0);
     37 }
     38 #endif  // defined(OS_WIN)
     39 
     40 #if defined(OS_MACOSX)
     41 bool SwitchKeyboardLayout(const std::string& input_source_id) {
     42   base::ScopedCFTypeRef<CFMutableDictionaryRef> filter_dict(
     43       CFDictionaryCreateMutable(kCFAllocatorDefault,
     44                                 1,
     45                                 &kCFTypeDictionaryKeyCallBacks,
     46                                 &kCFTypeDictionaryValueCallBacks));
     47   base::ScopedCFTypeRef<CFStringRef> id_ref(CFStringCreateWithCString(
     48       kCFAllocatorDefault, input_source_id.c_str(), kCFStringEncodingUTF8));
     49   CFDictionaryAddValue(filter_dict, kTISPropertyInputSourceID, id_ref);
     50   base::ScopedCFTypeRef<CFArrayRef> sources(
     51       TISCreateInputSourceList(filter_dict, true));
     52   if (CFArrayGetCount(sources) != 1)
     53     return false;
     54   TISInputSourceRef source = (TISInputSourceRef)CFArrayGetValueAtIndex(
     55       sources, 0);
     56   return TISSelectInputSource(source) == noErr;
     57 }
     58 #endif  // defined(OS_MACOSX)
     59