Show correct URL based on env

cr/api.py

@@ -89,6 +89,14 @@ def __init__(self, handle: str, token: str):
                 db_type=DatabaseType(dbdict["db_type"]),
             )
 
+    def url(self, env: Env) -> str:
+        """
+        Return the URL of this website based on environment.
+        """
+        if env == Env.STAGING:
+            return f"https://{self.handle}.staging.codered.cloud/"
+        return self.primary_url
+
     def local_check_path(self, p: Path, c: Optional[Console]) -> None:
         """
         Validity check for a provided Path ``p``. If Console ``c`` is provided,

cr/cli.py

@@ -309,7 +309,7 @@ def run(self, args: argparse.Namespace):
             )
             pbar.update(t, total=1, completed=1)
 
-        CONSOLE.print(f"\nYour site is live at: {w.primary_url}\n")
+        CONSOLE.print(f"\nYour site is live at: {w.url(args.env)}\n")
 
 
 class Restart(Command):
@@ -332,9 +332,8 @@ def run(self, args: argparse.Namespace):
         w = self.get_webapp(args)
         w.api_queue_restart(args.env)
         CONSOLE.print(
-            "Restart in progress. You will receive an email when complete."
-            "\n\n"
-            f"Restarting: {w.primary_url}"
+            f"Restarting: {w.url(args.env)}\n"
+            "You will receive an email when complete."
         )