frontend/client/src/autotest/moblab/rpc/MoblabRpcHelper.java

package autotest.moblab.rpc;

import com.google.gwt.json.client.JSONNumber;
import com.google.gwt.json.client.JSONObject;
import com.google.gwt.json.client.JSONString;
import com.google.gwt.json.client.JSONValue;

import autotest.common.JsonRpcCallback;
import autotest.common.JsonRpcProxy;
import autotest.common.SimpleCallback;

import com.google.gwt.user.client.Window;
import java.util.Map;

/**
 * A helper class for moblab RPC call.
 */
public class MoblabRpcHelper {
  public static final String RPC_PARAM_CLOUD_STORAGE_INFO = "cloud_storage_info";

  private MoblabRpcHelper() {}

  /**
   * Fetches config data.
   */
  public static void fetchConfigData(final SimpleCallback callback) {
    fetchConfigData(new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        if (callback != null)
          callback.doCallback(result);
      }
    });
  }

  /**
   * Fetch config data.
   */
  public static void fetchConfigData(final JsonRpcCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("get_config_values", null, callback);
  }

  /**
   * Submits config data.
   */
  public static void submitConfigData(JSONObject configValues, JsonRpcCallback callback) {
    JSONObject params = new JSONObject();
    params.put("config_values", configValues);
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("update_config_handler", params, callback);
  }

  /**
   * Resets config data on Moblab.
   */
  public static void resetConfigData(final JsonRpcCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("reset_config_settings", null, callback);
  }

  /**
   * Reboot Moblab device.
   */
  public static void rebootMoblab(final JsonRpcCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("reboot_moblab", null, callback);
  }

  /**
   * Fetches the server network information.
   */
  public static void fetchNetworkInfo(
      final MoblabRpcCallbacks.FetchNetworkInfoRpcCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("get_network_info", null, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        NetworkInfo networkInfo = new NetworkInfo();
        networkInfo.fromJson(result.isObject());
        callback.onNetworkInfoFetched(networkInfo);
      }
    });
  }

  /**
   * Fetches the cloud storage configuration information.
   */
  public static void fetchCloudStorageInfo(
      final MoblabRpcCallbacks.FetchCloudStorageInfoCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("get_cloud_storage_info", null, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        CloudStorageInfo info = new CloudStorageInfo();
        info.fromJson(result.isObject());
        callback.onCloudStorageInfoFetched(info);
      }
    });
  }

  /**
   * Validates the cloud storage configuration information.
   */
  public static void validateCloudStorageInfo(CloudStorageInfo cloudStorageInfo,
      final MoblabRpcCallbacks.ValidateCloudStorageInfoCallback callback) {
    JSONObject params = new JSONObject();
    params.put(RPC_PARAM_CLOUD_STORAGE_INFO, cloudStorageInfo.toJson());
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("validate_cloud_storage_info", params, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        OperationStatus status = new OperationStatus();
        status.fromJson(result.isObject());
        callback.onCloudStorageInfoValidated(status);
      }
    });
  }

  /**
   * Submits the wizard configuration data.
   */
  public static void submitWizardConfigData(Map<String, JSONObject> configDataMap,
      final MoblabRpcCallbacks.SubmitWizardConfigInfoCallback callback) {
    JSONObject params = new JSONObject();
    if (configDataMap.containsKey(RPC_PARAM_CLOUD_STORAGE_INFO)) {
      params.put(RPC_PARAM_CLOUD_STORAGE_INFO, configDataMap.get(RPC_PARAM_CLOUD_STORAGE_INFO));
    } else {
      params.put(RPC_PARAM_CLOUD_STORAGE_INFO, new JSONObject());
    }
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("submit_wizard_config_info", params, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        OperationStatus status = new OperationStatus();
        status.fromJson(result.isObject());
        callback.onWizardConfigInfoSubmitted(status);
      }
    });
  }

  /**
   * Fetches the version information.
   */
  public static void fetchMoblabBuildVersionInfo(
      final MoblabRpcCallbacks.FetchVersionInfoCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("get_version_info", null, new JsonRpcCallback() {
          @Override
          public void onSuccess(JSONValue result) {
            VersionInfo info = new VersionInfo();
            info.fromJson(result.isObject());
            callback.onVersionInfoFetched(info);
          }
        });
  }

   /**
   * Get information about the DUT's connected to the moblab.
   */
  public static void fetchDutInformation(
      final MoblabRpcCallbacks.FetchConnectedDutInfoCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    rpcProxy.rpcCall("get_connected_dut_info", null, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        ConnectedDutInfo info = new ConnectedDutInfo();
        info.fromJson(result.isObject());
        callback.onFetchConnectedDutInfoSubmitted(info);
      }
    });
  }

  /**
   * Enroll a device into the autotest syste.
   * @param dutIpAddress ipAddress of the new DUT.
   * @param callback Callback execute when the RPC is complete.
   */
  public static void addMoblabDut(String dutIpAddress,
      final MoblabRpcCallbacks.LogActionCompleteCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    JSONObject params = new JSONObject();
    params.put("ipaddress", new JSONString(dutIpAddress));
    rpcProxy.rpcCall("add_moblab_dut", params, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        boolean didSucceed = result.isArray().get(0).isBoolean().booleanValue();
        String information = result.isArray().get(1).isString().stringValue();
        callback.onLogActionComplete(didSucceed, information);
      }
    });
  }

  /**
   * Remove a device from the autotest system.
   * @param dutIpAddress ipAddress of the DUT to remove.
   * @param callback Callback to execute when the RPC is complete.
   */
  public static void removeMoblabDut(String dutIpAddress,
      final MoblabRpcCallbacks.LogActionCompleteCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    JSONObject params = new JSONObject();
    params.put("ipaddress", new JSONString(dutIpAddress));
    rpcProxy.rpcCall("remove_moblab_dut", params, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        boolean didSucceed = result.isArray().get(0).isBoolean().booleanValue();
        String information = result.isArray().get(1).isString().stringValue();
        callback.onLogActionComplete(didSucceed, information);
      }
    });
  }

  /**
   * Add a label to a specific DUT.
   * @param dutIpAddress  ipAddress of the device to have the new label applied.
   * @param labelName the label to apply
   * @param callback callback to execute when the RPC is complete.
   */
  public static void addMoblabLabel(String dutIpAddress, String labelName,
      final MoblabRpcCallbacks.LogActionCompleteCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    JSONObject params = new JSONObject();
    params.put("ipaddress", new JSONString(dutIpAddress));
    params.put("label_name", new JSONString(labelName));
    rpcProxy.rpcCall("add_moblab_label", params, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        boolean didSucceed = result.isArray().get(0).isBoolean().booleanValue();
        String information = result.isArray().get(1).isString().stringValue();
        callback.onLogActionComplete(didSucceed, information);
      }
    });
  }

  /**
   * Remove a label from a specific DUT.
   * @param dutIpAddress ipAddress of the device to have the label removed.
   * @param labelName the label to remove
   * @param callback callback to execute when the RPC is complete.
   */
  public static void removeMoblabLabel(String dutIpAddress, String labelName,
      final MoblabRpcCallbacks.LogActionCompleteCallback callback) {
    JsonRpcProxy rpcProxy = JsonRpcProxy.getProxy();
    JSONObject params = new JSONObject();
    params.put("ipaddress", new JSONString(dutIpAddress));
    params.put("label_name", new JSONString(labelName));
    rpcProxy.rpcCall("remove_moblab_label", params, new JsonRpcCallback() {
      @Override
      public void onSuccess(JSONValue result) {
        boolean didSucceed = result.isArray().get(0).isBoolean().booleanValue();
        String information = result.isArray().get(1).isString().stringValue();
        callback.onLogActionComplete(didSucceed, information);
      }
    });
  }
}