dot_graph 0.2.3

A library for generating Graphviz DOT language files.
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
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

use crate::{
    arrow::{Arrow},
    style::{Style},
    utils::{quote_string},
};

/// `Graph`'s edge.
#[derive(Clone)]
pub struct Edge {
    from: String,
    to: String,
    label: String,
    style: Style,
    start_arrow: Arrow,
    end_arrow: Arrow,
    color: Option<String>,
}

impl Edge {
    pub fn new(from: &str, to: &str, label: &str) -> Self {
        Edge { from: String::from(from), to: String::from(to), label: String::from(label), style: Style::None, start_arrow: Arrow::default(), end_arrow: Arrow::default(), color: None }
    }

    pub fn label(&mut self, label: &str) -> Self {
        let mut edge = self.clone();
        edge.label = String::from(label);
        edge
    }

    pub fn style(&mut self, style: Style) -> Self {
        let mut edge = self.clone();
        edge.style = style;
        edge
    }

    pub fn color(&mut self, color: Option<&str>) -> Self {
        let mut edge = self.clone();
        edge.color = match color {
            Some(c) => Some(String::from(c)),
            None => None
        };
        edge
    }

    pub fn start_arrow(&mut self, arrow: Arrow) -> Self {
        let mut edge = self.clone();
        edge.start_arrow = arrow;
        edge
    }

    pub fn end_arrow(&mut self, arrow: Arrow) -> Self {
        let mut edge = self.clone();
        edge.end_arrow = arrow;
        edge
    }

    pub fn to_dot_string(&self, edge_symbol: &str) -> String {
        let colorstring: String;
        let escaped_label: &String = &quote_string(self.label.clone());
        let start_arrow_s: String = self.start_arrow.to_dot_string();
        let end_arrow_s: String = self.end_arrow.to_dot_string();

        let mut text = vec!["\"", self.from.as_str(), "\" ",
                            edge_symbol, " ",
                            "\"", self.to.as_str(), "\"",];

        text.push("[label=");
        text.push(escaped_label.as_str());
        text.push("]");

        if self.style != Style::None {
            text.push("[style=\"");
            text.push(self.style.as_slice());
            text.push("\"]");
        }

        let color: Option<String> = match &self.color {
            Some(l) => {
                Some((*l).clone())
            },
            None => None,
        };
        if let Some(c) = color {
            colorstring = quote_string(c);
            text.push("[color=");
            text.push(&colorstring);
            text.push("]");
        }

        let mut arrow_text: Vec<String> = vec![];
        let mut arrow_str: String = String::new();
        if !self.start_arrow.is_default() || !self.end_arrow.is_default() {
            if !self.end_arrow.is_default() {
                arrow_text.push(vec!["arrowhead=\"", &end_arrow_s, "\""].into_iter().collect());
            }
            if !self.start_arrow.is_default() {
                arrow_text.push(vec!["arrowtail=\"", &start_arrow_s, "\""].into_iter().collect());
            }
            if !self.start_arrow.is_default() && !self.end_arrow.is_default() {
                arrow_text.push(String::from("dir=\"both\""));
            }
        }
        if arrow_text.len() > 0 {
            arrow_str.push_str(&arrow_text.join(" "));
            arrow_str.insert(0, '[');
            arrow_str.push_str("]");
            text.push(arrow_str.as_str());
        }

        text.push(";");
        return text.into_iter().collect();
    }
}