gdritter repos picoml / master src / pretty.rs
master

Tree @master (Download .tar.gz)

pretty.rs @master — raw · history · blame 

  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
114
115
116
117
118
119
120
use types::*;

pub trait Show {
    fn show(&self, buf: &mut String);
}

impl Show for Stmt {
    fn show(&self, buf: &mut String) {
        match *self {
            Stmt::Let(ref p, _, _, ref e) => {
                p.show(buf);
                buf.push_str(" = ");
                e.show(buf);
            }
            Stmt::Expr(ref e) =>
                e.show(buf),
        }
        buf.push_str(";\n");
    }
}


impl Show for Pat {
    fn show(&self, buf: &mut String) {
        match *self {
            Pat::Var(ref i) => buf.push_str(i),
            Pat::Tuple(ref ps) => {
                buf.push_str("(");
                for (n, t) in ps.iter().enumerate() {
                    if n > 0 {
                        buf.push_str(", ");
                    }
                    t.show(buf);
                }
                buf.push_str(")");
            }
        }
    }
}


impl Show for Value {
    fn show(&self, buf: &mut String) {
        match *self {
            Value::Unit => buf.push_str("()"),
            Value::Lit(ref l) => l.show(buf),
            Value::Tuple(ref ts) => {
                buf.push_str("(");
                for (n, t) in ts.iter().enumerate() {
                    if n > 0 {
                        buf.push_str(", ");
                    }
                    t.show(buf);
                }
                buf.push_str(")");
            }
            Value::Closure(ref p, ref e, _) => {
                buf.push_str("(\\");
                p.show(buf);
                buf.push_str(".");
                e.show(buf);
                buf.push_str(")");
            }
        }
    }
}

impl Show for Literal {
    fn show(&self, buf: &mut String) {
        match *self {
            Literal::Int(i) =>
                buf.push_str(&format!("{}", i)),
        }
    }
}

impl Show for Expr {
    fn show(&self, buf: &mut String) {
        match *self {
            Expr::Var(ref v) => buf.push_str(v),
            Expr::Lit(ref l) => l.show(buf),
            Expr::Fn(ref p, ref e) => {
                buf.push_str("fn ");
                p.show(buf);
                buf.push_str(" => ");
                e.show(buf);
            }
            Expr::App(ref e1, ref e2) => {
                buf.push_str("(");
                e1.show(buf);
                buf.push_str(")(");
                e2.show(buf);
                buf.push_str(")");
            }
            Expr::Do(ref ss) => {
                buf.push_str("do\n");
                for s in ss.iter() {
                    s.show(buf);
                }
                buf.push_str("end\n");
            }
            Expr::Tuple(ref ts) => {
                buf.push_str("(");
                for (n, t) in ts.iter().enumerate() {
                    if n > 0 {
                        buf.push_str(", ");
                    }
                    t.show(buf);
                }
                buf.push_str(")");
            }
        }
    }
}

pub fn show<S: Show>(s: &S) -> String {
    let mut buf = String::new();
    s.show(&mut buf);
    buf
}