switch to ruff for better perf

packages/pyright-internal/src/languageServerBase.ts

@@ -76,7 +76,7 @@ import {
 import { ResultProgressReporter, attachWorkDone } from 'vscode-languageserver/lib/common/progress';
 
 import { TextDocument } from 'vscode-languageserver-textdocument';
-import { formatBufferWithYapf } from '../../pyright-yapf';
+import { formatBuffer } from '../../pyright-yapf';
 import { AnalysisResults } from './analyzer/analysis';
 import { BackgroundAnalysisProgram, InvalidatedReason } from './analyzer/backgroundAnalysisProgram';
 import { ImportResolver } from './analyzer/importResolver';
@@ -849,7 +849,7 @@ export abstract class LanguageServerBase implements LanguageServerInterface, Dis
 
         const buf = workspace.service.getSourceFile(filePath)?.getOpenFileContents();
         if (!buf) return;
-        return formatBufferWithYapf(buf, params.options.tabSize, !params.options.insertSpaces);
+        return formatBuffer(buf, params.options.tabSize, !params.options.insertSpaces);
     }
 
     protected async onDeclaration(

packages/pyright-yapf/index.ts

@@ -1,38 +1,25 @@
 import { TextEdit, uinteger } from 'vscode-languageserver';
-import { writeFile, open } from "fs/promises";
-import { interpreter, PyModule } from "node-calls-python";
-let yapf: PyModule;
-const YAPF_CONF = `[style]
-indent_width: `;
-const style_config = "/dev/shm/yapf.ini";
-(async()=>{yapf = await interpreter.import("yapf.yapflib.yapf_api", false);
-await writeFile(style_config, YAPF_CONF);})();
-async function _runYapf(buf: string, indentWidth: number, useTabs: boolean): Promise<string[]> {
-    const handle = await open(style_config, "a+");
-    await handle.truncate(YAPF_CONF.length);
-    await handle.write(`${indentWidth}`);
-    if (useTabs)
-        await handle.write(`
-use_tabs: true`);
-    return interpreter.call(yapf, "FormatCode", buf, {style_config, __kwargs: true}) as unknown as string[];
-}
-export async function formatBufferWithYapf(buf: string, indentWidth: number, useTabs: boolean): Promise<TextEdit[]> {
-    const [newText, changed] = await _runYapf(buf, indentWidth, useTabs);
-		const changes = [];
-		if (changed)
-			changes.push({
-					// range may seem sus but this is what the official ruff lsp actually does https://github.com/astral-sh/ruff-lsp/blob/main/ruff_lsp/server.py#L735-L740
-					range: {
-							start: {
-									line: 0,
-									character: 0,
-							},
-							end: {
-									line: uinteger.MAX_VALUE,
-									character: 0,
-							},
-					},
-					newText,
-      })
-    return changes;
+import { readFile } from "fs/promises";
+import init, { format } from "@wasm-fmt/ruff_fmt";
+/** @deprecated formatBuffer */
+export const formatBufferWithYapf = formatBuffer;
+export async function formatBuffer(buf: string, indent_width: number, useTabs: boolean): Promise<TextEdit[]> {
+    const newText = format(buf, "", {
+    indent_style: useTabs? "tab" : "space", indent_width });
+    return [
+        {
+            // range may seem sus but this is what the official ruff lsp actually does https://github.com/astral-sh/ruff-lsp/blob/main/ruff_lsp/server.py#L735-L740
+            range: {
+                start: {
+                    line: 0,
+                    character: 0,
+                },
+                end: {
+                    line: uinteger.MAX_VALUE,
+                    character: 0,
+                },
+            },
+            newText,
+        },
+    ];
 }

packages/pyright-yapf/package.json

@@ -3,8 +3,9 @@
     "version": "0.1.0",
     "private": true,
     "dependencies": {
-        "vscode-languageserver": "8.1.0",
+        "@wasm-fmt/ruff_fmt": "^0.5.0",
         "node-calls-python": "^1.8.0",
+        "vscode-languageserver": "8.1.0",
         "vscode-uri": "^3.0.7"
     },
     "files": [

packages/pyright-yapf/tsconfig.json

@@ -1,5 +1,6 @@
 {
     "extends": "../../tsconfig.json",
+    "target": "es2017",
     "compilerOptions": {
         "outDir": "./out"
     },

packages/pyright/langserver.index.js

@@ -4,5 +4,16 @@
 
 // Stash the base directory into a global variable.
 global.__rootDirectory = __dirname + '/dist/';
-
+const {readFile} = require("fs/promises");
+oldFetch = global.fetch;
+/**Workaround to get WASM loading to work*/
+global.fetch = async function(url) {
+  try {
+    let res = new Response(await readFile(url.pathname));
+    // node doesn't support passing headers in the options param?
+    res.headers.set("Content-Type", "application/wasm");
+    return res;
+  } catch {}
+  return oldFetch(url);
+};
 require('./dist/pyright-langserver');