26 lines
604 B
Go
26 lines
604 B
Go
package web
|
|
|
|
import (
|
|
"github.com/gin-contrib/static"
|
|
"github.com/gin-gonic/gin"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// A ModuleRegisterFunc is a module.
|
|
type ModuleRegisterFunc func(*gin.Engine, *Service)
|
|
|
|
// ModuleRegister adds f to ws's list of modules.
|
|
func (ws *Service) ModuleRegister(f ModuleRegisterFunc) {
|
|
ws.modules = append(ws.modules, f)
|
|
}
|
|
|
|
// Bind executes all of ws's modules with r.
|
|
func (ws *Service) Bind(r *gin.Engine) {
|
|
for _, f := range ws.modules {
|
|
f(r, ws)
|
|
}
|
|
|
|
ws.log.Info("bind modules", zap.Int("count", len(ws.modules)))
|
|
r.Use(static.Serve("/", static.LocalFile(ws.Webroot, false)))
|
|
}
|