libperl-macrogen 0.1.4

//! TinyCC Macro Bindgen CLI
//!
//! CファイルをパースしてS-expression形式で出力する

use std::fs::File;
use std::io::{self, BufWriter, Write};
use std::path::PathBuf;

use std::ops::ControlFlow;

use clap::Parser as ClapParser;
use libperl_macrogen::{
    ApidocDict, BlockItem, CompileError, FieldsDict, FileId,
    ParseResult, Parser, Preprocessor, RustDeclDict, SexpPrinter,
    SourceLocation, TokenKind, TypedSexpPrinter,
    // Pipeline API
    Pipeline, PipelineError,
};

/// コマンドライン引数
#[derive(ClapParser)]
#[command(name = "libperl-macrogen")]
#[command(version, about = "C to Rust macro bindgen tool")]
struct Cli {
    /// 入力Cファイル（--parse-rust-bindings使用時は不要）
    input: Option<PathBuf>,

    /// インクルードパス (-I)
    #[arg(short = 'I', long = "include")]
    include: Vec<PathBuf>,

    /// マクロ定義 (-D)
    #[arg(short = 'D', long = "define")]
    define: Vec<String>,

    /// 出力ファイル（省略時は標準出力）
    #[arg(short = 'o', long = "output")]
    output: Option<PathBuf>,

    /// プリプロセッサ出力のみ (cc -E 相当)
    #[arg(short = 'E')]
    preprocess_only: bool,

    /// プリプロセッサデバッグ出力
    #[arg(long = "debug-pp")]
    debug_pp: bool,

    /// Perl Config.pm から設定を自動取得
    #[arg(long = "auto")]
    auto: bool,

    /// GCC互換の出力形式 (-E と併用)
    #[arg(long = "gcc-format")]
    gcc_format: bool,

    /// ストリーミングモード（逐次パース、エラー時にソースコード表示）
    #[arg(long = "streaming")]
    streaming: bool,

    /// 型注釈付きS-expression出力
    #[arg(long = "typed-sexp")]
    typed_sexp: bool,

    /// S-expression出力（マクロ型推論を行わない）
    #[arg(long = "sexp")]
    sexp: bool,

    /// 構造体フィールド辞書をダンプ
    #[arg(long = "dump-fields-dict")]
    dump_fields_dict: bool,

    /// マクロ定義をダンプ（gcc -E -dM 相当）
    /// フィルタ文字列を指定可能（省略時は全マクロ）
    #[arg(long = "dump-macros", value_name = "FILTER")]
    dump_macros: Option<Option<String>>,

    /// ApidocファイルをJSONに変換して出力（入力ファイルはapidoc）
    #[arg(long = "apidoc-to-json")]
    apidoc_to_json: bool,

    /// コンパクトなJSON出力（--apidoc-to-json用）
    #[arg(long = "compact")]
    compact: bool,

    /// ターゲットディレクトリ（デフォルト: /usr/lib64/perl5/CORE）
    #[arg(long = "target-dir")]
    target_dir: Option<PathBuf>,

    /// Rustバインディングファイルから宣言を抽出
    #[arg(long = "parse-rust-bindings")]
    parse_rust_bindings: Option<PathBuf>,

    // 廃止予定: --analyze-macros (MacroAnalyzer2 使用)
    // #[arg(long = "analyze-macros")]
    // analyze_macros: bool,

    // 廃止予定: --gen-rust-fns (macrogen 使用)
    // #[arg(long = "gen-rust-fns")]
    // gen_rust_fns: bool,

    /// Rustバインディングファイル（マクロ型推論用）
    #[arg(long = "bindings")]
    bindings: Option<PathBuf>,

    /// Perl apidocファイル (embed.fnc)
    #[arg(long = "apidoc")]
    apidoc: Option<PathBuf>,

    // 廃止予定: --debug-macro-gen (MacroAnalyzer2, RustCodeGen 使用)
    // #[arg(long = "debug-macro-gen")]
    // debug_macro_gen: bool,

    /// 進行状況を表示
    #[arg(long = "progress")]
    progress: bool,

    /// マクロ展開マーカーを出力（デバッグ用）
    #[arg(long = "emit-macro-markers")]
    emit_macro_markers: bool,

    /// 生成コードにマクロ定義位置コメントを追加
    #[arg(long = "macro-comments")]
    macro_comments: bool,

    /// Rust コード生成（マクロと inline 関数）
    #[arg(long = "gen-rust")]
    gen_rust: bool,

    /// apidoc マージ後にダンプして終了（フィルタ文字列を指定可能）
    #[arg(long = "dump-apidoc-after-merge", value_name = "FILTER")]
    dump_apidoc_after_merge: Option<Option<String>>,

    /// rustfmt に渡す Rust edition (デフォルト: 2024)
    #[arg(long = "rust-edition", default_value = "2024")]
    rust_edition: String,

    /// rustfmt 失敗時にエラー終了する
    #[arg(long = "strict-rustfmt")]
    strict_rustfmt: bool,

