Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 1 | // Copyright 2014 The Chromium OS 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 "buffet/dbus_command_proxy.h" |
| 6 | |
| 7 | #include <chromeos/dbus/async_event_sequencer.h> |
| 8 | #include <chromeos/dbus/exported_object_manager.h> |
Vitaly Buka | e2713ac | 2015-08-03 13:50:01 -0700 | [diff] [blame] | 9 | #include <weave/enum_to_string.h> |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 10 | |
| 11 | #include "buffet/dbus_conversion.h" |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 12 | #include "buffet/weave_error_conversion.h" |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 13 | |
| 14 | using chromeos::dbus_utils::AsyncEventSequencer; |
| 15 | using chromeos::dbus_utils::ExportedObjectManager; |
| 16 | |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 17 | namespace errors { |
| 18 | namespace commands { |
| 19 | const char kDomain[] = "weaved"; |
| 20 | const char kCommandDestroyed[] = "command_destroyed"; |
| 21 | } // namespace commands |
| 22 | } // namespace errors |
| 23 | |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 24 | namespace buffet { |
| 25 | |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 26 | namespace { |
| 27 | |
| 28 | bool ReportDestroyedError(chromeos::ErrorPtr* error) { |
| 29 | chromeos::Error::AddTo(error, FROM_HERE, errors::commands::kDomain, |
| 30 | errors::commands::kCommandDestroyed, |
| 31 | "Command has been destroyed"); |
| 32 | return false; |
| 33 | } |
| 34 | |
| 35 | } // anonymous namespace |
| 36 | |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 37 | DBusCommandProxy::DBusCommandProxy(ExportedObjectManager* object_manager, |
| 38 | const scoped_refptr<dbus::Bus>& bus, |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 39 | const std::weak_ptr<weave::Command>& command, |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 40 | std::string object_path) |
| 41 | : command_{command}, |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 42 | dbus_object_{object_manager, bus, dbus::ObjectPath{object_path}} {} |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 43 | |
| 44 | void DBusCommandProxy::RegisterAsync( |
| 45 | const AsyncEventSequencer::CompletionAction& completion_callback) { |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 46 | auto command = command_.lock(); |
| 47 | if (!command) |
| 48 | return; |
| 49 | |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 50 | dbus_adaptor_.RegisterWithDBusObject(&dbus_object_); |
| 51 | |
| 52 | // Set the initial property values before registering the DBus object. |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 53 | dbus_adaptor_.SetName(command->GetName()); |
| 54 | dbus_adaptor_.SetId(command->GetID()); |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 55 | dbus_adaptor_.SetState(EnumToString(command->GetState())); |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 56 | dbus_adaptor_.SetProgress( |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 57 | DictionaryToDBusVariantDictionary(*command->GetProgress())); |
| 58 | dbus_adaptor_.SetOrigin(EnumToString(command->GetOrigin())); |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 59 | dbus_adaptor_.SetParameters( |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 60 | DictionaryToDBusVariantDictionary(*command->GetParameters())); |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 61 | dbus_adaptor_.SetResults( |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 62 | DictionaryToDBusVariantDictionary(*command->GetResults())); |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 63 | |
| 64 | // Register the command DBus object and expose its methods and properties. |
| 65 | dbus_object_.RegisterAsync(completion_callback); |
| 66 | } |
| 67 | |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 68 | bool DBusCommandProxy::SetProgress( |
| 69 | chromeos::ErrorPtr* error, |
| 70 | const chromeos::VariantDictionary& progress) { |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 71 | auto command = command_.lock(); |
| 72 | if (!command) |
| 73 | return ReportDestroyedError(error); |
| 74 | |
| 75 | LOG(INFO) << "Received call to Command<" << command->GetName() |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 76 | << ">::SetProgress()"; |
| 77 | auto dictionary = DictionaryFromDBusVariantDictionary(progress, error); |
| 78 | if (!dictionary) |
| 79 | return false; |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 80 | weave::ErrorPtr weave_error; |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 81 | if (!command->SetProgress(*dictionary, &weave_error)) { |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 82 | ConvertError(*weave_error, error); |
| 83 | return false; |
| 84 | } |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 85 | dbus_adaptor_.SetProgress( |
| 86 | DictionaryToDBusVariantDictionary(*command->GetProgress())); |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 87 | dbus_adaptor_.SetState(EnumToString(command->GetState())); |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 88 | return true; |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 89 | } |
| 90 | |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 91 | bool DBusCommandProxy::Complete(chromeos::ErrorPtr* error, |
| 92 | const chromeos::VariantDictionary& results) { |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 93 | auto command = command_.lock(); |
| 94 | if (!command) |
| 95 | return ReportDestroyedError(error); |
| 96 | |
| 97 | LOG(INFO) << "Received call to Command<" << command->GetName() |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 98 | << ">::Complete()"; |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 99 | auto dictionary = DictionaryFromDBusVariantDictionary(results, error); |
| 100 | if (!dictionary) |
| 101 | return false; |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 102 | weave::ErrorPtr weave_error; |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 103 | if (!command->Complete(*dictionary, &weave_error)) { |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 104 | ConvertError(*weave_error, error); |
| 105 | return false; |
| 106 | } |
Vitaly Buka | baf7a95 | 2015-10-10 01:44:45 -0700 | [diff] [blame^] | 107 | dbus_adaptor_.SetResults( |
| 108 | DictionaryToDBusVariantDictionary(*command->GetResults())); |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 109 | dbus_adaptor_.SetState(EnumToString(command->GetState())); |
Vitaly Buka | 4f77153 | 2015-08-14 14:58:39 -0700 | [diff] [blame] | 110 | return true; |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 111 | } |
| 112 | |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 113 | bool DBusCommandProxy::Abort(chromeos::ErrorPtr* error, |
| 114 | const std::string& code, |
| 115 | const std::string& message) { |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 116 | auto command = command_.lock(); |
| 117 | if (!command) |
| 118 | return ReportDestroyedError(error); |
| 119 | |
| 120 | LOG(INFO) << "Received call to Command<" << command->GetName() |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 121 | << ">::Abort()"; |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 122 | weave::ErrorPtr cmd_error; |
| 123 | weave::Error::AddTo(&cmd_error, FROM_HERE, "command_error", code, message); |
| 124 | weave::ErrorPtr weave_error; |
| 125 | if (!command->Abort(cmd_error.get(), &weave_error)) { |
| 126 | ConvertError(*weave_error, error); |
| 127 | return false; |
| 128 | } |
| 129 | dbus_adaptor_.SetState(EnumToString(command->GetState())); |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 130 | return true; |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 131 | } |
| 132 | |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 133 | bool DBusCommandProxy::Cancel(chromeos::ErrorPtr* error) { |
| 134 | auto command = command_.lock(); |
| 135 | if (!command) |
| 136 | return ReportDestroyedError(error); |
| 137 | |
| 138 | LOG(INFO) << "Received call to Command<" << command->GetName() |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 139 | << ">::Cancel()"; |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 140 | weave::ErrorPtr weave_error; |
| 141 | if (!command->Cancel(&weave_error)) { |
| 142 | ConvertError(*weave_error, error); |
| 143 | return false; |
| 144 | } |
| 145 | dbus_adaptor_.SetState(EnumToString(command->GetState())); |
Alex Vakulenko | 2915a7b | 2015-10-07 17:04:00 -0700 | [diff] [blame] | 146 | return true; |
Vitaly Buka | a0305d3 | 2015-07-27 16:08:51 -0700 | [diff] [blame] | 147 | } |
| 148 | |
| 149 | } // namespace buffet |