hauchiwa 0.21.0

use std::any::type_name;
use std::borrow::Cow;
use std::marker::PhantomData;
use std::sync::Arc;

use camino::Utf8PathBuf;
use glob::Pattern;
use petgraph::Graph;

use crate::core::{Environment, Mode, Store};
use crate::engine::{
    Dependencies, Many, NodeGather, NodeMap, NodeScatter, One, Task, TypedCoarse, TypedFine,
    run_once_parallel,
};
use crate::error::HauchiwaError;
use crate::loader::Input;
use crate::utils::ProgressStyles;
use crate::{Diagnostics, TaskContext};

/// The blueprint for your static site.
///
/// `Blueprint` is used to define the Task graph of your website. You add tasks
/// (including loaders) to the config, and wire them together using their
/// references like [`One`](crate::One) or [`Many`](crate::Many).
///
/// Once configured, you convert this into a [`Website`] to execute the build.
///
/// # Example
///
/// ```rust,no_run
/// use hauchiwa::Blueprint;
///
/// let mut config: Blueprint<()> = Blueprint::new();
/// // Add tasks here...
/// ```
pub struct Blueprint<G: Send + Sync = ()> {
    pub(crate) graph: Graph<Task<G>, ()>,
    pub(crate) copied: Vec<(String, String)>,
    pub(crate) out_dir: Utf8PathBuf,
    pub(crate) cache_dir: Utf8PathBuf,
    pub(crate) progress: ProgressStyles,
}

impl<G: Send + Sync + 'static> Blueprint<G> {
    /// Creates a new, empty configuration.
    pub fn new() -> Self {
        Self::default()
    }

    /// Sets the output directory (default: `"dist"`).
    #[must_use]
    pub fn set_dir_dist(mut self, dir: impl Into<Utf8PathBuf>) -> Self {
        self.out_dir = dir.into();
        self
    }

    /// Sets the cache directory (default: `".cache"`).
    #[must_use]
    pub fn set_dir_cache(mut self, dir: impl Into<Utf8PathBuf>) -> Self {
        self.cache_dir = dir.into();
        self
    }

    /// Overrides the progress bar styles used during builds.
    ///
    /// See [`ProgressStyles`] for the available fields and their defaults.
    #[must_use]
    pub fn set_progress_styles(mut self, styles: ProgressStyles) -> Self {
        self.progress = styles;
        self
    }

    #[must_use]
    pub fn copy_static(mut self, src: impl Into<String>, dest: impl Into<String>) -> Self {
        self.copied.push((dest.into(), src.into()));
        self
    }

    pub fn task(&mut self) -> TaskDef<'_, G> {
        TaskDef {
            blueprint: self,
            name: None,
        }
    }

    #[must_use]
    pub fn finish(self) -> Website<G> {
        Website {
            graph: self.graph,
            copied: self.copied,
            out_dir: self.out_dir,
            cache_dir: self.cache_dir,
            progress: self.progress,
        }
    }

    pub(crate) fn add_task_fine<O, T>(&mut self, task: T) -> Many<O>
    where
        O: 'static,
        T: TypedFine<G, Output = O> + 'static,
    {
        let dependencies = task.dependencies();
        let index = self.graph.add_node(Task::F(Arc::new(task)));

        for dependency in dependencies {
            self.graph.add_edge(dependency, index, ());
        }

        Many::new(index)
    }

    pub(crate) fn add_task_coarse<O, T>(&mut self, task: T) -> One<O>
    where
        O: 'static,
        T: TypedCoarse<G, Output = O> + 'static,
    {
        let dependencies = task.dependencies();
        let index = self.graph.add_node(Task::C(Arc::new(task)));

        for dependency in dependencies {
            self.graph.add_edge(dependency, index, ());
        }

        One::new(index)
    }
}

impl<G: Send + Sync> Default for Blueprint<G> {
    fn default() -> Self {
        Self {
            copied: Vec::default(),
            graph: Graph::default(),
            out_dir: Utf8PathBuf::from("dist"),
            cache_dir: Utf8PathBuf::from(".cache"),
            progress: ProgressStyles::default(),
        }
    }
}

