mCaptcha/utils/cache-bust/src/main.rs

// Copyright (C) 2021  Aravinth Manivannan <realaravinth@batsense.net>
// SPDX-FileCopyrightText: 2023 Aravinth Manivannan <realaravinth@batsense.net>
//
// SPDX-License-Identifier: AGPL-3.0-or-later

use std::fs;
use std::path::Path;
use std::collections::HashMap;

use libcachebust::{BusterBuilder, CACHE_BUSTER_DATA_FILE, NoHashCategory};
use serde::{Serialize, Deserialize};

#[derive(Deserialize, Serialize)]
struct  FileMap {
    map: HashMap<String, String>,
  base_dir: String,
}

fn main() {
    cache_bust();
    process_file_map();
}

fn cache_bust() {
    //    until APPLICATION_WASM gets added to mime crate
    //    PR: https://github.com/hyperium/mime/pull/138
    //    let types = vec![
    //        mime::IMAGE_PNG,
    //        mime::IMAGE_SVG,
    //        mime::IMAGE_JPEG,
    //        mime::IMAGE_GIF,
    //        mime::APPLICATION_JAVASCRIPT,
    //        mime::TEXT_CSS,
    //    ];

    println!("[*] Cache busting");
    let no_hash = vec![NoHashCategory::FileExtentions(vec!["wasm"])];

    let config = BusterBuilder::default()
        .source("../../static/cache/")
        .result("./../../assets")
        .no_hash(no_hash)
        .follow_links(true)
        .build()
        .unwrap();

    config.process().unwrap();
}

fn process_file_map() {
    let contents = fs::read_to_string(CACHE_BUSTER_DATA_FILE).unwrap();
    let files: FileMap = serde_json::from_str(&contents).unwrap();
    let mut map = HashMap::with_capacity(files.map.len()); 
    for (k, v) in files.map.iter() {
        map.insert(k.strip_prefix("../.").unwrap().to_owned(),
        v.strip_prefix("./../.").unwrap().to_owned()
        );
    }

    let new_filemap = FileMap{
        map,
        base_dir: files.base_dir.strip_prefix("./../.").unwrap().to_owned(),
    };

    let dest = Path::new("../../").join(CACHE_BUSTER_DATA_FILE);
    fs::write(&dest, serde_json::to_string(&new_filemap).unwrap()).unwrap();
}