frontend/client/src/autotest/afe/HostDataSource.java

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));
    }
}