From 8d7a59952349bbd6234dabe5280b6204733f84eb Mon Sep 17 00:00:00 2001
From: Carsten Munk <carsten@zippie.com>
Date: Mon, 13 May 2024 14:36:00 +0200
Subject: [PATCH] feat: add create_machine_snapshot

Signed-off-by: Carsten Munk <carsten@zippie.com>
---
 .changeset/neat-items-impress.md     |  5 ++
 packages/sdk/Dockerfile              |  1 +
 packages/sdk/create_machine_snapshot | 82 ++++++++++++++++++++++++++++
 3 files changed, 88 insertions(+)
 create mode 100644 .changeset/neat-items-impress.md
 create mode 100755 packages/sdk/create_machine_snapshot

diff --git a/.changeset/neat-items-impress.md b/.changeset/neat-items-impress.md
new file mode 100644
index 00000000..4b5628ca
--- /dev/null
+++ b/.changeset/neat-items-impress.md
@@ -0,0 +1,5 @@
+---
+"@cartesi/sdk": patch
+---
+
+feat: script to generate a machine
diff --git a/packages/sdk/Dockerfile b/packages/sdk/Dockerfile
index 9811f3c3..802c2493 100644
--- a/packages/sdk/Dockerfile
+++ b/packages/sdk/Dockerfile
@@ -105,6 +105,7 @@ COPY devnet /usr/local/bin
 COPY eth_isready /usr/local/bin
 COPY eth_dump /usr/local/bin
 COPY eth_load /usr/local/bin
+COPY create_machine_snapshot /usr/local/bin
 
 COPY entrypoint.sh /usr/local/bin/
 COPY --from=su-exec /usr/local/src/su-exec /usr/local/bin/
diff --git a/packages/sdk/create_machine_snapshot b/packages/sdk/create_machine_snapshot
new file mode 100755
index 00000000..11e3a7d4
--- /dev/null
+++ b/packages/sdk/create_machine_snapshot
@@ -0,0 +1,82 @@
+#!/bin/bash
+set -e
+
+# Define function to display usage
+usage() {
+    echo "Usage: $0 [options]"
+    echo "Options:"
+    echo "  -r, --ram-length <size>        Specify RAM size"
+    echo "  -d, --drive-label <label>      Specify drive label, filename for input"
+    echo "  -m, --drive-filename <path>    Specify raw drive image filename"
+    echo "  -o, --output <path>            Specify output directory"
+    echo "  -w, --workdir <path>           Set the working directory"
+    echo "  -e, --entrypoint <cmd>         Set the entrypoint command"
+    echo "  -v, --env <var=value>          Add an environment variable"
+    echo "  -h, --help                     Display this help and exit"
+}
+
+# Initialize variables with default values
+ram_length=""
+drive_label=""
+drive_filename=""
+output_path=""
+workdir=""
+entrypoint=""
+declare -a env_vars
+
+# Parse command line options using getopt
+TEMP=$(getopt -o r:d:m:o:w:e:v:h --long ram-length:,drive-label:,drive-filename:,output:,workdir:,entrypoint:,env:,help -- "$@")
+eval set -- "$TEMP"
+
+# Extract options and their arguments into variables.
+while true ; do
+    case "$1" in
+        -r|--ram-length)
+            ram_length="$2"
+            shift 2 ;;
+        -d|--drive-label)
+            drive_label="$2"
+            shift 2 ;;
+        -m|--drive-filename)
+            drive_filename="$2"
+            shift 2 ;;
+        -o|--output)
+            output_path="$2"
+            shift 2 ;;
+        -w|--workdir)
+            workdir="$2"
+            shift 2 ;;
+        -e|--entrypoint)
+            entrypoint="$2"
+            shift 2 ;;
+        -v|--env)
+            env_vars+=("$2")
+            shift 2 ;;
+        --)
+            shift ; break ;;
+        -h|--help)
+            usage
+            exit 0 ;;
+        *)
+            echo "Invalid option: $1" >&2
+            usage
+            exit 1 ;;
+    esac
+done
+
+# Construct the command line for cartesi-machine
+cmd=("cartesi-machine")
+for env_var in "${env_vars[@]}"; do
+    cmd+=("--append-entrypoint=export $env_var")
+done
+[[ -n "$ram_length" ]] && cmd+=("--ram-length=$ram_length")
+[[ -n "$drive_label" ]] && [[ -n $drive_filename ]] && cmd+=("--flash-drive=label:$drive_label,filename:$drive_filename")
+[[ -n "$output_path" ]] && cmd+=("--store=$output_path")
+[[ -n "$workdir" ]] && cmd+=("--append-init=WORKDIR=$workdir")
+[[ -n "$entrypoint" ]] && cmd+=("--append-entrypoint=$entrypoint")
+cmd+=("--append-bootargs=no4lvl")
+cmd+=("--assert-rolling-template")
+cmd+=("--final-hash")
+
+# Execute the command
+"${cmd[@]}"