From 8a1b718709c0800a7b1eb5747a6a1767d026605e Mon Sep 17 00:00:00 2001
From: Shimun <shimun@shimun.net>
Date: Sat, 8 Jun 2019 20:04:24 +0200
Subject: [PATCH] prevent user from overrriding source file

---
 src/main.rs | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 774aeff..fb5e640 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,13 +11,13 @@ mod conf;
 mod opts;
 
 use conf::*;
+use dirs;
 use opts::*;
 use std::env;
 use std::fs::File;
-use std::io::{self, Read, Write};
+use std::io::{self, Read, Stdin, Write};
 use std::io::{BufRead, BufReader};
-
-use dirs;
+use std::process::exit;
 use xml::writer::{EmitterConfig, EventWriter, Result as XResult, XmlEvent};
 
 #[derive(Builder, Debug)]
@@ -82,6 +82,14 @@ fn config() -> Option<Config> {
 
 fn main() {
     let opts = Opts::from_args();
+    if opts.input == opts.output {
+        eprintln!("Input and output are the same file! {:?}\nContinue(y/n): ", opts.input);
+        match io::stdin().bytes().next() {
+            Some(Ok(r)) if r as char == 'y' => (),
+            Some(Ok(_)) => exit(1),
+            _ => (), //Stdin closed, assume in == out is intentional
+        }
+    }
     let file = File::open(opts.input).expect("Failed to open file!");
     let mut out = File::create(opts.output).expect("Failed to open OUTPUT file!");
     let mut writer = EmitterConfig::new().perform_indent(true).create_writer(&mut out);
@@ -108,8 +116,8 @@ fn convert<I: Iterator<Item = Track>, W: Write>(html: bool, tracks: I, w: &mut E
                         .attr("type", "text/javascript")
                         .attr("src", url),
                 )?;
-                 w.write(XmlEvent::characters(""))?;
-                            w.write(XmlEvent::end_element())?;
+                w.write(XmlEvent::characters(""))?;
+                w.write(XmlEvent::end_element())?;
             }
             for url in CONFIG.include.css.iter() {
                 w.write(
@@ -118,7 +126,7 @@ fn convert<I: Iterator<Item = Track>, W: Write>(html: bool, tracks: I, w: &mut E
                         .attr("href", url),
                 )?;
                 w.write(XmlEvent::characters(""))?;
-                            w.write(XmlEvent::end_element())?;
+                w.write(XmlEvent::end_element())?;
             }
             w.write(XmlEvent::end_element())?;
         }