...
This commit is contained in:
25
pkg/system/builders/hetznerinstall/cmd/build.sh
Executable file
25
pkg/system/builders/hetznerinstall/cmd/build.sh
Executable file
@@ -0,0 +1,25 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
|
||||
# Change to the script's directory to ensure relative paths work
|
||||
cd "$(dirname "$0")"
|
||||
|
||||
echo "Building Hetzner Installer for Linux on AMD64..."
|
||||
|
||||
# Create build directory if it doesn't exist
|
||||
mkdir -p build
|
||||
|
||||
# Build the Hetzner installer binary
|
||||
echo "Building Hetzner installer..."
|
||||
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build \
|
||||
-ldflags="-s -w" \
|
||||
-trimpath \
|
||||
-o build/hetzner_installer \
|
||||
main.go # Reference main.go in the current directory
|
||||
|
||||
# Set executable permissions
|
||||
chmod +x build/hetzner_installer
|
||||
|
||||
# Output binary info
|
||||
echo "Build complete!"
|
||||
ls -lh build/
|
||||
53
pkg/system/builders/hetznerinstall/cmd/main.go
Normal file
53
pkg/system/builders/hetznerinstall/cmd/main.go
Normal file
@@ -0,0 +1,53 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/freeflowuniverse/heroagent/pkg/system/builders/hetznerinstall"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Define command-line flags
|
||||
hostname := flag.String("hostname", "", "Target hostname for the server (required)")
|
||||
image := flag.String("image", hetznerinstall.DefaultImage, "OS image to install (e.g., Ubuntu-2404)")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
// Validate required flags
|
||||
if *hostname == "" {
|
||||
fmt.Fprintln(os.Stderr, "Error: -hostname flag is required.")
|
||||
flag.Usage()
|
||||
os.Exit(1)
|
||||
}
|
||||
// Drives are now always auto-detected by the builder
|
||||
|
||||
// Create a new HetznerInstall builder
|
||||
builder := hetznerinstall.NewBuilder().
|
||||
WithHostname(*hostname).
|
||||
WithImage(*image)
|
||||
|
||||
// Example: Add custom partitions (optional, overrides default)
|
||||
// builder.WithPartitions(
|
||||
// hetznerinstall.Partition{MountPoint: "/boot", FileSystem: "ext4", Size: "1G"},
|
||||
// hetznerinstall.Partition{MountPoint: "swap", FileSystem: "swap", Size: "4G"},
|
||||
// hetznerinstall.Partition{MountPoint: "/", FileSystem: "ext4", Size: "all"},
|
||||
// )
|
||||
|
||||
// Example: Enable Software RAID 1 (optional)
|
||||
// builder.WithSoftwareRAID(true, 1)
|
||||
|
||||
// Run the Hetzner installation process
|
||||
// The builder will handle drive detection/validation internally if drives were not set
|
||||
fmt.Printf("Starting Hetzner installation for hostname %s using image %s...\n",
|
||||
*hostname, *image)
|
||||
if err := builder.RunInstall(); err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error during Hetzner installation: %v\n", err)
|
||||
os.Exit(1) // Ensure we exit with non-zero status on error
|
||||
}
|
||||
|
||||
// Note: If RunInstall succeeds, the system typically reboots,
|
||||
// so this message might not always be seen.
|
||||
fmt.Println("Hetzner installation process initiated successfully!")
|
||||
}
|
||||
134
pkg/system/builders/hetznerinstall/cmd/run.sh
Executable file
134
pkg/system/builders/hetznerinstall/cmd/run.sh
Executable file
@@ -0,0 +1,134 @@
|
||||
#!/bin/bash
|
||||
set -e # Exit immediately if a command exits with a non-zero status.
|
||||
|
||||
# --- Configuration ---
|
||||
# Required Environment Variables:
|
||||
# SERVER: IPv4 or IPv6 address of the target Hetzner server (already in Rescue Mode).
|
||||
# HOSTNAME: The desired hostname for the installed system.
|
||||
# Drives are now always auto-detected by the installer binary.
|
||||
|
||||
LOG_FILE="hetzner_install_$(date +%Y%m%d_%H%M%S).log"
|
||||
REMOTE_USER="root" # Hetzner Rescue Mode typically uses root
|
||||
REMOTE_DIR="/tmp/hetzner_installer_$$" # Temporary directory on the remote server
|
||||
BINARY_NAME="hetzner_installer"
|
||||
BUILD_DIR="build"
|
||||
|
||||
# --- Helper Functions ---
|
||||
log() {
|
||||
local timestamp=$(date +"%Y-%m-%d %H:%M:%S")
|
||||
echo "[$timestamp] $1" | tee -a "$LOG_FILE"
|
||||
}
|
||||
|
||||
cleanup_remote() {
|
||||
if [ -n "$SERVER" ]; then
|
||||
log "Cleaning up remote directory $REMOTE_DIR on $SERVER..."
|
||||
ssh "$REMOTE_USER@$SERVER" "rm -rf $REMOTE_DIR" || log "Warning: Failed to clean up remote directory (might be okay if server rebooted)."
|
||||
fi
|
||||
}
|
||||
|
||||
# --- Main Script ---
|
||||
cd "$(dirname "$0")"
|
||||
|
||||
log "=== Starting Hetzner Installimage Deployment ==="
|
||||
log "Log file: $LOG_FILE"
|
||||
log "IMPORTANT: Ensure the target server ($SERVER) is booted into Hetzner Rescue Mode!"
|
||||
|
||||
# Check required environment variables
|
||||
if [ -z "$SERVER" ]; then
|
||||
log "❌ ERROR: SERVER environment variable is not set."
|
||||
log "Please set it to the IP address of the target server (in Rescue Mode)."
|
||||
exit 1
|
||||
fi
|
||||
if [ -z "$HOSTNAME" ]; then
|
||||
log "❌ ERROR: HOSTNAME environment variable is not set."
|
||||
log "Please set it to the desired hostname for the installed system."
|
||||
exit 1
|
||||
fi
|
||||
# Drives are auto-detected by the binary.
|
||||
|
||||
# Validate SERVER IP (basic check)
|
||||
if ! [[ "$SERVER" =~ ^[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+$ ]] && \
|
||||
! [[ "$SERVER" =~ ^[0-9a-fA-F:]+$ ]]; then
|
||||
log "❌ ERROR: SERVER ($SERVER) does not look like a valid IPv4 or IPv6 address."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log "Target Server: $SERVER"
|
||||
log "Target Hostname: $HOSTNAME"
|
||||
log "Target Drives: Auto-detected by the installer."
|
||||
|
||||
# Build the Hetzner installer binary
|
||||
log "Building $BINARY_NAME binary..."
|
||||
./build.sh | tee -a "$LOG_FILE"
|
||||
|
||||
# Check if binary exists
|
||||
BINARY_PATH="$BUILD_DIR/$BINARY_NAME"
|
||||
if [ ! -f "$BINARY_PATH" ]; then
|
||||
log "❌ ERROR: $BINARY_NAME binary not found at $BINARY_PATH after build."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log "Binary size:"
|
||||
ls -lh "$BINARY_PATH" | tee -a "$LOG_FILE"
|
||||
|
||||
# Set up trap for cleanup
|
||||
trap cleanup_remote EXIT
|
||||
|
||||
# Create deployment directory on server
|
||||
log "Creating temporary directory $REMOTE_DIR on server..."
|
||||
# Use -t to force pseudo-terminal allocation for mkdir (less critical but consistent)
|
||||
ssh -t "$REMOTE_USER@$SERVER" "mkdir -p $REMOTE_DIR" 2>&1 | tee -a "$LOG_FILE"
|
||||
if [ $? -ne 0 ]; then
|
||||
log "❌ ERROR: Failed to create remote directory $REMOTE_DIR on $SERVER."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Transfer the binary to the server
|
||||
log "Transferring $BINARY_NAME binary to $SERVER:$REMOTE_DIR/ ..."
|
||||
rsync -avz --progress "$BINARY_PATH" "$REMOTE_USER@$SERVER:$REMOTE_DIR/" 2>&1 | tee -a "$LOG_FILE"
|
||||
if [ $? -ne 0 ]; then
|
||||
log "❌ ERROR: Failed to transfer binary to $SERVER."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Ensure binary is executable on the server
|
||||
log "Setting permissions on server..."
|
||||
# Use -t
|
||||
ssh -t "$REMOTE_USER@$SERVER" "chmod +x $REMOTE_DIR/$BINARY_NAME" 2>&1 | tee -a "$LOG_FILE" || { log "❌ ERROR: Failed to set permissions on remote binary."; exit 1; }
|
||||
# Use -t
|
||||
ssh -t "$REMOTE_USER@$SERVER" "ls -la $REMOTE_DIR/" 2>&1 | tee -a "$LOG_FILE"
|
||||
|
||||
# Construct remote command arguments (only hostname needed now)
|
||||
# Note: The binary expects -hostname
|
||||
REMOTE_CMD_ARGS="-hostname \"$HOSTNAME\""
|
||||
|
||||
# Run the Hetzner installer (Go binary) on the server
|
||||
log "Running Go installer binary $BINARY_NAME on server $SERVER..."
|
||||
REMOTE_FULL_CMD="cd $REMOTE_DIR && ./$BINARY_NAME $REMOTE_CMD_ARGS"
|
||||
log "Command: $REMOTE_FULL_CMD"
|
||||
|
||||
# Execute the command and capture output. Use -t for better output.
|
||||
INSTALL_OUTPUT=$(ssh -t "$REMOTE_USER@$SERVER" "$REMOTE_FULL_CMD" 2>&1)
|
||||
INSTALL_EXIT_CODE=$?
|
||||
|
||||
log "--- Go Installer Binary Output ---"
|
||||
echo "$INSTALL_OUTPUT" | tee -a "$LOG_FILE"
|
||||
log "--- End Go Installer Binary Output ---"
|
||||
log "Go installer binary exit code: $INSTALL_EXIT_CODE"
|
||||
|
||||
# Analyze results - relies on Go binary output now
|
||||
if [[ "$INSTALL_OUTPUT" == *"installimage command finished. System should reboot shortly if successful."* ]]; then
|
||||
log "✅ SUCCESS: Go installer reported successful initiation. The server should be rebooting into the new OS."
|
||||
log "Verification of the installed OS must be done manually after reboot."
|
||||
elif [[ "$INSTALL_OUTPUT" == *"Error during Hetzner installation"* || $INSTALL_EXIT_CODE -ne 0 ]]; then
|
||||
log "❌ ERROR: Go installer reported an error or exited with code $INSTALL_EXIT_CODE."
|
||||
log "Check the output above for details. Common issues include installimage errors or config problems."
|
||||
# Don't exit immediately, allow cleanup trap to run
|
||||
else
|
||||
# This might happen if the SSH connection is abruptly closed by the reboot during installimage
|
||||
log "⚠️ WARNING: The Go installer finished with exit code $INSTALL_EXIT_CODE, but the output might be incomplete due to server reboot."
|
||||
log "Assuming the installimage process was initiated. Manual verification is required after reboot."
|
||||
fi
|
||||
|
||||
log "=== Hetzner Installimage Deployment Script Finished ==="
|
||||
# Cleanup trap will run on exit
|
||||
Reference in New Issue
Block a user