Add visit creation option in TUI

Author: d-j-hatton

src/murfey/client/tui/app.py

@@ -25,6 +25,7 @@
     MainScreen,
     ProcessingForm,
     SessionSelection,
+    VisitCreation,
     VisitSelection,
     WaitingScreen,
     determine_default_destination,
@@ -682,8 +683,12 @@ async def on_mount(self) -> None:
         exisiting_sessions = requests.get(
             f"{self._environment.url.geturl()}/sessions"
         ).json()
-        self.install_screen(VisitSelection(self.visits), "visit-select-screen")
-        self.push_screen("visit-select-screen")
+        if self.visits:
+            self.install_screen(VisitSelection(self.visits), "visit-select-screen")
+            self.push_screen("visit-select-screen")
+        else:
+            self.install_screen(VisitCreation(), "visit-creation-screen")
+            self.push_screen("visit-creation-screen")
         if exisiting_sessions:
             self.install_screen(
                 SessionSelection(

src/murfey/client/tui/controller.css

@@ -46,6 +46,12 @@ SessionSelection {
   border: hidden;
 }
 
+VisitCreation {
+  layout: grid;
+  grid-size: 2;
+  border: hidden;
+}
+
 VisitSelection {
   layout: grid;
   grid-size: 4;
@@ -235,6 +241,22 @@ RadioSet {
   background: darkslateblue;
 }
 
+.btn-visit-create {
+  width: 100%;
+  height: 90fr;
+  column-span: 2;
+  background: teal;
+  border: solid black;
+}
+
+.btn-visit-create:hover {
+  background: purple;
+}
+
+.btn-visit-create:focus {
+  background: darkslateblue;
+}
+
 .btn-session {
   width: 100%;
   height: 20%;
@@ -468,6 +490,16 @@ RadioSet {
   background: black;
 }
 
+.input-visit-name {
+  width: 100%;
+  height: 100%;
+  column-span: 2;
+  row-span: 1;
+  content-align: left middle;
+  text-style: bold;
+  background: blueviolet;
+}
+
 #log_book {
   width: 100%;
   height: 100%;

src/murfey/client/tui/screens.py

@@ -760,6 +760,70 @@ def on_button_pressed(self, event: Button.Pressed):
             self.app.push_screen("upstream-downloads")
 
 
+class VisitCreation(Screen):
+    visit_name: reactive[str] = reactive("")
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def compose(self):
+        yield Input(placeholder="Visit name", classes="input-visit-name")
+        yield Button("Create visit", classes="btn-visit-create")
+
+    def on_input_changed(self, event):
+        self.visit_name = event.value
+
+    def on_button_pressed(self, event: Button.Pressed):
+        text = str(self.visit_name)
+        self.app._visit = text
+        self.app._environment.visit = text
+        response = requests.post(
+            f"{self.app._environment.url.geturl()}/visits/{text}",
+            json={"id": self.app._environment.client_id},
+        )
+        log.info(f"Posted visit registration: {response.status_code}")
+        machine_data = requests.get(
+            f"{self.app._environment.url.geturl()}/machine"
+        ).json()
+
+        self.app.install_screen(
+            DirectorySelection(
+                [
+                    p[0]
+                    for p in machine_data.get("data_directories", {}).items()
+                    if p[1] == "detector" and Path(p[0]).exists()
+                ]
+            ),
+            "directory-select",
+        )
+        self.app.pop_screen()
+
+        if machine_data.get("gain_reference_directory"):
+            self.app.install_screen(
+                GainReference(
+                    determine_gain_ref(Path(machine_data["gain_reference_directory"])),
+                    True,
+                ),
+                "gain-ref-select",
+            )
+            self.app.push_screen("gain-ref-select")
+        else:
+            if self._switch_status:
+                self.app.push_screen("directory-select")
+            else:
+                self.app.install_screen(LaunchScreen(basepath=Path("./")), "launcher")
+                self.app.push_screen("launcher")
+
+        if machine_data.get("upstream_data_directories"):
+            upstream_downloads = requests.get(
+                f"{self.app._environment.url.geturl()}/sessions/{self.app._environment.murfey_session}/upstream_visits"
+            ).json()
+            self.app.install_screen(
+                UpstreamDownloads(upstream_downloads), "upstream-downloads"
+            )
+            self.app.push_screen("upstream-downloads")
+
+
 class UpstreamDownloads(Screen):
     def __init__(self, connected_visits: Dict[str, Path], *args, **kwargs):
         super().__init__(*args, **kwargs)