From c78dc531a571da2a2c369eb889aa9a8c768258ed Mon Sep 17 00:00:00 2001
From: Alex Flores <domain@me.com>
Date: Sat, 24 Feb 2024 11:53:14 -0500
Subject: [PATCH] adds starnet2

---
 Casks/s/starnet2.rb | 64 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 Casks/s/starnet2.rb

diff --git a/Casks/s/starnet2.rb b/Casks/s/starnet2.rb
new file mode 100644
index 0000000000000..ac28bbd17259f
--- /dev/null
+++ b/Casks/s/starnet2.rb
@@ -0,0 +1,64 @@
+cask "starnet2" do
+  version "02,2023"
+  sha256 "d566cd0a59d15894a3b027354812707b4408ca0917275af36602520f4ddec918"
+
+  url "https://www.starnetastro.com/wp-content/uploads/#{version.csv.second}/#{version.csv.first}/StarNet2T_MacOS.zip"
+  name "starnet2"
+  desc "Removes stars from astrophotography images using ML models"
+  homepage "https://www.starnetastro.com/"
+
+  livecheck do
+    url "https://www.starnetastro.com/experimental/"
+    regex(%r{uploads/(\d+)/(\d+)/StarNet2T_MacOS.zip}i)
+    strategy :page_match do |page, regex|
+      page.scan(regex).map { |match| "#{match[1]},#{match[0]}" }
+    end
+  end
+
+  depends_on arch: :arm64
+
+  bin_path = "#{staged_path}/StarNet2T_MacOS"
+  installer = "#{bin_path}/installer.sh"
+  shim = "#{bin_path}/shim.sh"
+
+  installer script: {
+    executable: installer,
+    sudo:       true,
+  }
+  binary shim, target: "starnet2"
+
+  preflight do
+    File.write installer, <<~EOS
+      #!/bin/sh
+
+      chmod 0755 #{bin_path}/lib/*
+      mkdir -p /usr/local/lib
+      cp #{bin_path}/lib/* /usr/local/lib/
+    EOS
+
+    File.write shim, <<~EOS
+      #!/bin/sh
+
+      # delete the symlink on process exit
+      cleanup() {
+        rm -f StarNet2_weights.pt
+      }
+      trap cleanup RETURN EXIT SIGINT SIGKILL
+
+      # the binary hardcodes the weights path so we have to symlink it to the CWD
+      ln -sf #{bin_path}/StarNet2_weights.pt .
+      #{bin_path}/starnet2 $@
+    EOS
+  end
+
+  uninstall delete: [
+    "/usr/local/lib/libc10.dylib",
+    "/usr/local/lib/libopencv_*",
+    "/usr/local/lib/libtorch.dylib",
+    "/usr/local/lib/libtorch_cpu.dylib",
+  ]
+
+  caveats do
+    files_in_usr_local
+  end
+end