Add option for archive and restore VMs

lib/virtual-machine.rb

@@ -2,13 +2,46 @@ require 'downloader'
 require 'system'
 require_relative 'data/resources/iso-images'
 require 'vm/qemu'
+require 'vm/archive'
 
 module VirtualMachine 
     def self.distro(name, arch, type = :install)
-        ISO_URLS[:debian][:arm64][:install]
+        ISO_URLS[name][arch][type]
     end
 
-    def self.setup(options)
+    def self.vm_root_path
+        ENV["DAT_VM_DATA"] || File.join(User.cache_path, "vm")
+    end
+
+    def self.vm_dir(options)
+        File.join(vm_root_path, "image", "#{options[:distro].to_s}-#{options[:arch].to_s}", options[:name])
+    end
+
+    def self.archive_dir(options)
+        File.join(vm_root_path, "archive", "#{options[:distro].to_s}-#{options[:arch].to_s}", options[:name])
+    end
+    
+    def self.create_archive_path(options)
+        File.join(archive_dir(options), "archive.tar.zst")
+    end
+
+    def self.get_recent_archive_path(options)
+        File.join(archive_dir(options), "archive.tar.zst")
+    end
+
+    # https://artur.gurgul.pro/vm/
+    # https://artur.gurgul.pro/vm/debian-arm64/debian/archive.tar.zst
+
+    def self.vm_archive_url(options)
+        return nil unless ENV["DAT_VM_REPO_URL"]
+        
+    end
+
+    def self.root_img_path(options)
+        File.join(vm_dir(options), "root.img")
+    end
+
+    def self.fill_defaults(options)
         if options[:name] == nil 
             options[:name] = options[:distro]
         end
@@ -18,27 +51,51 @@ module VirtualMachine
         else 
             options[:arch] = System.arch_to_symbol(options[:arch])
         end
+    end
 
-        puts options
+    def self.archive(options)
+        fill_defaults(options)
+        Archive.create(vm_dir(options), out: create_archive_path(options))
+    end
 
-        distro = options[:distro]
-        type = :install
-        arch = options[:arch]
-        url = distro(name, arch, type)
+    def self.restore(options)
+        fill_defaults(options)
+        Archive.restore(get_recent_archive_path(options), vm_dir(options))
+    end
+
+    def self.run(options)
+        fill_defaults(options)
+
+        disk_img_path = root_img_path(options)
+        # TODO:
+        #    - if image path not exists, check the cache
+        #    - if cache do not exists try to download
+        puts "Starting image: #{disk_img_path}"
+
+        Qemu.launch(
+            options[:arch],
+            disk_img_path,
+            cpus: [1, System.cpus - 2].max,
+            detach: true
+        )
+    end
+
+    def self.setup(options)
+        fill_defaults(options)
         
-        Downloader.get(url) do |path|
-            disk_img_path = File.join(User.cache_path, "vm", distro.to_s, arch.to_s, options[:name], "root.img")
-            create_disk_image(disk_img_path, 15000)
-            
-            puts path
-            puts disk_img_path
+        url = distro(options[:name], options[:arch], :install)
+        disk_img_path = root_img_path(options)
 
+        Downloader.get(url) do |path|
+            
+            create_disk_image(disk_img_path, 64000)
+            
             Qemu.launch(
-                arch,
+                options[:arch],
                 disk_img_path,
                 cpus: [1, System.cpus - 2].max,
                 cdrom: path,
-                detach: false
+                detach: true
             )
         end
     end

lib/vm/archive.rb

@@ -0,0 +1,37 @@
+require "fileutils"
+
+module Archive
+    def self.cmd(*cmd)
+        puts "cmd: #{cmd.join(" ")}"
+        # return 
+        stdout, stderr, status = Open3.capture3(*cmd)
+        unless status.success?
+            warn "Command failed: #{cmd.join(' ')}"
+            warn stderr
+            exit status.exitstatus || 1
+        end
+        stdout
+    end
+
+    # it preserved sparsiveness 
+    #     gtar -S --sparse-version=2 -I 'zstd -T0 -19' -cpf test.tar.zst /Volumes/Projs/VM/VM-data/image/debian-arm64/debian/
+    #     tar --zstd -xpf test.tar.zst -C ./r
+
+    def self.create(path, **options)
+        out = options[:out] || File.join(Dir.pwd, "archive.tar.zst")
+        
+        FileUtils.mkdir_p(File.dirname(out))
+        cmd "gtar", "-S", "--sparse-version=2", "-I",
+            "zstd -T0 -19", "-cpf", out, "-C" , path, "."
+    end
+
+    def self.restore(file, path, **options)
+        puts file
+        puts path
+        puts options
+
+        FileUtils.mkdir_p(path)
+        cmd "gtar", "-S", "--sparse-version=2", "-I", "zstd", "-xpf",
+            file, "-C", path
+    end
+end