wolfree-dockerfile/rust/wolfree_sed_in_place/src/main.rs

75 lines
2.8 KiB
Rust
Raw Normal View History

2023-09-27 13:34:23 +02:00
/* SPDX-License-Identifier: AGPL-3.0-or-later */
//! This crate provides a function for performing in-place replacements of patterns in multiple files using regular expressions.
2023-08-11 07:09:24 +02:00
//! It allows you to recursively process all files within a specified directory and replace occurrences of a given pattern with a specified replacement string.
#![allow(clippy::blanket_clippy_restriction_lints)]
#![allow(clippy::implicit_return)]
2023-09-27 13:34:23 +02:00
#![allow(clippy::question_mark_used)]
2023-08-11 07:09:24 +02:00
use regex::Regex;
2023-09-27 13:34:23 +02:00
use std::error;
2023-08-11 07:09:24 +02:00
use std::fs;
use walkdir::WalkDir;
/// Perform in-place replacements on files in a given directory, using regular expressions.
///
/// This function walks through all the files in the specified `directory_path`,
/// filters out directories, and applies a regular expression-based replacement
/// on the contents of each file.
///
/// * `directory_path`: A string slice representing the path of the directory to traverse.
/// * `pattern`: A string slice containing the regular expression pattern to search for.
/// * `replacement`: A string slice representing the replacement for each matched pattern.
2023-09-27 13:34:23 +02:00
fn sed(
directory_path: &str,
pattern: &str,
replacement: &str
) -> Result<(), Box<dyn error::Error>> {
// Compile the regular expression pattern.
let regex = Regex::new(pattern)?;
// Create a directory walker and filter out non-file entries.
for entry in WalkDir::new(directory_path)
2023-08-11 07:09:24 +02:00
.into_iter()
.filter_map(Result::ok)
2023-09-27 13:34:23 +02:00
.filter(|entry| !entry.file_type().is_dir()) {
2023-08-11 07:09:24 +02:00
let file_path = entry.path();
2023-09-27 13:34:23 +02:00
// Read the file's contents into a string.
// Perform the regex replacement on the file's contents.
// Write the modified contents back to the file, overwriting its previous contents.
fs::write(file_path, &*regex.replace(&fs::read_to_string(file_path)?, replacement))?;
2023-08-11 07:09:24 +02:00
}
2023-09-27 13:34:23 +02:00
Ok(())
2023-08-11 07:09:24 +02:00
}
/// Entry point of the program.
2023-09-27 13:34:23 +02:00
/// Demonstrates using `sed` function to perform in-place replacements on files in specific directories.
fn main() -> Result<(), Box<dyn error::Error>> {
sed("./docusaurus/static/input/", "</head><body>", include_str!("include_str.html"))?;
2023-08-11 07:09:24 +02:00
2023-09-27 13:34:23 +02:00
sed(
"./docusaurus/static/_next/static/chunks/",
2023-08-11 07:09:24 +02:00
r"try(.{0,100}?)generateEncodedJSONFromValue(.*?)unescapeForUrl(.*?)catch(.*?)\{}",
2023-09-27 13:34:23 +02:00
include_str!("include_str.js")
)?;
2023-08-11 07:09:24 +02:00
// console error:
// WebSocket connection to 'wss://localhost/n/v1/api/fetcher/results' failed:
// fix:
2023-09-27 13:34:23 +02:00
// override the hostname
sed(
"./docusaurus/static/_next/static/chunks/pages/",
2023-08-11 07:09:24 +02:00
"window.location.hostname",
2023-09-27 13:34:23 +02:00
"'www.wolframalpha.com'"
)?;
// block Wolfram trackers
// https://wal.wolfram.com/js/3.0.0/wal.js
sed("./docusaurus/static/input/", "wal.wolfram.com", "data:,//")?;
2023-09-27 13:34:23 +02:00
Ok(())
2023-08-11 07:09:24 +02:00
}
// regex101: build, test, and debug regex
// https://regex101.com/