221 lines
7.9 KiB
Bash
221 lines
7.9 KiB
Bash
|
#!/bin/bash
|
||
|
|
# Enhance guest agent verification in all VM YAML templates
|
||
|
|
# Adds detailed verification commands matching the check script
|
||
|
|
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
PROJECT_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||
|
|
TEMPLATES_DIR="$PROJECT_ROOT/examples/production"
|
||
|
|
|
||
|
|
# Enhanced verification block
|
||
|
|
ENHANCED_VERIFICATION=' # Verify packages are installed
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Verifying required packages are installed..."
|
||
|
|
echo "=========================================="
|
||
|
|
for pkg in qemu-guest-agent curl wget net-tools chrony unattended-upgrades; do
|
||
|
|
if ! dpkg -l | grep -q "^ii.*$pkg"; then
|
||
|
|
echo "ERROR: Package $pkg is not installed"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
echo "✅ Package $pkg is installed"
|
||
|
|
done
|
||
|
|
echo "All required packages verified"
|
||
|
|
|
||
|
|
# Verify qemu-guest-agent package details
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Checking qemu-guest-agent package details..."
|
||
|
|
echo "=========================================="
|
||
|
|
if dpkg -l | grep -q "^ii.*qemu-guest-agent"; then
|
||
|
|
echo "✅ qemu-guest-agent package IS installed"
|
||
|
|
dpkg -l | grep qemu-guest-agent
|
||
|
|
else
|
||
|
|
echo "❌ qemu-guest-agent package is NOT installed"
|
||
|
|
echo "Attempting to install..."
|
||
|
|
apt-get update
|
||
|
|
apt-get install -y qemu-guest-agent
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Enable and start QEMU Guest Agent
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Enabling and starting QEMU Guest Agent..."
|
||
|
|
echo "=========================================="
|
||
|
|
systemctl enable qemu-guest-agent
|
||
|
|
systemctl start qemu-guest-agent
|
||
|
|
echo "QEMU Guest Agent enabled and started"
|
||
|
|
|
||
|
|
# Verify guest agent service is running
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Verifying QEMU Guest Agent service status..."
|
||
|
|
echo "=========================================="
|
||
|
|
for i in {1..30}; do
|
||
|
|
if systemctl is-active --quiet qemu-guest-agent; then
|
||
|
|
echo "✅ QEMU Guest Agent service IS running"
|
||
|
|
systemctl status qemu-guest-agent --no-pager -l
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
echo "Waiting for QEMU Guest Agent to start... ($i/30)"
|
||
|
|
sleep 1
|
||
|
|
done
|
||
|
|
echo "⚠️ WARNING: QEMU Guest Agent may not have started properly"
|
||
|
|
systemctl status qemu-guest-agent --no-pager -l || true
|
||
|
|
echo "Attempting to restart..."
|
||
|
|
systemctl restart qemu-guest-agent
|
||
|
|
sleep 3
|
||
|
|
if systemctl is-active --quiet qemu-guest-agent; then
|
||
|
|
echo "✅ QEMU Guest Agent started after restart"
|
||
|
|
else
|
||
|
|
echo "❌ QEMU Guest Agent failed to start"
|
||
|
|
fi'
|
||
|
|
|
||
|
|
# Old verification block pattern (to be replaced)
|
||
|
|
OLD_PATTERN_START=' # Verify packages are installed'
|
||
|
|
OLD_PATTERN_END=' systemctl status qemu-guest-agent --no-pager || true'
|
||
|
|
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Enhancing Guest Agent Verification"
|
||
|
|
echo "=========================================="
|
||
|
|
echo ""
|
||
|
|
echo "Target directory: $TEMPLATES_DIR"
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# Find all YAML files (excluding backups)
|
||
|
|
FILES=$(find "$TEMPLATES_DIR" -name "*.yaml" -type f ! -name "*.backup*" | sort)
|
||
|
|
|
||
|
|
if [ -z "$FILES" ]; then
|
||
|
|
echo "No YAML files found in $TEMPLATES_DIR"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
UPDATED_COUNT=0
|
||
|
|
SKIPPED_COUNT=0
|
||
|
|
|
||
|
|
for file in $FILES; do
|
||
|
|
# Check if file contains the old verification pattern
|
||
|
|
if ! grep -q "Verifying required packages are installed" "$file"; then
|
||
|
|
echo "⏭️ Skipping $file (no guest agent verification found)"
|
||
|
|
((SKIPPED_COUNT++))
|
||
|
|
continue
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Check if already enhanced
|
||
|
|
if grep -q "Checking qemu-guest-agent package details" "$file"; then
|
||
|
|
echo "✅ Already enhanced: $file"
|
||
|
|
continue
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "📝 Processing: $file"
|
||
|
|
|
||
|
|
# Create backup
|
||
|
|
cp "$file" "${file}.backup-$(date +%Y%m%d-%H%M%S)"
|
||
|
|
|
||
|
|
# Use Python for more reliable YAML manipulation
|
||
|
|
python3 <<PYTHON_SCRIPT
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
|
||
|
|
file_path = "$file"
|
||
|
|
|
||
|
|
with open(file_path, 'r') as f:
|
||
|
|
content = f.read()
|
||
|
|
|
||
|
|
# Find the old verification block
|
||
|
|
old_pattern = r' # Verify packages are installed.*? systemctl status qemu-guest-agent --no-pager \|\| true'
|
||
|
|
|
||
|
|
# Enhanced verification block (properly indented)
|
||
|
|
enhanced = ''' # Verify packages are installed
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Verifying required packages are installed..."
|
||
|
|
echo "=========================================="
|
||
|
|
for pkg in qemu-guest-agent curl wget net-tools chrony unattended-upgrades; do
|
||
|
|
if ! dpkg -l | grep -q "^ii.*\\$pkg"; then
|
||
|
|
echo "ERROR: Package \\$pkg is not installed"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
echo "✅ Package \\$pkg is installed"
|
||
|
|
done
|
||
|
|
echo "All required packages verified"
|
||
|
|
|
||
|
|
# Verify qemu-guest-agent package details
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Checking qemu-guest-agent package details..."
|
||
|
|
echo "=========================================="
|
||
|
|
if dpkg -l | grep -q "^ii.*qemu-guest-agent"; then
|
||
|
|
echo "✅ qemu-guest-agent package IS installed"
|
||
|
|
dpkg -l | grep qemu-guest-agent
|
||
|
|
else
|
||
|
|
echo "❌ qemu-guest-agent package is NOT installed"
|
||
|
|
echo "Attempting to install..."
|
||
|
|
apt-get update
|
||
|
|
apt-get install -y qemu-guest-agent
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Enable and start QEMU Guest Agent
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Enabling and starting QEMU Guest Agent..."
|
||
|
|
echo "=========================================="
|
||
|
|
systemctl enable qemu-guest-agent
|
||
|
|
systemctl start qemu-guest-agent
|
||
|
|
echo "QEMU Guest Agent enabled and started"
|
||
|
|
|
||
|
|
# Verify guest agent service is running
|
||
|
|
- |
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Verifying QEMU Guest Agent service status..."
|
||
|
|
echo "=========================================="
|
||
|
|
for i in {1..30}; do
|
||
|
|
if systemctl is-active --quiet qemu-guest-agent; then
|
||
|
|
echo "✅ QEMU Guest Agent service IS running"
|
||
|
|
systemctl status qemu-guest-agent --no-pager -l
|
||
|
|
exit 0
|
||
|
|
fi
|
||
|
|
echo "Waiting for QEMU Guest Agent to start... (\\$i/30)"
|
||
|
|
sleep 1
|
||
|
|
done
|
||
|
|
echo "⚠️ WARNING: QEMU Guest Agent may not have started properly"
|
||
|
|
systemctl status qemu-guest-agent --no-pager -l || true
|
||
|
|
echo "Attempting to restart..."
|
||
|
|
systemctl restart qemu-guest-agent
|
||
|
|
sleep 3
|
||
|
|
if systemctl is-active --quiet qemu-guest-agent; then
|
||
|
|
echo "✅ QEMU Guest Agent started after restart"
|
||
|
|
else
|
||
|
|
echo "❌ QEMU Guest Agent failed to start"
|
||
|
|
fi'''
|
||
|
|
|
||
|
|
# Try to match and replace
|
||
|
|
if re.search(old_pattern, content, re.DOTALL):
|
||
|
|
new_content = re.sub(old_pattern, enhanced, content, flags=re.DOTALL)
|
||
|
|
with open(file_path, 'w') as f:
|
||
|
|
f.write(new_content)
|
||
|
|
print(f"✅ Updated: {file_path}")
|
||
|
|
sys.exit(0)
|
||
|
|
else:
|
||
|
|
print(f"⚠️ Pattern not found in {file_path}")
|
||
|
|
sys.exit(1)
|
||
|
|
PYTHON_SCRIPT
|
||
|
|
|
||
|
|
if [ $? -eq 0 ]; then
|
||
|
|
((UPDATED_COUNT++))
|
||
|
|
else
|
||
|
|
echo "⚠️ Failed to update: $file"
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
echo "=========================================="
|
||
|
|
echo "Summary"
|
||
|
|
echo "=========================================="
|
||
|
|
echo "✅ Updated: $UPDATED_COUNT files"
|
||
|
|
echo "⏭️ Skipped: $SKIPPED_COUNT files"
|
||
|
|
echo ""
|
||
|
|
echo "Done!"
|
||
|
|
|