Introduce rbwasm pack command as an alias of wasi-vfs pack

To remove wasi-vfs CLI binary dependency

Author: kateinoigakukun

Cargo.lock

@@ -2,8 +2,8 @@
 #
 # Example runs
 # $ ruby vm_deep_call.rb
-# $ RUBY_EXE="wasmtime run --mapdir /::/ head-wasm32-unknown-wasi-minimal/usr/local/bin/ruby --" ruby vm_deep_call.rb
-# $ RUBY_EXE="wasmtime run --env RUBY_FIBER_MACHINE_STACK_SIZE=20971520 --mapdir /::/ head-wasm32-unknown-wasi-minimal/usr/local/bin/ruby --" ruby vm_deep_call.rb
+# $ RUBY_EXE="wasmtime run --dir /::/ head-wasm32-unknown-wasi-minimal/usr/local/bin/ruby --" ruby vm_deep_call.rb
+# $ RUBY_EXE="wasmtime run --env RUBY_FIBER_MACHINE_STACK_SIZE=20971520 --dir /::/ head-wasm32-unknown-wasi-minimal/usr/local/bin/ruby --" ruby vm_deep_call.rb
 
 def vm_rec n
   vm_rec n - 1 if n > 0

benchmarks/vm_deep_call.rb

@@ -15,3 +15,4 @@ wizer = "3.0.0"
 wasmtime-wasi = "9.0.4"
 wasi-cap-std-sync = "9.0.4"
 wasi-vfs-cli = { git = "https://github.com/kateinoigakukun/wasi-vfs/", rev = "b1e4e5d9cd6322e8745e67c092b495973835a94f" }
+structopt = "0.3.26"

ext/ruby_wasm/Cargo.toml

@@ -7,6 +7,7 @@ use magnus::{
     wrap, Error, ExceptionClass, RModule, Ruby,
 };
 use wizer::Wizer;
+use structopt::StructOpt;
 
 static RUBY_WASM: value::Lazy<RModule> =
     value::Lazy::new(|ruby| ruby.define_module("RubyWasmExt").unwrap());
@@ -35,6 +36,15 @@ struct WasiVfsInner {
 struct WasiVfs(std::cell::RefCell<WasiVfsInner>);
 
 impl WasiVfs {
+    fn run_cli(args: Vec<String>) -> Result<(), Error> {
+        wasi_vfs_cli::App::from_iter(args).execute().map_err(|e| {
+            Error::new(
+                exception::standard_error(),
+                format!("failed to run wasi vfs cli: {}", e),
+            )
+        })
+    }
+
     fn new() -> Self {
         Self(std::cell::RefCell::new(WasiVfsInner { map_dirs: vec![] }))
     }
@@ -63,6 +73,7 @@ fn init(ruby: &Ruby) -> Result<(), Error> {
 
     let wasi_vfs = module.define_class("WasiVfs", ruby.class_object())?;
     wasi_vfs.define_singleton_method("new", function!(WasiVfs::new, 0))?;
+    wasi_vfs.define_singleton_method("run_cli", function!(WasiVfs::run_cli, 1))?;
     wasi_vfs.define_method("map_dir", method!(WasiVfs::map_dir, 2))?;
     wasi_vfs.define_method("pack", method!(WasiVfs::pack, 1))?;
     Ok(())

ext/ruby_wasm/src/lib.rs

@@ -9,7 +9,7 @@ def initialize(stdout:, stderr:)
     end
 
     def run(args)
-      available_commands = %w[build]
+      available_commands = %w[build pack]
       parser =
         OptionParser.new do |opts|
           opts.banner = <<~USAGE
@@ -32,6 +32,8 @@ def run(args)
       case command
       when "build"
         build(args)
+      when "pack"
+        pack(args)
       else
         @stderr.puts parser
         exit
@@ -141,6 +143,11 @@ def build(args)
       end
     end
 
+    def pack(args)
+      self.require_extension
+      RubyWasmExt::WasiVfs::run_cli(["pack", *args])
+    end
+
     private
 
     def build_config(options)
@@ -180,13 +187,7 @@ def do_print_ruby_cache_key(packager)
     end
 
     def do_build(executor, tmpdir, packager, options)
-      # Tries to require the extension for the given Ruby version first
-      begin
-        RUBY_VERSION =~ /(\d+\.\d+)/
-        require_relative "#{Regexp.last_match(1)}/ruby_wasm.so"
-      rescue LoadError
-        require_relative "ruby_wasm.so"
-      end
+      self.require_extension
       wasm_bytes = packager.package(executor, tmpdir, options)
       RubyWasm.logger.info "Size: #{SizeFormatter.format(wasm_bytes.size)}"
       case options[:output]
@@ -197,5 +198,15 @@ def do_build(executor, tmpdir, packager, options)
         RubyWasm.logger.debug "Wrote #{options[:output]}"
       end
     end
+
+    def require_extension
+      # Tries to require the extension for the given Ruby version first
+      begin
+        RUBY_VERSION =~ /(\d+\.\d+)/
+        require_relative "#{Regexp.last_match(1)}/ruby_wasm.so"
+      rescue LoadError
+        require_relative "ruby_wasm.so"
+      end
+    end
   end
 end

lib/ruby_wasm/cli.rb

@@ -22,6 +22,6 @@ cp -R "$ruby_root" "$workdir/ruby-root"
 (
   cd "$workdir" && \
   "$WASMOPT" --strip-debug ruby-root/usr/local/bin/ruby -o ./ruby-root/ruby.wasm && \
-  "$WASI_VFS_CLI" pack ./ruby-root/ruby.wasm --mapdir /usr::./ruby-root/usr --mapdir /gems::$package_dir/gems -o "$dist_dir/irb.wasm" && \
+  "$WASI_VFS_CLI" pack ./ruby-root/ruby.wasm --dir ./ruby-root/usr::/usr --dir $package_dir/gems::/gems -o "$dist_dir/irb.wasm" && \
   wasi-preset-args "$dist_dir/irb.wasm" -o "$dist_dir/irb.wasm" -- -I/gems/lib /gems/libexec/irb --prompt default
 )

packages/standalone/irb/build-package.sh

@@ -22,5 +22,5 @@ cp -R "$ruby_root" "$workdir/ruby-root"
 (
   cd "$workdir" && \
   "$WASMOPT" --strip-debug ruby-root/usr/local/bin/ruby -o ./ruby-root/ruby.wasm && \
-  "$WASI_VFS_CLI" pack ./ruby-root/ruby.wasm --mapdir /usr::./ruby-root/usr -o "$dist_dir/ruby.wasm"
+  "$WASI_VFS_CLI" pack ./ruby-root/ruby.wasm --dir ./ruby-root/usr::/usr -o "$dist_dir/ruby.wasm"
 )

packages/standalone/ruby/build-package.sh

@@ -1,7 +1,7 @@
 wasi_vfs = RubyWasm::WasiVfsProduct.new(File.join(Dir.pwd, "build"))
 wasi_sdk = TOOLCHAINS["wasi-sdk"]
 tools = {
-  "WASI_VFS_CLI" => wasi_vfs.cli_bin_path,
+  "WASI_VFS_CLI" => File.expand_path(File.join(__dir__, "..", "exe", "rbwasm")),
   "WASMOPT" => wasi_sdk.wasm_opt
 }