org.sonarsource.sonarlint.omnisharp.protocol.OmnisharpResponseProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sonarlint-omnisharp-plugin Show documentation
Code Analyzer based on Omnisharp
There is a newer version: 1.26.0.100277
Show newest version
/*
 * SonarOmnisharp
 * Copyright (C) 2021-2024 SonarSource SA
 * mailto:info AT sonarsource DOT com
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package org.sonarsource.sonarlint.omnisharp.protocol;

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CountDownLatch;
import org.sonar.api.utils.log.Logger;
import org.sonar.api.utils.log.Loggers;
import org.sonarsource.api.sonarlint.SonarLintSide;


@SonarLintSide(lifespan = "MODULE")
public class OmnisharpResponseProcessor {

  private static final Logger LOG = Loggers.get(OmnisharpResponseProcessor.class);

  private final ConcurrentHashMap responseLatchQueue = new ConcurrentHashMap<>();

  public void handleOmnisharpOutput(CompletableFuture startFuture, CompletableFuture loadProjectsFuture, String line) {
    JsonObject jsonObject;
    try {
      jsonObject = JsonParser.parseString(line).getAsJsonObject();
    } catch (Exception e) {
      LOG.debug(line);
      return;
    }
    handleJsonMessage(startFuture, loadProjectsFuture, line, jsonObject);
  }

  private void handleJsonMessage(CompletableFuture startFuture, CompletableFuture loadProjectsFuture, String line, JsonObject jsonObject) {
    String type = jsonObject.get("Type").getAsString();
    switch (type) {
      case "response":
        long reqSeq = jsonObject.get("Request_seq").getAsLong();
        OmnisharpResponseHandler omnisharpResponseHandler = responseLatchQueue.get(reqSeq);
        if (omnisharpResponseHandler != null) {
          omnisharpResponseHandler.response = jsonObject;
          omnisharpResponseHandler.responseLatch.countDown();
        }
        break;
      case "event":
        String eventType = jsonObject.get("Event").getAsString();
        switch (eventType) {
          case "log":
            handleLog(jsonObject.get("Body").getAsJsonObject());
            break;
          case "started":
            LOG.debug(line);
            startFuture.complete(null);
            break;
          case "ProjectAdded":
          case "ProjectChanged":
          case "ProjectRemoved":
            LOG.debug(line);
            loadProjectsFuture.complete(null);
            break;
          case "Diagnostic":
            // For now we ignore diagnostics "pushed" by Omnisharp
            break;
          case "MsBuildProjectDiagnostics":
            var msbuildErrors = jsonObject.get("Body").getAsJsonObject().get("Errors").getAsJsonArray();
            if (!msbuildErrors.isEmpty()) {
              LOG.error("MSBuild failed to load the project");
              // No need to wait for project loading, it might never happen
              // firstUpdateProjectLatch.countDown();
            }
            LOG.debug(line);
            break;
          default:
            LOG.debug(line);
        }
        break;
      default:
        LOG.debug(line);
    }
  }

  private static void handleLog(JsonObject jsonObject) {
    String level = jsonObject.get("LogLevel").getAsString();
    String message = jsonObject.get("Message").getAsString();
    LOG.debug("Omnisharp: [" + level + "] " + message);
  }

  static class OmnisharpResponseHandler {
    JsonObject response;
    CountDownLatch responseLatch = new CountDownLatch(1);
  }

  public OmnisharpResponseHandler registerResponseHandler(long id) {
    OmnisharpResponseHandler omnisharpResponseHandler = new OmnisharpResponseHandler();
    responseLatchQueue.put(id, omnisharpResponseHandler);
    return omnisharpResponseHandler;
  }

  public void removeResponseHandler(long id) {
    responseLatchQueue.remove(id);
  }

}