fnx_web/webcontroller/template_data.go

package webcontroller

import (
	"html/template"
	"net/http"
	"net/url"
	"time"

	"fornaxian.com/pixeldrain-web/pixelapi"
	"fornaxian.com/pixeldrain-web/webcontroller/forms"
	"github.com/Fornaxian/log"
)

// TemplateData is a struct that every template expects when being rendered. In
// the field Other you can pass your own template-specific variables.
type TemplateData struct {
	Authenticated bool
	Username      string
	UserStyle     template.CSS
	APIEndpoint   template.URL
	PixelAPI      *pixelapi.PixelAPI

	// Only used on file viewer page
	Title  string
	OGData OGData

	Other    interface{}
	URLQuery url.Values

	// Only used for pages containing forms
	Form forms.Form
}

func (wc *WebController) newTemplateData(w http.ResponseWriter, r *http.Request) *TemplateData {
	var t = &TemplateData{
		Authenticated: false,
		Username:      "",
		UserStyle:     userStyle(r),
		APIEndpoint:   template.URL(wc.conf.APIURLExternal),
		URLQuery:      r.URL.Query(),
	}

	if key, err := wc.getAPIKey(r); err == nil {
		t.PixelAPI = pixelapi.New(wc.conf.APIURLInternal, key)
		uinf, err := t.PixelAPI.UserInfo()
		if err != nil {
			// This session key doesn't work, or the backend is down, user
			// cannot be authenticated
			log.Debug("Session check for key '%s' failed: %s", key, err)

			if err.Error() == "authentication_required" || err.Error() == "authentication_failed" {
				// This key is invalid, delete it
				log.Debug("Deleting invalid API key")
				http.SetCookie(w, &http.Cookie{
					Name:    "pd_auth_key",
					Value:   "",
					Path:    "/",
					Expires: time.Unix(0, 0),
				})
			}
			return t
		}

		// Authentication succeeded
		t.Authenticated = true
		t.Username = uinf.Username
	} else {
		t.PixelAPI = pixelapi.New(wc.conf.APIURLInternal, "")
	}

	return t
}