    /// マクロ型推論のデバッグ出力（カンマ区切りでマクロ名を指定）
    #[arg(long = "debug-type-inference", value_name = "MACROS")]
    debug_type_inference: Option<String>,

    /// 指定した関数のコード生成時に入力 AST をコメントとしてダンプ（デバッグ用）
    #[arg(long = "dump-ast-for", value_name = "FUNC")]
    dump_ast_for: Option<String>,

    /// 指定した関数のコード生成時に型推論結果を stderr にダンプ（デバッグ用）
    #[arg(long = "dump-types-for", value_name = "FUNC")]
    dump_types_for: Option<String>,

    /// codegen をスキップする関数名リストファイル。
    /// 1 行 1 名、`#` コメント可、空行無視。複数指定可。
    #[arg(long = "skip-codegen-list", value_name = "FILE")]
    skip_codegen_list: Vec<PathBuf>,

    /// 対象 perl の build mode（threaded / non-threaded / auto）
    /// 省略時は auto（実行時の `perl Config{usethreads}` から自動検出）
    #[arg(long = "perl-build-mode", value_name = "MODE", value_parser = parse_perl_build_mode)]
    perl_build_mode: Option<libperl_macrogen::perl_config::PerlBuildMode>,
}

/// `--perl-build-mode` の値パーサー
///
/// `auto` は `Ok(None)` 相当だが clap の Option パーサーでは少し扱いが
/// 異なるため、明示的にパースして「auto は明示指定なし」として扱う。
fn parse_perl_build_mode(s: &str) -> Result<libperl_macrogen::perl_config::PerlBuildMode, String> {
    use libperl_macrogen::perl_config::PerlBuildMode;
    match s {
        "threaded" | "thr" => Ok(PerlBuildMode::Threaded),
        "non-threaded" | "nonthreaded" | "nothr" => Ok(PerlBuildMode::NonThreaded),
        // `auto` は Option<PerlBuildMode> の None として扱いたいので、
        // ここではエラーで弾き、`--perl-build-mode auto` の代わりに
        // フラグ省略を案内する。
        "auto" => Err(
            "use omission of --perl-build-mode for auto-detection".to_string()
        ),
        other => Err(format!(
            "unknown perl build mode: {} (expected 'threaded' or 'non-threaded')", other
        )),
    }
}

fn main() {
    if let Err(e) = run() {
        eprintln!("Error: {}", e);
        std::process::exit(1);
    }
}

