fire-scope 0.1.3

use crate::common::{IpFamily, OutputFormat};
use crate::output_common::{make_header, write_list_nft, write_list_txt};
use chrono::Local;
use ipnet::IpNet;
use std::{collections::BTreeSet, error::Error};

/// IPv4/IPv6リストをファイルに書き出す (国コード用)
pub fn write_ip_lists_to_files(
    country_code: &str,
    ipv4_list: &BTreeSet<IpNet>,
    ipv6_list: &BTreeSet<IpNet>,
    mode: &str,
    format_enum: OutputFormat,
) -> Result<(), Box<dyn Error + Send + Sync>> {
    match format_enum {
        OutputFormat::Txt => {
            let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
            // IPv4
            let ipv4_file = format!("IPv4_{}.txt", country_code);
            let header_v4 = make_header(&now_str, country_code, "N/A");
            write_list_txt(&ipv4_file, ipv4_list, mode, &header_v4)?;

            // IPv6
            let ipv6_file = format!("IPv6_{}.txt", country_code);
            let header_v6 = make_header(&now_str, country_code, "N/A");
            write_list_txt(&ipv6_file, ipv6_list, mode, &header_v6)?;
        }
        OutputFormat::Nft => {
            let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
            // IPv4
            let ipv4_file = format!("IPv4_{}.nft", country_code);
            let header_v4 = make_header(&now_str, country_code, "N/A");
            write_list_nft(&ipv4_file, ipv4_list, mode, &header_v4)?;

            // IPv6
            let ipv6_file = format!("IPv6_{}.nft", country_code);
            let header_v6 = make_header(&now_str, country_code, "N/A");
            write_list_nft(&ipv6_file, ipv6_list, mode, &header_v6)?;
        }
    }
    Ok(())
}

/// IPv4/IPv6リストをファイルに書き出す (AS番号用)
pub fn write_as_ip_list_to_file(
    as_number: &str,
    family: IpFamily,
    ipnets: &BTreeSet<IpNet>,
    mode: &str,
    format_enum: OutputFormat,
) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
    let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
    let header = make_header(&now_str, "N/A", as_number);

    match format_enum {
        OutputFormat::Txt => {
            let file_name = format!("AS_{}_{}.txt", as_number, family.as_str());
            write_list_txt(&file_name, ipnets, mode, &header)?;
            println!(
                "[output] Wrote/append TXT for AS_{} {}",
                as_number,
                family.as_str()
            );
        }
        OutputFormat::Nft => {
            let file_name = format!("AS_{}_{}.nft", as_number, family.as_str());
            write_list_nft(&file_name, ipnets, mode, &header)?;
            println!(
                "[output] Wrote/append NFT for AS_{} {}",
                as_number,
                family.as_str()
            );
        }
    }
    Ok(())
}

/// 国コード+AS番号の重複CIDRリストを書き出す (Overlap)
pub fn write_overlap_to_file(
    country_code: &str,
    as_number: &str,
    overlaps: &BTreeSet<IpNet>,
    mode: &str,
    format_enum: OutputFormat,
) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
    // IPv4 / IPv6 を仕分け
    let overlaps_v4: BTreeSet<IpNet> = overlaps
        .iter()
        .cloned()
        .filter(|net| matches!(net, IpNet::V4(_)))
        .collect();

    let overlaps_v6: BTreeSet<IpNet> = overlaps
        .iter()
        .cloned()
        .filter(|net| matches!(net, IpNet::V6(_)))
        .collect();

    if overlaps_v4.is_empty() && overlaps_v6.is_empty() {
        println!(
            "[overlap] No overlap found for country={} and AS={}",
            country_code, as_number
        );
        return Ok(());
    }

    // ファイル名の作成
    // overlap_{}_{}_(IPv4|IPv6).(txt|nft)
    match format_enum {
        OutputFormat::Txt => {
            // IPv4
            if !overlaps_v4.is_empty() {
                let filename_v4 = format!("overlap_{}_{}_IPv4.txt", country_code, as_number);
                let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
                let header_v4 = make_header(&now_str, country_code, as_number);
                write_list_txt(&filename_v4, &overlaps_v4, mode, &header_v4)?;
                println!("[overlap] Wrote/append IPv4 overlaps to: {}", filename_v4);
            }
            // IPv6
            if !overlaps_v6.is_empty() {
                let filename_v6 = format!("overlap_{}_{}_IPv6.txt", country_code, as_number);
                let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
                let header_v6 = make_header(&now_str, country_code, as_number);
                write_list_txt(&filename_v6, &overlaps_v6, mode, &header_v6)?;
                println!("[overlap] Wrote/append IPv6 overlaps to: {}", filename_v6);
            }
        }
        OutputFormat::Nft => {
            // IPv4
            if !overlaps_v4.is_empty() {
                let filename_v4 = format!("overlap_{}_{}_IPv4.nft", country_code, as_number);
                let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
                let header_v4 = make_header(&now_str, country_code, as_number);
                write_list_nft(&filename_v4, &overlaps_v4, mode, &header_v4)?;
                println!(
                    "[overlap] Wrote/append IPv4 overlaps nft to: {}",
                    filename_v4
                );
            }
            // IPv6
            if !overlaps_v6.is_empty() {
                let filename_v6 = format!("overlap_{}_{}_IPv6.nft", country_code, as_number);
                let now_str = Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
                let header_v6 = make_header(&now_str, country_code, as_number);
                write_list_nft(&filename_v6, &overlaps_v6, mode, &header_v6)?;
                println!(
                    "[overlap] Wrote/append IPv6 overlaps nft to: {}",
                    filename_v6
                );
            }
        }
    }

    Ok(())
}