Simple web server for showing original messages.

This commit is contained in:
Bill Thiede 2014-03-30 21:53:33 -07:00
parent d7c789863e
commit 8fa9825b8e
2 changed files with 49 additions and 4 deletions

45
cmd/xwebmail/xwebmail.go Normal file
View File

@ -0,0 +1,45 @@
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))
}

View File

@ -14,22 +14,22 @@ var (
sslmode = flag.String("dbsslmode", "disable", "PostgreSQL sslmode setting")
)
type conn struct {
type Conn struct {
*sql.DB
}
// NewDB creates connection to PostgreSQL DB. If dsn is empty, the flags
// -dbuser, -dbname, -dbsslmode are used.
func NewConn(dsn string) (*conn, error) {
func NewConn(dsn string) (*Conn, error) {
if dsn == "" {
dsn = fmt.Sprintf("user=%s dbname=%s sslmode=%s", *user, *name,
*sslmode)
}
db, err := sql.Open("postgres", dsn)
return &conn{DB: db}, err
return &Conn{DB: db}, err
}
func (c *conn) OriginalBlobByHash(hash string, blob *[]byte) error {
func (c *Conn) OriginalBlobByHash(hash string, blob *[]byte) error {
row := c.QueryRow(`
SELECT
blob