dioxus_cli_opt/
file.rs

1use anyhow::Context;
2use manganis::{AssetOptions, CssModuleAssetOptions, FolderAssetOptions};
3use manganis_core::{AssetVariant, CssAssetOptions, ImageAssetOptions, JsAssetOptions};
4use std::path::Path;
5
6use crate::css::{process_css_module, process_scss};
7
8use super::{
9    css::process_css, folder::process_folder, image::process_image, js::process_js,
10    json::process_json,
11};
12
13/// Process a specific file asset with the given options reading from the source and writing to the output path
14pub fn process_file_to(
15    options: &AssetOptions,
16    source: &Path,
17    output_path: &Path,
18) -> anyhow::Result<()> {
19    process_file_to_with_options(options, source, output_path, false)
20}
21
22/// Process a specific file asset with additional options
23pub(crate) fn process_file_to_with_options(
24    options: &AssetOptions,
25    source: &Path,
26    output_path: &Path,
27    in_folder: bool,
28) -> anyhow::Result<()> {
29    // If the file already exists and this is a hashed asset, then we must have a file
30    // with the same hash already. The hash has the file contents and options, so if we
31    // find a file with the same hash, we probably already created this file in the past
32    if output_path.exists() && options.hash_suffix() {
33        return Ok(());
34    }
35    if let Some(parent) = output_path.parent() {
36        if !parent.exists() {
37            std::fs::create_dir_all(parent).context("Failed to create directory")?;
38        }
39    }
40
41    // Processing can be slow. Write to a temporary file first and then rename it to the final output path. If everything
42    // goes well. Without this, the user could quit in the middle of processing and the file will look complete to the
43    // caching system even though it is empty.
44    let temp_path = output_path.with_file_name(format!(
45        "partial.{}",
46        output_path
47            .file_name()
48            .unwrap_or_default()
49            .to_string_lossy()
50    ));
51    let resolved_options = resolve_asset_options(source, options.variant());
52
53    match &resolved_options {
54        ResolvedAssetType::Css(options) => {
55            process_css(options, source, &temp_path)?;
56        }
57        ResolvedAssetType::CssModule(options) => {
58            process_css_module(options, source, output_path, &temp_path)?;
59        }
60        ResolvedAssetType::Scss(options) => {
61            process_scss(options, source, &temp_path)?;
62        }
63        ResolvedAssetType::Js(options) => {
64            process_js(options, source, &temp_path, !in_folder)?;
65        }
66        ResolvedAssetType::Image(options) => {
67            process_image(options, source, &temp_path)?;
68        }
69        ResolvedAssetType::Json => {
70            process_json(source, &temp_path)?;
71        }
72        ResolvedAssetType::Folder(_) => {
73            process_folder(source, &temp_path)?;
74        }
75        ResolvedAssetType::File => {
76            let source_file = std::fs::File::open(source)?;
77            let mut reader = std::io::BufReader::new(source_file);
78            let output_file = std::fs::File::create(&temp_path)?;
79            let mut writer = std::io::BufWriter::new(output_file);
80            std::io::copy(&mut reader, &mut writer).with_context(|| {
81                format!(
82                    "Failed to write file to output location: {}",
83                    temp_path.display()
84                )
85            })?;
86        }
87    }
88
89    // Remove the existing output file if it exists
90    if output_path.exists() {
91        if output_path.is_file() {
92            std::fs::remove_file(output_path).context("Failed to remove previous output file")?;
93        } else if output_path.is_dir() {
94            std::fs::remove_dir_all(output_path)
95                .context("Failed to remove previous output file")?;
96        }
97    }
98
99    // If everything was successful, rename the temp file to the final output path
100    std::fs::rename(temp_path, output_path).context("Failed to rename output file")?;
101
102    Ok(())
103}
104
105pub(crate) enum ResolvedAssetType {
106    /// An image asset
107    Image(ImageAssetOptions),
108    /// A css asset
109    Css(CssAssetOptions),
110    /// A css module asset
111    CssModule(CssModuleAssetOptions),
112    /// A SCSS asset
113    Scss(CssAssetOptions),
114    /// A javascript asset
115    Js(JsAssetOptions),
116    /// A json asset
117    Json,
118    /// A folder asset
119    Folder(FolderAssetOptions),
120    /// A generic file
121    File,
122}
123
124pub(crate) fn resolve_asset_options(source: &Path, options: &AssetVariant) -> ResolvedAssetType {
125    match options {
126        AssetVariant::Image(image) => ResolvedAssetType::Image(*image),
127        AssetVariant::Css(css) => ResolvedAssetType::Css(*css),
128        AssetVariant::CssModule(css) => ResolvedAssetType::CssModule(*css),
129        AssetVariant::Js(js) => ResolvedAssetType::Js(*js),
130        AssetVariant::Folder(folder) => ResolvedAssetType::Folder(*folder),
131        AssetVariant::Unknown => resolve_unknown_asset_options(source),
132        _ => {
133            tracing::warn!("Unknown asset options... you may need to update the Dioxus CLI. Defaulting to a generic file: {:?}", options);
134            resolve_unknown_asset_options(source)
135        }
136    }
137}
138
139fn resolve_unknown_asset_options(source: &Path) -> ResolvedAssetType {
140    match source.extension().map(|e| e.to_string_lossy()).as_deref() {
141        Some("scss" | "sass") => ResolvedAssetType::Scss(CssAssetOptions::default()),
142        Some("css") => ResolvedAssetType::Css(CssAssetOptions::default()),
143        Some("js") => ResolvedAssetType::Js(JsAssetOptions::default()),
144        Some("json") => ResolvedAssetType::Json,
145        Some("jpg" | "jpeg" | "png" | "webp" | "avif") => {
146            ResolvedAssetType::Image(ImageAssetOptions::default())
147        }
148        _ if source.is_dir() => ResolvedAssetType::Folder(FolderAssetOptions::default()),
149        _ => ResolvedAssetType::File,
150    }
151}
dioxus_cli_opt/file.rs

dioxus_cli_opt/
file.rs
