From 4ec6acf7312160c10be1f8015d467210712f7d43 Mon Sep 17 00:00:00 2001
From: Mehran Kholdi <semekh@hamravesh.com>
Date: Fri, 24 Apr 2020 20:08:18 +0430
Subject: [PATCH] Store metadata file

---
 rawfile_servicer.py | 12 ++++--------
 rawfile_util.py     | 30 ++++++++++++++++++++++++++++++
 remote.py           | 23 +++++++++++++++++------
 3 files changed, 51 insertions(+), 14 deletions(-)
 create mode 100644 rawfile_util.py

diff --git a/rawfile_servicer.py b/rawfile_servicer.py
index dda7bf8..150d793 100644
--- a/rawfile_servicer.py
+++ b/rawfile_servicer.py
@@ -1,13 +1,11 @@
 import grpc
-from pathlib import Path
-
 from google.protobuf.wrappers_pb2 import BoolValue
 
-from consts import DATA_DIR
+import rawfile_util
 from csi import csi_pb2, csi_pb2_grpc
 from orchestrator.k8s import volume_to_node, run_on_node
-from util import log_grpc_request, run
 from remote import init_rawfile, scrub
+from util import log_grpc_request, run
 
 NODE_NAME_TOPOLOGY_KEY = "hostname"
 
@@ -49,9 +47,7 @@ class RawFileNodeServicer(csi_pb2_grpc.NodeServicer):
     @log_grpc_request
     def NodePublishVolume(self, request, context):
         mount_path = request.target_path
-        img_dir = Path(f"{DATA_DIR}/{request.volume_id}")
-        img_file = Path(f"{img_dir}/raw.img")
-
+        img_file = rawfile_util.img_file(request.volume_id)
         run(f"mount {img_file} {mount_path}")
         return csi_pb2.NodePublishVolumeResponse()
 
@@ -112,7 +108,7 @@ class RawFileControllerServicer(csi_pb2_grpc.ControllerServicer):
             )
 
         size = request.capacity_range.required_bytes
-        size = min(size, 10 * 1024 * 1024)  # At least 10MB
+        size = max(size, 10 * 1024 * 1024)  # At least 10MB
 
         try:
             node_name = request.accessibility_requirements.preferred[0].segments[
diff --git a/rawfile_util.py b/rawfile_util.py
new file mode 100644
index 0000000..60d53f8
--- /dev/null
+++ b/rawfile_util.py
@@ -0,0 +1,30 @@
+import json
+from pathlib import Path
+
+from consts import DATA_DIR
+
+
+def img_dir(volume_id):
+    return Path(f"{DATA_DIR}/{volume_id}")
+
+
+def meta_file(volume_id):
+    return Path(f"{img_dir(volume_id)}/disk.meta")
+
+
+def metadata(volume_id):
+    try:
+        return json.loads(meta_file(volume_id).read_text())
+    except FileNotFoundError:
+        return {}
+
+
+def img_file(volume_id):
+    return Path(metadata(volume_id)["img_file"])
+
+
+def patch_metadata(volume_id, obj):
+    old_data = metadata(volume_id)
+    new_data = {**old_data, **obj}
+    meta_file(volume_id).write_text(json.dumps(new_data))
+    return new_data
diff --git a/remote.py b/remote.py
index fe1b47b..270b302 100644
--- a/remote.py
+++ b/remote.py
@@ -3,18 +3,29 @@ from util import remote_fn
 
 @remote_fn
 def scrub(volume_id):
-    # TODO: stub
-    pass
+    import time
+    import rawfile_util
+
+    rawfile_util.patch_metadata(volume_id, {"deleted_at": time.time()})
 
 
 @remote_fn
 def init_rawfile(volume_id, size):
+    import time
+    import rawfile_util
     from util import run
-    from consts import DATA_DIR
-    from pathlib import Path
 
-    img_dir = Path(f"{DATA_DIR}/{volume_id}")
+    img_dir = rawfile_util.img_dir(volume_id)
     img_dir.mkdir(parents=False, exist_ok=False)
-    img_file = Path(f"{img_dir}/raw.img")
+    img_file = f"{img_dir}/disk.img"
+    rawfile_util.patch_metadata(
+        volume_id,
+        {
+            "volume_id": volume_id,
+            "created_at": time.time(),
+            "img_file": img_file,
+            "size": size,
+        },
+    )
     run(f"truncate -s {size} {img_file}")
     run(f"mkfs.ext4 {img_file}")