init

web/app/app.go

@@ -0,0 +1,49 @@
+package app
+
+import (
+	"git.sleepycat.moe/sam/mercury/config"
+	"git.sleepycat.moe/sam/mercury/internal/database/sql"
+	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
+)
+
+type App struct {
+	Router chi.Router
+
+	Config   config.Config
+	Database *sql.Base
+}
+
+func NewApp(cfg config.Config, db *sql.Base) *App {
+	app := &App{
+		Router:   chi.NewRouter(),
+		Config:   cfg,
+		Database: db,
+	}
+
+	app.Router.Use(app.Logger)
+	app.Router.Use(middleware.Recoverer)
+
+	return app
+}
+
+func (a *App) Account(q ...sql.Querier) *sql.AccountStore {
+	if len(q) == 0 || q[0] == nil {
+		return sql.NewAccountStore(a.Database.PoolQuerier())
+	}
+	return sql.NewAccountStore(q[0])
+}
+
+func (a *App) Blog(q ...sql.Querier) *sql.BlogStore {
+	if len(q) == 0 || q[0] == nil {
+		return sql.NewBlogStore(a.Database.PoolQuerier())
+	}
+	return sql.NewBlogStore(q[0])
+}
+
+func (a *App) Post(q ...sql.Querier) *sql.PostStore {
+	if len(q) == 0 || q[0] == nil {
+		return sql.NewPostStore(a.Database.PoolQuerier())
+	}
+	return sql.NewPostStore(q[0])
+}

web/app/errors.go

@@ -0,0 +1,33 @@
+package app
+
+import (
+	"net/http"
+
+	"github.com/go-chi/render"
+)
+
+// WriteError writes an error message to w.
+// If one or more messages are passed, a JSON response is also sent,
+// with the first message becoming `error` and the second becoming `error_description`.
+func (a *App) WriteError(w http.ResponseWriter, r *http.Request, status int, messages ...string) {
+	render.Status(r, status)
+
+	switch len(messages) {
+	case 0:
+		return
+	case 1:
+		render.JSON(w, r, errorMessage{
+			Error: messages[0],
+		})
+	default:
+		render.JSON(w, r, errorMessage{
+			Error:            messages[0],
+			ErrorDescription: messages[1],
+		})
+	}
+}
+
+type errorMessage struct {
+	Error            string `json:"error"`
+	ErrorDescription string `json:"error_description,omitempty"`
+}

web/app/log.go

@@ -0,0 +1,48 @@
+package app
+
+import (
+	"net/http"
+	"runtime/debug"
+	"time"
+
+	"github.com/go-chi/chi/v5/middleware"
+	"github.com/rs/zerolog/log"
+)
+
+func (a *App) Logger(next http.Handler) http.Handler {
+	fn := func(w http.ResponseWriter, r *http.Request) {
+		ww := middleware.NewWrapResponseWriter(w, r.ProtoMajor)
+
+		t1 := time.Now()
+		defer func() {
+			t2 := time.Now()
+
+			// Recover and record stack traces in case of a panic
+			if rec := recover(); rec != nil {
+				log.Error().
+					Str("type", "error").
+					Timestamp().
+					Interface("recover_info", rec).
+					Bytes("debug_stack", debug.Stack()).
+					Msg("Handler panicked")
+				http.Error(ww, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
+			}
+
+			// log end request
+			log.Info().
+				Timestamp().
+				Str("remote_ip", r.RemoteAddr).
+				Str("url", r.URL.Path).
+				Str("proto", r.Proto).
+				Str("method", r.Method).
+				Int("status", ww.Status()).
+				Dur("elapsed", t2.Sub(t1)).
+				Int64("bytes_in", r.ContentLength).
+				Int("bytes_out", ww.BytesWritten()).
+				Msg(r.Method + " " + r.URL.Path)
+		}()
+
+		next.ServeHTTP(ww, r)
+	}
+	return http.HandlerFunc(fn)
+}