fn run() -> Result<(), Box<dyn std::error::Error>> {
    let cli = Cli::parse();

    // --parse-rust-bindings: Rustファイルのみ処理（プリプロセッサ不要）
    if let Some(ref rust_file) = cli.parse_rust_bindings {
        return run_parse_rust_bindings(rust_file);
    }

    // --apidoc-to-json: ApidocファイルをJSONに変換（プリプロセッサ不要）
    if cli.apidoc_to_json {
        let input = cli.input.ok_or("Input file (apidoc) is required for --apidoc-to-json")?;
        return run_apidoc_to_json(&input, cli.output.as_ref(), cli.compact);
    }

    // 入力ファイルが必要
    let input = cli.input.ok_or("Input file is required")?;

    // PipelineBuilder を構築
    let mut builder = Pipeline::builder(&input);

    if cli.auto {
        // --auto: Perl Config.pm から設定を取得
        if !cli.include.is_empty() {
            return Err("--auto cannot be used with -I options".into());
        }
        builder = builder.with_auto_perl_config()
            .map_err(|e| format!("Failed to get Perl config: {}", e))?;

        // 追加の -D オプションがあればマージ
        for (name, value) in parse_defines(&cli.define) {
            builder = builder.with_define(name, value);
        }
    } else {
        // 従来通り CLI 引数から
        for path in &cli.include {
            builder = builder.with_include(path);
        }
        for (name, value) in parse_defines(&cli.define) {
            builder = builder.with_define(name, value);
        }
    }

    // target_dir: CLI 指定があればそれを使用
    if let Some(ref target_dir) = cli.target_dir {
        builder = builder.with_target_dir(target_dir);
    }

    // オプション設定
    if cli.debug_pp {
        builder = builder.with_debug_pp();
    }
    if cli.emit_macro_markers {
        builder = builder.with_emit_markers();
    }

    // コード生成用のデフォルト設定（assert ラップなど）
    builder = builder.with_codegen_defaults();

    // bindings があれば設定
    if let Some(ref bindings_path) = cli.bindings {
        builder = builder.with_bindings(bindings_path);
    }

    // apidoc があれば設定
    if let Some(ref apidoc_path) = cli.apidoc {
        builder = builder.with_apidoc(apidoc_path);
    }

    // skip-codegen リストファイル（複数指定可）
    for path in &cli.skip_codegen_list {
        builder = builder.with_skip_codegen_list(path);
    }

    // 対象 perl の build mode（省略時は auto-detect）
    if let Some(mode) = cli.perl_build_mode {
        builder = builder.with_perl_build_mode(mode);
    }

    // デバッグオプション: dump_apidoc_after_merge
    if let Some(ref filter_opt) = cli.dump_apidoc_after_merge {
        let filter = filter_opt.as_deref().unwrap_or("").to_string();
        builder = builder.with_dump_apidoc(filter);
    }

    // デバッグオプション: debug_type_inference
    if let Some(ref macros) = cli.debug_type_inference {
        let macro_list: Vec<String> = macros
            .split(',')
            .map(|s| s.trim().to_string())
            .filter(|s| !s.is_empty())
            .collect();
        builder = builder.with_debug_type_inference(macro_list);
    }

    // Codegen 設定
    builder = builder.with_rust_edition(&cli.rust_edition);
    if cli.strict_rustfmt {
        builder = builder.with_strict_rustfmt();
    }
    if cli.macro_comments {
        builder = builder.with_macro_comments();
    }
    if let Some(ref name) = cli.dump_ast_for {
        builder = builder.with_dump_ast_for(name);
    }
    if let Some(ref name) = cli.dump_types_for {
        builder = builder.with_dump_types_for(name);
    }

    // Pipeline を構築してプリプロセスを実行
    let mut preprocessed = builder.build()?.preprocess()
        .map_err(|e| format_pipeline_error(&e))?;

    if cli.preprocess_only {
        // -E: プリプロセス結果のみ出力
        output_preprocessed(preprocessed.preprocessor_mut(), cli.output.as_ref(), cli.gcc_format)?;
    } else if cli.streaming {
        // --streaming: ストリーミングモード
        run_streaming(preprocessed.preprocessor_mut())?;
    } else if cli.typed_sexp {
        // --typed-sexp: 型注釈付きS-expression出力
        run_typed_sexp(preprocessed.preprocessor_mut(), cli.output.as_ref())?;
    } else if cli.dump_fields_dict {
        // --dump-fields-dict: 構造体フィールド辞書をダンプ
        run_dump_fields_dict(preprocessed.preprocessor_mut(), cli.target_dir.as_ref())?;
    } else if let Some(ref filter_opt) = cli.dump_macros {
        // --dump-macros: マクロ定義をダンプ
        // まず全トークンを消費してマクロ定義を収集
        let filter = filter_opt.as_deref().unwrap_or("");
        let pp = preprocessed.preprocessor_mut();
        loop {
            match pp.next_token() {
                Ok(token) if matches!(token.kind, libperl_macrogen::TokenKind::Eof) => break,
                Ok(_) => continue,
                Err(e) => return Err(format_error(&e, pp).into()),
            }
        }
        pp.macros().dump_filtered(filter, pp.interner());
    } else if cli.sexp {
        // --sexp: S-expression出力（マクロ型推論なし）
        let pp = preprocessed.preprocessor_mut();
        let mut parser = match Parser::new(pp) {
            Ok(p) => p,
            Err(e) => return Err(format_error(&e, pp).into()),
        };
        let tu = match parser.parse() {
            Ok(tu) => tu,
            Err(e) => return Err(format_error(&e, pp).into()),
        };

        // 出力
        if let Some(output_path) = cli.output {
            let file = File::create(&output_path)?;
            let mut writer = BufWriter::new(file);
            let mut printer = SexpPrinter::new(&mut writer, pp.interner());
            printer.print_translation_unit(&tu)?;
            writer.flush()?;
        } else {
            let stdout = io::stdout();
            let mut handle = stdout.lock();
            let mut printer = SexpPrinter::new(&mut handle, pp.interner());
            printer.print_translation_unit(&tu)?;
            handle.flush()?;
        }
    } else if cli.gen_rust {
        // --gen-rust: Rust コード生成（Pipeline API を使用）
        run_gen_rust_pipeline(preprocessed, cli.output.as_ref(), cli.auto, &cli.rust_edition, cli.strict_rustfmt)?;
    } else {
        // デフォルト: マクロ型推論（統計出力）
        run_infer_macro_types_pipeline(preprocessed, cli.auto)?;
    }

    Ok(())
}

/// ストリーミングモードで実行
fn run_streaming(pp: &mut Preprocessor) -> Result<(), Box<dyn std::error::Error>> {
    let mut parser = match Parser::new(pp) {
        Ok(p) => p,
        Err(e) => return Err(format_error(&e, pp).into()),
    };

    // ストリーミング出力用
    let stdout = io::stdout();
    let mut handle = stdout.lock();
    let mut count = 0usize;

    // parse_each でパースし、即座に出力
    let parse_result = parser.parse_each(|decl, _loc, _path, interner| {
        let mut printer = SexpPrinter::new(&mut handle, interner);
        if let Err(e) = printer.print_external_decl(decl) {
            eprintln!("Output error: {}", e);
            return ControlFlow::Break(());
        }
        if let Err(e) = printer.writeln() {
            eprintln!("Output error: {}", e);
            return ControlFlow::Break(());
        }
        count += 1;
        ControlFlow::Continue(())
    });

    drop(handle);

    // エラーがあった場合、詳細を表示
    if let Err(error) = parse_result {
        // エラー内の実際の位置を使う
        let error_loc = error.loc();

        eprintln!("\n=== Parse Error ===");
        eprintln!("Location: {}:{}:{}",
            pp.files().get_path(error_loc.file_id).display(),
            error_loc.line,
            error_loc.column);
        eprintln!("Error: {}", format_error(&error, pp));

        // ソースコードのコンテキストを表示
        show_source_context(pp, error_loc);

        return Err("Parse failed".into());
    }

    eprintln!("\nSuccessfully parsed {} declarations", count);
    Ok(())
}

