rwalk 0.9.0

A blazing fast web directory scanner
Documentation 
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

use super::color_for_status_code;
use colored::Colorize;

pub fn display_range_status(mut status: String) -> String {
    if status.contains('-') {
        status = status
            .split('-')
            .map(|x| match x.parse::<u16>() {
                Ok(x) => color_for_status_code(x.to_string(), x),
                Err(_) => x.to_string(),
            })
            .collect::<Vec<_>>()
            .join("-")
            .to_string();
    } else if let Some(stripped) = status.strip_prefix('>') {
        status = format!(
            ">{}",
            color_for_status_code(stripped.to_string(), stripped.parse().unwrap_or_default())
        );
    } else if let Some(stripped) = status.strip_prefix('<') {
        status = format!(
            "<{}",
            color_for_status_code(stripped.to_string(), stripped.parse().unwrap_or_default())
        );
    } else {
        status = color_for_status_code(status.to_string(), status.parse().unwrap_or_default());
    }

    status
}

pub fn display_range(range: String) -> String {
    range
        .split(',')
        .map(|x| {
            if let Some(stripped) = x.strip_prefix('>') {
                format!(">{}", stripped.blue())
            } else if let Some(stripped) = x.strip_prefix('<') {
                format!("<{}", stripped.blue())
            } else {
                let parts = x.split('-').collect::<Vec<_>>();
                if parts.len() == 2 {
                    let start = parts[0].parse::<u16>().unwrap_or_default();
                    let end = parts[1].parse::<u16>().unwrap_or_default();
                    format!("{}-{}", start.to_string().blue(), end.to_string().blue())
                } else if let Ok(x) = x.parse::<u16>() {
                    x.to_string().blue().to_string()
                } else {
                    x.to_string()
                }
            }
        })
        .collect::<Vec<_>>()
        .join(", ")
}

pub fn color_n(s: String, n: usize) -> String {
    match n % 5 {
        0 => s.bold().green().to_string(),
        1 => s.bold().yellow().to_string(),
        2 => s.bold().red().to_string(),
        3 => s.bold().cyan().to_string(),
        _ => s.bold().magenta().to_string(),
    }
}