Allow import by filename.

This commit is contained in:
Bill Thiede 2014-03-29 22:32:03 -07:00
parent c266e04053
commit 8021820311

View File

@ -218,9 +218,24 @@ func main() {
}
glog.Infoln("Using uid", uid, "for", *username)
if *maildir == "" {
if err := LoadReader(db, uid, os.Stdin); err != nil {
glog.Fatal(err)
if *maildir == "" || flag.NArg() > 0 {
if flag.NArg() == 0 {
if err := LoadReader(db, uid, os.Stdin); err != nil {
glog.Errorln(err)
}
return
}
for _, fn := range flag.Args() {
r, err := os.Open(fn)
if err != nil {
glog.Error(err)
continue
}
if err := LoadReader(db, uid, r); err != nil {
glog.Errorln(fn, err)
}
r.Close()
}
return
}