/// 構造体フィールド辞書をダンプ
fn run_dump_fields_dict(pp: &mut Preprocessor, _target_dir: Option<&PathBuf>) -> Result<(), Box<dyn std::error::Error>> {
    // フィールド辞書を作成
    let mut fields_dict = FieldsDict::new();

    let mut parser = match Parser::new(pp) {
        Ok(p) => p,
        Err(e) => return Err(format_error(&e, pp).into()),
    };

    // パースしながらフィールド情報を収集
    parser.parse_each(|decl, _loc, _path, interner| {
        fields_dict.collect_from_external_decl(decl, decl.is_target(), interner);
        std::ops::ControlFlow::Continue(())
    })?;

    // 統計情報を表示
    let stats = fields_dict.stats();
    let interner = parser.interner();
    eprintln!("=== Fields Dictionary Stats ===");
    eprintln!("Total fields: {}", stats.total_fields);
    eprintln!("Unique fields (can infer struct): {}", stats.unique_fields);
    eprintln!("Ambiguous fields: {}", stats.ambiguous_fields);
    eprintln!("Field types collected: {}", fields_dict.field_types_count());
    eprintln!();

    // 一意なフィールドをダンプ
    println!("{}", fields_dict.dump_unique(interner));

    Ok(())
}

/// マクロ型推論（Pipeline API 使用）
///
/// Pipeline を使用して型推論を実行し、結果を出力する。
fn run_infer_macro_types_pipeline(
    preprocessed: libperl_macrogen::PreprocessedPipeline,
    auto_mode: bool,
) -> Result<(), Box<dyn std::error::Error>> {
    // 自動ロード時の通知（省略 - Pipeline 内部で解決）
    let _ = auto_mode;

    // 推論を実行
    let inferred = match preprocessed.infer() {
        Ok(r) => r,
        Err(e) => {
            // デバッグダンプで早期終了の場合は成功扱い
            if matches!(e, PipelineError::Io(ref io_err) if io_err.kind() == std::io::ErrorKind::Interrupted) {
                return Ok(());
            }
            return Err(format_pipeline_error(&e).into());
        }
    };

    // 結果から必要な情報を取り出す
    let result = inferred.result();
    let infer_ctx = &result.infer_ctx;
    let interner = result.preprocessor.interner();
    let stats = &result.stats;

    // パース結果の統計を収集
    let mut expr_count = 0;
    let mut stmt_count = 0;
    let mut unparseable_count = 0;
    for info in infer_ctx.macros.values() {
        if !info.is_target {
            continue;
        }
        if info.is_expression() {
            expr_count += 1;
        } else if info.is_statement() {
            stmt_count += 1;
        } else {
            unparseable_count += 1;
        }
    }

    // 統計情報を出力
    let macro_stats = infer_ctx.stats();
    eprintln!("=== Macro Type Inference Stats ===");
    eprintln!("Total macros analyzed: {}", macro_stats.total);
    eprintln!("  - Expression macros: {}", expr_count);
    eprintln!("  - Statement macros: {}", stmt_count);
    eprintln!("  - Unparseable: {}", unparseable_count);
    eprintln!("Confirmed (type complete): {}", macro_stats.confirmed);
    eprintln!("Unconfirmed (pending): {}", macro_stats.unconfirmed);
    eprintln!("Args unknown: {}", macro_stats.args_unknown);
    eprintln!("Return unknown: {}", macro_stats.return_unknown);
    eprintln!();

    // コメントから収集した apidoc 数
    eprintln!("Apidoc from comments: {}", stats.apidoc_from_comments);
    // THX 依存マクロ数（解析済みターゲットマクロのうち）
    eprintln!("THX-dependent macros: {}", stats.thx_dependent_count);
    // C 関数宣言統計
    eprintln!("C function declarations: {} ({} THX)", stats.c_fn_decl_count, stats.c_fn_thx_count);
    // SV family 検出統計
    eprintln!("SV family members: {}", result.fields_dict.sv_family_members_count());
    eprintln!("typeName -> struct mapping: {}", result.fields_dict.sv_head_type_mapping_count());
    eprintln!();

    // 各マクロの詳細を出力（辞書順）
    println!("=== Macro Analysis Results ===");
    let mut parseable_count = 0;
    let mut has_constraints_count = 0;

    // マクロ名でソートするためにベクターに収集
    // - 関数形式マクロ、または THX 依存のオブジェクトマクロを出力
    // - 空のトークン列を持つマクロ（条件コンパイルフラグ等）は除外
    let mut sorted_macros: Vec<_> = infer_ctx
        .macros
        .iter()
        .filter(|(_, info)| info.is_target && info.has_body && (info.is_function || info.is_thx_dependent))
        .collect();
    sorted_macros.sort_by_key(|(name, _)| interner.get(**name));

    for (name, info) in sorted_macros {
        let name_str = interner.get(*name);

        let parse_status = if info.is_expression() {
            parseable_count += 1;
            "expression"
        } else if info.is_statement() {
            parseable_count += 1;
            "statement"
        } else {
            "unparseable"
        };

        let thx_marker = if info.is_thx_dependent { " [THX]" } else { "" };
        let pasting_marker = if info.has_token_pasting { " [##]" } else { "" };
        let constraint_count = info.type_env.total_constraint_count();

        if constraint_count > 0 {
            has_constraints_count += 1;
        }

        println!(
            "{}: {} ({} constraints, {} uses){}{}",
            name_str,
            parse_status,
            constraint_count,
            info.uses.len(),
            thx_marker,
            pasting_marker
        );

        // 型付き S 式を追加出力（pretty print）
        match &info.parse_result {
            ParseResult::Expression(expr) => {
                let stdout = io::stdout();
                let mut handle = stdout.lock();
                let mut printer = TypedSexpPrinter::new(&mut handle, interner);
                printer.set_type_env(&info.type_env);
                printer.set_param_map(&info.params);
                printer.set_pretty(true);
                printer.set_indent(1);  // 行頭にスペース1文字分のインデント
                printer.set_skip_first_newline(true);  // 先頭の空行を抑制
                let _ = printer.print_expr(expr);
                let _ = writeln!(handle);
            }
            ParseResult::Statement(block_items) => {
                let stdout = io::stdout();
                let mut handle = stdout.lock();
                let _ = write!(handle, " ");  // 最初の行頭スペース
                let mut printer = TypedSexpPrinter::new(&mut handle, interner);
                printer.set_type_env(&info.type_env);
                printer.set_param_map(&info.params);
                printer.set_pretty(true);
                printer.set_indent(1);  // 行頭にスペース1文字分のインデント
                printer.set_skip_first_newline(true);  // 先頭の空行を抑制
                for item in block_items {
                    if let BlockItem::Stmt(stmt) = item {
                        let _ = printer.print_stmt(stmt);
                    }
                }
                let _ = writeln!(handle);
            }
            ParseResult::Unparseable(Some(err_msg)) => {
                println!("  error: {}", err_msg);
            }
            _ => {}
        }

        // 型制約の詳細
        if constraint_count > 0 {
            for (expr_id, constraints) in &info.type_env.expr_constraints {
                for c in constraints {
                    println!("  expr#{}: {} ({})", expr_id.0, c.ty.to_display_string(interner), c.context);
                }
            }
        }
    }

    eprintln!();
    eprintln!("Parseable macros: {}", parseable_count);
    eprintln!("Macros with type constraints: {}", has_constraints_count);

    Ok(())
}

