codeframe 0.1.1

Capture pretty codeframes.
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

//! Pretty Capture codeframes
//!
//! Ex:
//! let raw_lines = "let a: i64 = 12;";
//! let codeframe = Codeframe::new(raw_lines, 1).set_color("red").capture();
//!
//!
//! Colors Supported
//! Black
//! Red
//! Green
//! Yellow
//! Blue
//! Magenta or (Purple)
//! Cyan
//! White

use crate::capture;

pub struct Codeframe<'a> {
    color: &'a str,
    line: i64,
    raw_lines: &'a str,
}

impl<'a> Codeframe<'a> {
    pub fn new(raw_lines: &'a str, line: i64) -> Codeframe<'a> {
        Codeframe {
            color: "red",
            line,
            raw_lines,
        }
    }

    pub fn set_color(mut self, color: &'a str) -> Self {
        self.color = color;
        self
    }

    pub fn capture(self) -> Option<String> {
        let vec_lines = self.raw_lines.split('\n').collect();
        capture::capture_code_frame(vec_lines, self.line, self.color)
    }
}