impl<G> std::fmt::Display for Blueprint<G>
where
    G: Send + Sync + 'static,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        writeln!(f, "graph LR")?;

        for index in self.graph.node_indices() {
            let task = &self.graph[index];
            let name = task.name().replace('"', "\\\""); // Simple escape
            writeln!(f, "    {:?}[\"{}\"]", index.index(), name)?;

            if task.is_output() {
                writeln!(f, "    {:?} --> Output", index.index())?;
            }
        }

        writeln!(f, "    Output[Output]")?;

        for edge in self.graph.edge_indices() {
            #[allow(clippy::unwrap_used)] // edge_indices() only returns valid indices
            let (source, target) = self.graph.edge_endpoints(edge).unwrap();
            let source_task = &self.graph[source];
            let type_name = source_task
                .type_name_output()
                .replace('<', "&lt;")
                .replace('>', "&gt;");
            writeln!(
                f,
                "    {:?} -- \"{}\" --> {:?}",
                source.index(),
                type_name,
                target.index()
            )?;
        }

        Ok(())
    }
}

/// Entry point for defining a new task. Created by [`Blueprint::task`].
///
/// Chain `.glob()`, `.each()`, `.using()`, or `.run()` to configure the task
/// and produce a [`One`] or [`Many`] handle.
#[must_use]
pub struct TaskDef<'a, G: Send + Sync> {
    blueprint: &'a mut Blueprint<G>,
    name: Option<Cow<'static, str>>,
}

impl<'a, G: Send + Sync + 'static> TaskDef<'a, G> {
    pub fn name(mut self, name: impl Into<Cow<'static, str>>) -> Self {
        self.name = Some(name.into());
        self
    }

    /// Load assets from file system using glob pattern.
    pub fn glob(self, glob: impl Into<String>) -> Result<TaskBinderGlob<'a, G>, HauchiwaError> {
        let glob = glob.into();
        let pattern = Pattern::new(&glob)?;
        Ok(TaskBinderGlob {
            blueprint: self.blueprint,
            name: self.name,
            entry: vec![glob],
            watch: vec![pattern],
        })
    }

    /// Perform a map on a collection.
    pub fn each<T>(self, each: Many<T>) -> TaskBinderEach<'a, G, T, ()> {
        TaskBinderEach {
            blueprint: self.blueprint,
            name: self.name,
            primary: each,
            secondary: (),
        }
    }

    /// Set task dependencies.
    pub fn using<D>(self, dependencies: D) -> TaskBinder<'a, G, D>
    where
        D: Dependencies,
    {
        TaskBinder {
            blueprint: self.blueprint,
            name: self.name,
            dependencies,
        }
    }

    /// Immediately run a task with no dependencies.
    pub fn run<F, R>(self, callback: F) -> One<R>
    where
        F: Fn(&TaskContext<'_, G>) -> anyhow::Result<R> + Send + Sync + 'static,
        R: Send + Sync + 'static,
    {
        self.blueprint.add_task_coarse(NodeGather {
            name: self.name.unwrap_or(type_name::<F>().into()),
            dependencies: (),
            callback: move |ctx, ()| callback(ctx),
            _phantom: PhantomData,
        })
    }
}

/// A task builder that loads files matching one or more glob patterns.
///
/// Created by [`TaskDef::glob`]. Call `.map()` to process each matched file
/// and produce a [`Many`] handle.
#[must_use]
pub struct TaskBinderGlob<'a, G: Send + Sync> {
    blueprint: &'a mut Blueprint<G>,
    name: Option<Cow<'static, str>>,
    entry: Vec<String>,
    watch: Vec<Pattern>,
}

