src/rustfmt_diff.rs

   1 use std::collections::VecDeque;
   2 use diff;
   3 use term;
   4 use std::io;
   5
   6 #[derive(Debug, PartialEq)]
   7 pub enum DiffLine {
   8     Context(String),
   9     Expected(String),
  10     Resulting(String),
  11 }
  12
  13 #[derive(Debug, PartialEq)]
  14 pub struct Mismatch {
  15     pub line_number: u32,
  16     pub lines: Vec<DiffLine>,
  17 }
  18
  19 impl Mismatch {
  20     fn new(line_number: u32) -> Mismatch {
  21         Mismatch {
  22             line_number: line_number,
  23             lines: Vec::new(),
  24         }
  25     }
  26 }
  27
  28 // Produces a diff between the expected output and actual output of rustfmt.
  29 pub fn make_diff(expected: &str, actual: &str, context_size: usize) -> Vec<Mismatch> {
  30     let mut line_number = 1;
  31     let mut context_queue: VecDeque<&str> = VecDeque::with_capacity(context_size);
  32     let mut lines_since_mismatch = context_size + 1;
  33     let mut results = Vec::new();
  34     let mut mismatch = Mismatch::new(0);
  35
  36     for result in diff::lines(expected, actual) {
  37         match result {
  38             diff::Result::Left(str) => {
  39                 if lines_since_mismatch >= context_size {
  40                     results.push(mismatch);
  41                     mismatch = Mismatch::new(line_number - context_queue.len() as u32);
  42                 }
  43
  44                 while let Some(line) = context_queue.pop_front() {
  45                     mismatch.lines.push(DiffLine::Context(line.to_owned()));
  46                 }
  47
  48                 mismatch.lines.push(DiffLine::Resulting(str.to_owned()));
  49                 lines_since_mismatch = 0;
  50             }
  51             diff::Result::Right(str) => {
  52                 if lines_since_mismatch >= context_size {
  53                     results.push(mismatch);
  54                     mismatch = Mismatch::new(line_number - context_queue.len() as u32);
  55                 }
  56
  57                 while let Some(line) = context_queue.pop_front() {
  58                     mismatch.lines.push(DiffLine::Context(line.to_owned()));
  59                 }
  60
  61                 mismatch.lines.push(DiffLine::Expected(str.to_owned()));
  62                 line_number += 1;
  63                 lines_since_mismatch = 0;
  64             }
  65             diff::Result::Both(str, _) => {
  66                 if context_queue.len() >= context_size {
  67                     let _ = context_queue.pop_front();
  68                 }
  69
  70                 if lines_since_mismatch < context_size {
  71                     mismatch.lines.push(DiffLine::Context(str.to_owned()));
  72                 } else {
  73                     context_queue.push_back(str);
  74                 }
  75
  76                 line_number += 1;
  77                 lines_since_mismatch += 1;
  78             }
  79         }
  80     }
  81
  82     results.push(mismatch);
  83     results.remove(0);
  84
  85     results
  86 }
  87
  88 pub fn print_diff<F>(diff: Vec<Mismatch>, get_section_title: F)
  89     where F: Fn(u32) -> String
  90 {
  91     if let Some(t) = term::stdout() {
  92         print_diff_fancy(diff, get_section_title, t);
  93     } else {
  94         print_diff_basic(diff, get_section_title);
  95     }
  96 }
  97
  98 fn print_diff_fancy<F>(diff: Vec<Mismatch>,
  99                        get_section_title: F,
 100                        mut t: Box<term::Terminal<Output = io::Stdout>>)
 101     where F: Fn(u32) -> String
 102 {
 103     for mismatch in diff {
 104         let title = get_section_title(mismatch.line_number);
 105         writeln!(t, "{}", title).unwrap();
 106
 107         for line in mismatch.lines {
 108             match line {
 109                 DiffLine::Context(ref str) => {
 110                     t.reset().unwrap();
 111                     writeln!(t, " {}⏎", str).unwrap();
 112                 }
 113                 DiffLine::Expected(ref str) => {
 114                     t.fg(term::color::GREEN).unwrap();
 115                     writeln!(t, "+{}⏎", str).unwrap();
 116                 }
 117                 DiffLine::Resulting(ref str) => {
 118                     t.fg(term::color::RED).unwrap();
 119                     writeln!(t, "-{}⏎", str).unwrap();
 120                 }
 121             }
 122         }
 123         t.reset().unwrap();
 124     }
 125 }
 126
 127 pub fn print_diff_basic<F>(diff: Vec<Mismatch>, get_section_title: F)
 128     where F: Fn(u32) -> String
 129 {
 130     for mismatch in diff {
 131         let title = get_section_title(mismatch.line_number);
 132         println!("{}", title);
 133
 134         for line in mismatch.lines {
 135             match line {
 136                 DiffLine::Context(ref str) => {
 137                     println!(" {}⏎", str);
 138                 }
 139                 DiffLine::Expected(ref str) => {
 140                     println!("+{}⏎", str);
 141                 }
 142                 DiffLine::Resulting(ref str) => {
 143                     println!("-{}⏎", str);
 144                 }
 145             }
 146         }
 147     }
 148 }