| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "mash/task_viewer/task_viewer.h" | 5 #include "mash/task_viewer/task_viewer.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <stdint.h> | 8 #include <stdint.h> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 67 kill_button_->SetStyle(views::Button::STYLE_BUTTON); | 67 kill_button_->SetStyle(views::Button::STYLE_BUTTON); |
| 68 AddChildView(kill_button_); | 68 AddChildView(kill_button_); |
| 69 } | 69 } |
| 70 ~TaskViewerContents() override { | 70 ~TaskViewerContents() override { |
| 71 table_view_->SetModel(nullptr); | 71 table_view_->SetModel(nullptr); |
| 72 task_viewer_->RemoveWindow(GetWidget()); | 72 task_viewer_->RemoveWindow(GetWidget()); |
| 73 } | 73 } |
| 74 | 74 |
| 75 private: | 75 private: |
| 76 struct InstanceInfo { | 76 struct InstanceInfo { |
| 77 InstanceInfo(uint32_t id, | 77 InstanceInfo(const shell::Identity& identity, base::ProcessId pid) |
| 78 const std::string& url, | 78 : identity(identity), pid(pid) {} |
| 79 base::ProcessId pid) | 79 shell::Identity identity; |
| 80 : id(id), url(url), pid(pid) {} | |
| 81 uint32_t id; | |
| 82 std::string url; | |
| 83 uint32_t pid; | 80 uint32_t pid; |
| 84 std::string name; | 81 std::string display_name; |
| 85 }; | 82 }; |
| 86 | 83 |
| 87 | 84 |
| 88 // Overridden from views::WidgetDelegate: | 85 // Overridden from views::WidgetDelegate: |
| 89 views::View* GetContentsView() override { return this; } | 86 views::View* GetContentsView() override { return this; } |
| 90 base::string16 GetWindowTitle() const override { | 87 base::string16 GetWindowTitle() const override { |
| 91 // TODO(beng): use resources. | 88 // TODO(beng): use resources. |
| 92 return base::ASCIIToUTF16("Tasks"); | 89 return base::ASCIIToUTF16("Tasks"); |
| 93 } | 90 } |
| 94 bool CanResize() const override { return true; } | 91 bool CanResize() const override { return true; } |
| (...skipping 22 matching lines...) Expand all Loading... |
| 117 } | 114 } |
| 118 | 115 |
| 119 // Overridden from ui::TableModel: | 116 // Overridden from ui::TableModel: |
| 120 int RowCount() override { | 117 int RowCount() override { |
| 121 return static_cast<int>(instances_.size()); | 118 return static_cast<int>(instances_.size()); |
| 122 } | 119 } |
| 123 base::string16 GetText(int row, int column_id) override { | 120 base::string16 GetText(int row, int column_id) override { |
| 124 switch(column_id) { | 121 switch(column_id) { |
| 125 case 0: | 122 case 0: |
| 126 DCHECK(row < static_cast<int>(instances_.size())); | 123 DCHECK(row < static_cast<int>(instances_.size())); |
| 127 return base::UTF8ToUTF16(instances_[row]->name); | 124 return base::UTF8ToUTF16(instances_[row]->display_name); |
| 128 case 1: | 125 case 1: |
| 129 DCHECK(row < static_cast<int>(instances_.size())); | 126 DCHECK(row < static_cast<int>(instances_.size())); |
| 130 return base::UTF8ToUTF16(instances_[row]->url); | 127 return base::UTF8ToUTF16(instances_[row]->identity.name()); |
| 131 case 2: | 128 case 2: |
| 132 DCHECK(row < static_cast<int>(instances_.size())); | 129 DCHECK(row < static_cast<int>(instances_.size())); |
| 133 return base::IntToString16(instances_[row]->pid); | 130 return base::IntToString16(instances_[row]->pid); |
| 134 default: | 131 default: |
| 135 NOTREACHED(); | 132 NOTREACHED(); |
| 136 break; | 133 break; |
| 137 } | 134 } |
| 138 return base::string16(); | 135 return base::string16(); |
| 139 } | 136 } |
| 140 void SetObserver(ui::TableModelObserver* observer) override { | 137 void SetObserver(ui::TableModelObserver* observer) override { |
| 141 observer_ = observer; | 138 observer_ = observer; |
| 142 } | 139 } |
| 143 | 140 |
| 144 // Overridden from views::ButtonListener: | 141 // Overridden from views::ButtonListener: |
| 145 void ButtonPressed(views::Button* sender, const ui::Event& event) override { | 142 void ButtonPressed(views::Button* sender, const ui::Event& event) override { |
| 146 DCHECK_EQ(sender, kill_button_); | 143 DCHECK_EQ(sender, kill_button_); |
| 147 DCHECK_EQ(table_view_->SelectedRowCount(), 1); | 144 DCHECK_EQ(table_view_->SelectedRowCount(), 1); |
| 148 int row = table_view_->FirstSelectedRow(); | 145 int row = table_view_->FirstSelectedRow(); |
| 149 DCHECK(row < static_cast<int>(instances_.size())); | 146 DCHECK(row < static_cast<int>(instances_.size())); |
| 150 base::Process process = base::Process::Open(instances_[row]->pid); | 147 base::Process process = base::Process::Open(instances_[row]->pid); |
| 151 process.Terminate(9, true); | 148 process.Terminate(9, true); |
| 152 } | 149 } |
| 153 | 150 |
| 154 // Overridden from shell::mojom::ServiceManagerListener: | 151 // Overridden from shell::mojom::ServiceManagerListener: |
| 155 void OnInit(mojo::Array<ServiceInfoPtr> instances) override { | 152 void OnInit(mojo::Array<ServiceInfoPtr> instances) override { |
| 156 // This callback should only be called with an empty model. | 153 // This callback should only be called with an empty model. |
| 157 DCHECK(instances_.empty()); | 154 DCHECK(instances_.empty()); |
| 158 mojo::Array<mojo::String> names; | 155 mojo::Array<mojo::String> names; |
| 159 for (size_t i = 0; i < instances.size(); ++i) { | 156 for (size_t i = 0; i < instances.size(); ++i) { |
| 160 InsertInstance(instances[i]->id, instances[i]->identity->name, | 157 shell::Identity identity = instances[i]->identity.To<shell::Identity>(); |
| 161 instances[i]->pid); | 158 InsertInstance(identity, instances[i]->pid); |
| 162 names.push_back(instances[i]->identity->name); | 159 names.push_back(identity.name()); |
| 163 } | 160 } |
| 164 catalog_->GetEntries(std::move(names), | 161 catalog_->GetEntries(std::move(names), |
| 165 base::Bind(&TaskViewerContents::OnGotCatalogEntries, | 162 base::Bind(&TaskViewerContents::OnGotCatalogEntries, |
| 166 weak_ptr_factory_.GetWeakPtr())); | 163 weak_ptr_factory_.GetWeakPtr())); |
| 167 } | 164 } |
| 168 void OnServiceCreated(ServiceInfoPtr instance) override { | 165 void OnServiceCreated(ServiceInfoPtr instance) override { |
| 169 DCHECK(!ContainsId(instance->id)); | 166 shell::Identity identity = instance->identity.To<shell::Identity>(); |
| 170 InsertInstance(instance->id, instance->identity->name, instance->pid); | 167 DCHECK(!ContainsIdentity(identity)); |
| 168 InsertInstance(identity, instance->pid); |
| 171 observer_->OnItemsAdded(static_cast<int>(instances_.size()), 1); | 169 observer_->OnItemsAdded(static_cast<int>(instances_.size()), 1); |
| 172 mojo::Array<mojo::String> names; | 170 mojo::Array<mojo::String> names; |
| 173 names.push_back(instance->identity->name); | 171 names.push_back(identity.name()); |
| 174 catalog_->GetEntries(std::move(names), | 172 catalog_->GetEntries(std::move(names), |
| 175 base::Bind(&TaskViewerContents::OnGotCatalogEntries, | 173 base::Bind(&TaskViewerContents::OnGotCatalogEntries, |
| 176 weak_ptr_factory_.GetWeakPtr())); | 174 weak_ptr_factory_.GetWeakPtr())); |
| 177 } | 175 } |
| 178 void OnServiceStarted(uint32_t id, uint32_t pid) override { | 176 void OnServiceStarted(shell::mojom::IdentityPtr identity_ptr, |
| 177 uint32_t pid) override { |
| 178 shell::Identity identity = identity_ptr.To<shell::Identity>(); |
| 179 for (auto it = instances_.begin(); it != instances_.end(); ++it) { | 179 for (auto it = instances_.begin(); it != instances_.end(); ++it) { |
| 180 if ((*it)->id == id) { | 180 if ((*it)->identity == identity) { |
| 181 (*it)->pid = pid; | 181 (*it)->pid = pid; |
| 182 observer_->OnItemsChanged( | 182 observer_->OnItemsChanged( |
| 183 static_cast<int>(it - instances_.begin()), 1); | 183 static_cast<int>(it - instances_.begin()), 1); |
| 184 return; | 184 return; |
| 185 } | 185 } |
| 186 } | 186 } |
| 187 } | 187 } |
| 188 void OnServiceStopped(uint32_t id) override { | 188 void OnServiceStopped(shell::mojom::IdentityPtr identity_ptr) override { |
| 189 shell::Identity identity = identity_ptr.To<shell::Identity>(); |
| 189 for (auto it = instances_.begin(); it != instances_.end(); ++it) { | 190 for (auto it = instances_.begin(); it != instances_.end(); ++it) { |
| 190 if ((*it)->id == id) { | 191 if ((*it)->identity == identity) { |
| 191 observer_->OnItemsRemoved( | 192 observer_->OnItemsRemoved( |
| 192 static_cast<int>(it - instances_.begin()), 1); | 193 static_cast<int>(it - instances_.begin()), 1); |
| 193 instances_.erase(it); | 194 instances_.erase(it); |
| 194 return; | 195 return; |
| 195 } | 196 } |
| 196 } | 197 } |
| 197 NOTREACHED(); | 198 NOTREACHED(); |
| 198 } | 199 } |
| 199 | 200 |
| 200 bool ContainsId(uint32_t id) const { | 201 bool ContainsIdentity(const shell::Identity& identity) const { |
| 201 for (auto& it : instances_) { | 202 for (auto& it : instances_) { |
| 202 if (it->id == id) | 203 if (it->identity == identity) |
| 203 return true; | 204 return true; |
| 204 } | 205 } |
| 205 return false; | 206 return false; |
| 206 } | 207 } |
| 207 | 208 |
| 208 void InsertInstance(uint32_t id, const std::string& url, uint32_t pid) { | 209 void InsertInstance(const shell::Identity& identity, uint32_t pid) { |
| 209 instances_.push_back(base::WrapUnique(new InstanceInfo(id, url, pid))); | 210 instances_.push_back( |
| 211 base::WrapUnique(new InstanceInfo(identity, pid))); |
| 210 } | 212 } |
| 211 | 213 |
| 212 void OnGotCatalogEntries(mojo::Array<catalog::mojom::EntryPtr> entries) { | 214 void OnGotCatalogEntries(mojo::Array<catalog::mojom::EntryPtr> entries) { |
| 213 for (auto it = instances_.begin(); it != instances_.end(); ++it) { | 215 for (auto it = instances_.begin(); it != instances_.end(); ++it) { |
| 214 for (auto& entry : entries) { | 216 for (auto& entry : entries) { |
| 215 if (entry->name == (*it)->url) { | 217 if (entry->name == (*it)->identity.name()) { |
| 216 (*it)->name = entry->display_name; | 218 (*it)->display_name = entry->display_name; |
| 217 observer_->OnItemsChanged( | 219 observer_->OnItemsChanged( |
| 218 static_cast<int>(it - instances_.begin()), 1); | 220 static_cast<int>(it - instances_.begin()), 1); |
| 219 break; | 221 break; |
| 220 } | 222 } |
| 221 } | 223 } |
| 222 } | 224 } |
| 223 } | 225 } |
| 224 | 226 |
| 225 static std::vector<ui::TableColumn> GetColumns() { | 227 static std::vector<ui::TableColumn> GetColumns() { |
| 226 std::vector<ui::TableColumn> columns; | 228 std::vector<ui::TableColumn> columns; |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 276 TaskViewer::~TaskViewer() {} | 278 TaskViewer::~TaskViewer() {} |
| 277 | 279 |
| 278 void TaskViewer::RemoveWindow(views::Widget* widget) { | 280 void TaskViewer::RemoveWindow(views::Widget* widget) { |
| 279 auto it = std::find(windows_.begin(), windows_.end(), widget); | 281 auto it = std::find(windows_.begin(), windows_.end(), widget); |
| 280 DCHECK(it != windows_.end()); | 282 DCHECK(it != windows_.end()); |
| 281 windows_.erase(it); | 283 windows_.erase(it); |
| 282 if (windows_.empty()) | 284 if (windows_.empty()) |
| 283 base::MessageLoop::current()->QuitWhenIdle(); | 285 base::MessageLoop::current()->QuitWhenIdle(); |
| 284 } | 286 } |
| 285 | 287 |
| 286 void TaskViewer::OnStart(shell::Connector* connector, | 288 void TaskViewer::OnStart(const shell::Identity& identity) { |
| 287 const shell::Identity& identity, | 289 tracing_.Initialize(connector(), identity.name()); |
| 288 uint32_t id) { | |
| 289 connector_ = connector; | |
| 290 tracing_.Initialize(connector, identity.name()); | |
| 291 | 290 |
| 292 aura_init_.reset(new views::AuraInit(connector, "views_mus_resources.pak")); | 291 aura_init_.reset( |
| 292 new views::AuraInit(connector(), "views_mus_resources.pak")); |
| 293 window_manager_connection_ = | 293 window_manager_connection_ = |
| 294 views::WindowManagerConnection::Create(connector, identity); | 294 views::WindowManagerConnection::Create(connector(), identity); |
| 295 } | 295 } |
| 296 | 296 |
| 297 bool TaskViewer::OnConnect(shell::Connection* connection) { | 297 bool TaskViewer::OnConnect(shell::Connection* connection) { |
| 298 connection->AddInterface<mojom::Launchable>(this); | 298 connection->AddInterface<mojom::Launchable>(this); |
| 299 return true; | 299 return true; |
| 300 } | 300 } |
| 301 | 301 |
| 302 void TaskViewer::Launch(uint32_t what, mojom::LaunchMode how) { | 302 void TaskViewer::Launch(uint32_t what, mojom::LaunchMode how) { |
| 303 bool reuse = how == mojom::LaunchMode::REUSE || | 303 bool reuse = how == mojom::LaunchMode::REUSE || |
| 304 how == mojom::LaunchMode::DEFAULT; | 304 how == mojom::LaunchMode::DEFAULT; |
| 305 if (reuse && !windows_.empty()) { | 305 if (reuse && !windows_.empty()) { |
| 306 windows_.back()->Activate(); | 306 windows_.back()->Activate(); |
| 307 return; | 307 return; |
| 308 } | 308 } |
| 309 | 309 |
| 310 shell::mojom::ServiceManagerPtr service_manager; | 310 shell::mojom::ServiceManagerPtr service_manager; |
| 311 connector_->ConnectToInterface("mojo:shell", &service_manager); | 311 connector()->ConnectToInterface("mojo:shell", &service_manager); |
| 312 | 312 |
| 313 shell::mojom::ServiceManagerListenerPtr listener; | 313 shell::mojom::ServiceManagerListenerPtr listener; |
| 314 shell::mojom::ServiceManagerListenerRequest request = GetProxy(&listener); | 314 shell::mojom::ServiceManagerListenerRequest request = GetProxy(&listener); |
| 315 service_manager->AddListener(std::move(listener)); | 315 service_manager->AddListener(std::move(listener)); |
| 316 | 316 |
| 317 catalog::mojom::CatalogPtr catalog; | 317 catalog::mojom::CatalogPtr catalog; |
| 318 connector_->ConnectToInterface("mojo:catalog", &catalog); | 318 connector()->ConnectToInterface("mojo:catalog", &catalog); |
| 319 | 319 |
| 320 TaskViewerContents* task_viewer = new TaskViewerContents( | 320 TaskViewerContents* task_viewer = new TaskViewerContents( |
| 321 this, std::move(request), std::move(catalog)); | 321 this, std::move(request), std::move(catalog)); |
| 322 views::Widget* window = views::Widget::CreateWindowWithContextAndBounds( | 322 views::Widget* window = views::Widget::CreateWindowWithContextAndBounds( |
| 323 task_viewer, nullptr, gfx::Rect(10, 10, 500, 500)); | 323 task_viewer, nullptr, gfx::Rect(10, 10, 500, 500)); |
| 324 window->Show(); | 324 window->Show(); |
| 325 windows_.push_back(window); | 325 windows_.push_back(window); |
| 326 } | 326 } |
| 327 | 327 |
| 328 void TaskViewer::Create(const shell::Identity& remote_identity, | 328 void TaskViewer::Create(const shell::Identity& remote_identity, |
| 329 mojom::LaunchableRequest request) { | 329 mojom::LaunchableRequest request) { |
| 330 bindings_.AddBinding(this, std::move(request)); | 330 bindings_.AddBinding(this, std::move(request)); |
| 331 } | 331 } |
| 332 | 332 |
| 333 } // namespace task_viewer | 333 } // namespace task_viewer |
| 334 } // namespace main | 334 } // namespace main |
| OLD | NEW |