systemprompt-generator 0.21.1

//! Reorganise the freshly-built `dist/` directory by moving CSS and JS files
//! from the root into per-extension subdirectories so the generated site has
//! a tidy `dist/css/*.css` and `dist/js/*.js` layout.
//!
//! Copyright (c) systemprompt.io — Business Source License 1.1.
//! See <https://systemprompt.io> for licensing details.

use std::path::Path;
use tokio::fs;

use crate::error::{GeneratorResult, PublishError};

pub async fn organize_dist_assets(dist_dir: &Path) -> GeneratorResult<(u32, u32)> {
    let css_count = organize_assets_by_extension(dist_dir, "css").await?;
    let js_count = organize_assets_by_extension(dist_dir, "js").await?;
    Ok((css_count, js_count))
}

async fn organize_assets_by_extension(dist_dir: &Path, ext: &str) -> GeneratorResult<u32> {
    let target_dir = dist_dir.join(ext);
    fs::create_dir_all(&target_dir)
        .await
        .map_err(|e| PublishError::io_context(format!("Failed to create {ext} directory"), e))?;
    copy_files_by_extension(dist_dir, &target_dir, ext).await
}

async fn copy_files_by_extension(
    source_dir: &Path,
    dest_dir: &Path,
    ext: &str,
) -> GeneratorResult<u32> {
    let mut copied = 0;
    let mut entries = fs::read_dir(source_dir)
        .await
        .map_err(|e| PublishError::io_context("Failed to read source directory", e))?;

    while let Some(entry) = entries
        .next_entry()
        .await
        .map_err(|e| PublishError::io_context("Failed to read entry", e))?
    {
        let path = entry.path();
        let matches_ext = path.extension().is_some_and(|e| e == ext);

        if matches_ext && let Some(file_name) = path.file_name() {
            let dest = dest_dir.join(file_name);
            fs::copy(&path, &dest).await.map_err(|e| {
                PublishError::io_context(format!("Failed to copy {}", file_name.display()), e)
            })?;
            copied += 1;
        }
    }

    Ok(copied)
}