Gestion de l'image de base

package

@@ -1,10 +1,10 @@
 #!/usr/bin/env python3
 
-import argparse, sys, shutil, os, json
+import argparse, sys, shutil, os
 
 sys.path.append(os.path.dirname(__file__) + '/lib')
 
-import tamarin, system
+import tamarin, system, rkt
 
 def configure_args_parser():
 
@@ -57,45 +57,46 @@ if __name__ == "__main__":
     base_image = profile['profile']['default_image']
     # If the base image is Docker-based, preload it and get its name from the store
     if base_image.startswith('docker://'):
-        tamarin.run_rkt([
+        rkt.run([
             "fetch",
             "--insecure-options=image",
             base_image
         ] + rkt_flags)
-        output = tamarin.run_rkt([
-            "image",
-            "list",
-            "--format=json"
-        ] + rkt_flags, captureOutput=True)
-        # Fetch the list of installed images
-        images_list = json.loads(output.decode('utf-8'))
-        base_image = images_list[0]['name']
+
+    name_pattern = base_image.split('/')[-1] + '$'
+    image = rkt.find_image_by_name(name_pattern, rkt_flags=rkt_flags)
+
+
+    base_aci_name = image['id'][:21]+'.aci'
+    base_aci_file = workspace_tmp+'/'+base_aci_name
+
+    if not os.path.isfile(base_aci_file):
+        rkt.export_image(image['id'], workspace_tmp+'/'+base_aci_name, rkt_flags=rkt_flags);
 
     # Start building image
-    tamarin.run_acbuild(acbuild_flags+["begin"])
+    tamarin.run_acbuild(acbuild_flags+["begin", base_aci_file])
     tamarin.run_acbuild(acbuild_flags+["set-name", "image_{:d}".format(pid)])
-    tamarin.run_acbuild(acbuild_flags+["dependency", "add", base_image])
-    tamarin.run_acbuild(acbuild_flags+["set-exec", "--", "/bin/sh", "-c", "echo Hello World"])
+    tamarin.run_acbuild(acbuild_flags+["set-exec", "--", "/bin/sh", "-c", "uname -a"])
     tamarin.run_acbuild(acbuild_flags+["write", image_path])
     tamarin.run_acbuild(acbuild_flags+["end"])
 
     # rkt run --insecure-options=image ./nginx.aci --volume html,kind=host,source=/path/to/test --net=host
-    tamarin.run_rkt([
+    rkt.run([
         "run",
         "--insecure-options=image",
         image_path, "--net=host"
-    ] + rkt_flags, asRoot=True)
+    ] + rkt_flags, as_root=True)
 
     # Cleanup
 
-    tamarin.run_rkt([
+    rkt.run([
         "gc",
         "--grace-period=0"
-    ] + rkt_flags, asRoot=True)
+    ] + rkt_flags, as_root=True)
 
-    tamarin.run_rkt([
+    rkt.run([
         "image",
         "gc"
-    ] + rkt_flags, asRoot=True)
+    ] + rkt_flags, as_root=True)
 
     shutil.rmtree(acbuild_workspace, ignore_errors=True)