impl<'a, G: Send + Sync + 'static> TaskBinderGlob<'a, G> {
    pub fn name(mut self, name: impl Into<Cow<'static, str>>) -> Self {
        self.name = Some(name.into());
        self
    }

    pub fn glob(mut self, glob: impl Into<String>) -> Result<Self, HauchiwaError> {
        let glob = glob.into();
        let pattern = Pattern::new(&glob)?;
        self.entry.push(glob);
        self.watch.push(pattern);
        Ok(self)
    }

    pub fn map<F, R>(self, callback: F) -> Many<R>
    where
        F: Fn(&TaskContext<G>, &mut Store, Input) -> anyhow::Result<R> + Send + Sync + 'static,
        R: Send + Sync + 'static,
    {
        let task =
            crate::loader::GlobFiles::new(self.entry, self.watch, move |ctx, store, input| {
                let path = input.path.clone();
                let res = callback(ctx, store, input)?;
                Ok((path, res))
            });

        self.blueprint.add_task_fine(task)
    }
}

/// A task builder that maps a function over every item in a [`Many`] collection.
///
/// Created by [`TaskDef::each`]. Optionally add side dependencies with `.using()`,
/// then call `.map()` to produce a new [`Many`] handle.
#[must_use]
pub struct TaskBinderEach<'a, G, T, D>
where
    G: Send + Sync,
{
    blueprint: &'a mut Blueprint<G>,
    name: Option<Cow<'static, str>>,
    primary: Many<T>,
    secondary: D,
}

impl<'a, G, T, D> TaskBinderEach<'a, G, T, D>
where
    G: Send + Sync + 'static,
    T: Send + Sync + 'static,
    D: Dependencies + Send + Sync + 'static,
{
    pub fn name(mut self, name: impl Into<Cow<'static, str>>) -> Self {
        self.name = Some(name.into());
        self
    }

    /// Add secondary dependencies (context) that are needed for every item mapping.
    pub fn using<D2>(self, dependencies: D2) -> TaskBinderEach<'a, G, T, D2>
    where
        D2: Dependencies,
    {
        TaskBinderEach {
            blueprint: self.blueprint,
            name: self.name,
            primary: self.primary,
            secondary: dependencies,
        }
    }

    /// Execute the mapping function.
    /// The callback receives the context, the individual item `&T`, and the resolved secondary dependencies.
    pub fn map<F, R>(self, callback: F) -> Many<R>
    where
        F: for<'b> Fn(&TaskContext<'b, G>, &T, D::Output<'b>) -> anyhow::Result<R>
            + Send
            + Sync
            + 'static,
        R: Send + Sync + Clone + 'static,
    {
        self.blueprint.add_task_fine(NodeMap {
            name: self.name.unwrap_or(type_name::<F>().into()),
            dep_primary: self.primary,
            dep_secondary: self.secondary,
            callback,
            _phantom: PhantomData,
        })
    }
}

/// A task builder with explicit dependencies.
///
/// Created by [`TaskDef::using`]. Call `.merge()` to produce a single [`One`]
/// output, or `.spread()` to produce a keyed [`Many`] collection.
#[must_use]
pub struct TaskBinder<'a, G: Send + Sync, D> {
    blueprint: &'a mut Blueprint<G>,
    name: Option<Cow<'static, str>>,
    dependencies: D,
}

impl<'a, G, D> TaskBinder<'a, G, D>
where
    G: Send + Sync + 'static,
    D: Dependencies + Send + Sync + 'static,
{
    pub fn name(mut self, name: impl Into<Cow<'static, str>>) -> Self {
        self.name = Some(name.into());
        self
    }

    pub fn merge<F, R>(self, callback: F) -> One<R>
    where
        F: for<'b> Fn(&TaskContext<'b, G>, D::Output<'b>) -> anyhow::Result<R>
            + Send
            + Sync
            + 'static,
        R: Send + Sync + 'static,
    {
        self.blueprint.add_task_coarse(NodeGather {
            name: self.name.unwrap_or(type_name::<F>().into()),
            dependencies: self.dependencies,
            callback,
            _phantom: PhantomData,
        })
    }

    pub fn spread<F, R>(self, callback: F) -> Many<R>
    where
        F: for<'b> Fn(&TaskContext<'b, G>, D::Output<'b>) -> anyhow::Result<Vec<(String, R)>>
            + Send
            + Sync
            + 'static,
        R: Send + Sync + std::hash::Hash + 'static,
    {
        self.blueprint.add_task_fine(NodeScatter {
            name: self.name.unwrap_or(type_name::<F>().into()),
            dependencies: self.dependencies,
            callback,
            _phantom: PhantomData,
        })
    }
}

