ensure pio-home gets installed (#21)

Author: Jason2866

platformio/home/cli.py

@@ -20,6 +20,7 @@
 
 from platformio.compat import IS_WINDOWS
 from platformio.home.run import run_server
+from platformio.package.manager.core import get_core_package_dir
 from platformio.project.config import ProjectConfig
 
 
@@ -53,11 +54,13 @@
 def cli(port, host, no_open, shutdown_timeout, session_id):
     # hook for `platformio-node-helpers`
     if host == "__do_not_start__":
-        # download all dependent packages
-        os.path.join(
-            ProjectConfig.get_instance().get("platformio","packages_dir"),
-            "contrib-piohome"
-        )
+        # Ensure contrib-piohome package is installed
+        try:
+            get_core_package_dir("contrib-piohome")
+        except Exception as exc:  # pylint: disable=broad-except
+            click.secho(
+                "Warning! Could not install contrib-piohome: %s" % exc, fg="yellow"
+            )
         return
 
     # Ensure PIO Home mimetypes are known

platformio/home/run.py

@@ -36,6 +36,7 @@
 from platformio.home.rpc.handlers.project import ProjectRPC
 from platformio.home.rpc.handlers.registry import RegistryRPC
 from platformio.home.rpc.server import WebSocketJSONRPCServerFactory
+from platformio.package.manager.core import get_core_package_dir
 from platformio.project.config import ProjectConfig
 from platformio.proc import force_exit
 
@@ -62,10 +63,12 @@ async def protected_page(_):
 
 
 def run_server(host, port, no_open, shutdown_timeout, home_url):
-    packages_dir = ProjectConfig.get_instance().get("platformio", "packages_dir")
-    contrib_dir = os.path.join(packages_dir, "contrib-piohome")
+    # Ensure contrib-piohome is installed before starting the server
+    contrib_dir = get_core_package_dir("contrib-piohome")
     if not os.path.isdir(contrib_dir):
-        raise PlatformioException("Invalid path to PIO Home Contrib")
+        raise PlatformioException(
+            "Invalid path to PIO Home Contrib: %s" % contrib_dir
+        )
 
     ws_rpc_factory = WebSocketJSONRPCServerFactory(shutdown_timeout)
     ws_rpc_factory.add_object_handler(AccountRPC(), namespace="account")