#!/bin/bash

# ============================================================================
# Upgrade Existing SystemD Services to Production-Grade
# ============================================================================

set -e

# Colors for output
GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color

AITBC_ROOT="${AITBC_ROOT:-/opt/aitbc}"
VENV_DIR="$AITBC_ROOT/venv"

echo -e "${BLUE}🔧 UPGRADING EXISTING SYSTEMD SERVICES${NC}"
echo "=================================="
echo "Upgrading existing services to production-grade"
echo ""

# Step 1: Upgrade blockchain service
echo -e "${CYAN}⛓️  Step 1: Upgrade Blockchain Service${NC}"
echo "=================================="

# Backup original service
cp /opt/aitbc/systemd/aitbc-blockchain-node.service /opt/aitbc/systemd/aitbc-blockchain-node.service.backup

# Create production-grade blockchain service
cat > /opt/aitbc/systemd/aitbc-blockchain-node.service << 'EOF'
[Unit]
Description=AITBC Production Blockchain Node
After=network.target postgresql.service redis.service
Wants=postgresql.service redis.service

[Service]
Type=simple
User=root
Group=root
WorkingDirectory=/opt/aitbc
Environment=PATH=/usr/bin:/usr/local/bin:/usr/bin:/bin
Environment=NODE_ID=aitbc
Environment=PYTHONPATH=/opt/aitbc/production/services
EnvironmentFile=/opt/aitbc/production/.env

# Production execution
ExecStart=/opt/aitbc/venv/bin/python /opt/aitbc/production/services/blockchain_simple.py
ExecReload=/bin/kill -HUP $MAINPID
KillMode=mixed
TimeoutStopSec=10

# Production reliability
Restart=always
RestartSec=5
StartLimitBurst=5
StartLimitIntervalSec=60

# Production logging
StandardOutput=journal
StandardError=journal
SyslogIdentifier=aitbc-blockchain-production

# Production security
NoNewPrivileges=true
ProtectSystem=strict
ProtectHome=true
ReadWritePaths=/opt/aitbc/production/data/blockchain /opt/aitbc/production/logs/blockchain

# Production performance
LimitNOFILE=65536
LimitNPROC=4096
MemoryMax=2G
CPUQuota=50%

[Install]
WantedBy=multi-user.target
EOF

echo "✅ Blockchain service upgraded to production-grade"

# Step 2: Upgrade marketplace service
echo -e "${CYAN}🏪 Step 2: Upgrade Marketplace Service${NC}"
echo "===================================="

# Backup original service
cp /opt/aitbc/systemd/aitbc-marketplace.service /opt/aitbc/systemd/aitbc-marketplace.service.backup

# Create production-grade marketplace service
cat > /opt/aitbc/systemd/aitbc-marketplace.service << 'EOF'
[Unit]
Description=AITBC Production Marketplace Service
After=network.target aitbc-blockchain-node.service postgresql.service redis.service
Wants=aitbc-blockchain-node.service postgresql.service redis.service

[Service]
Type=simple
User=root
Group=root
WorkingDirectory=/opt/aitbc
Environment=PATH=/usr/bin:/usr/local/bin:/usr/bin:/bin
Environment=NODE_ID=aitbc
Environment=MARKETPLACE_PORT=8002
Environment=WORKERS=4
Environment=PYTHONPATH=/opt/aitbc/production/services
EnvironmentFile=/opt/aitbc/production/.env

# Production execution
ExecStart=/opt/aitbc/venv/bin/python /opt/aitbc/production/services/marketplace.py
ExecReload=/bin/kill -HUP $MAINPID
KillMode=mixed
TimeoutStopSec=10

# Production reliability
Restart=always
RestartSec=5
StartLimitBurst=5
StartLimitIntervalSec=60

# Production logging
StandardOutput=journal
StandardError=journal
SyslogIdentifier=aitbc-marketplace-production

# Production security
NoNewPrivileges=true
ProtectSystem=strict
ProtectHome=true
ReadWritePaths=/opt/aitbc/production/data/marketplace /opt/aitbc/production/logs/marketplace

# Production performance
LimitNOFILE=65536
LimitNPROC=4096
MemoryMax=1G
CPUQuota=25%

[Install]
WantedBy=multi-user.target
EOF

echo "✅ Marketplace service upgraded to production-grade"

# Step 3: Upgrade GPU service
echo -e "${CYAN}🖥️  Step 3: Upgrade GPU Service${NC}"
echo "=============================="

# Backup original service
cp /opt/aitbc/systemd/aitbc-gpu.service /opt/aitbc/systemd/aitbc-gpu.service.backup

# Create production-grade GPU service
cat > /opt/aitbc/systemd/aitbc-gpu.service << 'EOF'
[Unit]
Description=AITBC Production GPU Marketplace Service
After=network.target aitbc-marketplace.service nvidia-persistenced.service
Wants=aitbc-marketplace.service nvidia-persistenced.service

