release script

Author: okaris

scripts/release.sh

@@ -9,6 +9,27 @@ fi
 
 VERSION_TYPE=$1
 
+# Store the original version before any changes
+ORIGINAL_VERSION=$(poetry version -s)
+
+# Function to rollback changes
+rollback() {
+    echo "Error occurred. Rolling back changes..."
+    # Reset version in pyproject.toml
+    poetry version "$ORIGINAL_VERSION"
+    # Remove local tag if it exists
+    if git tag | grep -q "v${NEW_VERSION}"; then
+        git tag -d "v${NEW_VERSION}"
+    fi
+    # Reset any changes in git
+    git reset --hard HEAD^
+    echo "Rollback complete. Version restored to ${ORIGINAL_VERSION}"
+    exit 1
+}
+
+# Set up trap to catch errors
+trap rollback ERR
+
 # Ensure we're on main branch
 current_branch=$(git branch --show-current)
 if [ "$current_branch" != "main" ]; then
@@ -50,3 +71,6 @@ gh release create "v${NEW_VERSION}" \
     --generate-notes
 
 echo "Released v${NEW_VERSION} successfully!"
+
+# Remove the trap since we succeeded
+trap - ERR