channels-console 0.3.3

use crate::{get_channel_logs, get_metrics_json};
use serde::Serialize;
use std::fmt::Display;
use tiny_http::{Header, Request, Response, Server};

pub(crate) fn start_metrics_server(addr: &str) {
    let server = match Server::http(addr) {
        Ok(s) => s,
        Err(e) => {
            panic!("Failed to bind metrics server to {}: {}. Customize the port using the CHANNELS_CONSOLE_METRICS_PORT environment variable.", addr, e);
        }
    };

    println!("Channel metrics server listening on http://{}", addr);

    for request in server.incoming_requests() {
        handle_request(request);
    }
}

fn handle_request(request: Request) {
    let path = request.url().split('?').next().unwrap_or("/");

    match path {
        "/metrics" => {
            let metrics = get_metrics_json();
            respond_json(request, &metrics);
        }
        _ => {
            if let Some(id_str) = path.strip_prefix("/logs/") {
                match id_str.parse::<u64>() {
                    Ok(channel_id) => {
                        let channel_id_str = channel_id.to_string();
                        match get_channel_logs(&channel_id_str) {
                            Some(logs) => respond_json(request, &logs),
                            None => respond_error(request, 404, "Channel not found"),
                        }
                    }
                    Err(_) => {
                        respond_error(request, 400, "Invalid channel ID: must be a valid number")
                    }
                }
            } else {
                respond_error(request, 404, "Not found");
            }
        }
    }
}

fn respond_json<T: Serialize>(request: Request, value: &T) {
    match serde_json::to_vec(value) {
        Ok(body) => {
            let mut response = Response::from_data(body);
            response.add_header(
                Header::from_bytes(b"Content-Type".as_slice(), b"application/json".as_slice())
                    .unwrap(),
            );
            let _ = request.respond(response);
        }
        Err(e) => respond_internal_error(request, e),
    }
}

fn respond_error(request: Request, code: u16, msg: &str) {
    let _ = request.respond(Response::from_string(msg).with_status_code(code));
}

fn respond_internal_error(request: Request, e: impl Display) {
    eprintln!("Internal server error: {}", e);
    let _ = request.respond(
        Response::from_string(format!("Internal server error: {}", e)).with_status_code(500),
    );
}