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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#[derive(Debug)]
pub enum RenderError {
BinaryFile(String),
}
pub fn render(project_name: &str, files: &[(&str, &[u8])]) -> Result<String, RenderError> {
let mut output = format!("# {}\n", project_name);
if !files.is_empty() {
output.push_str("\n## Files\n");
}
for (filename, bytes) in files {
let content = std::str::from_utf8(bytes)
.map_err(|_| RenderError::BinaryFile(filename.to_string()))?;
let outer_backticks = outer_backticks(content);
output.push_str(&format!(
"\n### {}\n{}\n{}\n{}\n",
filename, outer_backticks, content, outer_backticks
));
}
Ok(output)
}
fn outer_backticks(contents: &str) -> String {
let mut max_ticks = 0;
let mut current_count = 0;
for char in contents.chars() {
if char == '`' {
current_count += 1;
if current_count > max_ticks {
max_ticks = current_count;
}
} else {
current_count = 0;
}
}
let fence_len = std::cmp::max(3, max_ticks + 1);
"`".repeat(fence_len)
}
#[cfg(test)]
mod tests {
use super::{RenderError, render};
#[test]
fn empty_project_renders_only_title() {
let output = render("Project name", &[]);
assert_eq!(output.unwrap(), "# Project name\n");
}
#[test]
fn single_file_is_rendered() {
let files: Vec<(&str, &[u8])> = vec![("main.rs", b"fn main() {}")];
let output = render("Project name", &files);
assert_eq!(
output.unwrap(),
"# Project name\n\n\
## Files\n\n\
### main.rs\n\
```\n\
fn main() {}\n\
```\n"
);
}
#[test]
fn multiple_files_are_rendered_in_order() {
let files: Vec<(&str, &[u8])> = vec![
("main.rs", b"fn main() {}"),
("lib.rs", b"pub fn hello() {}"),
];
let output = render("Project name", &files);
assert_eq!(
output.unwrap(),
"# Project name\n\n\
## Files\n\n\
### main.rs\n\
```\n\
fn main() {}\n\
```\n\n\
### lib.rs\n\
```\n\
pub fn hello() {}\n\
```\n"
);
}
#[test]
fn file_with_backticks_is_handled_safely() {
let files: Vec<(&str, &[u8])> =
vec![("example.rs", b"fn main() { println!(\"``` inside\"); }")];
let output = render("Project name", &files);
assert_eq!(
output.unwrap(),
"# Project name\n\n\
## Files\n\n\
### example.rs\n\
````\n\
fn main() { println!(\"``` inside\"); }\n\
````\n"
);
}
#[test]
fn binary_file_is_rejected() {
let files: Vec<(&str, &[u8])> = vec![("image.png", &[0x00, 0x01, 0x02, 0xc3])];
let result = render("Project name", &files);
assert!(matches!(
result,
Err(RenderError::BinaryFile(name)) if name == "image.png"
));
}
}
|
