code2prompt_core 4.2.0

//! Built-in templates embedded as static resources.
//!
//! This module provides access to all built-in templates that are embedded
//! directly into the binary, making them available even when the crate is
//! installed from crates.io without access to the source file structure.

use std::{collections::HashMap, sync::OnceLock};

/// Information about a built-in template
#[derive(Debug, Clone, Copy)]
pub struct BuiltinTemplate {
    pub name: &'static str,
    pub content: &'static str,
    pub description: &'static str,
}

/// All built-in templates embedded as static strings
pub struct BuiltinTemplates;

static TEMPLATES: OnceLock<HashMap<&'static str, BuiltinTemplate>> = OnceLock::new();

impl BuiltinTemplates {
    /// Get all available built-in templates
    pub fn get_all() -> &'static HashMap<&'static str, BuiltinTemplate> {
        TEMPLATES.get_or_init(|| {
            HashMap::from([
                (
                    "default-markdown",
                    BuiltinTemplate {
                        name: "Default (Markdown)",
                        content: include_str!("default_template_md.hbs"),
                        description: "Default markdown template for code analysis",
                    },
                ),
                (
                    "default-xml",
                    BuiltinTemplate {
                        name: "Default (XML)",
                        content: include_str!("default_template_xml.hbs"),
                        description: "Default XML template for code analysis",
                    },
                ),
                (
                    "binary-exploitation-ctf-solver",
                    BuiltinTemplate {
                        name: "Binary Exploitation CTF Solver",
                        content: include_str!("../templates/binary-exploitation-ctf-solver.hbs"),
                        description: "Template for solving binary exploitation CTF challenges",
                    },
                ),
                (
                    "clean-up-code",
                    BuiltinTemplate {
                        name: "Clean Up Code",
                        content: include_str!("../templates/clean-up-code.hbs"),
                        description: "Template for code cleanup and refactoring",
                    },
                ),
                (
                    "cryptography-ctf-solver",
                    BuiltinTemplate {
                        name: "Cryptography CTF Solver",
                        content: include_str!("../templates/cryptography-ctf-solver.hbs"),
                        description: "Template for solving cryptography CTF challenges",
                    },
                ),
                (
                    "document-the-code",
                    BuiltinTemplate {
                        name: "Document the Code",
                        content: include_str!("../templates/document-the-code.hbs"),
                        description: "Template for generating code documentation",
                    },
                ),
                (
                    "find-security-vulnerabilities",
                    BuiltinTemplate {
                        name: "Find Security Vulnerabilities",
                        content: include_str!("../templates/find-security-vulnerabilities.hbs"),
                        description: "Template for security vulnerability analysis",
                    },
                ),
                (
                    "fix-bugs",
                    BuiltinTemplate {
                        name: "Fix Bugs",
                        content: include_str!("../templates/fix-bugs.hbs"),
                        description: "Template for bug fixing and debugging",
                    },
                ),
                (
                    "improve-performance",
                    BuiltinTemplate {
                        name: "Improve Performance",
                        content: include_str!("../templates/improve-performance.hbs"),
                        description: "Template for performance optimization",
                    },
                ),
                (
                    "refactor",
                    BuiltinTemplate {
                        name: "Refactor",
                        content: include_str!("../templates/refactor.hbs"),
                        description: "Template for code refactoring",
                    },
                ),
                (
                    "reverse-engineering-ctf-solver",
                    BuiltinTemplate {
                        name: "Reverse Engineering CTF Solver",
                        content: include_str!("../templates/reverse-engineering-ctf-solver.hbs"),
                        description: "Template for solving reverse engineering CTF challenges",
                    },
                ),
                (
                    "web-ctf-solver",
                    BuiltinTemplate {
                        name: "Web CTF Solver",
                        content: include_str!("../templates/web-ctf-solver.hbs"),
                        description: "Template for solving web CTF challenges",
                    },
                ),
                (
                    "write-git-commit",
                    BuiltinTemplate {
                        name: "Write Git Commit",
                        content: include_str!("../templates/write-git-commit.hbs"),
                        description: "Template for generating git commit messages",
                    },
                ),
                (
                    "write-github-pull-request",
                    BuiltinTemplate {
                        name: "Write GitHub Pull Request",
                        content: include_str!("../templates/write-github-pull-request.hbs"),
                        description: "Template for generating GitHub pull request descriptions",
                    },
                ),
                (
                    "write-github-readme",
                    BuiltinTemplate {
                        name: "Write GitHub README",
                        content: include_str!("../templates/write-github-readme.hbs"),
                        description: "Template for generating GitHub README files",
                    },
                ),
            ])
        })
    }

    /// Get a specific template by its key
    pub fn get_template(key: &str) -> Option<BuiltinTemplate> {
        Self::get_all().get(key).cloned()
    }

    /// Get all template keys
    pub fn get_template_keys() -> Vec<&'static str> {
        Self::get_all().keys().copied().collect()
    }

    /// Check if a template exists
    pub fn has_template(key: &str) -> bool {
        Self::get_all().contains_key(key)
    }
}