auspex/config.go

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
	"strings"
)

// =============================================================================
// Глобальная конфигурация Auspex
// =============================================================================
//
// Конфигурация загружается из /etc/auspex/auspex.env
// Fallback значения используются если env файл не найден
// =============================================================================

const (
	// Путь к файлу конфигурации
	envFilePath = "/etc/auspex/auspex.env"
)

// --- Telegram ---
var (
	cfgTelegramBotToken = ""
	cfgTelegramChatID   = ""
)

// --- PostgreSQL Database ---
var (
	cfgDBUser     = ""
	cfgDBPassword = ""
	cfgDBName     = "waf_info" // Значение по умолчанию
	cfgDBPort     = 5432       // Значение по умолчанию
)

// --- Пути к хостам БД ---
var (
	cfgPrimaryDBHost   = "10.100.10.8"
	cfgSecondaryDBHost = "10.100.13.5"
)

// --- Пороги сертификатов (дни) ---
var (
	cfgCertCriticalDays = 3
	cfgCertWarningDays  = 15
	cfgCertInfoDays     = 30
)

// --- Пути ---
var (
	cfgPTAFBaseDir     = "/home/install/ptaf"
	cfgAngieConfDir    = "/etc/angie/http.d"
	cfgSSLDir          = "/etc/ssl"
	cfgNginxConfigGlob = "/home/install/conf/ptaf-nginx/*/*/*/*.conf"
	cfgStateFilePath   = "/tmp/port_checker_state.json"
)

// --- Параметры проверки портов ---
var (
	cfgCheckTimeoutSec = 10
	cfgMaxRetries      = 3
	cfgRetryDelaySec   = 2
)

// cfgMaxWorkers — количество горутин для параллельной обработки
var cfgMaxWorkers = 10

// --- Log Check ---
var (
	cfgLogDir          = "/var/log/ptaf_nginx"
	cfgLogStaleMinutes = 61 // Порог в минутах для определения устаревшего лога
)

// =============================================================================
// Загрузка конфигурации из файла
// =============================================================================

// loadConfig загружает конфигурацию из /etc/auspex/auspex.env
func loadConfig() error {
	file, err := os.Open(envFilePath)
	if err != nil {
		// Файл не найден - используем значения по умолчанию
		return fmt.Errorf("файл конфигурации не найден: %s", envFilePath)
	}
	defer file.Close()

	envVars := make(map[string]string)
	scanner := bufio.NewScanner(file)
	lineNum := 0

	for scanner.Scan() {
		lineNum++
		line := strings.TrimSpace(scanner.Text())

		// Пропускаем пустые строки и комментарии
		if line == "" || strings.HasPrefix(line, "#") {
			continue
		}

		// Парсим KEY=VALUE
		parts := strings.SplitN(line, "=", 2)
		if len(parts) != 2 {
			fmt.Printf("⚠️  Некорректная строка %d в %s: %s\n", lineNum, envFilePath, line)
			continue
		}

		key := strings.TrimSpace(parts[0])
		value := strings.TrimSpace(parts[1])

		// Убираем кавычки если есть
		value = strings.Trim(value, `"'`)

		envVars[key] = value
	}

	if err := scanner.Err(); err != nil {
		return fmt.Errorf("ошибка чтения файла: %v", err)
	}

	// Применяем значения
	applyConfig(envVars)

	return nil
}