/// Rust コード生成（Pipeline API 使用）
///
/// 型推論結果から Rust コードを生成する。
fn run_gen_rust_pipeline(
    preprocessed: libperl_macrogen::PreprocessedPipeline,
    output_path: Option<&PathBuf>,
    auto_mode: bool,
    rust_edition: &str,
    strict_rustfmt: bool,
) -> Result<(), Box<dyn std::error::Error>> {
    // 自動ロード時はパスを表示
    let infer_config = preprocessed.infer_config();
    if auto_mode && infer_config.apidoc_path.is_none() {
        // apidoc パスは infer() 内で解決されるので、ここでは通知のみ
        // Pipeline 内部で解決した apidoc を表示するためには
        // resolve_apidoc_path を呼び出すか、Pipeline に機能を追加する必要がある
        // 今回は省略（ユーザーへの通知を一旦削除）
    }

    // 推論を実行
    let inferred = match preprocessed.infer() {
        Ok(r) => r,
        Err(e) => {
            // デバッグダンプで早期終了の場合は成功扱い
            if matches!(e, PipelineError::Io(ref io_err) if io_err.kind() == std::io::ErrorKind::Interrupted) {
                return Ok(());
            }
            return Err(format_pipeline_error(&e).into());
        }
    };

    // まずバッファに生成
    let mut buffer = Vec::new();
    let generated = inferred.generate(&mut buffer)
        .map_err(|e| format_pipeline_error(&e))?;
    let stats = generated.stats();

    // rustfmt を適用
    let formatted = match apply_rustfmt(&buffer, rust_edition) {
        Ok(code) => code,
        Err(e) => {
            if strict_rustfmt {
                return Err(format!("rustfmt failed: {}", e).into());
            } else {
                eprintln!("Warning: {}", e);
                buffer
            }
        }
    };

    // 出力先に書き込み
    if let Some(path) = output_path {
        let mut file = File::create(path)?;
        file.write_all(&formatted)?;
        file.flush()?;
        eprintln!("Output: {}", path.display());
    } else {
        let stdout = io::stdout();
        let mut handle = stdout.lock();
        handle.write_all(&formatted)?;
        handle.flush()?;
    }

    // 統計情報を出力
    eprintln!("=== Rust Code Generation Stats ===");
    eprintln!("Macros: {} success, {} parse failed, {} type incomplete, {} cascade unavailable, {} unresolved names",
        stats.macros_success, stats.macros_parse_failed, stats.macros_type_incomplete,
        stats.macros_cascade_unavailable, stats.macros_unresolved_names);
    eprintln!("Inline functions: {} success, {} type incomplete, {} cascade unavailable, {} unresolved names, {} contains goto",
        stats.inline_fns_success, stats.inline_fns_type_incomplete,
        stats.inline_fns_cascade_unavailable, stats.inline_fns_unresolved_names,
        stats.inline_fns_contains_goto);

    Ok(())
}

