46 lines
794 B
Go
46 lines
794 B
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"net/http"
|
|
|
|
"github.com/golang/glog"
|
|
"github.com/gorilla/mux"
|
|
|
|
"xinu.tv/email/db"
|
|
)
|
|
|
|
var addr = flag.String("addr", ":8080", "address:port to listen on")
|
|
|
|
type handler struct {
|
|
c *db.Conn
|
|
}
|
|
|
|
func (h *handler) OriginalHandler(w http.ResponseWriter, r *http.Request) {
|
|
var blob []byte
|
|
hash := mux.Vars(r)["hash"]
|
|
if err := h.c.OriginalBlobByHash(hash, &blob); err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
if _, err := w.Write(blob); err != nil {
|
|
glog.Error(err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
defer glog.Flush()
|
|
flag.Parse()
|
|
|
|
c, err := db.NewConn("")
|
|
if err != nil {
|
|
glog.Fatal(err)
|
|
}
|
|
|
|
h := &handler{c: c}
|
|
|
|
r := mux.NewRouter()
|
|
r.HandleFunc("/raw/{hash}", h.OriginalHandler)
|
|
glog.Fatal(http.ListenAndServe(*addr, r))
|
|
}
|