alef_e2e/
escape.rs

1//! Language-specific string escaping for e2e test code generation.
2
3/// Escape a string for embedding in a Python string literal.
4pub fn escape_python(s: &str) -> String {
5    s.replace('\\', "\\\\")
6        .replace('"', "\\\"")
7        .replace('\n', "\\n")
8        .replace('\r', "\\r")
9        .replace('\t', "\\t")
10}
11
12/// Escape a string for embedding in a Rust string literal.
13pub fn escape_rust(s: &str) -> String {
14    s.replace('\\', "\\\\")
15        .replace('"', "\\\"")
16        .replace('\n', "\\n")
17        .replace('\r', "\\r")
18        .replace('\t', "\\t")
19}
20
21/// Compute the number of # needed for a Rust raw string literal.
22pub fn raw_string_hashes(s: &str) -> usize {
23    let mut max_hashes = 0;
24    let mut current = 0;
25    let mut after_quote = false;
26    for ch in s.chars() {
27        if ch == '"' {
28            after_quote = true;
29            current = 0;
30        } else if ch == '#' && after_quote {
31            current += 1;
32            max_hashes = max_hashes.max(current);
33        } else {
34            after_quote = false;
35            current = 0;
36        }
37    }
38    max_hashes + 1
39}
40
41/// Format a string as a Rust raw string literal (r#"..."#).
42pub fn rust_raw_string(s: &str) -> String {
43    let hashes = raw_string_hashes(s);
44    let h: String = "#".repeat(hashes);
45    format!("r{h}\"{s}\"{h}")
46}
47
48/// Escape a string for embedding in a JavaScript/TypeScript string literal.
49pub fn escape_js(s: &str) -> String {
50    s.replace('\\', "\\\\")
51        .replace('"', "\\\"")
52        .replace('\n', "\\n")
53        .replace('\r', "\\r")
54        .replace('\t', "\\t")
55        .replace('`', "\\`")
56        .replace('$', "\\$")
57}
58
59/// Format a string as a Go string literal (backtick or quoted).
60pub fn go_string_literal(s: &str) -> String {
61    if !s.contains('`') {
62        format!("`{s}`")
63    } else {
64        format!("\"{}\"", escape_go(s))
65    }
66}
67
68/// Escape a string for embedding in a Go double-quoted string.
69pub fn escape_go(s: &str) -> String {
70    s.replace('\\', "\\\\")
71        .replace('"', "\\\"")
72        .replace('\n', "\\n")
73        .replace('\r', "\\r")
74        .replace('\t', "\\t")
75}
76
77/// Escape a string for embedding in a Java string literal.
78pub fn escape_java(s: &str) -> String {
79    s.replace('\\', "\\\\")
80        .replace('"', "\\\"")
81        .replace('\n', "\\n")
82        .replace('\r', "\\r")
83        .replace('\t', "\\t")
84}
85
86/// Escape a string for embedding in a C# string literal.
87pub fn escape_csharp(s: &str) -> String {
88    s.replace('\\', "\\\\")
89        .replace('"', "\\\"")
90        .replace('\n', "\\n")
91        .replace('\r', "\\r")
92        .replace('\t', "\\t")
93}
94
95/// Escape a string for embedding in a PHP string literal.
96pub fn escape_php(s: &str) -> String {
97    s.replace('\\', "\\\\")
98        .replace('"', "\\\"")
99        .replace('$', "\\$")
100        .replace('\n', "\\n")
101        .replace('\r', "\\r")
102        .replace('\t', "\\t")
103}
104
105/// Escape a string for embedding in a double-quoted Ruby string literal.
106pub fn escape_ruby(s: &str) -> String {
107    s.replace('\\', "\\\\")
108        .replace('"', "\\\"")
109        .replace('#', "\\#")
110        .replace('\n', "\\n")
111        .replace('\r', "\\r")
112        .replace('\t', "\\t")
113}
114
115/// Escape a string for embedding in a single-quoted Ruby string literal.
116/// Single-quoted Ruby strings only interpret `\\` and `\'`.
117pub fn escape_ruby_single(s: &str) -> String {
118    s.replace('\\', "\\\\").replace('\'', "\\'")
119}
120
121/// Returns true if the string needs double quotes (contains control characters
122/// that require escape sequences only available in double-quoted strings).
123pub fn ruby_needs_double_quotes(s: &str) -> bool {
124    s.contains('\n') || s.contains('\r') || s.contains('\t') || s.contains('\0')
125}
126
127/// Format a string as a Ruby literal, preferring single quotes.
128pub fn ruby_string_literal(s: &str) -> String {
129    if ruby_needs_double_quotes(s) {
130        format!("\"{}\"", escape_ruby(s))
131    } else {
132        format!("'{}'", escape_ruby_single(s))
133    }
134}
135
136/// Escape a string for embedding in an Elixir string literal.
137pub fn escape_elixir(s: &str) -> String {
138    s.replace('\\', "\\\\")
139        .replace('"', "\\\"")
140        .replace('#', "\\#")
141        .replace('\n', "\\n")
142        .replace('\r', "\\r")
143        .replace('\t', "\\t")
144}
145
146/// Escape a string for embedding in an R string literal.
147pub fn escape_r(s: &str) -> String {
148    s.replace('\\', "\\\\")
149        .replace('"', "\\\"")
150        .replace('\n', "\\n")
151        .replace('\r', "\\r")
152        .replace('\t', "\\t")
153}
154
155/// Escape a string for embedding in a C string literal.
156pub fn escape_c(s: &str) -> String {
157    s.replace('\\', "\\\\")
158        .replace('"', "\\\"")
159        .replace('\n', "\\n")
160        .replace('\r', "\\r")
161        .replace('\t', "\\t")
162}
163
164/// Sanitize an identifier for use as a test function name.
165/// Replaces non-alphanumeric characters with underscores, strips leading digits.
166pub fn sanitize_ident(s: &str) -> String {
167    let mut result = String::with_capacity(s.len());
168    for ch in s.chars() {
169        if ch.is_ascii_alphanumeric() || ch == '_' {
170            result.push(ch);
171        } else {
172            result.push('_');
173        }
174    }
175    // Strip leading digits
176    let trimmed = result.trim_start_matches(|c: char| c.is_ascii_digit());
177    if trimmed.is_empty() {
178        "_".to_string()
179    } else {
180        trimmed.to_string()
181    }
182}
183
184/// Convert a category name to a sanitized filename component.
185pub fn sanitize_filename(s: &str) -> String {
186    s.chars()
187        .map(|c| if c.is_ascii_alphanumeric() || c == '_' { c } else { '_' })
188        .collect::<String>()
189        .to_lowercase()
190}
alef_e2e/escape.rs

alef_e2e/
escape.rs