/// rustfmt を適用
///
/// 成功時はフォーマット済みコードを返す。
/// 失敗時はエラーメッセージを返す。
fn apply_rustfmt(code: &[u8], edition: &str) -> Result<Vec<u8>, String> {
    use std::process::{Command, Stdio};

    let mut child = match Command::new("rustfmt")
        .arg("--edition")
        .arg(edition)
        .stdin(Stdio::piped())
        .stdout(Stdio::piped())
        .stderr(Stdio::piped())
        .spawn()
    {
        Ok(child) => child,
        Err(_) => {
            return Err("rustfmt not found, skipping formatting".to_string());
        }
    };

    // stdin に書き込み
    if let Some(mut stdin) = child.stdin.take() {
        if let Err(e) = stdin.write_all(code) {
            return Err(format!("failed to write to rustfmt stdin: {}", e));
        }
    }

    // 結果を取得
    match child.wait_with_output() {
        Ok(output) if output.status.success() => Ok(output.stdout),
        Ok(output) => {
            let mut msg = "rustfmt failed".to_string();
            if !output.stderr.is_empty() {
                if let Ok(stderr_str) = std::str::from_utf8(&output.stderr) {
                    msg = format!("rustfmt failed: {}", stderr_str.trim());
                }
            }
            Err(msg)
        }
        Err(e) => {
            Err(format!("rustfmt failed: {}", e))
        }
    }
}

// 廃止予定: run_gen_rust_fns_lib (macrogen 使用)
// fn run_gen_rust_fns_lib(...) -> Result<(), Box<dyn std::error::Error>> {
//     ...
// }

// 廃止予定: run_debug_macro_gen (MacroAnalyzer2, RustCodeGen 使用)
// fn run_debug_macro_gen(pp: &mut Preprocessor) -> Result<(), Box<dyn std::error::Error>> {
//     ...
// }

/// 型注釈付きS-expression出力モードで実行
fn run_typed_sexp(pp: &mut Preprocessor, output: Option<&PathBuf>) -> Result<(), Box<dyn std::error::Error>> {
    let mut parser = match Parser::new(pp) {
        Ok(p) => p,
        Err(e) => return Err(format_error(&e, pp).into()),
    };

    let tu = match parser.parse() {
        Ok(tu) => tu,
        Err(e) => return Err(format_error(&e, pp).into()),
    };

    // 出力
    if let Some(output_path) = output {
        let file = File::create(output_path)?;
        let mut writer = BufWriter::new(file);
        let mut printer = TypedSexpPrinter::new(&mut writer, pp.interner());
        for decl in &tu.decls {
            printer.print_external_decl(decl)?;
        }
        writer.flush()?;
    } else {
        let stdout = io::stdout();
        let mut handle = stdout.lock();
        let mut printer = TypedSexpPrinter::new(&mut handle, pp.interner());
        for decl in &tu.decls {
            printer.print_external_decl(decl)?;
        }
        handle.flush()?;
    }

    Ok(())
}

/// Rustバインディングファイルから宣言を抽出
fn run_parse_rust_bindings(rust_file: &PathBuf) -> Result<(), Box<dyn std::error::Error>> {
    let dict = RustDeclDict::parse_file(rust_file)?;

    // 統計情報を表示
    let stats = dict.stats();
    eprintln!("=== Rust Declarations Stats ===");
    eprintln!("Constants: {}", stats.const_count);
    eprintln!("Type aliases: {}", stats.type_count);
    eprintln!("Functions: {}", stats.fn_count);
    eprintln!("Structs: {}", stats.struct_count);
    eprintln!();

    // ダンプ
    println!("{}", dict.dump());

    Ok(())
}

