justcaptcha/internal/handlers/handlers.go

package handlers

import (
	"fmt"
	"image/png"
	"justcaptcha/internal/captcha"
	pcaptcha "justcaptcha/pkg/captcha"
	"justcaptcha/pkg/server"
	"net/http"
)

type CaptchaHandlers struct{}

func New() *CaptchaHandlers {
	return &CaptchaHandlers{}
}

func (h *CaptchaHandlers) New(w http.ResponseWriter, r *http.Request) {
	dc := captcha.NewDwellingCaptcha(captcha.GetExpiry())
	_, id := captcha.New(r.RemoteAddr, dc)

	fmt.Fprint(w, id)
}

func (h *CaptchaHandlers) Image(w http.ResponseWriter, r *http.Request) {
	captchaID := pcaptcha.ID(server.GetURLParam(r, "captcha"))
	captchaStyle := r.URL.Query().Get("style")

	captchaImage, err := captcha.Image(captchaID, captchaStyle)
	if err != nil {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	if captchaImage == nil {
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	w.Header().Add("Content-Disposition", "inline; filename=\""+string(captchaID)+"\"")

	png.Encode(w, captchaImage)
}

func (h *CaptchaHandlers) Solve(w http.ResponseWriter, r *http.Request) {
	r.ParseForm()

	captchaID := pcaptcha.ID(server.GetURLParam(r, "captcha"))
	answer := pcaptcha.Answer(r.FormValue("answer"))

	ok, err := captcha.Solve(captchaID, answer)
	if err != nil {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	if !ok {
		w.WriteHeader(http.StatusForbidden)
	}
}

func (h *CaptchaHandlers) IsSolved(w http.ResponseWriter, r *http.Request) {
	captchaID := pcaptcha.ID(server.GetURLParam(r, "captcha"))

	solved, err := captcha.IsSolved(captchaID)
	if err != nil {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	if !solved {
		w.WriteHeader(http.StatusForbidden)
	}
}