storj/satellite/marketing/marketingweb/server.go

117 lines
3.0 KiB
Go
Raw Normal View History

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package marketingweb
import (
"context"
"html/template"
"net"
"net/http"
"github.com/gorilla/mux"
"github.com/zeebo/errs"
"go.uber.org/zap"
"golang.org/x/sync/errgroup"
)
// Error is satellite marketing error type
var Error = errs.Class("satellite marketing error")
// Config contains configuration for marketing offersweb server
type Config struct {
Address string `help:"server address of the marketing Admin GUI" default:"0.0.0.0:8090"`
StaticDir string `help:"path to static resources" default:""`
}
// Server represents marketing offersweb server
type Server struct {
log *zap.Logger
config Config
listener net.Listener
server http.Server
}
// The three pages contained in addPages are pages all templates require
// This exists in order to limit handler verbosity
func (s *Server) addPages(assets []string) []string {
rp := s.config.StaticDir + "/pages/"
pages := []string{rp + "base.html", rp + "index.html", rp + "banner.html"}
for _, page := range assets {
pages = append(pages, page)
}
return pages
}
// NewServer creates new instance of offersweb server
func NewServer(logger *zap.Logger, config Config, listener net.Listener) *Server {
server := Server{
log: logger,
config: config,
listener: listener,
}
logger.Sugar().Debugf("Starting Marketing Admin UI on %s...", server.listener.Addr().String())
fs := http.FileServer(http.Dir(server.config.StaticDir))
mux := mux.NewRouter()
if server.config.StaticDir != "" {
mux.Handle("/static/", http.StripPrefix("/static", fs))
mux.Handle("/", http.HandlerFunc(server.appHandler))
}
server.server = http.Server{
Handler: mux,
}
return &server
}
// appHandler is web app http handler function
func (s *Server) appHandler(w http.ResponseWriter, req *http.Request) {
if req.URL.Path != "/" {
s.serveError(w, req)
return
}
rp := s.config.StaticDir + "/pages/"
pages := []string{rp + "home.html", rp + "refOffers.html", rp + "freeOffers.html", rp + "roModal.html", rp + "foModal.html"}
files := s.addPages(pages)
home := template.Must(template.New("landingPage").ParseFiles(files...))
err := home.ExecuteTemplate(w, "base", nil)
if err != nil {
s.serveError(w, req)
}
}
func (s *Server) serveError(w http.ResponseWriter, req *http.Request) {
rp := s.config.StaticDir + "/pages/"
files := s.addPages([]string{rp + "404.html"})
unavailable := template.Must(template.New("404").ParseFiles(files...))
err := unavailable.ExecuteTemplate(w, "base", nil)
if err != nil {
s.serveError(w, req)
}
}
// Run starts the server that host admin web app and api endpoint
func (s *Server) Run(ctx context.Context) error {
ctx, cancel := context.WithCancel(ctx)
var group errgroup.Group
group.Go(func() error {
<-ctx.Done()
return Error.Wrap(s.server.Shutdown(nil))
})
group.Go(func() error {
defer cancel()
return Error.Wrap(s.server.Serve(s.listener))
})
return group.Wait()
}
// Close closes server and underlying listener
func (s *Server) Close() error {
return Error.Wrap(s.server.Close())
}