From c1e1be2ab2f278cf80aa5c52dbb996a62823fe2b Mon Sep 17 00:00:00 2001
From: admin <admin@sechpoint.com>
Date: Sat, 1 Aug 2026 15:34:02 +0000
Subject: [PATCH] fix: deploy.sh copies source to /opt/wallarm/source/ for
 local rebuilds

- Binary + source installed to /opt/wallarm/
- Clients can rebuild: cd /opt/wallarm/source && go build ./cmd/wallarm/
- Source stays at repo root, not in a subdirectory
---
 deploy.sh | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/deploy.sh b/deploy.sh
index e9108d2..8504279 100644
--- a/deploy.sh
+++ b/deploy.sh
@@ -77,14 +77,24 @@ fi
 
 # ── Install ───────────────────────────────────────────────────────────
 mkdir -p "$WALLARM_DIR"
-cp wallarm "$WALLARM_DIR/wallarm"
+cp "$BUILD_DIR/wallarm" "$WALLARM_DIR/wallarm"
 chmod +x "$WALLARM_DIR/wallarm"
 
+# Copy source for local rebuilds
+rm -rf "$WALLARM_DIR/source"
+cp -r "$BUILD_DIR/cmd" "$BUILD_DIR/internal" "$BUILD_DIR/go.mod" "$BUILD_DIR/go.sum" "$WALLARM_DIR/source/" 2>/dev/null || true
+mkdir -p "$WALLARM_DIR/source/cmd" "$WALLARM_DIR/source/internal" 2>/dev/null || true
+cp -r "$BUILD_DIR/cmd" "$BUILD_DIR/internal" "$WALLARM_DIR/source/" 2>/dev/null || true
+cp "$BUILD_DIR/go.mod" "$BUILD_DIR/go.sum" "$WALLARM_DIR/source/" 2>/dev/null || true
+
 echo
 echo -e "${GREEN}${BOLD}Ready!${NC}"
 echo
-echo -e "  ${CYAN}sudo ${WALLARM_DIR}/wallarm${NC}     — Start the TUI"
-echo -e "  ${CYAN}${WALLARM_DIR}/wallarm --help${NC}   — Show all commands"
-echo -e "  ${CYAN}${WALLARM_DIR}/wallarm --tunnel${NC} — Start remote tunnel"
+echo -e "  ${CYAN}sudo ${WALLARM_DIR}/wallarm${NC}        — Start the TUI"
+echo -e "  ${CYAN}${WALLARM_DIR}/wallarm --help${NC}      — Show all commands"
+echo -e "  ${CYAN}${WALLARM_DIR}/wallarm --tunnel${NC}    — Start remote tunnel"
+echo
+echo -e "  ${CYAN}${WALLARM_DIR}/source/${NC}            — Source for local rebuilds"
+echo -e "  ${CYAN}cd ${WALLARM_DIR}/source && go build ./cmd/wallarm/${NC}"
 echo
 ls -lh "$WALLARM_DIR/wallarm"