chrome/browser/ui/libgtk2ui/unity_service.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 "chrome/browser/ui/libgtk2ui/unity_service.h"
   6
   7 #include <dlfcn.h>
   8 #include <string>
   9
  10 #include <gtk/gtk.h>
  11
  12 #include "base/environment.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/nix/xdg_util.h"
  15 #include "chrome/browser/shell_integration_linux.h"
  16 #include "chrome/browser/ui/libgtk2ui/gtk2_util.h"
  17
  18 // Unity data typedefs.
  19 typedef struct _UnityInspector UnityInspector;
  20 typedef UnityInspector* (*unity_inspector_get_default_func)(void);
  21 typedef gboolean (*unity_inspector_get_unity_running_func)
  22     (UnityInspector* self);
  23
  24 typedef struct _UnityLauncherEntry UnityLauncherEntry;
  25 typedef UnityLauncherEntry* (*unity_launcher_entry_get_for_desktop_id_func)
  26     (const gchar* desktop_id);
  27 typedef void (*unity_launcher_entry_set_count_func)(UnityLauncherEntry* self,
  28                                                gint64 value);
  29 typedef void (*unity_launcher_entry_set_count_visible_func)
  30     (UnityLauncherEntry* self, gboolean value);
  31 typedef void (*unity_launcher_entry_set_progress_func)(UnityLauncherEntry* self,
  32                                                        gdouble value);
  33 typedef void (*unity_launcher_entry_set_progress_visible_func)
  34     (UnityLauncherEntry* self, gboolean value);
  35
  36
  37 namespace {
  38
  39 bool attempted_load = false;
  40
  41 // Unity has a singleton object that we can ask whether the unity is running.
  42 UnityInspector* inspector = NULL;
  43
  44 // A link to the desktop entry in the panel.
  45 UnityLauncherEntry* chrome_entry = NULL;
  46
  47 // Retrieved functions from libunity.
  48 unity_inspector_get_unity_running_func get_unity_running = NULL;
  49 unity_launcher_entry_set_count_func entry_set_count = NULL;
  50 unity_launcher_entry_set_count_visible_func entry_set_count_visible = NULL;
  51 unity_launcher_entry_set_progress_func entry_set_progress = NULL;
  52 unity_launcher_entry_set_progress_visible_func entry_set_progress_visible =
  53     NULL;
  54
  55 void EnsureMethodsLoaded() {
  56   using base::nix::GetDesktopEnvironment;
  57
  58   if (attempted_load)
  59     return;
  60   attempted_load = true;
  61
  62   scoped_ptr<base::Environment> env(base::Environment::Create());
  63   base::nix::DesktopEnvironment desktop_env =
  64       GetDesktopEnvironment(env.get());
  65
  66   // The "icon-tasks" KDE task manager also honors Unity Launcher API.
  67   if (desktop_env != base::nix::DESKTOP_ENVIRONMENT_UNITY &&
  68       desktop_env != base::nix::DESKTOP_ENVIRONMENT_KDE4)
  69     return;
  70
  71   // Ubuntu still hasn't given us a nice libunity.so symlink.
  72   void* unity_lib = dlopen("libunity.so.4", RTLD_LAZY);
  73   if (!unity_lib)
  74     unity_lib = dlopen("libunity.so.6", RTLD_LAZY);
  75   if (!unity_lib)
  76     unity_lib = dlopen("libunity.so.9", RTLD_LAZY);
  77   if (!unity_lib)
  78     return;
  79
  80   unity_inspector_get_default_func inspector_get_default =
  81       reinterpret_cast<unity_inspector_get_default_func>(
  82           dlsym(unity_lib, "unity_inspector_get_default"));
  83   if (inspector_get_default) {
  84     inspector = inspector_get_default();
  85
  86     get_unity_running =
  87         reinterpret_cast<unity_inspector_get_unity_running_func>(
  88             dlsym(unity_lib, "unity_inspector_get_unity_running"));
  89   }
  90
  91   unity_launcher_entry_get_for_desktop_id_func entry_get_for_desktop_id =
  92       reinterpret_cast<unity_launcher_entry_get_for_desktop_id_func>(
  93           dlsym(unity_lib, "unity_launcher_entry_get_for_desktop_id"));
  94   if (entry_get_for_desktop_id) {
  95     std::string desktop_id = libgtk2ui::GetDesktopName(env.get());
  96     chrome_entry = entry_get_for_desktop_id(desktop_id.c_str());
  97
  98     entry_set_count =
  99         reinterpret_cast<unity_launcher_entry_set_count_func>(
 100             dlsym(unity_lib, "unity_launcher_entry_set_count"));
 101
 102     entry_set_count_visible =
 103         reinterpret_cast<unity_launcher_entry_set_count_visible_func>(
 104             dlsym(unity_lib, "unity_launcher_entry_set_count_visible"));
 105
 106     entry_set_progress =
 107         reinterpret_cast<unity_launcher_entry_set_progress_func>(
 108             dlsym(unity_lib, "unity_launcher_entry_set_progress"));
 109
 110     entry_set_progress_visible =
 111         reinterpret_cast<unity_launcher_entry_set_progress_visible_func>(
 112             dlsym(unity_lib, "unity_launcher_entry_set_progress_visible"));
 113   }
 114 }
 115
 116 }  // namespace
 117
 118
 119 namespace unity {
 120
 121 bool IsRunning() {
 122   EnsureMethodsLoaded();
 123   if (inspector && get_unity_running)
 124     return get_unity_running(inspector);
 125
 126   return false;
 127 }
 128
 129 void SetDownloadCount(int count) {
 130   EnsureMethodsLoaded();
 131   if (chrome_entry && entry_set_count && entry_set_count_visible) {
 132     entry_set_count(chrome_entry, count);
 133     entry_set_count_visible(chrome_entry, count != 0);
 134   }
 135 }
 136
 137 void SetProgressFraction(float percentage) {
 138   EnsureMethodsLoaded();
 139   if (chrome_entry && entry_set_progress && entry_set_progress_visible) {
 140     entry_set_progress(chrome_entry, percentage);
 141     entry_set_progress_visible(chrome_entry,
 142                                percentage > 0.0 && percentage < 1.0);
 143   }
 144 }
 145
 146 }  // namespace unity