content/public/browser/desktop_media_id.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 "content/public/browser/desktop_media_id.h"
   6
   7 #include <vector>
   8
   9 #include "base/memory/singleton.h"
  10 #include "base/strings/string_split.h"
  11
  12 #if defined(USE_AURA)
  13 #include "ui/aura/window.h"
  14 #include "ui/aura/window_observer.h"
  15 #endif  // defined(USE_AURA)
  16
  17 namespace  {
  18
  19 #if defined(USE_AURA)
  20
  21 class AuraWindowRegistry : public aura::WindowObserver {
  22  public:
  23   static AuraWindowRegistry* GetInstance() {
  24     return Singleton<AuraWindowRegistry>::get();
  25   }
  26
  27   int RegisterWindow(aura::Window* window) {
  28     // First check if an Id is already assigned to the |window|.
  29     std::map<aura::Window*, int>::iterator it = window_to_id_map_.find(window);
  30     if (it != window_to_id_map_.end()) {
  31       return it->second;
  32     }
  33
  34     // If the windows doesn't have an Id yet assign it.
  35     int id;
  36     do {
  37       id = next_id_;
  38       next_id_ = (next_id_ == INT_MAX) ? 1 : (next_id_ + 1);
  39     } while (id_to_window_map_.find(id) != id_to_window_map_.end());
  40
  41     window_to_id_map_[window] = id;
  42     id_to_window_map_[id] = window;
  43     window->AddObserver(this);
  44     return id;
  45   }
  46
  47   aura::Window* GetWindowById(int id) {
  48     std::map<int, aura::Window*>::iterator it = id_to_window_map_.find(id);
  49     return (it != id_to_window_map_.end()) ? it->second : nullptr;
  50   }
  51
  52  private:
  53   friend struct DefaultSingletonTraits<AuraWindowRegistry>;
  54
  55   AuraWindowRegistry()
  56       : next_id_(1) {
  57   }
  58   ~AuraWindowRegistry() override {}
  59
  60   // WindowObserver overrides.
  61   void OnWindowDestroying(aura::Window* window) override {
  62     std::map<aura::Window*, int>::iterator it = window_to_id_map_.find(window);
  63     DCHECK(it != window_to_id_map_.end());
  64     id_to_window_map_.erase(it->second);
  65     window_to_id_map_.erase(it);
  66   }
  67
  68   int next_id_;
  69   std::map<aura::Window*, int> window_to_id_map_;
  70   std::map<int, aura::Window*> id_to_window_map_;
  71
  72   DISALLOW_COPY_AND_ASSIGN(AuraWindowRegistry);
  73 };
  74
  75 #endif  // defined(USE_AURA)
  76
  77 }  // namespace
  78
  79 namespace content {
  80
  81 const char kScreenPrefix[] = "screen";
  82 const char kWindowPrefix[] = "window";
  83 const char kAuraWindowPrefix[] = "aura_window";
  84
  85 #if defined(USE_AURA)
  86
  87 // static
  88 DesktopMediaID DesktopMediaID::RegisterAuraWindow(aura::Window* window) {
  89   return DesktopMediaID(
  90       TYPE_AURA_WINDOW,
  91       AuraWindowRegistry::GetInstance()->RegisterWindow(window));
  92 }
  93
  94 // static
  95 aura::Window* DesktopMediaID::GetAuraWindowById(const DesktopMediaID& id) {
  96   DCHECK_EQ(id.type, TYPE_AURA_WINDOW);
  97   return AuraWindowRegistry::GetInstance()->GetWindowById(id.id);
  98 }
  99
 100 #endif  // defined(USE_AURA)
 101
 102 // static
 103 DesktopMediaID DesktopMediaID::Parse(const std::string& str) {
 104   std::vector<std::string> parts;
 105   base::SplitString(str, ':', &parts);
 106
 107   if (parts.size() != 2)
 108     return DesktopMediaID(TYPE_NONE, 0);
 109
 110   Type type = TYPE_NONE;
 111   if (parts[0] == kScreenPrefix) {
 112     type = TYPE_SCREEN;
 113   } else if (parts[0] == kWindowPrefix) {
 114     type = TYPE_WINDOW;
 115   } else if (parts[0] == kAuraWindowPrefix) {
 116     type = TYPE_AURA_WINDOW;
 117   } else {
 118     return DesktopMediaID(TYPE_NONE, 0);
 119   }
 120
 121   int64 id;
 122   if (!base::StringToInt64(parts[1], &id))
 123     return DesktopMediaID(TYPE_NONE, 0);
 124
 125   return DesktopMediaID(type, id);
 126 }
 127
 128 std::string DesktopMediaID::ToString() {
 129   std::string prefix;
 130   switch (type) {
 131     case TYPE_NONE:
 132       NOTREACHED();
 133       return std::string();
 134     case TYPE_SCREEN:
 135       prefix = kScreenPrefix;
 136       break;
 137     case TYPE_WINDOW:
 138       prefix = kWindowPrefix;
 139       break;
 140     case TYPE_AURA_WINDOW:
 141       prefix = kAuraWindowPrefix;
 142       break;
 143   }
 144   DCHECK(!prefix.empty());
 145
 146   prefix.append(":");
 147   prefix.append(base::Int64ToString(id));
 148
 149   return prefix;
 150 }
 151
 152 }  // namespace content