139 lines
4.0 KiB
Go
139 lines
4.0 KiB
Go
/*
|
|
GoToSocial
|
|
Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package router
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/gin-contrib/sessions"
|
|
"github.com/gin-contrib/sessions/memstore"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/superseriousbusiness/gotosocial/internal/config"
|
|
)
|
|
|
|
// Router provides the REST interface for gotosocial, using gin.
|
|
type Router interface {
|
|
// Attach a gin handler to the router with the given method and path
|
|
AttachHandler(method string, path string, f gin.HandlerFunc)
|
|
// Attach a gin middleware to the router that will be used globally
|
|
AttachMiddleware(handler gin.HandlerFunc)
|
|
// Start the router
|
|
Start()
|
|
// Stop the router
|
|
Stop(ctx context.Context) error
|
|
}
|
|
|
|
// router fulfils the Router interface using gin and logrus
|
|
type router struct {
|
|
logger *logrus.Logger
|
|
engine *gin.Engine
|
|
srv *http.Server
|
|
}
|
|
|
|
// Start starts the router nicely
|
|
func (r *router) Start() {
|
|
go func() {
|
|
if err := r.srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
r.logger.Fatalf("listen: %s", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
// Stop shuts down the router nicely
|
|
func (r *router) Stop(ctx context.Context) error {
|
|
return r.srv.Shutdown(ctx)
|
|
}
|
|
|
|
// AttachHandler attaches the given gin.HandlerFunc to the router with the specified method and path.
|
|
// If the path is set to ANY, then the handlerfunc will be used for ALL methods at its given path.
|
|
func (r *router) AttachHandler(method string, path string, handler gin.HandlerFunc) {
|
|
if method == "ANY" {
|
|
r.engine.Any(path, handler)
|
|
} else {
|
|
r.engine.Handle(method, path, handler)
|
|
}
|
|
}
|
|
|
|
// AttachMiddleware attaches a gin middleware to the router that will be used globally
|
|
func (r *router) AttachMiddleware(middleware gin.HandlerFunc) {
|
|
r.engine.Use(middleware)
|
|
}
|
|
|
|
// New returns a new Router with the specified configuration, using the given logrus logger.
|
|
func New(config *config.Config, logger *logrus.Logger) (Router, error) {
|
|
lvl, err := logrus.ParseLevel(config.LogLevel)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("couldn't parse log level %s to set router level: %s", config.LogLevel, err)
|
|
}
|
|
switch lvl {
|
|
case logrus.TraceLevel, logrus.DebugLevel:
|
|
gin.SetMode(gin.DebugMode)
|
|
default:
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
engine := gin.Default()
|
|
|
|
// create a new session store middleware
|
|
store, err := sessionStore()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating session store: %s", err)
|
|
}
|
|
engine.Use(sessions.Sessions("gotosocial-session", store))
|
|
|
|
// load html templates for use by the router
|
|
cwd, err := os.Getwd()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error getting current working directory: %s", err)
|
|
}
|
|
tmPath := filepath.Join(cwd, fmt.Sprintf("%s*", config.TemplateConfig.BaseDir))
|
|
logger.Debugf("loading templates from %s", tmPath)
|
|
engine.LoadHTMLGlob(tmPath)
|
|
|
|
return &router{
|
|
logger: logger,
|
|
engine: engine,
|
|
srv: &http.Server{
|
|
Addr: ":8080",
|
|
Handler: engine,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
// sessionStore returns a new session store with a random auth and encryption key.
|
|
// This means that cookies using the store will be reset if gotosocial is restarted!
|
|
func sessionStore() (memstore.Store, error) {
|
|
auth := make([]byte, 32)
|
|
crypt := make([]byte, 32)
|
|
|
|
if _, err := rand.Read(auth); err != nil {
|
|
return nil, err
|
|
}
|
|
if _, err := rand.Read(crypt); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return memstore.NewStore(auth, crypt), nil
|
|
}
|