init, fighting with types

Author: Frederic de Zorzi

.gitignore

@@ -0,0 +1 @@
+target

Cargo.toml

@@ -0,0 +1,7 @@
+[package]
+name = "pg_dump_tool"
+version = "0.1.0"
+authors = ["Frederic de Zorzi <fredz@pimentech.fr>"]
+
+[dependencies]
+regex = "0.1"

src/main.rs

@@ -0,0 +1,39 @@
+use std::io;
+use std::io::prelude::*;
+extern crate regex;
+
+use regex::Regex;
+
+fn main() {
+    let stdin = io::stdin();
+    let mut stdin = stdin.lock();
+    let mut line = String::new();
+    let mut table_name = String::new();
+    let mut fields = String::new();
+    let re = match Regex::new(r"COPY (\w+) \(([\w, ]+)\) FROM stdin;") {
+        Ok(re) => re,
+        Err(err) => panic!("{}", err),
+    };
+    let mut insert_mode = false;
+    while stdin.read_line(&mut line).unwrap() > 0 {
+        if insert_mode {
+            if line == "\\.\n" {
+                insert_mode = false;
+            }
+            else {
+                print!("{} {}", table_name, fields);
+            }
+        }
+        else {
+            match re.captures(&line) {
+                None => print!("{}", line),
+                Some(caps) => {
+                    table_name = String::from(caps.at(1).unwrap());
+                    fields = String::from(caps.at(2).unwrap());
+                    insert_mode = true;
+                },
+            };
+        }
+        line.clear();
+    }
+}