import * as vscode from 'vscode';
import * as path from "path";
import * as os from "os";
-import { promises as fs } from "fs";
+import { promises as fs, PathLike } from "fs";
import * as commands from './commands';
import { activateInlayHints } from './inlay_hints';
-import { activateStatusDisplay } from './status_display';
import { Ctx } from './ctx';
-import { activateHighlighting } from './highlighting';
import { Config, NIGHTLY_TAG } from './config';
-import { log, assert } from './util';
+import { log, assert, isValidExecutable } from './util';
import { PersistentState } from './persistent_state';
import { fetchRelease, download } from './net';
-import { spawnSync } from 'child_process';
import { activateTaskProvider } from './tasks';
+import { setContextValue } from './util';
+import { exec } from 'child_process';
let ctx: Ctx | undefined;
+const RUST_PROJECT_CONTEXT_NAME = "inRustProject";
+
export async function activate(context: vscode.ExtensionContext) {
// Register a "dumb" onEnter command for the case where server fails to
// start.
const config = new Config(context);
const state = new PersistentState(context.globalState);
- const serverPath = await bootstrap(config, state);
+ const serverPath = await bootstrap(config, state).catch(err => {
+ let message = "bootstrap error. ";
+
+ if (err.code === "EBUSY" || err.code === "ETXTBSY") {
+ message += "Other vscode windows might be using rust-analyzer, ";
+ message += "you should close them and reload this window to retry. ";
+ }
+
+ message += 'Open "Help > Toggle Developer Tools > Console" to see the logs ';
+ message += '(enable verbose logs with "rust-analyzer.trace.extension")';
+
+ log.error("Bootstrap error", err);
+ throw new Error(message);
+ });
const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
if (workspaceFolder === undefined) {
// This a horribly, horribly wrong way to deal with this problem.
ctx = await Ctx.create(config, context, serverPath, workspaceFolder.uri.fsPath);
+ setContextValue(RUST_PROJECT_CONTEXT_NAME, true);
+
// Commands which invokes manually via command palette, shortcut, etc.
// Reloading is inspired by @DanTup maneuver: https://github.com/microsoft/vscode/issues/45774#issuecomment-373423895
ctx.registerCommand('syntaxTree', commands.syntaxTree);
ctx.registerCommand('expandMacro', commands.expandMacro);
ctx.registerCommand('run', commands.run);
+ ctx.registerCommand('debug', commands.debug);
+ ctx.registerCommand('newDebugConfig', commands.newDebugConfig);
defaultOnEnter.dispose();
ctx.registerCommand('onEnter', commands.onEnter);
ctx.registerCommand('ssr', commands.ssr);
ctx.registerCommand('serverVersion', commands.serverVersion);
+ ctx.registerCommand('toggleInlayHints', commands.toggleInlayHints);
// Internal commands which are invoked by the server.
ctx.registerCommand('runSingle', commands.runSingle);
ctx.registerCommand('debugSingle', commands.debugSingle);
ctx.registerCommand('showReferences', commands.showReferences);
- ctx.registerCommand('applySourceChange', commands.applySourceChange);
- ctx.registerCommand('selectAndApplySourceChange', commands.selectAndApplySourceChange);
+ ctx.registerCommand('applySnippetWorkspaceEdit', commands.applySnippetWorkspaceEditCommand);
+ ctx.registerCommand('resolveCodeAction', commands.resolveCodeAction);
+ ctx.registerCommand('applyActionGroup', commands.applyActionGroup);
+ ctx.registerCommand('gotoLocation', commands.gotoLocation);
ctx.pushCleanup(activateTaskProvider(workspaceFolder));
- activateStatusDisplay(ctx);
-
- if (!ctx.config.highlightingSemanticTokens) {
- activateHighlighting(ctx);
- }
activateInlayHints(ctx);
vscode.workspace.onDidChangeConfiguration(
}
export async function deactivate() {
+ setContextValue(RUST_PROJECT_CONTEXT_NAME, undefined);
await ctx?.client.stop();
ctx = undefined;
}
assert(!!artifact, `Bad release: ${JSON.stringify(release)}`);
const dest = path.join(config.globalStoragePath, "rust-analyzer.vsix");
- await download(artifact.browser_download_url, dest, "Downloading rust-analyzer extension");
+ await download({
+ url: artifact.browser_download_url,
+ dest,
+ progressTitle: "Downloading rust-analyzer extension",
+ });
await vscode.commands.executeCommand("workbench.extensions.installExtension", vscode.Uri.file(dest));
await fs.unlink(dest);
);
}
- const res = spawnSync(path, ["--version"], { encoding: 'utf8' });
- log.debug("Checked binary availability via --version", res);
- log.debug(res, "--version output:", res.output);
- if (res.status !== 0) {
+ log.debug("Using server binary at", path);
+
+ if (!isValidExecutable(path)) {
throw new Error(`Failed to execute ${path} --version`);
}
return path;
}
+async function patchelf(dest: PathLike): Promise<void> {
+ await vscode.window.withProgress(
+ {
+ location: vscode.ProgressLocation.Notification,
+ title: "Patching rust-analyzer for NixOS"
+ },
+ async (progress, _) => {
+ const expression = `
+ {src, pkgs ? import <nixpkgs> {}}:
+ pkgs.stdenv.mkDerivation {
+ name = "rust-analyzer";
+ inherit src;
+ phases = [ "installPhase" "fixupPhase" ];
+ installPhase = "cp $src $out";
+ fixupPhase = ''
+ chmod 755 $out
+ patchelf --set-interpreter "$(cat $NIX_CC/nix-support/dynamic-linker)" $out
+ '';
+ }
+ `;
+ const origFile = dest + "-orig";
+ await fs.rename(dest, origFile);
+ progress.report({ message: "Patching executable", increment: 20 });
+ await new Promise((resolve, reject) => {
+ const handle = exec(`nix-build -E - --arg src '${origFile}' -o ${dest}`,
+ (err, stdout, stderr) => {
+ if (err != null) {
+ reject(Error(stderr));
+ } else {
+ resolve(stdout);
+ }
+ });
+ handle.stdin?.write(expression);
+ handle.stdin?.end();
+ });
+ await fs.unlink(origFile);
+ }
+ );
+}
+
async function getServer(config: Config, state: PersistentState): Promise<string | undefined> {
const explicitPath = process.env.__RA_LSP_SERVER_DEBUG ?? config.serverPath;
if (explicitPath) {
const artifact = release.assets.find(artifact => artifact.name === binaryName);
assert(!!artifact, `Bad release: ${JSON.stringify(release)}`);
- await download(artifact.browser_download_url, dest, "Downloading rust-analyzer server", { mode: 0o755 });
+ // Unlinking the exe file before moving new one on its place should prevent ETXTBSY error.
+ await fs.unlink(dest).catch(err => {
+ if (err.code !== "ENOENT") throw err;
+ });
+
+ await download({
+ url: artifact.browser_download_url,
+ dest,
+ progressTitle: "Downloading rust-analyzer server",
+ mode: 0o755
+ });
+
+ // Patching executable if that's NixOS.
+ if (await fs.stat("/etc/nixos").then(_ => true).catch(_ => false)) {
+ await patchelf(dest);
+ }
+
await state.updateServerVersion(config.package.version);
return dest;
}