Simplify web router code (#541)

This commit is contained in:
6543
2021-11-26 09:50:56 +01:00
committed by GitHub
parent c1a8884d62
commit 8b476e7722
9 changed files with 34 additions and 167 deletions

View File

@@ -29,7 +29,7 @@ import (
"github.com/woodpecker-ci/woodpecker/version"
)
func WebConfig(c *gin.Context) {
func Config(c *gin.Context) {
user := session.User(c)
var csrf string

View File

@@ -1,42 +0,0 @@
// Copyright 2018 Drone.IO Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package web
import "time"
// Options defines website handler options.
type Options struct {
sync time.Duration
docs string
}
// Option configures the website handler.
type Option func(*Options)
// WithSync configures the website handler with the duration value
// used to determine if the user account requires synchronization.
func WithSync(d time.Duration) Option {
return func(o *Options) {
o.sync = d
}
}
// WithDocs configures the website handler with the documentation
// website address, which should be included in the user interface.
func WithDocs(s string) Option {
return func(o *Options) {
o.docs = s
}
}

View File

@@ -1,36 +0,0 @@
// Copyright 2018 Drone.IO Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package web
import (
"testing"
"time"
)
func TestWithSync(t *testing.T) {
opts := new(Options)
WithSync(time.Minute)(opts)
if got, want := opts.sync, time.Minute; got != want {
t.Errorf("Want sync duration %v, got %v", want, got)
}
}
func TestWithDocs(t *testing.T) {
opts := new(Options)
WithDocs("http://docs.drone.io")(opts)
if got, want := opts.docs, "http://docs.drone.io"; got != want {
t.Errorf("Want documentation url %q, got %q", want, got)
}
}

View File

@@ -15,7 +15,6 @@
package web
import (
"context"
"crypto/md5"
"fmt"
"net/http"
@@ -24,71 +23,40 @@ import (
"github.com/gin-gonic/gin"
"github.com/rs/zerolog/log"
"github.com/woodpecker-ci/woodpecker/server/model"
"github.com/woodpecker-ci/woodpecker/web"
)
// Endpoint provides the website endpoints.
type Endpoint interface {
// Register registers the provider endpoints.
Register(*gin.Engine)
// etag is an identifier for a resource version
// it lets caches determine if resource is still the same and not send it again
var etag = fmt.Sprintf("%x", md5.Sum([]byte(time.Now().String())))
// New returns a gin engine to serve the web frontend.
func New() *gin.Engine {
e := gin.New()
e.Use(setupCache)
h := http.FileServer(web.HttpFS())
e.GET("/favicon.svg", gin.WrapH(h))
e.GET("/assets/*filepath", gin.WrapH(h))
e.NoRoute(handleIndex)
return e
}
// New returns the default website endpoint.
func New(opt ...Option) Endpoint {
opts := new(Options)
for _, f := range opt {
f(opts)
}
return &website{
fs: web.HttpFS(),
opts: opts,
data: web.MustLookup("index.html"),
}
}
type website struct {
opts *Options
fs http.FileSystem
data []byte
}
func (w *website) Register(mux *gin.Engine) {
h := http.FileServer(w.fs)
h = setupCache(h)
mux.GET("/favicon.svg", gin.WrapH(h))
mux.GET("/assets/*filepath", gin.WrapH(h))
mux.NoRoute(gin.WrapF(w.handleIndex))
}
func (w *website) handleIndex(rw http.ResponseWriter, r *http.Request) {
func handleIndex(c *gin.Context) {
rw := c.Writer
data := web.MustLookup("index.html")
rw.Header().Set("Content-Type", "text/html; charset=UTF-8")
rw.WriteHeader(200)
if _, err := rw.Write(w.data); err != nil {
if _, err := rw.Write(data); err != nil {
log.Error().Err(err).Msg("can not write index.html")
}
}
func setupCache(h http.Handler) http.Handler {
data := []byte(time.Now().String())
etag := fmt.Sprintf("%x", md5.Sum(data))
return http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Cache-Control", "public, max-age=31536000")
w.Header().Del("Expires")
w.Header().Set("ETag", etag)
h.ServeHTTP(w, r)
},
)
func setupCache(c *gin.Context) {
c.Writer.Header().Set("Cache-Control", "public, max-age=31536000")
c.Writer.Header().Del("Expires")
c.Writer.Header().Set("ETag", etag)
}
// WithUser returns a context with the current authenticated user.
func WithUser(c context.Context, user *model.User) context.Context {
return context.WithValue(c, userKey, user)
}
type key int
const userKey key = 0