Split Handler into Handler and APIHandler

web/api_handler.go

@@ -0,0 +1,69 @@
+package web
+
+import (
+	"encoding/json"
+
+	"github.com/jimeh/ozu.io/shortener"
+	"github.com/qiangxue/fasthttp-routing"
+	"github.com/valyala/fasthttp"
+)
+
+// NewAPIHandler creates a new Handler object.
+func NewAPIHandler(s shortener.Shortener) *APIHandler {
+	return &APIHandler{s}
+}
+
+// APIHandler handle HTTP requests.
+type APIHandler struct {
+	shortener shortener.Shortener
+}
+
+// Shorten shortens given URL.
+func (h *APIHandler) Shorten(c *routing.Context) error {
+	uid, url, err := h.shortener.Shorten(c.FormValue("url"))
+	if err != nil {
+		return h.respondWithError(c, err)
+	}
+
+	r := makeURLResponse(c, uid, url)
+	return h.respond(c, &r)
+}
+
+// Lookup shortened UID.
+func (h *APIHandler) Lookup(c *routing.Context) error {
+	uid := c.FormValue("uid")
+	url, err := h.shortener.Lookup(uid)
+	if err != nil {
+		return h.respondWithError(c, err)
+	}
+
+	r := makeURLResponse(c, uid, url)
+	return h.respond(c, &r)
+}
+
+func (h *APIHandler) respond(c *routing.Context, r *URLResponse) error {
+	resp, err := json.Marshal(r)
+	if err != nil {
+		return err
+	}
+
+	c.SetContentType("application/json")
+	c.Write(resp)
+	return nil
+}
+
+func (h *APIHandler) respondWithError(c *routing.Context, err error) error {
+	r := ErrorResponse{
+		Error: err.Error(),
+	}
+
+	resp, err := json.Marshal(r)
+	if err != nil {
+		return err
+	}
+
+	c.SetStatusCode(fasthttp.StatusNotFound)
+	c.SetContentType("application/json")
+	c.Write(resp)
+	return nil
+}