126 lines
2.6 KiB
Go
126 lines
2.6 KiB
Go
package httpserver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"html/template"
|
|
"io/ioutil"
|
|
"log"
|
|
"mime"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/improbable-eng/grpc-web/go/grpcweb"
|
|
"go.uber.org/zap"
|
|
"golang.org/x/oauth2"
|
|
|
|
"go.c3c.cz/cv/app/server/internal/files"
|
|
)
|
|
|
|
func init() {
|
|
err := mime.AddExtensionType(".ico", "image/x-icon")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
err = mime.AddExtensionType(".json", "application/json; charset=utf-8")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
type Options struct {
|
|
BindAddr string
|
|
OAuth2 *oauth2.Config
|
|
GrpcWebServer *grpcweb.WrappedGrpcServer
|
|
FrontendConfig string
|
|
Logger *zap.Logger
|
|
LooseCORS bool
|
|
RefreshTokenCookieDomain string
|
|
RequireIndex bool
|
|
}
|
|
|
|
type server struct{ http.Server }
|
|
|
|
func New(options *Options) (*server, error) {
|
|
handler := &handler{
|
|
oauth2: options.OAuth2,
|
|
grpcWebServer: options.GrpcWebServer,
|
|
frontendConfig: options.FrontendConfig,
|
|
logger: options.Logger,
|
|
looseCORS: options.LooseCORS,
|
|
refreshTokenCookieDomain: options.RefreshTokenCookieDomain,
|
|
}
|
|
|
|
httpSrv := &server{
|
|
http.Server{
|
|
Addr: options.BindAddr,
|
|
Handler: handler,
|
|
},
|
|
}
|
|
|
|
index, err := files.Public.Open("data/public/index.html")
|
|
if err != nil {
|
|
if !options.RequireIndex {
|
|
return httpSrv, nil
|
|
}
|
|
|
|
return nil, fmt.Errorf("could not open index.html: %w", err)
|
|
}
|
|
|
|
fi, err := index.Stat()
|
|
if err != nil || fi.IsDir() {
|
|
if !options.RequireIndex {
|
|
return httpSrv, nil
|
|
}
|
|
|
|
return nil, fmt.Errorf("invalid index.html: %w", err)
|
|
}
|
|
|
|
tb, err := ioutil.ReadAll(index)
|
|
if err != nil {
|
|
if !options.RequireIndex {
|
|
return httpSrv, nil
|
|
}
|
|
|
|
return nil, fmt.Errorf("could not read index.html: %w", err)
|
|
}
|
|
|
|
handler.indexTemplate, err = template.New("index").Parse(string(tb))
|
|
if err != nil {
|
|
if !options.RequireIndex {
|
|
return httpSrv, nil
|
|
}
|
|
|
|
return nil, fmt.Errorf("could not parse index.html: index.html: %w", err)
|
|
}
|
|
|
|
return httpSrv, nil
|
|
}
|
|
|
|
func (s *server) ShutdownWithDelay(delay time.Duration, limit time.Duration) {
|
|
h, ok := s.Handler.(*handler)
|
|
|
|
var logger *zap.Logger
|
|
if ok {
|
|
logger = h.logger
|
|
} else {
|
|
logger = zap.L()
|
|
}
|
|
|
|
if delay > 0 {
|
|
logger.Info(fmt.Sprintf("Waiting %s for incoming requests\n", delay))
|
|
time.Sleep(delay)
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), limit)
|
|
defer cancel()
|
|
|
|
logger.Info("Gracefully shutting down http server")
|
|
if err := s.Shutdown(ctx); err != nil {
|
|
logger.Error("Failed to shutdown http server", zap.Error(err))
|
|
} else {
|
|
logger.Info("Http server was stopped")
|
|
}
|
|
}
|