host/libs/config/custom_actions.cpp

/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include "host/libs/config/custom_actions.h"

#include <android-base/logging.h>
#include <json/json.h>

#include <optional>
#include <string>
#include <vector>

#include "host/libs/config/cuttlefish_config.h"

namespace cuttlefish {
namespace {

const char* kCustomActionShellCommand = "shell_command";
const char* kCustomActionServer = "server";
const char* kCustomActionButton = "button";
const char* kCustomActionButtons = "buttons";
const char* kCustomActionButtonCommand = "command";
const char* kCustomActionButtonTitle = "title";
const char* kCustomActionButtonIconName = "icon_name";

} //namespace


CustomActionConfig::CustomActionConfig(const Json::Value& dictionary) {
  if (dictionary.isMember(kCustomActionShellCommand)) {
    if (dictionary.isMember(kCustomActionServer)) {
      LOG(ERROR) << "Custom action contains both shell command and action server.";
      return;
    }
    // Shell command with one button.
    Json::Value button_entry = dictionary[kCustomActionButton];
    buttons = {{button_entry[kCustomActionButtonCommand].asString(),
                button_entry[kCustomActionButtonTitle].asString(),
                button_entry[kCustomActionButtonIconName].asString()}};
    shell_command = dictionary[kCustomActionShellCommand].asString();
  } else if (dictionary.isMember(kCustomActionServer)) {
    // Action server with possibly multiple buttons.
    for (const Json::Value& button_entry : dictionary[kCustomActionButtons]) {
      ControlPanelButton button = {
          button_entry[kCustomActionButtonCommand].asString(),
          button_entry[kCustomActionButtonTitle].asString(),
          button_entry[kCustomActionButtonIconName].asString()};
      buttons.push_back(button);
    }
    server = dictionary[kCustomActionServer].asString();
  } else {
    LOG(ERROR) << "Unknown custom action format.";
  }
}

Json::Value CustomActionConfig::ToJson() const {
  Json::Value custom_action;
  if (shell_command) {
    // Shell command with one button.
    custom_action[kCustomActionShellCommand] = *shell_command;
    custom_action[kCustomActionButton] = Json::Value();
    custom_action[kCustomActionButton][kCustomActionButtonCommand] =
        buttons[0].command;
    custom_action[kCustomActionButton][kCustomActionButtonTitle] =
        buttons[0].title;
    custom_action[kCustomActionButton][kCustomActionButtonIconName] =
        buttons[0].icon_name;
  } else if (server) {
    // Action server with possibly multiple buttons.
    custom_action[kCustomActionServer] = *server;
    custom_action[kCustomActionButtons] = Json::Value(Json::arrayValue);
    for (const auto& button : buttons) {
      Json::Value button_entry;
      button_entry[kCustomActionButtonCommand] = button.command;
      button_entry[kCustomActionButtonTitle] = button.title;
      button_entry[kCustomActionButtonIconName] = button.icon_name;
      custom_action[kCustomActionButtons].append(button_entry);
    }
  } else {
    LOG(ERROR) << "Unknown custom action type.";
  }
  return custom_action;
}

}  // namespace cuttlefish