api_link.go

package shortpaste

import (
	"encoding/json"
	"fmt"
	"net/http"
	"strings"
	"text/template"
)

func (app *App) handleLink(w http.ResponseWriter, r *http.Request) {
	switch r.Method {
	case "GET":
		app.handleGetLink(w, r)
	default:
		w.WriteHeader(http.StatusMethodNotAllowed)
	}
}

func (app *App) handleGetLinks(w http.ResponseWriter, r *http.Request) {
	var links []Link
	app.db.Find(&links)
	json.NewEncoder(w).Encode(map[string][]Link{"links": links})
}

func (app *App) handleGetLink(w http.ResponseWriter, r *http.Request) {
	id := strings.TrimSuffix(strings.TrimPrefix(r.URL.Path, "/l/"), "/")
	if id == "" {
		onNotFound(w, "No ID found in request")
		return
	}
	var link Link
	if err := app.db.First(&link, "id = ?", id).Error; err != nil {
		w.WriteHeader(http.StatusNotFound)
		fmt.Fprintf(w, "Link for `%s` not found!\n", id)
		return
	}

	if app.link307Redirect {
		http.Redirect(w, r, link.Link, http.StatusTemporaryRedirect)
	} else {
		t, err := template.ParseFS(templateFS, "templates/link.html")
		if err != nil {
			onServerError(w, err, "failed to parse template")
			return
		}
		t.Execute(w, link)
	}

	link.HitCount += 1
	app.db.Save(&link)
}

func (app *App) handleCreateLink(w http.ResponseWriter, r *http.Request) {
	link := Link{}
	if err := json.NewDecoder(r.Body).Decode(&link); err != nil {
		onClientError(w, err, "check the input and try again")
		return
	}
	id := strings.TrimPrefix(r.URL.Path, "/l/")
	if id != "" && link.ID == "" {
		link.ID = id
	}
	if err := link.validate(); err != nil {
		onClientError(w, err, "check the input and try again")
		return
	}
	if err := app.db.Create(&link).Error; err != nil {
		onServerError(w, err, "failed to create DB entry")
		return
	}
	w.WriteHeader(http.StatusCreated)
	json.NewEncoder(w).Encode(map[string]string{"message": "created"})
}