#!/bin/bash
# SaaS Guard Agent Wrapper Script
# This script loads config and runs the agent

CONFIG_FILE="/etc/saas-guard/agent.conf"
AGENT_BIN="/opt/saas-guard/agent"

# Load configuration
if [[ -f "$CONFIG_FILE" ]]; then
    source "$CONFIG_FILE"
fi

# Export environment variables for agent
export SAAS_GUARD_SERVER_ADDR="${SERVER_ADDR:-localhost:50051}"
export SAAS_GUARD_LICENSE_KEY="${LICENSE_KEY:-}"
export SAAS_GUARD_HOSTNAME="${HOSTNAME:-$(hostname)}"
export SAAS_GUARD_HTTP_BASE="${HTTP_BASE:-}"
export SAAS_GUARD_SCAN_PATHS="${SCAN_PATHS:-/var/www,/home,/tmp}"
export SAAS_GUARD_QUARANTINE_DIR="${QUARANTINE_DIR:-/var/lib/saas-guard/quarantine}"
export SAAS_GUARD_IGNORE_LIST="${IGNORE_LIST:-}"
export SAAS_GUARD_EXCLUDE_PATHS="${EXCLUDE_PATHS:-}"
export SAAS_GUARD_SKIP_CPANEL_MAIL="${SKIP_CPANEL_MAIL:-true}"
export SAAS_GUARD_MAX_CPU_PERCENT="${MAX_CPU_PERCENT:-0}"
export SAAS_GUARD_MAX_MEMORY_MB="${MAX_MEMORY_MB:-0}"
export SAAS_GUARD_MAX_IO_READ_MB_PER_SEC="${MAX_IO_READ_MB_PER_SEC:-0}"
export SAAS_GUARD_MAX_IO_WRITE_MB_PER_SEC="${MAX_IO_WRITE_MB_PER_SEC:-0}"
export SAAS_GUARD_LOG_LEVEL="${LOG_LEVEL:-info}"

# Run agent
exec $AGENT_BIN