[forgive] Init

Init basic LSP. At the moment the extension doesn't do anything interesting, but it does compile successfully.

.gitignore

@@ -37,3 +37,4 @@ packages/react-devtools-fusebox/dist
 packages/react-devtools-inline/dist
 packages/react-devtools-shell/dist
 packages/react-devtools-timeline/dist
+

compiler/.gitignore

@@ -21,4 +21,8 @@ dist
 .spr.yml
 testfilter.txt
 
-bundle-oss.sh
+bundle-oss.sh
+
+# forgive
+*.vsix
+.vscode-test

compiler/packages/react-forgive/.vscode-test.mjs

@@ -0,0 +1,3 @@
+import {defineConfig} from '@vscode/test-cli';
+
+export default defineConfig({files: 'dist/test/**/*.test.js'});

compiler/packages/react-forgive/.vscodeignore

@@ -0,0 +1,3 @@
+**/node_modules
+client
+server

compiler/packages/react-forgive/.yarnrc

@@ -0,0 +1 @@
+ignore-engines true

compiler/packages/react-forgive/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) Meta Platforms, Inc. and affiliates.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

compiler/packages/react-forgive/client/package.json

@@ -11,7 +11,7 @@
   "repository": {
     "type": "git",
     "url": "git+https://github.com/facebook/react.git",
-    "directory": "compiler/packages/react-forgive-client"
+    "directory": "compiler/packages/react-forgive"
   },
   "dependencies": {
     "vscode-languageclient": "^9.0.1"

compiler/packages/react-forgive/client/src/extension.ts

@@ -0,0 +1,62 @@
+import * as path from 'path';
+import {ExtensionContext, window as Window} from 'vscode';
+
+import {
+  LanguageClient,
+  LanguageClientOptions,
+  ServerOptions,
+  TransportKind,
+} from 'vscode-languageclient/node';
+
+let client: LanguageClient;
+
+export function activate(context: ExtensionContext) {
+  const serverModule = context.asAbsolutePath(path.join('dist', 'server.js'));
+
+  // If the extension is launched in debug mode then the debug server options are used
+  // Otherwise the run options are used
+  const serverOptions: ServerOptions = {
+    run: {
+      module: serverModule,
+      transport: TransportKind.ipc,
+      options: {cwd: process.cwd()},
+    },
+    debug: {
+      module: serverModule,
+      transport: TransportKind.ipc,
+      options: {cwd: process.cwd()},
+    },
+  };
+
+  const clientOptions: LanguageClientOptions = {
+    documentSelector: [
+      {scheme: 'file', language: 'javascriptreact'},
+      {scheme: 'file', language: 'typescriptreact'},
+    ],
+    progressOnInitialization: true,
+  };
+
+  // Create the language client and start the client.
+  try {
+    client = new LanguageClient(
+      'react-forgive',
+      'React Analyzer',
+      serverOptions,
+      clientOptions,
+    );
+  } catch {
+    Window.showErrorMessage(
+      `React Analyzer couldn't be started. See the output channel for details.`,
+    );
+    return;
+  }
+
+  client.registerProposedFeatures();
+  client.start();
+}
+
+export function deactivate(): Thenable<void> | undefined {
+  if (client !== undefined) {
+    return client.stop();
+  }
+}

compiler/packages/react-forgive/package.json

@@ -35,25 +35,30 @@
     ]
   },
   "scripts": {
-    "compile": "yarn run esbuild-base -- --sourcemap",
+    "build": "yarn run compile",
+    "compile": "rimraf dist && concurrently -n server,client \"yarn run esbuild:server --sourcemap\" \"yarn run esbuild:client --sourcemap\"",
     "dev": "yarn run package && yarn run install-ext",
-    "esbuild-base": "esbuild ./src/extension.ts --bundle --outfile=dist/extension.js --external:vscode --format=cjs --platform=node",
-    "install-ext": "code --install-extension vscode-react-compiler-0.0.1.vsix",
-    "lint": "eslint src --ext ts",
-    "package": "vsce package",
+    "esbuild:client": "esbuild ./client/src/extension.ts --bundle --outfile=dist/extension.js --external:vscode --format=cjs --platform=node",
+    "esbuild:server": "esbuild ./server/src/index.ts --bundle --outfile=dist/server.js --external:vscode --format=cjs --platform=node",
+    "install-ext": "code --install-extension react-forgive-0.0.0.vsix",
+    "lint": "echo 'no tests'",
+    "package": "rm -f react-forgive-0.0.0.vsix && vsce package --yarn",
     "postinstall": "cd client && yarn install && cd ../server && yarn install && cd ..",
     "pretest": "yarn run compile && yarn run lint",
     "test": "vscode-test",
-    "test-compile": "tsc -p ./",
-    "vscode:prepublish": "yarn run esbuild-base -- --minify",
-    "watch": "yarn run esbuild-base -- --sourcemap --watch"
+    "vscode:prepublish": "yarn run compile",
+    "watch": "concurrently --kill-others -n server,client \"run esbuild:server --sourcemap --watch\" \"run esbuild:client --sourcemap --watch\""
   },
   "devDependencies": {
     "@eslint/js": "^9.13.0",
+    "@types/mocha": "^10.0.10",
     "@types/node": "^20",
+    "@types/vscode": "^1.96.0",
+    "@vscode/test-cli": "^0.0.10",
+    "@vscode/test-electron": "^2.4.1",
     "esbuild": "^0.24.0",
     "eslint": "^9.13.0",
-    "typescript": "^5.7.2",
+    "mocha": "^11.0.1",
     "typescript-eslint": "^8.16.0"
   }
 }

