#!/bin/bash
set -euo pipefail
ZARYA_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
source "${ZARYA_DIR}/lib/common.sh"

mode="${1:-all}"

pack_atlas() {
    log_info "==> [1/3] Упаковка чистой архитектуры Atlas (atlas-code.xml)..."
    (cd "${ATLAS_DIR}" && repomix --config "${ATLAS_DIR}/repomix.config.json")
    log_ok "Файл ${ATLAS_DIR}/atlas-code.xml готов."
}

pack_zarya() {
    log_info "==> [2/3] Упаковка станции оператора Zarya (zarya-code.xml)..."
    (cd "${ZARYA_DIR}" && repomix --config "${ZARYA_DIR}/repomix.config.json")
    log_ok "Файл ${ZARYA_DIR}/zarya-code.xml готов."
}

pack_state() {
    log_info "==> [3/3] Снятие живого аудита и упаковка телеметрии (zarya-state.xml)..."
    "${ZARYA_DIR}/state/collect_state.sh" all
    (cd "${ZARYA_DIR}" && repomix --config "${ZARYA_DIR}/repomix.state.json")
    log_ok "Файл ${ZARYA_DIR}/zarya-state.xml готов."
}

case "${mode}" in
    all)
        pack_atlas
        echo ""
        pack_zarya
        echo ""
        pack_state
        echo ""
        log_ok "Все три контекста Repomix собраны без ошибок."
        ;;
    atlas)
        pack_atlas
        ;;
    code)
        pack_zarya
        ;;
    state)
        pack_state
        ;;
    *)
        log_err "Использование: z pack [all|atlas|code|state]"
        exit 1
        ;;
esac