org.eclipse.corrosion/src/org/eclipse/corrosion/RLSClientImplementation.java

/*********************************************************************
 * Copyright (c) 2017, 2018 Red Hat Inc. and others.
 *
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *  Lucas Bullen (Red Hat Inc.) - Initial implementation
 *******************************************************************************/
package org.eclipse.corrosion;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.SubMonitor;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.lsp4e.LanguageClientImpl;
import org.eclipse.lsp4j.jsonrpc.services.JsonNotification;

@SuppressWarnings("restriction")
public class RLSClientImplementation extends LanguageClientImpl {

	private static Job diagnosticsJob;
	private String status;

	@JsonNotification("rustDocument/beginBuild")
	public void beginBuild() {
		status = "Building project";
		initializeJob();
	}

	@JsonNotification("rustDocument/diagnosticsBegin")
	public void diagnosticsBegin() {
		status = "Compiling diagnostics";
		initializeJob();
	}

	@JsonNotification("rustDocument/diagnosticsEnd")
	public void diagnosticsEnd() {
		status = null;
		if (diagnosticsJob != null) {
			diagnosticsJob.done(new Status(IStatus.OK, CorrosionPlugin.PLUGIN_ID, null));
			diagnosticsJob = null;
		}
	}

	private void initializeJob() {
		if (diagnosticsJob == null) {
			diagnosticsJob = Job.create("Compiling Rust project diagnostics", monitor -> {
				SubMonitor subMonitor = SubMonitor.convert(monitor, 2);
				subMonitor.worked(1);
				try {
					int maxWaitTime = 60000; // 1 minute
					subMonitor.beginTask("Building project", IProgressMonitor.UNKNOWN);
					while (maxWaitTime > 0 && !subMonitor.isCanceled()) {
						if (status == null || subMonitor.isCanceled()) {
							break;
						} else {
							subMonitor.subTask(status);
						}
						Thread.sleep(50);
						maxWaitTime -= 50;
					}
					diagnosticsJob = null;
				} catch (InterruptedException e) {
					// Exception ends the job
				}
			});
			diagnosticsJob.schedule();
		}
	}
}