/// ApidocファイルをJSONに変換
fn run_apidoc_to_json(
    input: &PathBuf,
    output: Option<&PathBuf>,
    compact: bool,
) -> Result<(), Box<dyn std::error::Error>> {
    // embed.fnc をパース
    let dict = ApidocDict::parse_embed_fnc(input)?;

    // 統計情報を表示
    let stats = dict.stats();
    eprintln!("Loaded {} entries from {:?}", stats.total, input);
    eprintln!("  Functions: {}", stats.function_count);
    eprintln!("  Macros: {}", stats.macro_count);
    eprintln!("  Inline: {}", stats.inline_count);
    eprintln!("  Public API: {}", stats.api_count);

    // JSONにシリアライズ
    let json = if compact {
        serde_json::to_string(&dict)?
    } else {
        serde_json::to_string_pretty(&dict)?
    };

    // 出力
    if let Some(output_path) = output {
        let file = File::create(output_path)?;
        let mut writer = BufWriter::new(file);
        writeln!(writer, "{}", json)?;
        writer.flush()?;
        eprintln!("Written to {:?}", output_path);
    } else {
        println!("{}", json);
    }

    Ok(())
}

/// エラー箇所のソースコードコンテキストを表示
fn show_source_context(pp: &Preprocessor, loc: &SourceLocation) {
    let path = pp.files().get_path(loc.file_id);

    // ファイルを読み込み
    let content = match std::fs::read_to_string(path) {
        Ok(c) => c,
        Err(e) => {
            eprintln!("Could not read source file: {}", e);
            return;
        }
    };

    let lines: Vec<&str> = content.lines().collect();
    let target_line = loc.line as usize;

    // エラー行の前後2行を表示
    let start = target_line.saturating_sub(3);
    let end = (target_line + 2).min(lines.len());

    eprintln!("\nSource context:");
    eprintln!("{}:{}:{}", path.display(), loc.line, loc.column);
    eprintln!("{}", "-".repeat(60));

    for i in start..end {
        let line_num = i + 1;
        let marker = if line_num == target_line as usize { ">>>" } else { "   " };
        if i < lines.len() {
            eprintln!("{} {:4} | {}", marker, line_num, lines[i]);

            // エラー行の場合、カラム位置を矢印で示す
            if line_num == target_line as usize && loc.column > 0 {
                let spaces = " ".repeat(loc.column as usize + 7);
                eprintln!("{}^", spaces);
            }
        }
    }
    eprintln!("{}", "-".repeat(60));
}

/// エラーをファイル名付きでフォーマット
fn format_error(e: &CompileError, pp: &Preprocessor) -> String {
    e.format_with_files(pp.files())
}

/// PipelineError をフォーマット
fn format_pipeline_error(e: &PipelineError) -> String {
    match e {
        PipelineError::PerlConfig(pe) => format!("Perl config error: {}", pe),
        PipelineError::Compile(ce) => format!("Compile error: {}", ce),
        PipelineError::Infer(ie) => format!("Inference error: {}", ie),
        PipelineError::Io(io_err) => format!("I/O error: {}", io_err),
    }
}

/// プリプロセス結果を出力
fn output_preprocessed(
    pp: &mut Preprocessor,
    output: Option<&PathBuf>,
    gcc_format: bool,
) -> Result<(), Box<dyn std::error::Error>> {
    let mut out: Box<dyn Write> = if let Some(path) = output {
        Box::new(BufWriter::new(File::create(path)?))
    } else {
        Box::new(io::stdout().lock())
    };

    if gcc_format {
        output_gcc_format(pp, &mut out)
    } else {
        output_debug_format(pp, &mut out)
    }
}

