93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/titpetric/factory/resputil"
|
|
|
|
"github.com/crusttech/crust/internal/auth"
|
|
"github.com/crusttech/crust/internal/db"
|
|
"github.com/crusttech/crust/internal/mail"
|
|
"github.com/crusttech/crust/internal/metrics"
|
|
migrate "github.com/crusttech/crust/messaging/db"
|
|
"github.com/crusttech/crust/messaging/internal/service"
|
|
)
|
|
|
|
var (
|
|
jwtVerifier func(http.Handler) http.Handler
|
|
jwtAuthenticator func(http.Handler) http.Handler
|
|
jwtEncoder auth.TokenEncoder
|
|
)
|
|
|
|
func Init(ctx context.Context) error {
|
|
// validate configuration
|
|
if err := flags.Validate(); err != nil {
|
|
return err
|
|
}
|
|
// JWT Auth
|
|
if jwtAuth, err := auth.JWT(); err != nil {
|
|
return errors.Wrap(err, "Error creating JWT Auth object")
|
|
} else {
|
|
jwtEncoder = jwtAuth
|
|
jwtVerifier = jwtAuth.Verifier()
|
|
jwtAuthenticator = jwtAuth.Authenticator()
|
|
}
|
|
|
|
mail.SetupDialer(flags.smtp)
|
|
|
|
if err := InitDatabase(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
// configure resputil options
|
|
resputil.SetConfig(resputil.Options{
|
|
Pretty: flags.http.Pretty,
|
|
Trace: flags.http.Tracing,
|
|
Logger: func(err error) {
|
|
// @todo: error logging
|
|
},
|
|
})
|
|
|
|
// Don't change this, it needs Database
|
|
service.Init()
|
|
|
|
return nil
|
|
}
|
|
|
|
func InitDatabase(ctx context.Context) error {
|
|
// start/configure database connection
|
|
db, err := db.TryToConnect(ctx, "messaging", flags.db.DSN, flags.db.Profiler)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not connect to database")
|
|
}
|
|
|
|
// migrate database schema
|
|
if err := migrate.Migrate(db); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func StartRestAPI(ctx context.Context) error {
|
|
log.Println("Starting http server on address " + flags.http.Addr)
|
|
listener, err := net.Listen("tcp", flags.http.Addr)
|
|
if err != nil {
|
|
return errors.Wrap(err, fmt.Sprintf("Can't listen on addr %s", flags.http.Addr))
|
|
}
|
|
|
|
if flags.monitor.Interval > 0 {
|
|
go metrics.NewMonitor(flags.monitor.Interval)
|
|
}
|
|
|
|
go http.Serve(listener, Routes(ctx))
|
|
<-ctx.Done()
|
|
|
|
return nil
|
|
}
|