org.jboss.tools.vscode.java.internal.handlers.DiagnosticsHandler.java Source code

Introduction

Here is the source code for org.jboss.tools.vscode.java.internal.handlers.DiagnosticsHandler.java

Source

/*******************************************************************************
 * Copyright (c) 2016 Red Hat Inc. and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Red Hat Inc. - initial API and implementation
 *******************************************************************************/
package org.jboss.tools.vscode.java.internal.handlers;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.core.resources.IResource;
import org.eclipse.jdt.core.IProblemRequestor;
import org.eclipse.jdt.core.compiler.IProblem;
import org.eclipse.jdt.internal.compiler.problem.DefaultProblem;
import org.jboss.tools.langs.Diagnostic;
import org.jboss.tools.langs.Position;
import org.jboss.tools.langs.PublishDiagnosticsParams;
import org.jboss.tools.langs.Range;
import org.jboss.tools.langs.base.LSPMethods;
import org.jboss.tools.langs.base.NotificationMessage;
import org.jboss.tools.vscode.java.internal.JDTUtils;
import org.jboss.tools.vscode.java.internal.JavaClientConnection;
import org.jboss.tools.vscode.java.internal.JavaLanguageServerPlugin;

public class DiagnosticsHandler implements IProblemRequestor {

    private final List<IProblem> problems;
    private final IResource resource;
    private final JavaClientConnection connection;

    public DiagnosticsHandler(JavaClientConnection conn, IResource resource) {
        problems = new ArrayList<IProblem>();
        this.resource = resource;
        this.connection = conn;
    }

    @Override
    public void acceptProblem(IProblem problem) {
        JavaLanguageServerPlugin.logInfo("accept problem for " + this.resource.getName());
        problems.add(problem);
    }

    @Override
    public void beginReporting() {
        JavaLanguageServerPlugin.logInfo("begin problem for " + this.resource.getName());
        problems.clear();

    }

    @Override
    public void endReporting() {
        JavaLanguageServerPlugin.logInfo("end reporting for " + this.resource.getName());
        NotificationMessage<PublishDiagnosticsParams> message = new NotificationMessage<PublishDiagnosticsParams>();
        message.setMethod(LSPMethods.DOCUMENT_DIAGNOSTICS.getMethod());
        message.setParams(new PublishDiagnosticsParams().withUri(JDTUtils.getFileURI(this.resource))
                .withDiagnostics(toDiagnosticsArray()));
        this.connection.send(message);
    }

    private List<Diagnostic> toDiagnosticsArray() {
        List<Diagnostic> array = new ArrayList<Diagnostic>();
        for (IProblem problem : problems) {
            Diagnostic diag = new Diagnostic();
            diag.setSource("Java");
            diag.setMessage(problem.getMessage());
            diag.setCode(Integer.valueOf(problem.getID()));
            diag.setSeverity(Double.valueOf(convertSeverity(problem).doubleValue()));
            diag.setRange(convertRange(problem));
            array.add(diag);
        }
        return array;
    }

    private Integer convertSeverity(IProblem problem) {
        if (problem.isError())
            return new Integer(1);
        if (problem.isWarning())
            return new Integer(2);
        return new Integer(3);
    }

    private Range convertRange(IProblem problem) {
        Range range = new Range();
        Position start = new Position();
        Position end = new Position();

        start.setLine(Double.valueOf(problem.getSourceLineNumber() - 1));// VSCode is 0-based
        end.setLine(Double.valueOf(problem.getSourceLineNumber() - 1));
        if (problem instanceof DefaultProblem) {
            DefaultProblem dProblem = (DefaultProblem) problem;
            start.setCharacter(Double.valueOf(dProblem.getSourceColumnNumber() - 1));
            int offset = 0;
            if (dProblem.getSourceStart() != -1 && dProblem.getSourceEnd() != -1) {
                offset = dProblem.getSourceEnd() - dProblem.getSourceStart() + 1;
            }
            end.setCharacter(Double.valueOf(dProblem.getSourceColumnNumber() - 1 + offset));
        }
        return range.withEnd(end).withStart(start);
    }

    @Override
    public boolean isActive() {
        return true;
    }

}