[Service]
Type=simple
User=root
Group=root
WorkingDirectory=/opt/aitbc
Environment=PATH=/usr/bin:/usr/local/bin:/usr/bin:/bin
Environment=NODE_ID=aitbc
Environment=GPU_MARKETPLACE_PORT=8003
Environment=PYTHONPATH=/opt/aitbc/production/services
EnvironmentFile=/opt/aitbc/production/.env

# GPU access
DeviceAllow=/dev/nvidia* rw
DevicePolicy=auto

# Production execution
ExecStart=/opt/aitbc/venv/bin/python -c "
import sys
sys.path.insert(0, '/opt/aitbc/production/services')
from marketplace import ProductionMarketplace
import uvicorn
import os

app = ProductionMarketplace().app
uvicorn.run(app, host='0.0.0.0', port=int(os.getenv('GPU_MARKETPLACE_PORT', 8003)))
"
ExecReload=/bin/kill -HUP $MAINPID
KillMode=mixed
TimeoutStopSec=10

# Production reliability
Restart=always
RestartSec=5
StartLimitBurst=5
StartLimitIntervalSec=60

# Production logging
StandardOutput=journal
StandardError=journal
SyslogIdentifier=aitbc-gpu-marketplace-production

# Production security
NoNewPrivileges=true
ProtectSystem=strict
ProtectHome=true
ReadWritePaths=/opt/aitbc/production/data/marketplace /opt/aitbc/production/logs/marketplace

# Production performance
LimitNOFILE=65536
LimitNPROC=4096
MemoryMax=2G
CPUQuota=75%

[Install]
WantedBy=multi-user.target
EOF

echo "✅ GPU service upgraded to production-grade"

# Step 4: Create production monitoring service
echo -e "${CYAN}📊 Step 4: Create Production Monitoring${NC}"
echo "======================================"

cat > /opt/aitbc/systemd/aitbc-production-monitor.service << 'EOF'
[Unit]
Description=AITBC Production Monitoring Service
After=network.target aitbc-blockchain-node.service aitbc-marketplace.service

[Service]
Type=simple
User=root
Group=root
WorkingDirectory=/opt/aitbc
Environment=PATH=/usr/bin:/usr/local/bin:/usr/bin:/bin
Environment=NODE_ID=aitbc
Environment=PYTHONPATH=/opt/aitbc/production/services
EnvironmentFile=/opt/aitbc/production/.env

# Production monitoring
ExecStart=/opt/aitbc/venv/bin/python -c "
import time
import logging
import json
from pathlib import Path

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger('production-monitor')

while True:
    try:
        # Monitor blockchain
        blockchain_file = Path('/opt/aitbc/production/data/blockchain/aitbc/blockchain.json')
        if blockchain_file.exists():
            with open(blockchain_file, 'r') as f:
                data = json.load(f)
                logger.info(f'Blockchain: {len(data.get(\"blocks\", []))} blocks')
        
        # Monitor marketplace
        marketplace_dir = Path('/opt/aitbc/production/data/marketplace')
        if marketplace_dir.exists():
            listings_file = marketplace_dir / 'gpu_listings.json'
            if listings_file.exists():
                with open(listings_file, 'r') as f:
                    listings = json.load(f)
                    logger.info(f'Marketplace: {len(listings)} GPU listings')
        
        # Monitor system resources
        import psutil
        cpu_percent = psutil.cpu_percent()
        memory_percent = psutil.virtual_memory().percent
        logger.info(f'System: CPU {cpu_percent}%, Memory {memory_percent}%')
        
        time.sleep(30)  # Monitor every 30 seconds
        
    except Exception as e:
        logger.error(f'Monitoring error: {e}')
        time.sleep(60)
"

# Production reliability
Restart=always
RestartSec=10

# Production logging
StandardOutput=journal
StandardError=journal
SyslogIdentifier=aitbc-production-monitor

# Production security
NoNewPrivileges=true
ProtectSystem=strict
ProtectHome=true
ReadWritePaths=/opt/aitbc/production/data /opt/aitbc/production/logs

[Install]
WantedBy=multi-user.target
EOF

echo "✅ Production monitoring service created"

# Step 5: Reload systemd and enable services
echo -e "${CYAN}🔄 Step 5: Reload SystemD and Enable${NC}"
echo "=================================="

# Reload systemd daemon
systemctl daemon-reload

# Enable production services
echo "Enabling production services..."
systemctl enable aitbc-blockchain-node.service
systemctl enable aitbc-marketplace.service
systemctl enable aitbc-gpu.service
systemctl enable aitbc-production-monitor.service

echo "✅ SystemD services reloaded and enabled"

# Step 6: Test production services on localhost
echo -e "${CYAN}🧪 Step 6: Test Production Services${NC}"
echo "==============================="

echo "Starting production services..."
systemctl start aitbc-blockchain-node.service
sleep 2
systemctl start aitbc-marketplace.service
sleep 2
systemctl start aitbc-gpu.service
sleep 2
systemctl start aitbc-production-monitor.service

# Check service status
echo "Checking service status..."
systemctl status aitbc-blockchain-node.service --no-pager -l | head -10
systemctl status aitbc-marketplace.service --no-pager -l | head -10
systemctl status aitbc-gpu.service --no-pager -l | head -10