/// Represents the configured site and provides methods for building and serving
/// it with a development server.
///
/// A [`Website`] is created from a [`Blueprint`] and is the primary interface
/// for executing the build process.
pub struct Website<G: Send + Sync = ()> {
    pub(crate) graph: Graph<Task<G>, ()>,
    pub(crate) copied: Vec<(String, String)>,
    pub(crate) out_dir: Utf8PathBuf,
    pub(crate) cache_dir: Utf8PathBuf,
    pub(crate) progress: ProgressStyles,
}

impl<G> Website<G>
where
    G: Send + Sync + 'static,
{
    fn run_preflight(&self) -> Result<(), crate::error::HauchiwaError> {
        use std::collections::HashSet;
        let mut seen = HashSet::new();
        let mut missing = Vec::new();

        for idx in self.graph.node_indices() {
            for req in self.graph[idx].requirements() {
                if seen.insert(req.clone()) && !req.check() {
                    missing.push(req);
                }
            }
        }

        if missing.is_empty() {
            return Ok(());
        }

        let list = missing
            .iter()
            .map(|r| format!("  - {r}"))
            .collect::<Vec<_>>()
            .join("\n");
        Err(crate::error::HauchiwaError::Preflight(list))
    }

    /// Runs the build process once.
    ///
    /// This will:
    /// 1. Clean the `dist` directory.
    /// 2. Copy static files.
    /// 3. Execute the task graph in parallel.
    /// 4. Save the generated [`Output`](crate::Output)s to `dist`.
    ///
    /// # Arguments
    ///
    /// * `data` - The global user data to pass to all tasks.
    pub fn build(&mut self, data: G) -> Result<Diagnostics, crate::error::HauchiwaError> {
        use crate::error::BuildError;
        self.run_preflight()?;

        let globals = Environment {
            generator: "hauchiwa",
            mode: Mode::Build,
            port: None,
            data,
        };

        let prev_meta =
            crate::snapshot::SnapshotMeta::load(&self.cache_dir).map_err(BuildError::Io)?;
        let static_files = crate::utils::collect_static(&self.copied, &self.out_dir)?;

        let (_, mut manifest, diagnostics) = run_once_parallel(self, &globals)?;

        for entry in &static_files {
            manifest.insert_static_file(entry.dist_rel.clone(), entry.source_utf8.clone())?;
        }
        crate::utils::copy_static_entries(&static_files, &self.progress.copy)?;

        match prev_meta {
            Some(ref prev) => manifest
                .commit_diff_meta(prev, &self.out_dir)
                .map_err(BuildError::Io)?,
            None => manifest.commit(&self.out_dir).map_err(BuildError::Io)?,
        }
        manifest
            .to_meta()
            .save(&self.cache_dir)
            .map_err(BuildError::Io)?;

        Ok(diagnostics)
    }

    /// Starts the development server in watch mode.
    ///
    /// This will perform an initial build and then watch for file changes.
    /// When a file changes, only the affected tasks are re-run.
    ///
    /// # Arguments
    ///
    /// * `data` - The global user data to pass to all tasks.
    #[cfg(feature = "live")]
    pub fn watch(&mut self, data: G) -> Result<(), crate::error::HauchiwaError> {
        self.run_preflight()?;

        let out_dir = self.out_dir.clone();
        let cache_dir = self.cache_dir.clone();
        let copied = self.copied.clone();

        crate::engine::watch(self, data, copied, &out_dir, &cache_dir)
            .map_err(crate::error::WatchError::Other)?;

        Ok(())
    }
}