Add postgres package indexing + cli options

indexer/src/main.rs

@@ -1,17 +1,71 @@
 #[macro_use] extern crate log;
 #[macro_use] extern crate lazy_static;
+#[macro_use] extern crate clap;
 extern crate env_logger;
 extern crate regex;
 extern crate libarchive3_sys;
 extern crate libc;
 extern crate ring;
 extern crate encoding;
+extern crate postgres;
 
 mod archive;
 mod archread;
 mod man;
+mod pkg;
+
+
+// Convenience function to get a system id by short-name. Panics if the system doesn't exist.
+fn sysbyshort(conn: &postgres::GenericConnection, short: &str) -> i32 {
+    let r = conn.query("SELECT id FROM systems WHERE short = $1", &[&short]).unwrap();
+    if r.is_empty() {
+        panic!("Invalid system: {}", short);
+    }
+    r.get(0).get(0)
+}
+
 
 fn main() {
-    env_logger::init().unwrap();
-    info!("Hello, world!");
+    let arg = clap_app!(indexer =>
+        (about: "Manned.org man page indexer")
+        (@arg v: -v +multiple "Increase verbosity")
+        (@arg host: -h +required +takes_value "PostgreSQL connection string")
+        (@subcommand pkg =>
+            (about: "Index a single package")
+            (@arg sys: --sys +required +takes_value "System short-name")
+            (@arg cat: --cat +required +takes_value "Package category")
+            (@arg pkg: --pkg +required +takes_value "Package name")
+            (@arg ver: --ver +required +takes_value "Package version")
+            (@arg date: --date +required +takes_value "Package release date")
+            (@arg FILE: +required "Package file")
+        )
+    ).get_matches();
+
+    let verbose = arg.occurrences_of("v");
+    env_logger::LogBuilder::new()
+        .filter(Some("indexer"), match verbose {
+            0 => log::LogLevelFilter::Warn,
+            1 => log::LogLevelFilter::Info,
+            2 => log::LogLevelFilter::Debug,
+            _ => log::LogLevelFilter::Trace,
+        })
+        .filter(Some("postgres"), if verbose >= 4 { log::LogLevelFilter::Trace } else { log::LogLevelFilter::Info })
+        .init().unwrap();
+
+    let db = match postgres::Connection::connect(arg.value_of("host").unwrap(), postgres::TlsMode::None) {
+        Ok(x) => x,
+        Err(x) => { error!("Can't connect to postgres: {}", x); return },
+    };
+    debug!("Connected to database");
+
+    if let Some(matches) = arg.subcommand_matches("pkg") {
+        pkg::pkg(&db, pkg::PkgOpt {
+            sys: sysbyshort(&db, matches.value_of("sys").unwrap()),
+            cat: matches.value_of("cat").unwrap(),
+            pkg: matches.value_of("pkg").unwrap(),
+            ver: matches.value_of("ver").unwrap(),
+            date: matches.value_of("date").unwrap(),
+            file: matches.value_of("FILE").unwrap()
+        });
+    }
 }