Function registerCodeLensCommands

lsp/src/codeLens.ts:322–345

(
  context: ExtensionContext,
  pythonEnv: PythonEnvironment,
)

}

export function registerCodeLensCommands(
  context: ExtensionContext,
  pythonEnv: PythonEnvironment,
): void {
  context.subscriptions.push(
    vscode.commands.registerCommand('pyrefly.runTest', async args => {
      const parsedArgs = parseRunTestArgs(args);
      if (!parsedArgs) {
        return;
      }
      await runTestAtLocation(parsedArgs, pythonEnv);
    }),
  );

  context.subscriptions.push(
    vscode.commands.registerCommand('pyrefly.runMain', async args => {
      const parsedArgs = parseRunMainArgs(args);
      if (!parsedArgs) {
        return;
      }
      await runMainFile(parsedArgs, pythonEnv);
    }),
  );
}