tmp

Author: SockworkOrange

ruby-wasm-poc/Gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+gem "ruby_wasm", "~> 2.6"
+gem "ruby-next", "~> 1.0"
+gem "google-protobuf", "~> 3.25"
+gem "grpc", "~> 1.63"

ruby-wasm-poc/Gemfile.lock

@@ -0,0 +1,42 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.4.2)
+    diff-lcs (1.5.1)
+    google-protobuf (3.25.3-arm64-darwin)
+    googleapis-common-protos-types (1.14.0)
+      google-protobuf (~> 3.18)
+    grpc (1.63.0-arm64-darwin)
+      google-protobuf (~> 3.25)
+      googleapis-common-protos-types (~> 1.0)
+    paco (0.2.3)
+    parser (3.3.1.0)
+      ast (~> 2.4.1)
+      racc
+    racc (1.7.3)
+    require-hooks (0.2.2)
+    ruby-next (1.0.3)
+      paco (~> 0.2)
+      require-hooks (~> 0.2)
+      ruby-next-core (= 1.0.3)
+      ruby-next-parser (>= 3.2.2.0)
+      unparser (~> 0.6.0)
+    ruby-next-core (1.0.3)
+    ruby-next-parser (3.2.2.0)
+      parser (>= 3.0.3.1)
+    ruby_wasm (2.6.1-arm64-darwin)
+    unparser (0.6.13)
+      diff-lcs (~> 1.3)
+      parser (>= 3.3.0)
+
+PLATFORMS
+  arm64-darwin
+
+DEPENDENCIES
+  google-protobuf (~> 3.25)
+  grpc (~> 1.63)
+  ruby-next (~> 1.0)
+  ruby_wasm (~> 2.6)
+
+BUNDLED WITH
+   2.5.10

ruby-wasm-poc/Makefile

@@ -0,0 +1,27 @@
+SHELL 		:= /bin/bash
+PWD 		:= $(shell pwd)
+RUBY_WASM 	:= ${HOME}/bin/ruby-wasm
+
+protobuf-generate:
+	./scripts/generate_protobuf.sh
+
+tebako-build:
+	tebako press \
+      --root='.' \
+      --entry=bin/plugin.rb \
+      --output=dist/plugin.tebako \
+      --Ruby=3.2.3 \
+      --log-level=debug # TODO if removed - fails on segmentation fault
+
+wasm-build:
+	rbwasm build --remake --reconfigure --ruby-version 3.2 -o dist/plugin.wasm
+	echo "WASM filesize:" && du -sh dist/plugin.wasm
+
+sync-wasm-sha: protobuf-generate wasm-build
+	./scripts/sync_wasm_sha.sh
+
+sqlc-generate-process: protobuf-generate tebako-build
+	sqlc -f sqlc.process.yaml generate
+
+sqlc-generate-wasm: sync-wasm-sha
+	SQLCCACHE=./; sqlc -f sqlc.wasm.yaml generate

ruby-wasm-poc/buf.gen.yaml

@@ -0,0 +1,4 @@
+version: v1
+plugins:
+  - plugin: buf.build/protocolbuffers/ruby
+    out: lib/gen

ruby-wasm-poc/lib/gen/plugin/codegen_pb.rb

@@ -0,0 +1,41 @@
+require_relative 'gen/plugin/codegen_pb'
+require 'json'
+
+module Plugin
+  class Runner
+    @options = {}
+    
+    def run
+      request_bytes = $stdin.read.bytes
+      generate_request = Plugin::GenerateRequest.decode(request_bytes)
+      generate_response = generate(generate_request)
+      $stdout.write(generate_response.encode)
+    end
+
+    private
+
+    def generate(generate_request)
+      @options = parse_options(generate_request['PluginOptions'])
+      files = generate_files_from_queries(generate_request)
+      generate_response = Plugin::GenerateResponse.new(:Files => files)
+      generate_response.encode
+    end
+
+    def parse_options(plugin_options)
+      plugin_options_str = plugin_options
+                             .bytes.pack('C*')
+                             .force_encoding('UTF-8')
+      JSON.parse(plugin_options_str)
+    end
+    
+    def generate_files_from_queries(generate_request)
+      generate_request['Queries']
+        .group_by { |query| query['Filename'] }
+        .map { |filename, queries| generate_file(filename, queries) }
+    end
+
+    def generate_file(filename, queries)
+      Plugin.OutputFile.new(:Name => filename, :Contents => "".bytes)
+    end
+  end
+end

ruby-wasm-poc/lib/plugin.rb

@@ -0,0 +1,25 @@
+#!/usr/bin/env bash
+
+set -ex
+
+replace_in_file () {
+  text_to_replace=$1
+  replace_with=$2
+  filename=$3
+  
+  if [[ "$(uname -s)" == "Linux" ]]; then
+    sed -i "s!${text_to_replace}!${replace_with}!g" "${filename}"
+  else 
+    sed -i '' "s!${text_to_replace}!${replace_with}!g" "${filename}"
+  fi 
+}
+
+if wget -q --spider http://google.com; then
+  buf generate --template buf.gen.yaml buf.build/sqlc/sqlc --path plugin/
+  
+  # File is a builtin class library in Ruby, and it's a bad practice to name a class as a builtin class.
+  # Since we have limited direct control on the generated protobuf code, we fix it post-generate.
+  replace_in_file 'File[[:blank:]]' 'OutputFile ' lib/gen/plugin/codegen_pb.rb
+else
+  echo "No internet connection - using pre-existing protobuf files.."
+fi

ruby-wasm-poc/scripts/generate_protobuf.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env bash
+
+set -ex
+
+# uninstall
+gem uninstall ruby_wasm --all --executables
+asdf uninstall ruby 3.2.2 || echo "ruby not installed"
+
+# install
+gem install ruby_wasm
+brew install libyaml
+gem install psych -- --with-libyaml-dir="$(brew --prefix libyaml)"
+RUBY_CONFIGURE_OPTS="--with-libyaml-dir=$(brew --prefix libyaml)"
+export RUBY_CONFIGURE_OPTS
+asdf install ruby 3.2.2 && asdf global ruby 3.2.2
+
+cd /tmp && curl -LO --silent \
+  https://github.com/ruby/ruby.wasm/releases/download/2.6.1/ruby-3.2-wasm32-unknown-wasip1-full.tar.gz
+tar xvf ruby-3.2-wasm32-unknown-wasip1-full.tar.gz
+mv /tmp/ruby-3.2-wasm32-unknown-wasip1-full/usr/local/bin/ruby ~/bin/ruby.wasm

ruby-wasm-poc/scripts/install_rbwasm.sh

@@ -0,0 +1,4 @@
+#!/usr/bin/env bash
+
+PLUGIN_SHA=$(shasum -a 256 dist/plugin.wasm | awk '{ print $1 }')
+yq -i ".plugins[0].wasm.sha256 = \"${PLUGIN_SHA}\"" sqlc.wasm.yaml