focalboard/server/web/webserver.go

110 lines
2.4 KiB
Go
Raw Normal View History

package web
2020-10-16 11:41:56 +02:00
import (
"fmt"
"log"
"net/http"
"os"
"path"
2020-10-16 11:41:56 +02:00
"path/filepath"
"github.com/gorilla/mux"
)
// RoutedService defines the interface that is needed for any service to
// register themself in the web server to provide new endpoints. (see
// AddRoutes).
2020-10-16 11:41:56 +02:00
type RoutedService interface {
RegisterRoutes(*mux.Router)
}
// Server is the structure responsible for managing our http web server.
type Server struct {
http.Server
2021-01-27 00:29:13 +01:00
rootPath string
port int
ssl bool
localOnly bool
2020-10-16 11:41:56 +02:00
}
// NewServer creates a new instance of the webserver.
2021-03-21 09:28:26 +01:00
func NewServer(rootPath string, port int, ssl, localOnly bool) *Server {
2020-10-16 11:41:56 +02:00
r := mux.NewRouter()
2021-01-27 00:29:13 +01:00
var addr string
if localOnly {
addr = fmt.Sprintf(`localhost:%d`, port)
} else {
addr = fmt.Sprintf(`:%d`, port)
}
ws := &Server{
Server: http.Server{
2021-01-27 00:29:13 +01:00
Addr: addr,
Handler: r,
},
2020-10-16 16:21:42 +02:00
rootPath: rootPath,
port: port,
ssl: ssl,
2020-10-16 11:41:56 +02:00
}
2020-10-16 16:21:42 +02:00
return ws
2020-10-16 11:41:56 +02:00
}
func (ws *Server) Router() *mux.Router {
return ws.Server.Handler.(*mux.Router)
}
// AddRoutes allows services to register themself in the webserver router and provide new endpoints.
func (ws *Server) AddRoutes(rs RoutedService) {
rs.RegisterRoutes(ws.Router())
2020-10-16 11:41:56 +02:00
}
func (ws *Server) registerRoutes() {
ws.Router().PathPrefix("/static").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir(filepath.Join(ws.rootPath, "static")))))
ws.Router().PathPrefix("/").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/html; charset=utf-8")
http.ServeFile(w, r, path.Join(ws.rootPath, "index.html"))
})
}
// Start runs the web server and start listening for charsetnnections.
2021-03-01 20:36:36 +01:00
func (ws *Server) Start() {
ws.registerRoutes()
2020-10-16 11:41:56 +02:00
isSSL := ws.ssl && fileExists("./cert/cert.pem") && fileExists("./cert/key.pem")
2020-10-16 11:41:56 +02:00
if isSSL {
log.Printf("https server started on :%d\n", ws.port)
2021-01-18 21:27:13 +01:00
go func() {
if err := ws.ListenAndServeTLS("./cert/cert.pem", "./cert/key.pem"); err != nil {
log.Fatalf("ListenAndServeTLS: %v", err)
}
}()
return
2020-10-16 11:41:56 +02:00
}
2020-11-09 21:53:25 +01:00
log.Printf("http server started on :%d\n", ws.port)
2021-01-18 21:27:13 +01:00
go func() {
if err := ws.ListenAndServe(); err != http.ErrServerClosed {
log.Fatalf("ListenAndServe: %v", err)
}
2021-03-01 20:36:36 +01:00
log.Println("http server stopped")
2021-01-18 21:27:13 +01:00
}()
2020-10-16 11:41:56 +02:00
}
func (ws *Server) Shutdown() error {
return ws.Close()
}
// fileExists returns true if a file exists at the path.
func fileExists(path string) bool {
_, err := os.Stat(path)
if os.IsNotExist(err) {
return false
}
return err == nil
}