win8/metro_driver/ime/input_source.cc

   1 // Copyright 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 "win8/metro_driver/ime/input_source.h"
   6
   7 #include <atlbase.h>
   8 #include <atlcom.h>
   9 #include <msctf.h>
  10
  11 #include "base/bind.h"
  12 #include "base/callback.h"
  13 #include "base/logging.h"
  14 #include "base/memory/ref_counted.h"
  15 #include "base/observer_list.h"
  16 #include "base/win/scoped_comptr.h"
  17 #include "ui/base/win/atl_module.h"
  18 #include "win8/metro_driver/ime/input_source_observer.h"
  19
  20 namespace metro_driver {
  21 namespace {
  22
  23 // An implementation of ITfLanguageProfileNotifySink interface, which will be
  24 // used to receive notifications when the text input source is changed.
  25 class ATL_NO_VTABLE InputSourceMonitor
  26     : public CComObjectRootEx<CComMultiThreadModel>,
  27       public ITfLanguageProfileNotifySink {
  28  public:
  29   InputSourceMonitor()
  30       : cookie_(TF_INVALID_COOKIE) {
  31   }
  32
  33   BEGIN_COM_MAP(InputSourceMonitor)
  34     COM_INTERFACE_ENTRY(ITfLanguageProfileNotifySink)
  35   END_COM_MAP()
  36
  37   bool Initialize(ITfSource* source) {
  38     DWORD cookie = TF_INVALID_COOKIE;
  39     HRESULT hr = source->AdviseSink(IID_ITfLanguageProfileNotifySink,
  40                                     this,
  41                                     &cookie);
  42     if (FAILED(hr)) {
  43       LOG(ERROR) << "ITfSource::AdviseSink failed. hr = " << hr;
  44       return false;
  45     }
  46     cookie_ = cookie;
  47     source_ = source;
  48     return true;
  49   }
  50
  51   void SetCallback(base::Closure on_language_chanaged) {
  52     on_language_chanaged_ = on_language_chanaged;
  53   }
  54
  55   void Unadvise() {
  56     if (cookie_ == TF_INVALID_COOKIE || !source_)
  57       return;
  58     if (FAILED(source_->UnadviseSink(cookie_)))
  59       return;
  60     cookie_ = TF_INVALID_COOKIE;
  61     source_.Release();
  62   }
  63
  64  private:
  65   // ITfLanguageProfileNotifySink overrides:
  66   STDMETHOD(OnLanguageChange)(LANGID langid, BOOL *accept) OVERRIDE {
  67     if (!accept)
  68       return E_INVALIDARG;
  69     *accept = TRUE;
  70     return S_OK;
  71   }
  72
  73   STDMETHOD(OnLanguageChanged)() OVERRIDE {
  74     if (!on_language_chanaged_.is_null())
  75       on_language_chanaged_.Run();
  76     return S_OK;
  77   }
  78
  79   base::Closure on_language_chanaged_;
  80   base::win::ScopedComPtr<ITfSource> source_;
  81   DWORD cookie_;
  82
  83   DISALLOW_COPY_AND_ASSIGN(InputSourceMonitor);
  84 };
  85
  86 class InputSourceImpl : public InputSource {
  87  public:
  88   InputSourceImpl(ITfInputProcessorProfileMgr* profile_manager,
  89                   InputSourceMonitor* monitor)
  90       : profile_manager_(profile_manager),
  91         monitor_(monitor) {
  92     monitor_->SetCallback(base::Bind(&InputSourceImpl::OnLanguageChanged,
  93                                      base::Unretained(this)));
  94   }
  95   virtual ~InputSourceImpl() {
  96     monitor_->SetCallback(base::Closure());
  97     monitor_->Unadvise();
  98   }
  99
 100  private:
 101   // InputSource overrides.
 102   virtual bool GetActiveSource(LANGID* langid, bool* is_ime) OVERRIDE {
 103     TF_INPUTPROCESSORPROFILE profile = {};
 104     HRESULT hr = profile_manager_->GetActiveProfile(GUID_TFCAT_TIP_KEYBOARD,
 105                                                     &profile);
 106     if (FAILED(hr)) {
 107       LOG(ERROR) << "ITfInputProcessorProfileMgr::GetActiveProfile failed."
 108                  << " hr = " << hr;
 109       return false;
 110     }
 111     *langid = profile.langid;
 112     *is_ime = profile.dwProfileType == TF_PROFILETYPE_INPUTPROCESSOR;
 113     return true;
 114   }
 115   virtual void AddObserver(InputSourceObserver* observer) OVERRIDE {
 116     observer_list_.AddObserver(observer);
 117   }
 118   virtual void RemoveObserver(InputSourceObserver* observer) OVERRIDE {
 119     observer_list_.RemoveObserver(observer);
 120   }
 121   void OnLanguageChanged() {
 122     FOR_EACH_OBSERVER(InputSourceObserver,
 123                       observer_list_,
 124                       OnInputSourceChanged());
 125   }
 126
 127   base::win::ScopedComPtr<ITfInputProcessorProfileMgr> profile_manager_;
 128   scoped_refptr<InputSourceMonitor> monitor_;
 129   ObserverList<InputSourceObserver> observer_list_;
 130
 131   DISALLOW_COPY_AND_ASSIGN(InputSourceImpl);
 132 };
 133
 134 }  // namespace
 135
 136 // static
 137 scoped_ptr<InputSource> InputSource::Create() {
 138   ui::win::CreateATLModuleIfNeeded();
 139
 140   base::win::ScopedComPtr<ITfInputProcessorProfileMgr> profile_manager;
 141   HRESULT hr = profile_manager.CreateInstance(CLSID_TF_InputProcessorProfiles);
 142   if (FAILED(hr)) {
 143     LOG(ERROR) << "Failed to instantiate CLSID_TF_InputProcessorProfiles."
 144                << " hr = " << hr;
 145     return scoped_ptr<InputSource>();
 146   }
 147   base::win::ScopedComPtr<ITfSource> profiles_source;
 148   hr = profiles_source.QueryFrom(profile_manager);
 149   if (FAILED(hr)) {
 150     LOG(ERROR) << "QueryFrom to ITfSource failed. hr = " << hr;
 151     return scoped_ptr<InputSource>();
 152   }
 153
 154   CComObject<InputSourceMonitor>* monitor = NULL;
 155   hr = CComObject<InputSourceMonitor>::CreateInstance(&monitor);
 156   if (FAILED(hr)) {
 157     LOG(ERROR) << "CComObject<InputSourceMonitor>::CreateInstance failed."
 158                << " hr = " << hr;
 159     return scoped_ptr<InputSource>();
 160   }
 161   if (!monitor->Initialize(profiles_source)) {
 162     LOG(ERROR) << "Failed to initialize the monitor.";
 163     return scoped_ptr<InputSource>();
 164   }
 165
 166   // Transfer the ownership.
 167   return scoped_ptr<InputSource>(new InputSourceImpl(profile_manager, monitor));
 168 }
 169
 170 }  // namespace metro_driver