codeframe 0.1.5

Capture pretty code-frames.
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

//! 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;
use crate::Color;

pub struct Codeframe {
    color: Color,
    line: i64,
    raw_lines: String,
}

impl Codeframe {
    pub fn new(raw_lines: String, line: i64) -> Codeframe {
        Codeframe {
            color: Color::Red,
            line,
            raw_lines,
        }
    }

    pub fn set_color(mut self, color: Color) -> Self {
        self.color = color;
        self
    }

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