sword-layers 0.1.6

Tower layers for the Sword web framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use super::ServeDirConfig;
use tower_http::services::{ServeDir, ServeFile};

/// ### Serve Directory Layer
///
/// This struct represents the Serve Directory Layer which
/// serves static files and directories.
///
/// The layer is a wrapper around `tower_http::services::ServeDir`.
/// Serves static content with optional compression and custom chunk sizing.
pub struct ServeDirLayer;

impl ServeDirLayer {
    pub fn new(config: &ServeDirConfig) -> ServeDir<ServeFile> {
        let mut fallback = ServeFile::new(format!("{}/404.html", config.static_dir));

        if let Some(not_found_file) = &config.not_found_file {
            fallback = ServeFile::new(format!("{}/{not_found_file}", config.static_dir));
        }

        let mut layer = ServeDir::new(&config.static_dir).fallback(fallback);

        if let Some(algorithm) = &config.compression_algorithm {
            match algorithm.as_str() {
                "br" => {
                    layer = layer.precompressed_br();
                }
                "gzip" => {
                    layer = layer.precompressed_gzip();
                }
                "deflate" => {
                    layer = layer.precompressed_deflate();
                }
                "zstd" => {
                    layer = layer.precompressed_zstd();
                }
                _ => {}
            }
        }

        if let Some(chunk_size) = &config.chunk_size {
            layer = layer.with_buf_chunk_size(chunk_size.parsed);
        }

        layer
    }
}