extended msd api for future otg

1 files changed, 13 insertions, 3 deletions

diff --git a/kvmd/plugins/msd/__init__.py b/kvmd/plugins/msd/__init__.py
index d5aa3064..bff60232 100644
--- a/kvmd/plugins/msd/__init__.py
+++ b/kvmd/plugins/msd/__init__.py
@@ -64,6 +64,11 @@ class MsdIsBusyError(MsdOperationError):
         super().__init__("Performing another MSD operation, please try again later")
 
 
+class MsdMultiNotSupported(MsdOperationError):
+    def __init__(self) -> None:
+        super().__init__("This MSD does not support storing multiple images")
+
+
 # =====
 class BaseMsd(BasePlugin):
     def get_state(self) -> Dict:
@@ -73,22 +78,27 @@ class BaseMsd(BasePlugin):
         yield {}
         raise NotImplementedError
 
+    async def reset(self) -> None:
+        raise NotImplementedError
+
     async def cleanup(self) -> None:
         pass
 
+    # =====
+
     async def connect(self) -> Dict:
         raise NotImplementedError
 
     async def disconnect(self) -> Dict:
         raise NotImplementedError
 
-    async def reset(self) -> None:
+    async def select(self, name: str) -> Dict:
         raise NotImplementedError
 
-    async def __aenter__(self) -> "BaseMsd":
+    async def remove(self, name: str) -> Dict:
         raise NotImplementedError
 
-    def get_chunk_size(self) -> int:
+    async def __aenter__(self) -> "BaseMsd":
         raise NotImplementedError
 
     async def write_image_info(self, name: str, complete: bool) -> None: