ssg 0.0.38

#![allow(clippy::unwrap_used, clippy::expect_used)]
// Copyright © 2023 - 2026 Static Site Generator (SSG). All rights reserved. All rights reserved.
// SPDX-License-Identifier: Apache-2.0 OR MIT

//! # Multilingual Example — 28-locale i18n showcase
//!
//! ## What this example demonstrates
//!
//! - **Per-locale search index + hreflang** — each language gets its own index and SEO tags
//! - **Auto-generated language switcher** — all 28 locales rendered at the site root
//! - **Accept-Language negotiation + `x-default` fallback** — browser picks the right locale
//!
//! ## When to use this pattern
//!
//! Use this example when shipping a site that must serve many languages with
//! correct SEO signals and a graceful fallback for unsupported locales.
//!
//! ## Run it
//!
//! ```sh
//! cargo run --release --example multilingual_example
//! ```
//!
//! Then open <http://127.0.0.1:3005> in your browser.
//!
//! ## What makes this different from other examples
//!
//! Unlike `blog` which is single-language, this example generates 28 parallel
//! locale trees with hreflang wiring and automatic language negotiation.

use anyhow::Result;
use http_handle::Server;
use ssg::plugin::{PluginContext, PluginManager};
use ssg::search::{LocalizedSearchPlugin, SearchLabels};
use ssg::seo::SeoPlugin;
use staticdatagen::compiler::service::compile;
use std::fs;
use std::path::Path;
use std::time::Instant;

/// Supported locales as (code, native name) pairs.
/// Matches the language set offered on bankstatementparser.com.
const LANGUAGES: &[(&str, &str)] = &[
    ("en", "English"),
    ("fr", "Français"),
    ("ar", "العربية"),
    ("bn", "বাংলা"),
    ("cs", "Čeština"),
    ("de", "Deutsch"),
    ("es", "Español"),
    ("ha", "Hausa"),
    ("he", "עברית"),
    ("hi", "हिन्दी"),
    ("id", "Indonesia"),
    ("it", "Italiano"),
    ("ja", "日本語"),
    ("ko", "한국어"),
    ("nl", "Nederlands"),
    ("pl", "Polski"),
    ("pt", "Português"),
    ("ro", "Română"),
    ("ru", "Русский"),
    ("sv", "Svenska"),
    ("th", "ไทย"),
    ("tl", "Filipino"),
    ("tr", "Türkçe"),
    ("uk", "Українська"),
    ("vi", "Tiếng Việt"),
    ("yo", "Yorùbá"),
    ("zh", "简体中文"),
    ("zh-tw", "繁體中文"),
];

fn main() -> Result<()> {
    // Define supported languages
    let languages: Vec<&str> =
        LANGUAGES.iter().map(|(code, _)| *code).collect();

    // Root directory for public files
    let public_root = Path::new("./examples/public");
    fs::create_dir_all(public_root)?;

    let start = Instant::now();

    // Generate sites for all languages
    for lang in &languages {
        println!("Processing language: {lang}");

        // Define paths specific to the language
        let build_dir = Path::new("./examples/build").join(lang);
        let site_dir = public_root.join(lang);
        let content_dir = Path::new("./examples/content").join(lang);
        let template_dir = Path::new("./examples/templates").join(lang);

        // Call the compile function to generate the website
        println!("    🔍 Compiling content for language: {lang}...");
        match compile(&build_dir, &content_dir, &site_dir, &template_dir) {
            Ok(()) => println!(
                "    ✅ Successfully compiled static site for language: {lang}"
            ),
            Err(e) => {
                println!("    ❌ Error compiling site for {lang}: {e:?}");
                return Err(e);
            }
        }

        // Run plugins (SEO + Search) for this language
        let mut plugins = PluginManager::new();
        plugins.register(SeoPlugin);
        plugins.register(LocalizedSearchPlugin::new(SearchLabels::for_locale(
            lang,
        )));
        let ctx = PluginContext::new(
            &content_dir,
            &build_dir,
            &site_dir,
            &template_dir,
        );
        plugins.run_after_compile(&ctx)?;
        plugins.run_fused_transforms(&ctx)?;
        println!("    🔌 Plugins complete for {lang}");
    }

    // Run the I18nPlugin once over the whole site. This single pass:
    //   - injects <link rel="alternate" hreflang="…"> into every page
    //   - generates per-locale XML sitemaps for search-engine discovery
    //   - replaces <!-- ssg:lang-switcher --> with a 28-locale dropdown
    //   - sets the `x-default` hreflang for Accept-Language negotiation
    {
        use ssg::i18n::{I18nConfig, I18nPlugin, UrlPrefixStrategy};
        let i18n_cfg = I18nConfig {
            default_locale: "en".to_string(),
            locales: languages.iter().map(|s| (*s).to_string()).collect(),
            url_prefix: UrlPrefixStrategy::SubPath,
        };
        let i18n_plugin = I18nPlugin::new(i18n_cfg);
        let ctx = PluginContext::new(
            Path::new("./examples/content"),
            Path::new("./examples/build"),
            public_root,
            Path::new("./examples/templates"),
        );
        use ssg::plugin::Plugin;
        i18n_plugin.after_compile(&ctx)?;
        println!("    🌍 I18nPlugin injected hreflang + language switcher");
    }

    // Promote English to the site root: copy every file from `public/en/`
    // into `public/` so that `/` serves English directly. Other locales remain
    // at `/<lang>/`. This mirrors the convention used by sites like
    // bankstatementparser.com where the default language has no path prefix.
    let en_root = public_root.join("en");
    if en_root.exists() {
        copy_dir_recursive(&en_root, public_root)?;
        println!("    ✅ Promoted English to site root");
    }

    let elapsed = start.elapsed();
    println!("    ⚡ Built in {elapsed:.0?}");

    // Serve the root public directory.
    //
    // Host/port can be overridden via env vars so WSL2, Codespaces and
    // dev-container users can opt into `0.0.0.0` without editing code:
    //   SSG_HOST=0.0.0.0 SSG_PORT=8080 cargo run --example multilingual
    let host = std::env::var("SSG_HOST")
        .ok()
        .filter(|v| !v.is_empty())
        .unwrap_or_else(|| "127.0.0.1".to_string());
    let port = std::env::var("SSG_PORT")
        .ok()
        .and_then(|v| v.parse::<u16>().ok())
        .unwrap_or(3005);
    let bind = format!("{host}:{port}");
    // Build the server with a Permissions-Policy header that opts the
    // page out of the Topics API. Suppresses the "Browsing Topics API
    // removed" Chrome console message in dev mode.
    let server = Server::builder()
        .address(&bind)
        .document_root(public_root.to_str().unwrap())
        .custom_header("Permissions-Policy", "browsing-topics=()")
        .build()
        .map_err(|e| anyhow::anyhow!("{e}"))?;
    println!("Serving site at http://{bind}");
    let _ = server.start();

    Ok(())
}

/// Recursively copies every file from `src` into `dst`, creating `dst` and
/// any intermediate directories as needed. Existing files in `dst` are
/// overwritten so that promoting a locale to the site root is idempotent.
fn copy_dir_recursive(src: &Path, dst: &Path) -> Result<()> {
    if !dst.exists() {
        fs::create_dir_all(dst)?;
    }
    for entry in fs::read_dir(src)? {
        let entry = entry?;
        let file_type = entry.file_type()?;
        let dst_path = dst.join(entry.file_name());
        if file_type.is_dir() {
            copy_dir_recursive(&entry.path(), &dst_path)?;
        } else {
            let _ = fs::copy(entry.path(), &dst_path)?;
        }
    }
    Ok(())
}