From fe40f9c5d66916cad4cee8d5b2d9e172fc1c0477 Mon Sep 17 00:00:00 2001
From: Alejandro Sirgo Rica <asirgo@soleta.eu>
Date: Thu, 8 Aug 2024 10:26:06 +0200
Subject: src: add POST cache/fetch method

Add API REST method to fetch an image.
Consolidate image fetch loging for cache/fetch and image/restore.

Resquest payload structure:
{
    'image': 'linux.img'
    'type': 'TIPTORRENT'
    'repository': '12.141.10.2'
}

The client will try to fetch'image' from 'repository' into cache.

Resquest response structure:
{
    'cache': [
        {'name': 'windows.img', 'size': 2432370213, checksum: '5d4dcc677bc19f40a647d0002f4ade90'},
        {'name': 'linux.img', 'size': 243234534213, checksum: '3eb22f888f88a55ad954f55644e1192e'}
    ]
}
---
 src/live/ogOperations.py | 55 ++++++++++++++++++++++++++++++++++--------------
 1 file changed, 39 insertions(+), 16 deletions(-)

(limited to 'src/live/ogOperations.py')

diff --git a/src/live/ogOperations.py b/src/live/ogOperations.py
index 9cc5b97..b22d8b2 100644
--- a/src/live/ogOperations.py
+++ b/src/live/ogOperations.py
@@ -178,14 +178,21 @@ class OgLiveOperations:
 
         return 0
 
-    def _copy_image_to_cache(self, image_name):
+    def _fetch_image_unicast(self, repo, image_name):
         """
         Copies /opt/opengnsys/image/{image_name} into
         /opt/opengnsys/cache/opt/opengnsys/images/
 
         Implies a unicast transfer. Does not use tiptorrent.
         """
+        if not get_cache_dev_path():
+                raise OgError('No cache partition is mounted')
+
         dst = f'{OG_CACHE_IMAGE_PATH}{image_name}.img'
+
+        if (os.path.exists(dst) and tip_check_csum(repo, image_name)):
+            return
+
         try:
             if os.path.exists(dst):
                 os.unlink(dst)
@@ -203,26 +210,19 @@ class OgLiveOperations:
         except (OSError, OgError) as e:
             raise OgError(f'Error copying image {image_name} to cache. Reported: {e}') from e
 
+        if (not os.path.exists(dst)):
+            raise OgError(f'could not find {dst} in cache')
+        if (not tip_check_csum(repo, image_name)):
+            raise OgError(f'Failed to validate checksum for {image_name}.img')
+
     def _restore_image_unicast(self, repo, name, devpath, cache=False):
         if ogChangeRepo(repo, smb_user=self._smb_user, smb_pass=self._smb_pass) != 0:
             self._restartBrowser(self._url)
             raise OgError(f'Cannot change repository to {repo}')
 
         if cache:
-            if not get_cache_dev_path():
-                raise OgError('No cache partition is mounted')
-
-            fetch = False
-            image_path = f'{OG_CACHE_IMAGE_PATH}{name}.img'
-            if (not os.path.exists(image_path) or not tip_check_csum(repo, name)):
-                self._copy_image_to_cache(name)
-                fetch = True
-
-            if fetch:
-                if (not os.path.exists(image_path)):
-                    raise OgError(f'could not find {image_path} in cache')
-                if (not tip_check_csum(repo, name)):
-                    raise OgError(f'Failed to validate checksum for {name}.img')
+            self._fetch_image_unicast(repo, name)
+            image_path = f'{OG_CACHE_IMAGE_PATH}{image_name}.img'
         else:
             if os.access(f'/opt/opengnsys/images', os.R_OK) == False:
                 raise OgError('Cannot access /opt/opengnsys/images in read mode, check permissions')
@@ -232,7 +232,7 @@ class OgLiveOperations:
 
         self._restore_image(image_path, devpath)
 
-    def _restore_image_tiptorrent(self, repo, name, devpath):
+    def _fetch_image_tiptorrent(self, repo, name):
         if not get_cache_dev_path():
             raise OgError('No cache partition is mounted')
 
@@ -251,7 +251,10 @@ class OgLiveOperations:
                 raise OgError(f'could not find {image_path} in cache')
             if (not tip_check_csum(repo, name)):
                 raise OgError(f'Failed to validate checksum for {name}.img')
+        return image_path
 
+    def _restore_image_tiptorrent(self, repo, name, devpath):
+        image_path = self._fetch_image_tiptorrent(repo, name)
         self._restore_image(image_path, devpath)
 
     def _restore_image(self, image_path, devpath):
@@ -677,6 +680,26 @@ class OgLiveOperations:
         logging.info('Delete file from cache request OK')
         return result
 
+    def cache_fetch(self, request, ogRest):
+        image = request.getImages()
+        repo = request.getRepo()
+        ctype = request.getType()
+
+        logging.info(f'Request to cache image {image}.img via {ctype} from {repo}')
+
+        if ctype == 'UNICAST':
+            self._fetch_image_unicast(repo, image)
+        elif ctype == 'TIPTORRENT':
+            self._fetch_image_tiptorrent(repo, image)
+        else:
+            raise OgError(f'Invalid image fetch type {ctype}')
+
+        logging.info('Cache fetch command OK')
+
+        result = {'cache': self._get_cache_contents()}
+
+        return result
+
     def refresh(self, ogRest):
         self._restartBrowser(self._url_log)
 
-- 
cgit v1.2.3-18-g5258