Add option for archive and restore VMs

bin/vm

@@ -4,6 +4,10 @@ require 'optparse'
 require 'ostruct'
 require 'virtual-machine'
 
+# Paramteters
+#      -env: DAT_VM_DATA=~/.cache/vm
+
+
 # vm setup debian --arch "<host:arch64>" --name "<image_name: debian>"
 
 options = OpenStruct.new
@@ -18,6 +22,8 @@ OptionParser.new do |opt|
   opt.on('--name NAME', 'Virtaul Machine name') do |name| 
     options.name = name
   end
+  # --port (ssh port)
+  # --attached (do not leave the shell process)
   # new from copy
   # fetch from the server
 end.parse!
@@ -25,11 +31,40 @@ end.parse!
 
 case subcommand
 when :create
-  puts "Creating image...."
+  puts "vm create debian-copy --name debian"
+  puts "Creating image base on existing one...."
+when :run
+  # same as setup but fetches image not
+  puts "Run the image or fetch and run...."
+  options[:distro] = parameter
+  VirtualMachine.run(options)
 
 when :setup
   options[:distro] = parameter
   VirtualMachine.setup(options)
+when :archive
+  # whole image or files that are not present in the archived
+  # --type files (zst all files by files)
+  # by default
+  #before start if should create index of files that are present on 
+  # the vm, so later we can copy new files or changed 
+
+  options[:distro] = parameter
+  VirtualMachine.archive(options)
+
+when :appy
+  puts "appy chnages from VM to the host os"
+when :restore
+  options[:distro] = parameter
+  VirtualMachine.restore(options)
+  #puts "download image from remote or if local copy exists then the local copy"
+when :daemon
+  puts "setup a pprocess so i can attach to it"
+  # paramaters exatly as run/setup
+when :status
+  puts "Print all images and status stopped/daemon(port)/running(port/)"
+when :attach
+  puts "if not paramteres attaches to last used"
 else
   puts "Error not found #{options.type}"
 end

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)
+        
+        url = distro(options[:name], options[:arch], :install)
+        disk_img_path = root_img_path(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
+            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