frontend/client/src/autotest/tko/embedded_spreadsheet/EmbeddedSpreadsheetPresenter.java

package autotest.tko.embedded_spreadsheet;

import autotest.common.JsonRpcCallback;
import autotest.common.JsonRpcProxy;
import autotest.common.spreadsheet.Spreadsheet;
import autotest.common.spreadsheet.Spreadsheet.CellInfo;
import autotest.common.spreadsheet.Spreadsheet.SpreadsheetListener;
import autotest.tko.HeaderField;
import autotest.tko.SpreadsheetDataProcessor;
import autotest.tko.TestGroupDataSource;
import autotest.tko.TestSet;
import autotest.tko.TkoSpreadsheetUtils;
import autotest.tko.TkoSpreadsheetUtils.DrilldownType;

import com.google.gwt.http.client.URL;
import com.google.gwt.json.client.JSONNumber;
import com.google.gwt.json.client.JSONObject;
import com.google.gwt.json.client.JSONValue;
import com.google.gwt.user.client.Command;
import com.google.gwt.user.client.Window;

import java.util.Collections;
import java.util.List;

public class EmbeddedSpreadsheetPresenter implements SpreadsheetListener {
    public interface Display {
        public void showNoResults();
        public void showSpreadsheet();
        public Spreadsheet getSpreadsheet();
        public Command getOnSpreadsheetRendered();
    }

    private static class SimpleHeaderField extends HeaderField {
        protected SimpleHeaderField(String name) {
            super(name, name);
        }

        @Override
        public String getSqlCondition(String value) {
            return getSimpleSqlCondition(name, value);
        }
    }

    public static final String ROW_HEADER = "hostname";
    public static final String COLUMN_HEADER = "test_name";
    public static final String DRILLDOWN_ROW_HEADER = "job_tag";
    public static final String DRILLDOWN_COLUMN_HEADER = "subdir";

    private List<HeaderField> rowHeader =
            Collections.singletonList((HeaderField) new SimpleHeaderField(ROW_HEADER));
    private List<HeaderField> columnHeader =
            Collections.singletonList((HeaderField) new SimpleHeaderField(COLUMN_HEADER));

    private String afeJobIdStr;
    private JSONObject condition;

    private Display display;

    public void bindDisplay(Display display) {
        this.display = display;
    }

    public void initialize(String afeJobIdStr) {
        JsonRpcProxy.setDefaultBaseUrl(JsonRpcProxy.TKO_BASE_URL);
        this.afeJobIdStr = afeJobIdStr;

        condition = getFilterCondition(afeJobIdStr);
        if (condition == null) {
            showNoResults();
        }

        JsonRpcProxy.getProxy().rpcCall("get_num_test_views", condition, new JsonRpcCallback() {
            @Override
            public void onSuccess(JSONValue result) {
                if (result.isNumber().doubleValue() != 0) {
                    renderSpreadsheet(condition);
                } else {
                    showNoResults();
                }
            }
        });
    }

    private JSONObject getFilterCondition(String afeJobIdStr) {
        if (afeJobIdStr == null) {
            return null;
        }

        int afeJobId;
        try {
            afeJobId = Integer.parseInt(afeJobIdStr);
        } catch (NumberFormatException e) {
            return null;
        }

        JSONObject condition = new JSONObject();
        condition.put("afe_job_id", new JSONNumber(afeJobId));
        return condition;
    }

    private void showNoResults() {
        display.showNoResults();
    }

    private void renderSpreadsheet(JSONObject condition) {
        display.showSpreadsheet();
        SpreadsheetDataProcessor spreadsheetProcessor =
                new SpreadsheetDataProcessor(display.getSpreadsheet());
        spreadsheetProcessor.setDataSource(TestGroupDataSource.getStatusCountDataSource());
        spreadsheetProcessor.setHeaders(rowHeader, columnHeader, new JSONObject());

        display.getSpreadsheet().setListener(this);
        spreadsheetProcessor.refresh(condition, display.getOnSpreadsheetRendered());
    }

    @Override
    public void onCellClicked(CellInfo cellInfo, boolean isRightClick) {
        TestSet testSet = TkoSpreadsheetUtils.getTestSet(
                cellInfo, condition, rowHeader, columnHeader);

        if (testSet.isSingleTest()) {
            openTestDetailView(testSet.getTestIndex());
        } else {
            openSpreadsheetView(testSet.getPartialSqlCondition(),
                    TkoSpreadsheetUtils.getDrilldownType(cellInfo));
        }
    }

    private void openTestDetailView(int testIdx) {
        openUrl("/new_tko/#tab_id=test_detail_view&object_id=" + testIdx);
    }

    private void openSpreadsheetView(String extraCondition, DrilldownType drilldownType) {
        String drilldownPath;

        switch (drilldownType) {
        case DRILLDOWN_ROW:
            drilldownPath = generatePath(DRILLDOWN_ROW_HEADER, COLUMN_HEADER, extraCondition);
            break;
        case DRILLDOWN_COLUMN:
            drilldownPath = generatePath(ROW_HEADER, DRILLDOWN_COLUMN_HEADER, extraCondition);
            break;
        case DRILLDOWN_BOTH:
            drilldownPath = generatePath(
                    DRILLDOWN_ROW_HEADER, DRILLDOWN_COLUMN_HEADER, extraCondition);
            break;
        default:
            throw new UnsupportedOperationException(
                    "DrilldownType " + drilldownType + " not supported");
        }

        openUrl(drilldownPath);
    }

    private String generatePath(String rowHeader, String columnHeader, String extraCondition) {
        String condition = "afe_job_id = " + afeJobIdStr + " AND " + extraCondition;

        return "/new_tko/#tab_id=spreadsheet_view&row=" + rowHeader + "&column=" + columnHeader +
                "&condition=" + URL.encodeComponent(condition, true) + "&show_incomplete=true";
    }

    private void openUrl(String url) {
        Window.open(url, "_blank", null);
    }
}