auth.go

package admin

import (
	"net/http"
	"time"
)

func (a *Admin) getUserSession(req *http.Request) *session {
	cookie, err := req.Cookie("admin")
	if err != nil {
		return nil
	}

	if sess, ok := a.sessions[cookie.Value]; ok {
		return sess
	}
	return nil
}

func (a *Admin) logIn(rw http.ResponseWriter, username, password string) bool {
	if username != a.username || password != a.password {
		return false
	}
	sessKey := randString(32)
	a.sessions[sessKey] = &session{
		time:     time.Now(),
		messages: []*flashMessage{},
	}

	http.SetCookie(rw, &http.Cookie{
		Name:  "admin",
		Value: sessKey,
		Path:  a.path,
	})
	return true
}

type session struct {
	time     time.Time
	messages []*flashMessage
}

func (s *session) addMessage(class, text string) {
	s.messages = append(s.messages, &flashMessage{class, text})
}

func (s *session) getMessages() []*flashMessage {
	// If empty, there's no need to create a new slice.
	if len(s.messages) == 0 {
		return s.messages
	}

	messages := s.messages
	s.messages = []*flashMessage{}
	return messages
}

type flashMessage struct {
	Class string
	Text  string
}