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
|
use std::io::{Read, Write};
use repo2markdown::{
fenced_md_generator::generate_fenced_markdown, logger::Logger,
md_generator::generate_markdown_from_paths,
};
use crate::cli::Cli;
pub fn run<R: Read, W: Write>(
cli: &Cli,
input: R,
output: W,
) -> Result<(), Box<dyn std::error::Error>> {
let logger = Logger::new(cli.verbosity());
if cli.fenced {
generate_fenced_markdown(
input,
output,
cli.render_options(),
&cli.root,
&cli.origin,
cli.name.as_deref(),
logger,
)
} else {
generate_markdown_from_paths(
input,
output,
cli.render_options(),
&cli.root,
&cli.origin,
cli.name.as_deref(),
logger,
)
}
}
#[cfg(test)]
mod tests {
use std::io::Cursor;
use clap::Parser;
use crate::cli::Cli;
use super::run;
#[test]
fn run_does_not_fence_output_by_default() {
let cli = Cli::try_parse_from(["repo2markdown", "--name", "Project name"]).unwrap();
let input = Cursor::new(b"");
let mut output = Vec::new();
run(&cli, input, &mut output).unwrap();
assert_eq!(String::from_utf8(output).unwrap(), "# Project name\n");
}
#[test]
fn run_uses_fenced_output_when_requested() {
let cli =
Cli::try_parse_from(["repo2markdown", "--fenced", "--name", "Project name"]).unwrap();
let input = Cursor::new(b"");
let mut output = Vec::new();
run(&cli, input, &mut output).unwrap();
assert_eq!(
String::from_utf8(output).unwrap(),
"```markdown\n# Project name\n```\n"
);
}
}
|
