Change Flyin VScode Default Extensions (flyteorg#2058)

Signed-off-by: Future Outlier <[email protected]> Co-authored-by: Future Outlier <[email protected]>
jasonlai1218 · Dec 21, 2023 · ad328a7 · ad328a7
1 parent 1087356
commit ad328a7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/config.py b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/config.py
@@ -34,8 +34,10 @@ def add_extensions(self, extensions: Union[str, List[str]]):
 COPILOT_EXTENSION = (
     "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/GitHub.copilot-1.138.563.vsix"
 )
-VIM_EXTENSION = "https://open-vsx.org/api/vscodevim/vim/1.27.0/file/vscodevim.vim-1.27.0.vsix"
-CODE_TOGETHER_EXTENSION = "https://openvsxorg.blob.core.windows.net/resources/genuitecllc/codetogether/2023.2.0/genuitecllc.codetogether-2023.2.0.vsix"
+VIM_EXTENSION = (
+    "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/vscodevim.vim-1.27.0.vsix"
+)
+CODE_TOGETHER_EXTENSION = "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/genuitecllc.codetogether-2023.2.0.vsix"
 
 # Predefined VSCode config with extensions
 VIM_CONFIG = VscodeConfig(

diff --git a/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/constants.py b/plugins/flytekit-flyin/flytekitplugins/flyin/vscode_lib/constants.py
@@ -10,8 +10,8 @@
     "arm64": "https://github.com/coder/code-server/releases/download/v4.18.0/code-server-4.18.0-linux-arm64.tar.gz",
 }
 DEFAULT_CODE_SERVER_EXTENSIONS = [
-    "https://open-vsx.org/api/ms-python/python/2023.20.0/file/ms-python.python-2023.20.0.vsix",
-    "https://open-vsx.org/api/ms-toolsai/jupyter/2023.9.100/file/ms-toolsai.jupyter-2023.9.100.vsix",
+    "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/ms-python.python-2023.20.0.vsix",
+    "https://raw.githubusercontent.com/flyteorg/flytetools/master/flytekitplugins/flyin/ms-toolsai.jupyter-2023.9.100.vsix",
 ]
 DEFAULT_CODE_SERVER_DIR_NAMES = {
     "amd64": "code-server-4.18.0-linux-amd64",

diff --git a/plugins/flytekit-flyin/tests/test_flyin_plugin.py b/plugins/flytekit-flyin/tests/test_flyin_plugin.py
@@ -237,6 +237,12 @@ def test_vscode_config():
     assert vim_config.code_server_dir_names == DEFAULT_CODE_SERVER_DIR_NAMES
     assert vim_config.extension_remote_paths == DEFAULT_CODE_SERVER_EXTENSIONS + [VIM_EXTENSION]
 
+    all_extensions_config = VscodeConfig()
+    all_extensions_config.add_extensions([CODE_TOGETHER_EXTENSION, COPILOT_EXTENSION, VIM_EXTENSION])
+    assert CODE_TOGETHER_EXTENSION in all_extensions_config.extension_remote_paths
+    assert COPILOT_EXTENSION in all_extensions_config.extension_remote_paths
+    assert VIM_EXTENSION in all_extensions_config.extension_remote_paths
+
 
 def test_vscode_config_add_extensions():
     additional_extensions = [COPILOT_EXTENSION, VIM_EXTENSION, CODE_TOGETHER_EXTENSION]