// applyConfig применяет загруженные значения к переменным конфигурации
func applyConfig(env map[string]string) {
	// Telegram
	if v, ok := env["TELEGRAM_BOT_TOKEN"]; ok && v != "" {
		cfgTelegramBotToken = v
	}
	if v, ok := env["TELEGRAM_CHAT_ID"]; ok && v != "" {
		cfgTelegramChatID = v
	}

	// Database
	if v, ok := env["DB_USER"]; ok && v != "" {
		cfgDBUser = v
	}
	if v, ok := env["DB_PASSWORD"]; ok && v != "" {
		cfgDBPassword = v
	}
	if v, ok := env["DB_NAME"]; ok && v != "" {
		cfgDBName = v
	}
	if v, ok := env["DB_PORT"]; ok && v != "" {
		if port, err := strconv.Atoi(v); err == nil {
			cfgDBPort = port
		}
	}
	if v, ok := env["PRIMARY_DB_HOST"]; ok && v != "" {
		cfgPrimaryDBHost = v
	}
	if v, ok := env["SECONDARY_DB_HOST"]; ok && v != "" {
		cfgSecondaryDBHost = v
	}

	// Paths
	if v, ok := env["PTAF_BASE_DIR"]; ok && v != "" {
		cfgPTAFBaseDir = v
	}
	if v, ok := env["ANGIE_CONF_DIR"]; ok && v != "" {
		cfgAngieConfDir = v
	}
	if v, ok := env["SSL_DIR"]; ok && v != "" {
		cfgSSLDir = v
	}
	if v, ok := env["NGINX_CONFIG_GLOB"]; ok && v != "" {
		cfgNginxConfigGlob = v
	}
	if v, ok := env["STATE_FILE_PATH"]; ok && v != "" {
		cfgStateFilePath = v
	}
	if v, ok := env["LOG_DIR"]; ok && v != "" {
		cfgLogDir = v
	}

	// Thresholds
	if v, ok := env["CERT_CRITICAL_DAYS"]; ok && v != "" {
		if days, err := strconv.Atoi(v); err == nil {
			cfgCertCriticalDays = days
		}
	}
	if v, ok := env["CERT_WARNING_DAYS"]; ok && v != "" {
		if days, err := strconv.Atoi(v); err == nil {
			cfgCertWarningDays = days
		}
	}
	if v, ok := env["CERT_INFO_DAYS"]; ok && v != "" {
		if days, err := strconv.Atoi(v); err == nil {
			cfgCertInfoDays = days
		}
	}
	if v, ok := env["LOG_STALE_MINUTES"]; ok && v != "" {
		if mins, err := strconv.Atoi(v); err == nil {
			cfgLogStaleMinutes = mins
		}
	}

	// Timeouts and retries
	if v, ok := env["CHECK_TIMEOUT_SEC"]; ok && v != "" {
		if sec, err := strconv.Atoi(v); err == nil {
			cfgCheckTimeoutSec = sec
		}
	}
	if v, ok := env["MAX_RETRIES"]; ok && v != "" {
		if retries, err := strconv.Atoi(v); err == nil {
			cfgMaxRetries = retries
		}
	}
	if v, ok := env["RETRY_DELAY_SEC"]; ok && v != "" {
		if sec, err := strconv.Atoi(v); err == nil {
			cfgRetryDelaySec = sec
		}
	}
	if v, ok := env["MAX_WORKERS"]; ok && v != "" {
		if workers, err := strconv.Atoi(v); err == nil {
			cfgMaxWorkers = workers
		}
	}
}

// =============================================================================
// Вспомогательные функции
// =============================================================================

func isTelegramConfigured() bool {
	return cfgTelegramBotToken != "" &&
		cfgTelegramBotToken != "YOUR_BOT_TOKEN" &&
		cfgTelegramChatID != "" &&
		cfgTelegramChatID != "YOUR_CHAT_ID"
}

// printConfigStatus выводит статус загруженной конфигурации
func printConfigStatus() {
	fmt.Println("📋 Статус конфигурации:")

	// Telegram
	if isTelegramConfigured() {
		fmt.Println("   ✅ Telegram: настроен")
	} else {
		fmt.Println("   ⚠️  Telegram: не настроен")
	}

	// Database
	if cfgDBUser != "" && cfgDBPassword != "" {
		fmt.Printf("   ✅ База данных: %s@%s:%d/%s\n", cfgDBUser, cfgPrimaryDBHost, cfgDBPort, cfgDBName)
	} else {
		fmt.Println("   ⚠️  База данных: не настроена")
	}

	fmt.Println()
}