# Restarts the vmagent deployment; errors are intentionally ignored.
function metricRestart() {
    k3sRun rollout restart deployment/"$metricKube"-vmagent || true
}

# Writes Prometheus metrics (build info, vhost counts, domain counts, pod memory) to $metricPromPath/kube.prom.
function metricKube() {
    local fileProm="$metricPromPath/kube.prom"
    local fileTmp="${fileProm}.tmp"

    mkdir -p -- "$metricPromPath" || return 1

    local vhostAllCount vhostUpCount
    vhostAllCount=$(openlitespeedConfigVhostList all | grep -c . || true)
    vhostUpCount=$(openlitespeedConfigVhostList up | grep -c . || true)

    local appTimestamp
    appTimestamp=$(date -d "$appDate ${appTime//-/:}" +%s)

    local domainAllCount=-1 domainFreeCount=-1
    [[ -f "$domainsList" ]] && domainAllCount=$(grep -cP '^(?!\s*$)' "$domainsList" || true)
    [[ -f "$domainsList" ]] && domainFreeCount=$(grep -c '^#' "$domainsList" || true)

    local masterMemReq=-1 masterMemLim=-1
    local slaveMemReq=-1  slaveMemLim=-1
    local olsMemReq=-1    olsMemLim=-1
    local redisMemReq=-1  redisMemLim=-1
    local _line _req _lim

    _line=$(k3sRun get pod "${mariadbMasterKube}-0" \
        -o jsonpath="{.spec.containers[?(@.name=='${mariadbMasterKube}')].resources.requests.memory} {.spec.containers[?(@.name=='${mariadbMasterKube}')].resources.limits.memory}" 2>/dev/null)
    read -r _req _lim <<< "$_line"
    masterMemReq=$(sizeToBytes "$_req"); masterMemLim=$(sizeToBytes "$_lim")

    _line=$(k3sRun get pod "${mariadbSlaveKube}-0" \
        -o jsonpath="{.spec.containers[?(@.name=='${mariadbSlaveKube}')].resources.requests.memory} {.spec.containers[?(@.name=='${mariadbSlaveKube}')].resources.limits.memory}" 2>/dev/null)
    read -r _req _lim <<< "$_line"
    slaveMemReq=$(sizeToBytes "$_req"); slaveMemLim=$(sizeToBytes "$_lim")

    _line=$(k3sRun get pods -l "app=$olsKube" \
        -o jsonpath="{.items[0].spec.containers[?(@.name=='${olsKube}')].resources.requests.memory} {.items[0].spec.containers[?(@.name=='${olsKube}')].resources.limits.memory}" 2>/dev/null)
    read -r _req _lim <<< "$_line"
    olsMemReq=$(sizeToBytes "$_req"); olsMemLim=$(sizeToBytes "$_lim")

    _line=$(k3sRun get pod "${redisKube}-0" \
        -o jsonpath="{.spec.containers[?(@.name=='${redisKube}')].resources.requests.memory} {.spec.containers[?(@.name=='${redisKube}')].resources.limits.memory}" 2>/dev/null)
    read -r _req _lim <<< "$_line"
    redisMemReq=$(sizeToBytes "$_req"); redisMemLim=$(sizeToBytes "$_lim")

    cat > "$fileTmp" <<EOF
# HELP kube_build_info Build and version info
# TYPE kube_build_info gauge
kube_build_info{version="${appVersion}"} 1

# HELP kube_start_time Time snapshot
# TYPE kube_start_time gauge
kube_start_time $appTimestamp

# HELP kube_vhost_all_count Number of virtual hosts
# TYPE kube_vhost_all_count gauge
kube_vhost_all_count $vhostAllCount

# HELP kube_vhost_up_count Number of active virtual hosts
# TYPE kube_vhost_up_count gauge
kube_vhost_up_count $vhostUpCount

# HELP kube_domain_all_count Number of domains
# TYPE kube_domain_all_count gauge
kube_domain_all_count $domainAllCount

# HELP kube_domain_use_count Number of use domains
# TYPE kube_domain_use_count gauge
kube_domain_use_count $domainFreeCount

# HELP kube_pod_memory_request_bytes Pod memory request in bytes
# TYPE kube_pod_memory_request_bytes gauge
kube_pod_memory_request_bytes{component="mariadb-master"} $masterMemReq
kube_pod_memory_request_bytes{component="mariadb-slave"} $slaveMemReq
kube_pod_memory_request_bytes{component="openlitespeed"} $olsMemReq
kube_pod_memory_request_bytes{component="redis"} $redisMemReq

# HELP kube_pod_memory_limit_bytes Pod memory limit in bytes
# TYPE kube_pod_memory_limit_bytes gauge
kube_pod_memory_limit_bytes{component="mariadb-master"} $masterMemLim
kube_pod_memory_limit_bytes{component="mariadb-slave"} $slaveMemLim
kube_pod_memory_limit_bytes{component="openlitespeed"} $olsMemLim
kube_pod_memory_limit_bytes{component="redis"} $redisMemLim
EOF
    mv "$fileTmp" "$fileProm"
}

