Merge branch 'docker.nu' of github.com:windmill-labs/windmill into docker.nu

Author: pyranota

docker/docker.nu

@@ -0,0 +1,99 @@
+#! /usr/bin/env nu
+# ------------------------------------ #
+# Build and Test docker images locally #
+#                run:                  #
+#        docker/docker.nu --help       #
+# ------------------------------------ #
+
+let branch = git branch --show-current;
+# TODO: Use podman
+let use_podman = false;
+
+# Build and Test docker images
+#
+# For more info run:
+# docker.nu up --help
+# docker.nu build --help
+#
+# Should be invoked in repo root
+def main [] {
+  ./docker/docker.nu --help;
+}
+
+# Run and Build (if not yet) docker image
+def "main up" [
+  custom_dockerfile?: path  # Any dockerfile that depends on root Dockerfile
+  --features(-f): string = "deno_core" # Features that will be passed to base Dockerfile to compile windmill
+  --dir(-d): path = ../windmill-ee-private # Path to EE repo
+] {
+  let ident = get_ident $custom_dockerfile -f $features;
+  if not (sudo docker images | into string | str contains $ident) {
+    print $"($ident) is not found, building..."
+    main build $custom_dockerfile -f $features -d $dir
+  } else {
+    print $"($ident) was found"
+  }
+  print $"Running: ($ident)"
+  sudo WM_IMAGE=($ident) docker compose up --pull never
+}
+
+# Build docker image
+def "main build" [
+  custom_dockerfile?: path  # Any dockerfile that depends on root Dockerfile
+  --features(-f): string = "deno_core" # Features that will be passed to base Dockerfile to compile windmill
+  --mock(-m) # Ignore building base image
+  --dir(-d): path = ../windmill-ee-private # Path to EE repo
+] {
+  if not $mock {
+    build_base -f $features -d $dir;
+  }
+  if $custom_dockerfile != null {
+    let ident = get_ident $custom_dockerfile -f $features;
+    let tmpfile = patch $custom_dockerfile -f $features;
+    print $"Building image by path ($custom_dockerfile) with ident: ($ident)"
+    sudo docker build -f $tmpfile -t $ident ../
+  }
+}
+
+def build_base [
+  --features(-f): string
+  --dir(-d): path
+] {
+  if not ($features | str contains "deno_core") {
+    print "Warning! deno_core feature not enabled"
+    print "Compilation may fail"
+  }
+
+  if ($features | str contains "enterprise") {
+    print "Building in EE mode"
+    print $"EE repo path: ($dir)"
+    (cd ./backend; ./substitute_ee_code.sh --copy --dir $dir)
+  }
+
+  let ident = get_ident -f $features;
+  print $"Building base image with ident: ($ident)"
+  sudo docker build -t ($ident) ../ --build-arg features=($features)
+  (cd ./backend; ./substitute_ee_code.sh --revert --dir $dir )
+}
+
+def get_ident [
+  custom_dockerfile?: path  # Any dockerfile that depends on root Dockerfile
+  --features(-f): string = "deno_core" # Features that will be passed to base Dockerfile to compile windmill
+] {
+    let feat_postfix = $features |  split row ',' | each { |e| $e | str trim --left --right } | sort | str join "-";
+    if $custom_dockerfile != null {
+      let df_postfix = $"($custom_dockerfile)" | split row "Dockerfile" | get 1 | str downcase;
+      return ($branch | append $df_postfix | append $feat_postfix | str join "__");
+    } else {
+      return ($branch | append $feat_postfix | str join "-");
+    }
+}
+
+def patch [
+  file: path
+  --features(-f): string
+] {
+  let tmpfile = $"(mktemp -d)/Dockerfile";
+  open $file | str replace 'ghcr.io/windmill-labs/windmill-ee:dev' (get_ident -f $features) | save $tmpfile;
+  return $tmpfile;
+}