# Test service endpoints
echo "Testing service endpoints..."
sleep 5
curl -s http://localhost:8002/health | head -5 || echo "Marketplace service not ready"
curl -s http://localhost:8003/health | head -5 || echo "GPU marketplace service not ready"

# Step 7: Deploy to aitbc1
echo -e "${CYAN}🚀 Step 7: Deploy to aitbc1${NC}"
echo "========================"

# Copy production services to aitbc1
echo "Copying production services to aitbc1..."
scp -r /opt/aitbc/production aitbc1:/opt/aitbc/
scp /opt/aitbc/systemd/aitbc-blockchain-node.service aitbc1:/opt/aitbc/systemd/
scp /opt/aitbc/systemd/aitbc-marketplace.service aitbc1:/opt/aitbc/systemd/
scp /opt/aitbc/systemd/aitbc-gpu.service aitbc1:/opt/aitbc/systemd/
scp /opt/aitbc/systemd/aitbc-production-monitor.service aitbc1:/opt/aitbc/systemd/

# Update services for aitbc1 node
echo "Configuring services for aitbc1..."
ssh aitbc1 "sed -i 's/NODE_ID=aitbc/NODE_ID=aitbc1/g' /opt/aitbc/systemd/aitbc-blockchain-node.service"
ssh aitbc1 "sed -i 's/NODE_ID=aitbc/NODE_ID=aitbc1/g' /opt/aitbc/systemd/aitbc-marketplace.service"
ssh aitbc1 "sed -i 's/NODE_ID=aitbc/NODE_ID=aitbc1/g' /opt/aitbc/systemd/aitbc-gpu.service"
ssh aitbc1 "sed -i 's/NODE_ID=aitbc/NODE_ID=aitbc1/g' /opt/aitbc/systemd/aitbc-production-monitor.service"

# Update ports for aitbc1
ssh aitbc1 "sed -i 's/MARKETPLACE_PORT=8002/MARKETPLACE_PORT=8004/g' /opt/aitbc/systemd/aitbc-marketplace.service"
ssh aitbc1 "sed -i 's/GPU_MARKETPLACE_PORT=8003/GPU_MARKETPLACE_PORT=8005/g' /opt/aitbc/systemd/aitbc-gpu.service"

# Deploy and start services on aitbc1
echo "Starting services on aitbc1..."
ssh aitbc1 "systemctl daemon-reload"
ssh aitbc1 "systemctl enable aitbc-blockchain-node.service aitbc-marketplace.service aitbc-gpu.service aitbc-production-monitor.service"
ssh aitbc1 "systemctl start aitbc-blockchain-node.service"
sleep 3
ssh aitbc1 "systemctl start aitbc-marketplace.service"
sleep 3
ssh aitbc1 "systemctl start aitbc-gpu.service"
sleep 3
ssh aitbc1 "systemctl start aitbc-production-monitor.service"

# Check aitbc1 services
echo "Checking aitbc1 services..."
ssh aitbc1 "systemctl status aitbc-blockchain-node.service --no-pager -l | head -5"
ssh aitbc1 "systemctl status aitbc-marketplace.service --no-pager -l | head -5"

# Test aitbc1 endpoints
echo "Testing aitbc1 endpoints..."
ssh aitbc1 "curl -s http://localhost:8004/health | head -5" || echo "aitbc1 marketplace not ready"
ssh aitbc1 "curl -s http://localhost:8005/health | head -5" || echo "aitbc1 GPU marketplace not ready"

echo ""
echo -e "${GREEN}🎉 PRODUCTION SYSTEMD SERVICES UPGRADED!${NC}"
echo "======================================"
echo ""
echo "✅ Upgraded Services:"
echo "   • aitbc-blockchain-node.service (Production blockchain)"
echo "   • aitbc-marketplace.service (Production marketplace)"
echo "   • aitbc-gpu.service (Production GPU marketplace)"
echo "   • aitbc-production-monitor.service (Production monitoring)"
echo ""
echo "✅ Production Features:"
echo "   • Real database persistence"
echo "   • Production logging and monitoring"
echo "   • Resource limits and security"
echo "   • Automatic restart and recovery"
echo "   • Multi-node deployment"
echo ""
echo "✅ Service Endpoints:"
echo "   • aitbc (localhost):"
echo "     - Blockchain: SystemD managed"
echo "     - Marketplace: http://localhost:8002"
echo "     - GPU Marketplace: http://localhost:8003"
echo "   • aitbc1 (remote):"
echo "     - Blockchain: SystemD managed"
echo "     - Marketplace: http://aitbc1:8004"
echo "     - GPU Marketplace: http://aitbc1:8005"
echo ""
echo "✅ Monitoring:"
echo "   • SystemD journal: journalctl -u aitbc-*"
echo "   • Production logs: /opt/aitbc/production/logs/"
echo "   • Service status: systemctl status aitbc-*"
echo ""
echo -e "${BLUE}🚀 Production SystemD services ready!${NC}"