# Collects lsphp CPU/memory/worker metrics per domain across OLS pods and writes to $metricPromPath/lsphp.prom.
function metricLsphp() {
    local fileProm="$metricPromPath/lsphp.prom"
    local fileTmp="${fileProm}.tmp"

    mkdir -p -- "$metricPromPath" || return 1

    local podList error
    run podList error k3sPodListStatus "$olsKube" || return 1

    local pod phase output cpu mem count domain
    {
        printf '# HELP lsphp_cpu_percent Total CPU percent used by lsphp workers per domain\n'
        printf '# TYPE lsphp_cpu_percent gauge\n'
        printf '# HELP lsphp_memory_percent Total memory percent used by lsphp workers per domain\n'
        printf '# TYPE lsphp_memory_percent gauge\n'
        printf '# HELP lsphp_worker_count Number of lsphp worker processes per domain\n'
        printf '# TYPE lsphp_worker_count gauge\n'

        while IFS='|' read -r pod phase; do
            [[ "$phase" == "Running" ]] || continue
            # run output error openlitespeedPodExec "$pod" sh -c "ps aux | grep lsphp | grep -v grep | awk '{u=\$1;cpu[u]+=\$3;mem[u]+=\$4;count[u]++} END {for(u in cpu){name=u;cmd=\"find $olsPodPrivatePath -maxdepth 1 -uid \"u\" -type d 2>/dev/null\";if((cmd|getline dir)>0&&dir!=\"\"){n=split(dir,a,\"/\");name=a[n]};close(cmd);print cpu[u],mem[u],count[u],name}}'" || continue
            run output error openlitespeedPodExec "$pod" sh -c "ps aux | awk '/lsphp/ && !/nobody/ {u=\$1;cpu[u]+=\$3;mem[u]+=\$4;count[u]++} END {for(u in cpu){name=u;cmd=\"find $olsPodPrivatePath -maxdepth 1 -uid \"u\" -type d 2>/dev/null\";if((cmd|getline dir)>0&&dir!=\"\"){n=split(dir,a,\"/\");name=a[n]};close(cmd);printf \"%.1f\\t%.1f\\t%d\\t%s\\n\",cpu[u],mem[u],count[u],name}}'" || continue

            while IFS=$'\t' read -r cpu mem count domain; do
                [[ -n "$domain" ]] || continue
                printf 'lsphp_cpu_percent{domain="%s",pod="%s"} %s\n' "$domain" "$pod" "$cpu"
                printf 'lsphp_memory_percent{domain="%s",pod="%s"} %s\n' "$domain" "$pod" "$mem"
                printf 'lsphp_worker_count{domain="%s",pod="%s"} %s\n' "$domain" "$pod" "$count"
            done <<< "$output"
        done < <(awk 'NF' <<< "$podList")
    } > "$fileTmp"

    mv "$fileTmp" "$fileProm"
}

# Collects disk usage per site and sends metrics to the API.
function metricSites() {
    local metricSitesApiUrl="https://api.sodew.ai/api/metrics/sites"
    local batchId batchTime metricsJson dataJson payload httpCode
    batchId="$(uuidgen | tr '[:upper:]' '[:lower:]')"
    batchTime="$(date +%s)"
    metricsJson='{
    "site_disk_usage_mb":  {"type":"number"}
    }'
    dataJson='{}'

    local error vhostList
    run vhostList error openlitespeedConfigVhostList || {
        appError "Error retrieving vhost list: $error"
        return 1
    }
    while IFS= read -r domain <&3; do
        local diskUsage domainJson
        diskUsage="$(pathSize "$olsVhostsPath/$domain" "mb" || true)"
        [[ -n "$diskUsage" ]] || continue

        domainJson="$(
            jq -n --arg diskUsage "$diskUsage" \
                '{
                    site_disk_usage_mb: { value: (if ($diskUsage | length) > 0 then ($diskUsage | tonumber) else null end) },
                }'
        )"
        dataJson="$(jq --arg domain "$domain" --argjson row "$domainJson" '. + {($domain): $row}' <<< "$dataJson")"
    done 3< <(awk 'NF' <<< "$vhostList")

    payload="$(
        jq -n \
            --arg source_type "worker" \
            --arg source_id "$hostUuid" \
            --arg batch_id "$batchId" \
            --argjson batch_time "$batchTime" \
            --argjson metrics "$metricsJson" \
            --argjson data "$dataJson" \
            '{
                source_type: $source_type,
                source_id: $source_id,
                batch_id: $batch_id,
                batch_time: $batch_time,
                metrics: $metrics,
                data: $data
            }'
    )"

    # Sending data...
    httpCode="$(curl -sS -o /tmp/metrics_sites_response.txt -w '%{http_code}' -X POST "$metricSitesApiUrl" -H 'Content-Type: application/json' --data-binary "$payload")"
    if [[ "$httpCode" != "204" ]]; then
        appError "Ingest failed, HTTP $httpCode"
        cat /tmp/metrics_sites_response.txt >&2 || true
        return 1
    fi
}