compiler/packages/react-forgive/server/package.json

@@ -4,16 +4,23 @@
   "version": "0.0.0",
   "description": "Experimental LSP server",
   "license": "MIT",
+  "main": "dist/index.js",
   "scripts": {
-    "build": "echo 'no build'",
+    "build": "rimraf dist && rollup --config --bundleConfigAsCjs",
     "test": "echo 'no tests'"
   },
   "repository": {
     "type": "git",
     "url": "git+https://github.com/facebook/react.git",
-    "directory": "compiler/packages/react-forgive-server"
+    "directory": "compiler/packages/react-forgive"
   },
   "dependencies": {
+    "@babel/core": "^7.26.0",
+    "@babel/parser": "^7.26.0",
+    "@babel/plugin-syntax-typescript": "^7.25.9",
+    "@babel/types": "^7.26.0",
+    "cosmiconfig": "^9.0.0",
+    "prettier": "^3.3.3",
     "vscode-languageserver": "^9.0.1",
     "vscode-languageserver-textdocument": "^1.0.12"
   }

compiler/packages/react-forgive/server/src/LRU.ts

@@ -0,0 +1,41 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+export default class LRU<TKey, TValue> {
+  #size: number;
+  #cache: Map<TKey, TValue> = new Map();
+
+  constructor(size: number) {
+    this.#size = size;
+  }
+
+  get(key: TKey): TValue | void {
+    let item = this.#cache.get(key);
+    if (item !== undefined) {
+      this.#cache.delete(key);
+      this.#cache.set(key, item);
+    }
+    return item;
+  }
+
+  set(key: TKey, val: TValue): void {
+    if (this.#cache.has(key)) {
+      this.#cache.delete(key);
+    } else if (this.#cache.size >= this.#size) {
+      this.#cache.delete(this.first());
+    }
+    this.#cache.set(key, val);
+  }
+
+  clear(): void {
+    this.#cache.clear();
+  }
+
+  first(): TKey {
+    return this.#cache.keys().next().value;
+  }
+}

compiler/packages/react-forgive/server/src/compiler/index.ts

@@ -0,0 +1,58 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+import type * as BabelCore from '@babel/core';
+import {parseAsync, transformFromAstAsync} from '@babel/core';
+import BabelPluginReactCompiler, {
+  type PluginOptions,
+} from 'babel-plugin-react-compiler/src';
+import * as babelParser from 'prettier/plugins/babel.js';
+import estreeParser from 'prettier/plugins/estree';
+import * as typescriptParser from 'prettier/plugins/typescript';
+import * as prettier from 'prettier/standalone';
+
+type CompileOptions = {
+  text: string;
+  file: string;
+  options: PluginOptions | null;
+};
+export async function compile({
+  text,
+  file,
+  options,
+}: CompileOptions): Promise<BabelCore.BabelFileResult> {
+  const ast = await parseAsync(text, {
+    sourceFileName: file,
+    parserOpts: {
+      plugins: ['typescript', 'jsx'],
+    },
+    sourceType: 'module',
+  });
+  const plugins =
+    options != null
+      ? [[BabelPluginReactCompiler, options]]
+      : [[BabelPluginReactCompiler]];
+  const result = await transformFromAstAsync(ast, text, {
+    filename: file,
+    highlightCode: false,
+    retainLines: true,
+    plugins,
+    sourceType: 'module',
+    sourceFileName: file,
+  });
+  if (result?.code == null) {
+    throw new Error(
+      `Expected BabelPluginReactCompiler to compile successfully, got ${result}`,
+    );
+  }
+  result.code = await prettier.format(result.code, {
+    semi: false,
+    parser: 'babel-ts',
+    plugins: [babelParser, estreeParser, typescriptParser],
+  });
+  return result;
+}

compiler/packages/react-forgive/server/src/compiler/options.ts

@@ -0,0 +1,25 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+import {
+  parsePluginOptions,
+  type PluginOptions,
+} from 'babel-plugin-react-compiler/src';
+import {cosmiconfigSync} from 'cosmiconfig';
+
+export function resolveReactConfig(projectPath: string): PluginOptions | null {
+  const explorerSync = cosmiconfigSync('react', {
+    searchStrategy: 'project',
+    cache: true,
+  });
+  const result = explorerSync.search(projectPath);
+  if (result != null) {
+    return parsePluginOptions(result.config);
+  } else {
+    return null;
+  }
+}