pw_ide: Add command to build VSC extension

Change-Id: If91f7363b93074afab6ee5c16a5fcdd7547382cc Reviewed-on: https://pigweed-review.googlesource.com/c/pigweed/pigweed/+/184992 Commit-Queue: Chad Norvell <[email protected]> Reviewed-by: Anthony DiGirolamo <[email protected]>
google · Dec 14, 2023 · 82c82d9 · 82c82d9
1 parent 8ffa93f
commit 82c82d9
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 0 deletions.
diff --git a/pw_ide/py/pw_ide/cli.py b/pw_ide/py/pw_ide/cli.py
@@ -363,6 +363,11 @@ def _build_argument_parser() -> argparse.ArgumentParser:
         metavar='SETTINGS_TYPE',
         help='do not update these settings types',
     )
+    parser_vscode.add_argument(
+        '--build-extension',
+        action='store_true',
+        help='build the extension from source',
+    )
 
     return parser_root
 

diff --git a/pw_ide/py/pw_ide/commands.py b/pw_ide/py/pw_ide/commands.py
@@ -52,6 +52,7 @@
 
 from pw_ide import vscode
 from pw_ide.vscode import (
+    build_extension as build_vscode_extension,
     VscSettingsManager,
     VscSettingsType,
 )
@@ -142,6 +143,7 @@ def cmd_setup(
 def cmd_vscode(
     include: Optional[List[VscSettingsType]] = None,
     exclude: Optional[List[VscSettingsType]] = None,
+    build_extension: bool = False,
     reporter: StatusReporter = StatusReporter(),
     pw_ide_settings: PigweedIdeSettings = PigweedIdeSettings(),
 ) -> None:
@@ -200,6 +202,16 @@ def cmd_vscode(
     Likewise, it can be enabled by setting that value to true. It is enabled by
     default.
     """
+    if build_extension:
+        reporter.info('Building the Visual Studio Code extension')
+
+        try:
+            build_vscode_extension(Path(env.PW_ROOT))
+        except subprocess.CalledProcessError:
+            reporter.err("Failed! See output for more info.")
+        else:
+            reporter.ok('Built successfully!')
+
     if not pw_ide_settings.editor_enabled('vscode'):
         reporter.wrn(
             'Visual Studio Code support is disabled in settings! If this is '

diff --git a/pw_ide/py/pw_ide/vscode.py b/pw_ide/py/pw_ide/vscode.py
@@ -65,6 +65,8 @@
 import os
 from pathlib import Path
 import platform
+import shutil
+import subprocess
 from typing import Any, Dict, List, Optional, OrderedDict
 
 from pw_cli.env import pigweed_environment
@@ -424,3 +426,25 @@ class VscSettingsManager(EditorSettingsManager[VscSettingsType]):
         VscSettingsType.EXTENSIONS: _default_extensions,
         VscSettingsType.LAUNCH: _default_launch,
     }
+
+
+def build_extension(pw_root: Path):
+    """Build the VS Code extension."""
+
+    license_path = pw_root / 'LICENSE'
+    icon_path = pw_root.parent / 'icon.png'
+
+    vsc_ext_path = pw_root / 'pw_ide' / 'ts' / 'pigweed-vscode'
+    temp_license_path = vsc_ext_path / 'LICENSE'
+    temp_icon_path = vsc_ext_path / 'icon.png'
+
+    shutil.copy(license_path, temp_license_path)
+    shutil.copy(icon_path, temp_icon_path)
+
+    try:
+        subprocess.run(['vsce', 'package'], check=True, cwd=vsc_ext_path)
+    except subprocess.CalledProcessError as e:
+        raise e
+    finally:
+        temp_license_path.unlink()
+        temp_icon_path.unlink()