Add delete version button

Author: frcroth

client/interactive/client.tcss

@@ -1,4 +1,4 @@
-ListKeysWidget #list-keys {
+DownloadNotification #list-keys {
     layout: horizontal;
     color: red;
 }

client/interactive/db_connection.py

@@ -79,6 +79,10 @@ def listVersions(stub, collection, key):
     assertSuccess(reply)
     return reply.versions
 
+def deleteVersion(stub, collection, key, version):
+    reply = stub.Delete(proto.DeleteRequest(collection=collection, key=key, version=version))
+    assertSuccess(reply)
+
 
 def main():
     stub = connect()

client/interactive/modal.tcss

@@ -0,0 +1,27 @@
+DownloadNotification {
+    align: center middle;
+}
+
+#dialog {
+    padding: 0 1;
+    margin: 5;
+    width: 100;
+    height: 11;
+    border: thick $background 80%;
+    background: $surface;
+}
+
+#question {
+    column-span: 2;
+    height: 1fr;
+    width: 1fr;
+    content-align: center middle;
+}
+
+#okay {
+    margin: 3 0 0 0;
+}
+
+Button {
+    width: 100%;
+}

client/interactive/record_explorer.py

@@ -1,9 +1,13 @@
 from protobuf_decoder.protobuf_decoder import Parser
-from textual.widgets import Label, Static, Collapsible, Rule, Button, TextArea
-from textual.containers import Horizontal, Vertical
+from textual.app import ComposeResult
+from textual.reactive import reactive
+from textual.screen import Screen
+from textual.widgets import Label, Static, Collapsible, Rule, Button, OptionList
+from textual.widgets.option_list import Option
+from textual.containers import Horizontal, Vertical, Grid
 import json
 
-from db_connection import getKey, listVersions
+from db_connection import deleteVersion, getKey, listVersions
 
 class ProtobufDecoder():
     def __init__(self):
@@ -28,9 +32,37 @@ def __init__(self, stub, key: str, collection: str, **kwargs):
         except Exception as e:
             print("Could not load versions: " + str(e))
 
+    selected_version = reactive(0)
+
+    cached_data = {}
     def get_data(self):
-        data = getKey(self.stub, self.collection, self.key, self.selected_version)
-        return data
+        if self.selected_version in self.cached_data:
+            return self.cached_data[self.selected_version]
+        self.cached_data[self.selected_version] = getKey(self.stub, self.collection, self.key, self.selected_version)
+        return self.cached_data[self.selected_version]
+    
+    def get_filename(self):
+        return f"{self.collection}_{self.key}_{self.selected_version}.bin"
+    
+    def download_data(self):
+        data = self.get_data()
+        with open(self.get_filename(), "wb") as f:
+            f.write(data)
+        self.app.push_screen(DownloadNotification(filename=self.get_filename()))
+
+    def delete_data(self):
+        async def delete_callback(result: bool):
+            if result:
+                deleteVersion(self.stub, self.collection, self.key, self.selected_version)
+                self.versions = listVersions(self.stub, self.collection, self.key)
+                self.versions.sort()
+                if len(self.versions) == 0:
+                    self.app.pop_screen()
+                else:
+                    self.selected_version = self.versions[-1]
+                    await self.recompose()
+
+        self.app.push_screen(DeleteModal(stub=self.stub, collection=self.collection, key=self.key, version=self.selected_version), delete_callback)
 
     def display_record(self):
         data = self.get_data()
@@ -97,7 +129,8 @@ def render_info_panel(self):
         return Vertical(
             Static(f"Exploring record/wire for {self.collection}:{self.key}. Currently viewing version {self.selected_version}"),
             Rule(),
-            Button("Download selected version"),
+            Button("Download selected version", id="download_button"),
+            Button("Delete selected version", id="delete_button"),
             Rule(),
             *self.render_version_buttons(),
             id = "record_explorer_info_panel"
@@ -109,7 +142,61 @@ def compose(self):
             yield self.display_record()
             yield self.render_info_panel()
 
-    
+    async def on_button_pressed(self, event):
+        if event.button.id.startswith("version_"):
+            version = int(event.button.id.split("_")[1])
+            self.selected_version = version
+            await self.recompose()
+        if event.button.id == "download_button":
+            self.download_data()
+        if event.button.id == "delete_button":
+            self.delete_data()
+
+
+class DownloadNotification(Screen):
+    """Screen with a note on where the downloaded file is stored."""
+
+    CSS_PATH = "modal.tcss"
+
+    def __init__(self, name = None, id = None, classes = None, filename = None, **kwargs):
+        super().__init__(name, id, classes)
+        self.filename = filename
+
+    def compose(self) -> ComposeResult:
+        yield Vertical(
+            Label(f"The version has been stored in {self.filename}", id="note"),
+            Button("Okay", variant="primary", id="okay"),
+            id="dialog",
+        )
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        if event.button.id == "okay":
+            self.app.pop_screen()
+
+class DeleteModal(Screen):
+    """Screen with a note on where the downloaded file is stored."""
+
+    CSS_PATH = "modal.tcss"
+
+    def __init__(self, name = None, id = None, classes = None,collection = None, key = None, version = None, **kwargs):
+        super().__init__(name, id, classes)
+        self.collection = collection
+        self.key = key
+        self.version = version
+
+    def compose(self) -> ComposeResult:
+        yield Vertical(
+            Label(f"Are you sure you want to delete version {self.version} of {self.collection}:{self.key}?", id="note"),
+            Button("Yes", variant="error", id="yes"),
+            Button("No", variant="primary", id="no"),
+            id="dialog",
+        )
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        if event.button.id == "yes":
+            self.dismiss(True)
+        if event.button.id == "no":
+            self.dismiss(False)
 
 if __name__ == "__main__":
     print("Running decoder demo!")