X-Git-Url: http://git.scottworley.com/tattlekey/blobdiff_plain/4f2bb592de3d26d989ce33c692c174a556885e22..0d4f2aa67ed92add9c371a49cc45daad8e529301:/server/src/main.rs

diff --git a/server/src/main.rs b/server/src/main.rs
index 62ef41e..5edadbb 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -80,13 +80,20 @@ impl Range {
     }
 }
 
-fn main() {
+fn open_log_for_writing() -> csv::Writer<std::fs::File> {
+    let log_file_exists = std::path::Path::new(LOGFILENAME).exists();
     let logfile = std::fs::OpenOptions::new()
-        .create(true)
+        .create_new(!log_file_exists)
         .append(true)
         .open(LOGFILENAME)
         .expect("Coudln't open log file");
-    let mut log = csv::Writer::from_writer(logfile);
+    csv::WriterBuilder::new()
+        .has_headers(!log_file_exists)
+        .from_writer(logfile)
+}
+
+fn main() {
+    let mut log = open_log_for_writing();
     let mut presses = HashMap::<MessageKey, Range>::new();
     let socket = UdpSocket::bind("0.0.0.0:29803").expect("couldn't bind to address");
     loop {