/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */
package org.apache.chemistry.opencmis.tck.report;

import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import org.apache.chemistry.opencmis.commons.SessionParameter;
import org.apache.chemistry.opencmis.commons.exceptions.CmisBaseException;
import org.apache.chemistry.opencmis.commons.impl.json.JSONArray;
import org.apache.chemistry.opencmis.commons.impl.json.JSONObject;
import org.apache.chemistry.opencmis.tck.CmisTest;
import org.apache.chemistry.opencmis.tck.CmisTestGroup;
import org.apache.chemistry.opencmis.tck.CmisTestResult;
import org.apache.chemistry.opencmis.tck.CmisTestResultStatus;

/**
 * JSON Report.
 */
public class JsonReport extends AbstractCmisTestReport {

    public JsonReport() {
    }

    @Override
    public void createReport(Map<String, String> parameters, List<CmisTestGroup> groups, Writer writer)
            throws IOException {

        JSONObject jsonReport = new JSONObject();

        JSONObject jsonParameters = new JSONObject();
        jsonReport.put("parameters", jsonParameters);

        if (parameters != null) {
            for (Map.Entry<String, String> p : (new TreeMap<String, String>(parameters)).entrySet()) {
                String value = p.getValue();
                if (SessionParameter.PASSWORD.endsWith(p.getKey())) {
                    value = "*****";
                }

                jsonParameters.put(p.getKey(), value);
            }
        }

        if (groups != null) {
            JSONArray jsonGroups = new JSONArray();
            jsonReport.put("groups", jsonGroups);

            for (CmisTestGroup group : groups) {
                printGroupResults(group, jsonGroups);
            }
        }

        jsonReport.writeJSONString(writer);
        writer.flush();
    }

    private void printGroupResults(CmisTestGroup group, JSONArray jsonGroups) throws IOException {
        if (!group.isEnabled()) {
            return;
        }

        JSONObject jsonGroup = new JSONObject();
        jsonGroups.add(jsonGroup);

        jsonGroup.put("name", group.getName());

        if (group.getTests() != null && !group.getTests().isEmpty()) {
            JSONArray jsonTests = new JSONArray();
            jsonGroup.put("tests", jsonTests);

            for (CmisTest test : group.getTests()) {
                printTestResults(test, jsonTests);
            }
        }
    }

    private void printTestResults(CmisTest test, JSONArray jsonTests) throws IOException {
        if (!test.isEnabled()) {
            return;
        }

        JSONObject jsonTest = new JSONObject();
        jsonTests.add(jsonTest);

        jsonTest.put("name", test.getName());
        jsonTest.put("time", test.getTime());

        if (test.getResults() != null && !test.getResults().isEmpty()) {
            JSONArray jsonResults = new JSONArray();
            jsonTest.put("results", jsonResults);

            for (CmisTestResult result : test.getResults()) {
                printResult(result, jsonResults);
            }
        }
    }

    private void printResult(CmisTestResult result, JSONArray results) throws IOException {
        JSONObject jsonResult = new JSONObject();
        results.add(jsonResult);

        jsonResult.put("status", result.getStatus().toString());
        jsonResult.put("message", result.getMessage());

        if (result.getStackTrace() != null && result.getStackTrace().length > 0) {
            jsonResult.put("file",
                    result.getStackTrace()[0].getFileName() + ":" + result.getStackTrace()[0].getLineNumber());
        }

        if (result.getStatus() == CmisTestResultStatus.UNEXPECTED_EXCEPTION && result.getException() != null) {
            StringWriter sw = new StringWriter();
            PrintWriter pw = new PrintWriter(sw);
            result.getException().printStackTrace(pw);

            jsonResult.put("stacktrace", sw.toString());

            if (result.getException() instanceof CmisBaseException) {
                CmisBaseException cbe = (CmisBaseException) result.getException();
                if (cbe.getErrorContent() != null) {
                    jsonResult.put("errorcontent", cbe.getErrorContent());
                }
            }
        }

        if (result.getException() != null) {
            jsonResult.put("exception", result.getException().getMessage());
        }

        if (result.getRequest() != null) {
            jsonResult.put("request", result.getRequest());
        }

        if (result.getRequest() != null) {
            jsonResult.put("response", result.getResponse());
        }

        if (!result.getChildren().isEmpty()) {
            JSONArray nextLevel = new JSONArray();
            jsonResult.put("results", nextLevel);

            for (CmisTestResult child : result.getChildren()) {
                printResult(child, nextLevel);
            }
        }
    }
}