Newer
Older
"github.com/jmoiron/sqlx"
_ "modernc.org/sqlite" // Imported for side effects: registers DB driver
)
//go:embed templates/*.tpl
var templateFS embed.FS
//go:embed static/*
var staticFS embed.FS
func httpError(w http.ResponseWriter, msg string, err error, status int) {
if err != nil {
msg += ": " + err.Error()
}
log.Println(msg)
http.Error(w, msg, status)
type authProvider struct {
db *sqlx.DB
}
func (a authProvider) Valid(ctx context.Context, user, pass string) (bool, error) {
query, args, err := squirrel.Select("password").
From("users").
Where(squirrel.Eq{"name": user}).
ToSql()
if err != nil {
return false, err
}
var dbPass string
err = a.db.GetContext(ctx, &dbPass, query, args...)
if errors.Is(err, sql.ErrNoRows) {
// User not found isn't an error, it's just an invalid auth.
return false, nil
}
if err != nil {
return false, err
}
}
func logRequest(r *http.Request) {
log.Println("handling", r.Method, r.URL, "from", r.RemoteAddr, "by", auth.User(r))
}
db, err := sqlx.Open("sqlite", "vino.sqlite")
if err != nil {
log.Fatalln("can't open DB:", err)
}
defer db.Close()
err = initDB(db)
if err != nil {
log.Fatalln("can't initialize DB:", err)
}
http.HandleFunc("/favicon.ico", http.NotFound)
http.HandleFunc("/details/img", auth.Require(http.HandlerFunc(handler.img), ap))
http.HandleFunc("/details/", auth.Require(http.HandlerFunc(handler.details), ap))
http.HandleFunc("/", auth.Require(http.HandlerFunc(handler.index), ap))