/// GCC互換形式で出力（diff比較用）
/// 行マーカーはファイル変更時と文の開始時のみ出力（文中のマクロ展開は無視）
fn output_gcc_format(
    pp: &mut Preprocessor,
    out: &mut dyn Write,
) -> Result<(), Box<dyn std::error::Error>> {
    let mut last_file: Option<FileId> = None;
    let mut last_output_line = 0u32;
    let mut need_space = false;
    let mut at_statement_start = true;
    let mut brace_depth = 0i32;
    let mut pending_block_end = false;
    let mut file_stack: Vec<FileId> = Vec::new();

    loop {
        let token = match pp.next_token() {
            Ok(t) => t,
            Err(e) => return Err(format_error(&e, pp).into()),
        };

        if matches!(token.kind, TokenKind::Eof) {
            break;
        }

        let current_file = token.loc.file_id;
        let current_line = token.loc.line;

        // 文の開始時のみファイル/行の変更をチェック（ブレース更新前）
        if at_statement_start && brace_depth == 0 {
            if last_file != Some(current_file) {
                // ファイル変更
                if need_space {
                    writeln!(out)?;
                }

                // GCCフラグを決定
                let flag = if last_file.is_none() {
                    "" // 最初のファイル
                } else if file_stack.contains(&current_file) {
                    // 以前のファイルに戻る
                    while file_stack.last() != Some(&current_file) {
                        file_stack.pop();
                    }
                    " 2"
                } else {
                    // 新しいファイルに入る
                    if let Some(prev) = last_file {
                        file_stack.push(prev);
                    }
                    " 1"
                };

                let path = pp.files().get_path(current_file);
                writeln!(out, "# {} \"{}\"{}", current_line, path.display(), flag)?;
                last_file = Some(current_file);
                last_output_line = current_line;
                need_space = false;
            } else if current_line > last_output_line {
                // 同一ファイル内で行が進んだ
                let gap = current_line - last_output_line;
                if gap <= 8 {
                    // 小さいギャップは空行で埋める
                    if need_space {
                        writeln!(out)?;
                    }
                    for _ in 1..gap {
                        writeln!(out)?;
                    }
                    need_space = false;
                } else {
                    // 大きいギャップはディレクティブを使う
                    if need_space {
                        writeln!(out)?;
                    }
                    let path = pp.files().get_path(current_file);
                    writeln!(out, "# {} \"{}\"", current_line, path.display())?;
                    need_space = false;
                }
                last_output_line = current_line;
            }
            at_statement_start = false;
        }

        // 前のトークンがブロック終了で、次がセミコロンでない場合は改行
        // （関数定義の終わり）
        if pending_block_end && !matches!(token.kind, TokenKind::Semi) {
            writeln!(out)?;
            last_output_line += 1;
            need_space = false;
            at_statement_start = true;
        }
        pending_block_end = false;

        // ブレース深度を更新
        let was_in_block = brace_depth > 0;
        match token.kind {
            TokenKind::LBrace => brace_depth += 1,
            TokenKind::RBrace => brace_depth -= 1,
            _ => {}
        }

        // トークン間のスペース（セミコロン、カンマ、閉じ括弧の前は不要）
        if need_space && !matches!(token.kind, TokenKind::Semi | TokenKind::Comma | TokenKind::RParen | TokenKind::RBracket | TokenKind::RBrace) {
            write!(out, " ")?;
        }

        // 開き括弧の後はスペース不要
        let suppress_next_space = matches!(token.kind, TokenKind::LParen | TokenKind::LBracket);

        // トークンを出力
        write!(out, "{}", token.kind.format(pp.interner()))?;
        need_space = !suppress_next_space;

        // トップレベルのセミコロンで改行
        if brace_depth == 0 && matches!(token.kind, TokenKind::Semi) {
            writeln!(out)?;
            last_output_line += 1;
            need_space = false;
            at_statement_start = true;
        }
        // トップレベルに戻った閉じブレースをマーク
        if brace_depth == 0 && was_in_block && matches!(token.kind, TokenKind::RBrace) {
            pending_block_end = true;
        }
    }

    if need_space {
        writeln!(out)?;
    }
    Ok(())
}

/// デバッグ用詳細形式で出力（行追跡あり）
fn output_debug_format(
    pp: &mut Preprocessor,
    out: &mut dyn Write,
) -> Result<(), Box<dyn std::error::Error>> {
    let mut last_line = 0u32;
    let mut last_file = None;
    let mut need_space = false;

    loop {
        let token = match pp.next_token() {
            Ok(t) => t,
            Err(e) => return Err(format_error(&e, pp).into()),
        };

        if matches!(token.kind, TokenKind::Eof) {
            break;
        }

        let current_file = Some(token.loc.file_id);
        let current_line = token.loc.line;

        // ファイルが変わった場合、または行が大きく変わった場合はディレクティブを出力
        if current_file != last_file {
            // ファイル変更
            if last_file.is_some() {
                writeln!(out)?;
            }
            let path = pp.files().get_path(token.loc.file_id);
            writeln!(out, "# {} \"{}\"", current_line, path.display())?;
            last_line = current_line;
            last_file = current_file;
            need_space = false;
        } else if current_line > last_line {
            // 行が進んだ
            let gap = current_line - last_line;
            if gap <= 8 {
                // 小さいギャップは空行で埋める
                for _ in 0..gap {
                    writeln!(out)?;
                }
            } else {
                // 大きいギャップはディレクティブを使う
                writeln!(out)?;
                let path = pp.files().get_path(token.loc.file_id);
                writeln!(out, "# {} \"{}\"", current_line, path.display())?;
            }
            last_line = current_line;
            need_space = false;
        }

        // トークン間のスペース
        if need_space {
            write!(out, " ")?;
        }

        // トークンを出力
        write!(out, "{}", token.kind.format(pp.interner()))?;
        need_space = true;
    }

    writeln!(out)?;
    Ok(())
}

/// -D オプションをパース（NAME または NAME=VALUE 形式）
fn parse_defines(defines: &[String]) -> Vec<(String, Option<String>)> {
    defines
        .iter()
        .map(|s| {
            if let Some(pos) = s.find('=') {
                let (name, value) = s.split_at(pos);
                (name.to_string(), Some(value[1..].to_string()))
            } else {
                (s.clone(), None)
            }
        })
        .collect()
}