kurz/web/get_short.go

package web

import (
	"io"
	"net/http"
	"strings"

	"code.osinet.fr/fgm/kurz/domain"
	"github.com/gorilla/mux"
)

/*
Template variables:

  - FullyQualifiedAssetsBaseURL http://plusvite-cdn.osinet.eu
  - FullyQualifiedShortURL      (short string)
  - FullyQualifiedSiteURL       http://plusvite.net/
  - FullyQualifiedTargetURL     (target string)
  - RefreshDelay                (seconds int)
  - SiteName                    PlusVite
*/

type bytes = []byte

func build403(w http.ResponseWriter, short string) error {
	data := struct {
		FullyQualifiedShortURL string
		Globals
	}{
		short,
		globals,
	}

	sw := &strings.Builder{}
	err := tmpl.ExecuteTemplate(sw, "403", data)
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
	} else {
		w.WriteHeader(http.StatusForbidden)
		io.Copy(w, strings.NewReader(sw.String()))
	}

	return err
}

func build404(w http.ResponseWriter, short string) error {
	data := struct {
		FullyQualifiedShortURL string
		Globals
	}{
		short,
		globals,
	}

	sw := &strings.Builder{}
	err := tmpl.ExecuteTemplate(sw, "404", data)
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
	} else {
		w.WriteHeader(http.StatusNotFound)
		io.Copy(w, strings.NewReader(sw.String()))
	}
	return err
}

func build451(w http.ResponseWriter, short string) error {
	return nil
}

// handleGetShort handles path /<short>
func handleGetShort(w http.ResponseWriter, r *http.Request) {
	short, ok := mux.Vars(r)["short"]
	if !ok {
		w.WriteHeader(http.StatusBadRequest)
		return
	}

	target, err := domain.GetTargetURL(short)
	// Happy path.
	if err == nil {
		w.Header().Set("Location", target)
		w.WriteHeader(http.StatusTemporaryRedirect)
		return
	}

	// Very sad path.
	domainErr, ok := err.(domain.Error)
	if !ok {
		// All errors return by the API should be domain-specific errors.
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	// Normal sad paths.
	switch domainErr.Kind {
	case domain.ShortNotFound:
		build404(w, short)

	case domain.TargetBlockedError:
		build403(w, short)

	case domain.TargetCensoredError:
		build451(w, short)

	default:
		// TargetInvalid is not supposed to happen in this case, so it is an internal error too.
		w.WriteHeader(http.StatusInternalServerError)
	}
}