dioxus_cli_opt/
lib.rs

1use anyhow::Context;
2use manganis::AssetOptions;
3use manganis_core::BundledAsset;
4use rayon::iter::{IntoParallelRefIterator, ParallelIterator};
5use serde::{Deserialize, Serialize};
6use std::collections::{HashMap, HashSet};
7use std::path::{Path, PathBuf};
8use std::sync::{Arc, RwLock};
9
10mod build_info;
11mod css;
12mod file;
13mod folder;
14mod hash;
15mod image;
16mod js;
17mod json;
18
19pub use file::process_file_to;
20pub use hash::add_hash_to_asset;
21
22/// A manifest of all assets collected from dependencies
23///
24/// This will be filled in primarily by incremental compilation artifacts.
25#[derive(Debug, PartialEq, Default, Clone, Serialize, Deserialize)]
26pub struct AssetManifest {
27    /// Map of bundled asset name to the asset itself
28    assets: HashMap<PathBuf, HashSet<BundledAsset>>,
29}
30
31impl AssetManifest {
32    /// Manually add an asset to the manifest
33    pub fn register_asset(
34        &mut self,
35        asset_path: &Path,
36        options: manganis::AssetOptions,
37    ) -> anyhow::Result<BundledAsset> {
38        let output_path_str = asset_path.to_str().ok_or(anyhow::anyhow!(
39            "Failed to convert wasm bindgen output path to string"
40        ))?;
41
42        let mut bundled_asset =
43            manganis::macro_helpers::create_bundled_asset(output_path_str, options);
44        add_hash_to_asset(&mut bundled_asset);
45
46        self.assets
47            .entry(asset_path.to_path_buf())
48            .or_default()
49            .insert(bundled_asset);
50
51        Ok(bundled_asset)
52    }
53
54    /// Insert an existing bundled asset to the manifest
55    pub fn insert_asset(&mut self, asset: BundledAsset) {
56        let asset_path = asset.absolute_source_path();
57        self.assets
58            .entry(asset_path.into())
59            .or_default()
60            .insert(asset);
61    }
62
63    /// Get any assets that are tied to a specific source file
64    pub fn get_assets_for_source(&self, path: &Path) -> Option<&HashSet<BundledAsset>> {
65        self.assets.get(path)
66    }
67
68    /// Get the first asset that matches the given source path
69    pub fn get_first_asset_for_source(&self, path: &Path) -> Option<&BundledAsset> {
70        self.assets
71            .get(path)
72            .and_then(|assets| assets.iter().next())
73    }
74
75    /// Check if the manifest contains a specific asset
76    pub fn contains(&self, asset: &BundledAsset) -> bool {
77        self.assets
78            .get(&PathBuf::from(asset.absolute_source_path()))
79            .is_some_and(|assets| assets.contains(asset))
80    }
81
82    /// Iterate over all the assets in the manifest
83    pub fn assets(&self) -> impl Iterator<Item = &BundledAsset> {
84        self.assets.values().flat_map(|assets| assets.iter())
85    }
86
87    pub fn load_from_file(path: &Path) -> anyhow::Result<Self> {
88        let src = std::fs::read_to_string(path)?;
89
90        serde_json::from_str(&src)
91            .with_context(|| format!("Failed to parse asset manifest from {path:?}\n{src}"))
92    }
93}
94
95/// Optimize a list of assets in parallel
96pub fn optimize_all_assets(
97    assets_to_transfer: Vec<(PathBuf, PathBuf, AssetOptions)>,
98    on_optimization_start: impl FnMut(&Path, &Path, &AssetOptions) + Sync + Send,
99    on_optimization_end: impl FnMut(&Path, &Path, &AssetOptions) + Sync + Send,
100) -> anyhow::Result<()> {
101    let on_optimization_start = Arc::new(RwLock::new(on_optimization_start));
102    let on_optimization_end = Arc::new(RwLock::new(on_optimization_end));
103    assets_to_transfer
104        .par_iter()
105        .try_for_each(|(from, to, options)| {
106            {
107                let mut on_optimization_start = on_optimization_start.write().unwrap();
108                on_optimization_start(from, to, options);
109            }
110
111            let res = process_file_to(options, from, to);
112            if let Err(err) = res.as_ref() {
113                tracing::error!("Failed to copy asset {from:?}: {err}");
114            }
115
116            {
117                let mut on_optimization_end = on_optimization_end.write().unwrap();
118                on_optimization_end(from, to, options);
119            }
120
121            res.map(|_| ())
122        })
123}
dioxus_cli_opt/lib.rs

dioxus_cli_opt/
lib.rs
