diff options
author | wxb1ank <[email protected]> | 2021-05-24 03:37:10 +0100 |
---|---|---|
committer | wxb1ank <[email protected]> | 2021-06-02 17:11:32 +0100 |
commit | 0448b7364666ba59b39bbd5564fe8a34b67b8f01 (patch) | |
tree | 483dcccf2033eaba79bef5865b0d929da073413c /editors | |
parent | 3ca7f61a8d736d8e56f0d413b109e6e6562c0a30 (diff) |
migrate from `fs` to `vscode.FileSystem` API
Diffstat (limited to 'editors')
-rw-r--r-- | editors/code/src/main.ts | 30 | ||||
-rw-r--r-- | editors/code/src/net.ts | 30 | ||||
-rw-r--r-- | editors/code/src/toolchain.ts | 19 |
3 files changed, 35 insertions, 44 deletions
diff --git a/editors/code/src/main.ts b/editors/code/src/main.ts index fb9d38a14..f58d26215 100644 --- a/editors/code/src/main.ts +++ b/editors/code/src/main.ts | |||
@@ -1,7 +1,5 @@ | |||
1 | import * as vscode from 'vscode'; | 1 | import * as vscode from 'vscode'; |
2 | import * as path from "path"; | ||
3 | import * as os from "os"; | 2 | import * as os from "os"; |
4 | import { promises as fs, PathLike } from "fs"; | ||
5 | 3 | ||
6 | import * as commands from './commands'; | 4 | import * as commands from './commands'; |
7 | import { activateInlayHints } from './inlay_hints'; | 5 | import { activateInlayHints } from './inlay_hints'; |
@@ -222,7 +220,7 @@ async function bootstrapExtension(config: Config, state: PersistentState): Promi | |||
222 | const artifact = latestNightlyRelease.assets.find(artifact => artifact.name === "rust-analyzer.vsix"); | 220 | const artifact = latestNightlyRelease.assets.find(artifact => artifact.name === "rust-analyzer.vsix"); |
223 | assert(!!artifact, `Bad release: ${JSON.stringify(latestNightlyRelease)}`); | 221 | assert(!!artifact, `Bad release: ${JSON.stringify(latestNightlyRelease)}`); |
224 | 222 | ||
225 | const dest = path.join(config.globalStorageUri.path, "rust-analyzer.vsix"); | 223 | const dest = vscode.Uri.joinPath(config.globalStorageUri, "rust-analyzer.vsix"); |
226 | 224 | ||
227 | await downloadWithRetryDialog(state, async () => { | 225 | await downloadWithRetryDialog(state, async () => { |
228 | await download({ | 226 | await download({ |
@@ -233,8 +231,8 @@ async function bootstrapExtension(config: Config, state: PersistentState): Promi | |||
233 | }); | 231 | }); |
234 | }); | 232 | }); |
235 | 233 | ||
236 | await vscode.commands.executeCommand("workbench.extensions.installExtension", vscode.Uri.file(dest)); | 234 | await vscode.commands.executeCommand("workbench.extensions.installExtension", dest); |
237 | await fs.unlink(dest); | 235 | await vscode.workspace.fs.delete(dest); |
238 | 236 | ||
239 | await state.updateNightlyReleaseId(latestNightlyRelease.id); | 237 | await state.updateNightlyReleaseId(latestNightlyRelease.id); |
240 | await state.updateLastCheck(now); | 238 | await state.updateLastCheck(now); |
@@ -259,7 +257,7 @@ async function bootstrapServer(config: Config, state: PersistentState): Promise< | |||
259 | return path; | 257 | return path; |
260 | } | 258 | } |
261 | 259 | ||
262 | async function patchelf(dest: PathLike): Promise<void> { | 260 | async function patchelf(dest: vscode.Uri): Promise<void> { |
263 | await vscode.window.withProgress( | 261 | await vscode.window.withProgress( |
264 | { | 262 | { |
265 | location: vscode.ProgressLocation.Notification, | 263 | location: vscode.ProgressLocation.Notification, |
@@ -279,11 +277,11 @@ async function patchelf(dest: PathLike): Promise<void> { | |||
279 | ''; | 277 | ''; |
280 | } | 278 | } |
281 | `; | 279 | `; |
282 | const origFile = dest + "-orig"; | 280 | const origFile = vscode.Uri.file(dest.path + "-orig"); |
283 | await fs.rename(dest, origFile); | 281 | await vscode.workspace.fs.rename(dest, origFile); |
284 | progress.report({ message: "Patching executable", increment: 20 }); | 282 | progress.report({ message: "Patching executable", increment: 20 }); |
285 | await new Promise((resolve, reject) => { | 283 | await new Promise((resolve, reject) => { |
286 | const handle = exec(`nix-build -E - --argstr srcStr '${origFile}' -o '${dest}'`, | 284 | const handle = exec(`nix-build -E - --argstr srcStr '${origFile.path}' -o '${dest.path}'`, |
287 | (err, stdout, stderr) => { | 285 | (err, stdout, stderr) => { |
288 | if (err != null) { | 286 | if (err != null) { |
289 | reject(Error(stderr)); | 287 | reject(Error(stderr)); |
@@ -294,7 +292,7 @@ async function patchelf(dest: PathLike): Promise<void> { | |||
294 | handle.stdin?.write(expression); | 292 | handle.stdin?.write(expression); |
295 | handle.stdin?.end(); | 293 | handle.stdin?.end(); |
296 | }); | 294 | }); |
297 | await fs.unlink(origFile); | 295 | await vscode.workspace.fs.delete(origFile); |
298 | } | 296 | } |
299 | ); | 297 | ); |
300 | } | 298 | } |
@@ -334,20 +332,20 @@ async function getServer(config: Config, state: PersistentState): Promise<string | |||
334 | platform = "x86_64-unknown-linux-musl"; | 332 | platform = "x86_64-unknown-linux-musl"; |
335 | } | 333 | } |
336 | const ext = platform.indexOf("-windows-") !== -1 ? ".exe" : ""; | 334 | const ext = platform.indexOf("-windows-") !== -1 ? ".exe" : ""; |
337 | const dest = path.join(config.globalStorageUri.path, `rust-analyzer-${platform}${ext}`); | 335 | const dest = vscode.Uri.joinPath(config.globalStorageUri, `rust-analyzer-${platform}${ext}`); |
338 | const exists = await fs.stat(dest).then(() => true, () => false); | 336 | const exists = await vscode.workspace.fs.stat(dest).then(() => true, () => false); |
339 | if (!exists) { | 337 | if (!exists) { |
340 | await state.updateServerVersion(undefined); | 338 | await state.updateServerVersion(undefined); |
341 | } | 339 | } |
342 | 340 | ||
343 | if (state.serverVersion === config.package.version) return dest; | 341 | if (state.serverVersion === config.package.version) return dest.path; |
344 | 342 | ||
345 | if (config.askBeforeDownload) { | 343 | if (config.askBeforeDownload) { |
346 | const userResponse = await vscode.window.showInformationMessage( | 344 | const userResponse = await vscode.window.showInformationMessage( |
347 | `Language server version ${config.package.version} for rust-analyzer is not installed.`, | 345 | `Language server version ${config.package.version} for rust-analyzer is not installed.`, |
348 | "Download now" | 346 | "Download now" |
349 | ); | 347 | ); |
350 | if (userResponse !== "Download now") return dest; | 348 | if (userResponse !== "Download now") return dest.path; |
351 | } | 349 | } |
352 | 350 | ||
353 | const releaseTag = config.package.releaseTag; | 351 | const releaseTag = config.package.releaseTag; |
@@ -374,7 +372,7 @@ async function getServer(config: Config, state: PersistentState): Promise<string | |||
374 | } | 372 | } |
375 | 373 | ||
376 | await state.updateServerVersion(config.package.version); | 374 | await state.updateServerVersion(config.package.version); |
377 | return dest; | 375 | return dest.path; |
378 | } | 376 | } |
379 | 377 | ||
380 | function serverPath(config: Config): string | null { | 378 | function serverPath(config: Config): string | null { |
@@ -383,7 +381,7 @@ function serverPath(config: Config): string | null { | |||
383 | 381 | ||
384 | async function isNixOs(): Promise<boolean> { | 382 | async function isNixOs(): Promise<boolean> { |
385 | try { | 383 | try { |
386 | const contents = await fs.readFile("/etc/os-release"); | 384 | const contents = (await vscode.workspace.fs.readFile(vscode.Uri.file("/etc/os-release"))).toString(); |
387 | return contents.indexOf("ID=nixos") !== -1; | 385 | return contents.indexOf("ID=nixos") !== -1; |
388 | } catch (e) { | 386 | } catch (e) { |
389 | return false; | 387 | return false; |
diff --git a/editors/code/src/net.ts b/editors/code/src/net.ts index 07ebc615c..747c02db9 100644 --- a/editors/code/src/net.ts +++ b/editors/code/src/net.ts | |||
@@ -73,14 +73,14 @@ export interface GithubRelease { | |||
73 | assets: Array<{ | 73 | assets: Array<{ |
74 | name: string; | 74 | name: string; |
75 | // eslint-disable-next-line camelcase | 75 | // eslint-disable-next-line camelcase |
76 | browser_download_url: string; | 76 | browser_download_url: vscode.Uri; |
77 | }>; | 77 | }>; |
78 | } | 78 | } |
79 | 79 | ||
80 | interface DownloadOpts { | 80 | interface DownloadOpts { |
81 | progressTitle: string; | 81 | progressTitle: string; |
82 | url: string; | 82 | url: vscode.Uri; |
83 | dest: string; | 83 | dest: vscode.Uri; |
84 | mode?: number; | 84 | mode?: number; |
85 | gunzip?: boolean; | 85 | gunzip?: boolean; |
86 | httpProxy?: string; | 86 | httpProxy?: string; |
@@ -90,9 +90,9 @@ export async function download(opts: DownloadOpts) { | |||
90 | // Put artifact into a temporary file (in the same dir for simplicity) | 90 | // Put artifact into a temporary file (in the same dir for simplicity) |
91 | // to prevent partially downloaded files when user kills vscode | 91 | // to prevent partially downloaded files when user kills vscode |
92 | // This also avoids overwriting running executables | 92 | // This also avoids overwriting running executables |
93 | const dest = path.parse(opts.dest); | ||
94 | const randomHex = crypto.randomBytes(5).toString("hex"); | 93 | const randomHex = crypto.randomBytes(5).toString("hex"); |
95 | const tempFile = path.join(dest.dir, `${dest.name}${randomHex}`); | 94 | const rawDest = path.parse(opts.dest.path); |
95 | const tempFilePath = vscode.Uri.joinPath(vscode.Uri.file(rawDest.dir), `${rawDest.name}${randomHex}`); | ||
96 | 96 | ||
97 | await vscode.window.withProgress( | 97 | await vscode.window.withProgress( |
98 | { | 98 | { |
@@ -102,7 +102,7 @@ export async function download(opts: DownloadOpts) { | |||
102 | }, | 102 | }, |
103 | async (progress, _cancellationToken) => { | 103 | async (progress, _cancellationToken) => { |
104 | let lastPercentage = 0; | 104 | let lastPercentage = 0; |
105 | await downloadFile(opts.url, tempFile, opts.mode, !!opts.gunzip, opts.httpProxy, (readBytes, totalBytes) => { | 105 | await downloadFile(opts.url, tempFilePath, opts.mode, !!opts.gunzip, opts.httpProxy, (readBytes, totalBytes) => { |
106 | const newPercentage = Math.round((readBytes / totalBytes) * 100); | 106 | const newPercentage = Math.round((readBytes / totalBytes) * 100); |
107 | if (newPercentage !== lastPercentage) { | 107 | if (newPercentage !== lastPercentage) { |
108 | progress.report({ | 108 | progress.report({ |
@@ -116,12 +116,12 @@ export async function download(opts: DownloadOpts) { | |||
116 | } | 116 | } |
117 | ); | 117 | ); |
118 | 118 | ||
119 | await fs.promises.rename(tempFile, opts.dest); | 119 | await vscode.workspace.fs.rename(tempFilePath, opts.dest); |
120 | } | 120 | } |
121 | 121 | ||
122 | async function downloadFile( | 122 | async function downloadFile( |
123 | url: string, | 123 | url: vscode.Uri, |
124 | destFilePath: fs.PathLike, | 124 | destFilePath: vscode.Uri, |
125 | mode: number | undefined, | 125 | mode: number | undefined, |
126 | gunzip: boolean, | 126 | gunzip: boolean, |
127 | httpProxy: string | null | undefined, | 127 | httpProxy: string | null | undefined, |
@@ -129,15 +129,15 @@ async function downloadFile( | |||
129 | ): Promise<void> { | 129 | ): Promise<void> { |
130 | const res = await (() => { | 130 | const res = await (() => { |
131 | if (httpProxy) { | 131 | if (httpProxy) { |
132 | log.debug(`Downloading ${url} via proxy: ${httpProxy}`); | 132 | log.debug(`Downloading ${url.path} via proxy: ${httpProxy}`); |
133 | return fetch(url, { agent: new HttpsProxyAgent(httpProxy) }); | 133 | return fetch(url.path, { agent: new HttpsProxyAgent(httpProxy) }); |
134 | } | 134 | } |
135 | 135 | ||
136 | return fetch(url); | 136 | return fetch(url.path); |
137 | })(); | 137 | })(); |
138 | 138 | ||
139 | if (!res.ok) { | 139 | if (!res.ok) { |
140 | log.error("Error", res.status, "while downloading file from", url); | 140 | log.error("Error", res.status, "while downloading file from", url.path); |
141 | log.error({ body: await res.text(), headers: res.headers }); | 141 | log.error({ body: await res.text(), headers: res.headers }); |
142 | 142 | ||
143 | throw new Error(`Got response ${res.status} when trying to download a file.`); | 143 | throw new Error(`Got response ${res.status} when trying to download a file.`); |
@@ -146,7 +146,7 @@ async function downloadFile( | |||
146 | const totalBytes = Number(res.headers.get('content-length')); | 146 | const totalBytes = Number(res.headers.get('content-length')); |
147 | assert(!Number.isNaN(totalBytes), "Sanity check of content-length protocol"); | 147 | assert(!Number.isNaN(totalBytes), "Sanity check of content-length protocol"); |
148 | 148 | ||
149 | log.debug("Downloading file of", totalBytes, "bytes size from", url, "to", destFilePath); | 149 | log.debug("Downloading file of", totalBytes, "bytes size from", url.path, "to", destFilePath.path); |
150 | 150 | ||
151 | let readBytes = 0; | 151 | let readBytes = 0; |
152 | res.body.on("data", (chunk: Buffer) => { | 152 | res.body.on("data", (chunk: Buffer) => { |
@@ -154,7 +154,7 @@ async function downloadFile( | |||
154 | onProgress(readBytes, totalBytes); | 154 | onProgress(readBytes, totalBytes); |
155 | }); | 155 | }); |
156 | 156 | ||
157 | const destFileStream = fs.createWriteStream(destFilePath, { mode }); | 157 | const destFileStream = fs.createWriteStream(destFilePath.path, { mode }); |
158 | const srcStream = gunzip ? res.body.pipe(zlib.createGunzip()) : res.body; | 158 | const srcStream = gunzip ? res.body.pipe(zlib.createGunzip()) : res.body; |
159 | 159 | ||
160 | await pipeline(srcStream, destFileStream); | 160 | await pipeline(srcStream, destFileStream); |
diff --git a/editors/code/src/toolchain.ts b/editors/code/src/toolchain.ts index 68826c478..ba1b8617a 100644 --- a/editors/code/src/toolchain.ts +++ b/editors/code/src/toolchain.ts | |||
@@ -1,9 +1,8 @@ | |||
1 | import * as cp from 'child_process'; | 1 | import * as cp from 'child_process'; |
2 | import * as os from 'os'; | 2 | import * as os from 'os'; |
3 | import * as path from 'path'; | 3 | import * as path from 'path'; |
4 | import * as fs from 'fs'; | ||
5 | import * as readline from 'readline'; | 4 | import * as readline from 'readline'; |
6 | import { OutputChannel } from 'vscode'; | 5 | import * as vscode from 'vscode'; |
7 | import { execute, log, memoize } from './util'; | 6 | import { execute, log, memoize } from './util'; |
8 | 7 | ||
9 | interface CompilationArtifact { | 8 | interface CompilationArtifact { |
@@ -19,7 +18,7 @@ export interface ArtifactSpec { | |||
19 | } | 18 | } |
20 | 19 | ||
21 | export class Cargo { | 20 | export class Cargo { |
22 | constructor(readonly rootFolder: string, readonly output: OutputChannel) { } | 21 | constructor(readonly rootFolder: string, readonly output: vscode.OutputChannel) { } |
23 | 22 | ||
24 | // Made public for testing purposes | 23 | // Made public for testing purposes |
25 | static artifactSpec(args: readonly string[]): ArtifactSpec { | 24 | static artifactSpec(args: readonly string[]): ArtifactSpec { |
@@ -158,9 +157,9 @@ export const getPathForExecutable = memoize( | |||
158 | try { | 157 | try { |
159 | // hmm, `os.homedir()` seems to be infallible | 158 | // hmm, `os.homedir()` seems to be infallible |
160 | // it is not mentioned in docs and cannot be infered by the type signature... | 159 | // it is not mentioned in docs and cannot be infered by the type signature... |
161 | const standardPath = path.join(os.homedir(), ".cargo", "bin", executableName); | 160 | const standardPath = vscode.Uri.joinPath(vscode.Uri.file(os.homedir()), ".cargo", "bin", executableName); |
162 | 161 | ||
163 | if (isFile(standardPath)) return standardPath; | 162 | if (isFile(standardPath.path)) return standardPath.path; |
164 | } catch (err) { | 163 | } catch (err) { |
165 | log.error("Failed to read the fs info", err); | 164 | log.error("Failed to read the fs info", err); |
166 | } | 165 | } |
@@ -181,12 +180,6 @@ function lookupInPath(exec: string): boolean { | |||
181 | return candidates.some(isFile); | 180 | return candidates.some(isFile); |
182 | } | 181 | } |
183 | 182 | ||
184 | function isFile(suspectPath: string): boolean { | 183 | async function isFile(path: string): Promise<boolean> { |
185 | // It is not mentionned in docs, but `statSync()` throws an error when | 184 | return ((await vscode.workspace.fs.stat(vscode.Uri.file(path))).type & vscode.FileType.File) != 0; |
186 | // the path doesn't exist | ||
187 | try { | ||
188 | return fs.statSync(suspectPath).isFile(); | ||
189 | } catch { | ||
190 | return false; | ||
191 | } | ||
192 | } | 185 | } |