package autotest.afe;
import autotest.afe.models.Host;
import autotest.common.JSONArrayList;
import autotest.common.Utils;
import autotest.common.table.RpcDataSource;
import com.google.gwt.json.client.JSONArray;
import com.google.gwt.json.client.JSONObject;
import com.google.gwt.json.client.JSONString;
import com.google.gwt.json.client.JSONValue;
import java.util.ArrayList;
import java.util.List;
public class HostDataSource extends RpcDataSource {
protected static final String LOCKED_TEXT = "locked_text";
protected static final String OTHER_LABELS = "other_labels";
protected static final String HOST_ACLS = "host_acls";
public HostDataSource() {
super("get_hosts", "get_num_hosts");
}
@Override
/**
* Convert the raw JSONObjects to Hosts.
*/
protected List<JSONObject> handleJsonResult(JSONValue result) {
List<JSONObject> resultList = super.handleJsonResult(result);
List<JSONObject> hosts = new ArrayList<JSONObject>();
for (JSONObject row : resultList) {
Host host = Host.fromJsonObject(row);
processHost(host);
hosts.add(host);
}
return hosts;
}
protected void processHost(JSONObject host) {
host.put(LOCKED_TEXT, AfeUtils.getLockedText(host));
JSONString jsonPlatform = host.get("platform").isString();
String platform = "";
if (jsonPlatform != null) {
platform = jsonPlatform.stringValue();
}
JSONArray labels = host.get("labels").isArray();
StringBuilder labelString = new StringBuilder();
for (int i = 0; i < labels.size(); i++) {
String label = labels.get(i).isString().stringValue();
if (label.equals(platform)) {
continue;
}
if (labelString.length() > 0) {
labelString.append(", ");
}
labelString.append(label);
}
host.put(OTHER_LABELS, new JSONString(labelString.toString()));
JSONArrayList<JSONString> aclsList =
new JSONArrayList<JSONString>(host.get("acls").isArray());
String aclString = Utils.joinStrings(",", aclsList);
host.put(HOST_ACLS, new JSONString(aclString));
}
}