qurl/pages/root.go

package pages

import (
	"fmt"
	"git.binarythought.com/cdramey/qurl/obj"
	"git.binarythought.com/cdramey/qurl/storage"
	"git.binarythought.com/cdramey/qurl/assets"
	"html/template"
	"net/http"
)

type RootHandler struct {
	Storage storage.Storage

	index  *StaticContent
	css    *StaticContent
	favi   *StaticContent
	usage  *StaticContent
	submit *template.Template
}

func (ctx *RootHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	fname := r.URL.Path

	switch fname {
	case "/index.html":
		fallthrough
	case "/":
		ctx.index.ServeHTTP(w, r)

	case "/submit.html":
		ctx.ServeSubmit(w, r)

	case "/qurl.css":
		ctx.css.ServeHTTP(w, r)

	case "/favicon.ico":
		ctx.favi.ServeHTTP(w, r)

	case "/api/usage.html":
		ctx.usage.ServeHTTP(w, r)

	case "/api/url":
		ctx.ServeAPI(w, r)

	default:
		id, err := obj.FromString(fname[1:len(fname)])
		if err != nil {
			http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
			return
		}

		q, err := ctx.Storage.GetQURLByID(id)
		if err != nil {
			http.Error(w, fmt.Sprintf("Database error: %s", err.Error()),
				http.StatusInternalServerError)
			return
		}

		if q == nil {
			http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
			return
		}

		http.Redirect(w, r, q.URL, http.StatusPermanentRedirect)
	}
}

func (ctx *RootHandler) Init() error {
	// Initialize the static content object for the index page
	data, err := assets.ReadFile("index.html")
	if err != nil {
		return err
	}
	ctx.index = &StaticContent{Type: "text/html", Content: data }
	ctx.index.Init()

	// Initialize the static content object for the css
	data, err = assets.ReadFile("qurl.css")
	if err != nil {
		return err
	}
	ctx.css = &StaticContent{Type: "text/css", Content: data }
	ctx.css.Init()

	// Initialize the static content object favicon
	data, err = assets.ReadFile("favicon.ico")
	if err != nil {
		return err
	}
	ctx.favi = &StaticContent{Type: "image/x-icon", Content: data}
	ctx.favi.Init()

	// Initialize the api usage instructions
	data, err = assets.ReadFile("usage.html")
	if err != nil {
		return err
	}
	ctx.usage = &StaticContent{Type: "text/html", Content: data}
	ctx.usage.Init()

	// Initialize submit page template
	data, err = assets.ReadFile("submit.html")
	if err != nil {
		return err
	}
	ctx.submit = template.New("submit.html")
	_, err = ctx.submit.Parse(string(data))
	if err != nil {
		return err
	}
	return nil
}