78 lines
1.9 KiB
Go
78 lines
1.9 KiB
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
|
|
Email string
|
|
UserAgent string
|
|
UserStyle template.CSS
|
|
APIEndpoint template.URL
|
|
PixelAPI *pixelapi.PixelAPI
|
|
|
|
// Only used on file viewer page
|
|
Title string
|
|
OGData template.HTML
|
|
|
|
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: "",
|
|
UserAgent: r.UserAgent(),
|
|
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),
|
|
Domain: wc.conf.SessionCookieDomain,
|
|
})
|
|
}
|
|
return t
|
|
}
|
|
|
|
// Authentication succeeded
|
|
t.Authenticated = true
|
|
t.Username = uinf.Username
|
|
t.Email = uinf.Email
|
|
} else {
|
|
t.PixelAPI = pixelapi.New(wc.conf.APIURLInternal, "")
|
|
}
|
|
|
|
return t
|
|
}
|