112 lines
2.4 KiB
Go
112 lines
2.4 KiB
Go
package messaging
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-chi/chi"
|
|
_ "github.com/joho/godotenv/autoload"
|
|
"github.com/spf13/cobra"
|
|
"github.com/titpetric/factory"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/cortezaproject/corteza-server/messaging/commands"
|
|
migrate "github.com/cortezaproject/corteza-server/messaging/db"
|
|
"github.com/cortezaproject/corteza-server/messaging/rest"
|
|
"github.com/cortezaproject/corteza-server/messaging/service"
|
|
"github.com/cortezaproject/corteza-server/messaging/service/event"
|
|
"github.com/cortezaproject/corteza-server/messaging/websocket"
|
|
"github.com/cortezaproject/corteza-server/pkg/app"
|
|
"github.com/cortezaproject/corteza-server/pkg/auth"
|
|
"github.com/cortezaproject/corteza-server/pkg/corredor"
|
|
"github.com/cortezaproject/corteza-server/pkg/scheduler"
|
|
)
|
|
|
|
type (
|
|
App struct {
|
|
Opts *app.Options
|
|
Log *zap.Logger
|
|
|
|
ws *websocket.Websocket
|
|
}
|
|
)
|
|
|
|
const SERVICE = "messaging"
|
|
|
|
func (app *App) Setup(log *zap.Logger, opts *app.Options) (err error) {
|
|
app.Log = log.Named(SERVICE)
|
|
app.Opts = opts
|
|
|
|
scheduler.Service().OnTick(
|
|
event.MessagingOnInterval(),
|
|
event.MessagingOnTimestamp(),
|
|
)
|
|
|
|
app.ws = websocket.New(&websocket.Config{
|
|
Timeout: opts.Websocket.Timeout,
|
|
PingTimeout: opts.Websocket.PingTimeout,
|
|
PingPeriod: opts.Websocket.PingPeriod,
|
|
})
|
|
|
|
// @todo Wire in cross-service JWT maker for Corredor
|
|
corredor.Service().SetUserFinder(nil)
|
|
corredor.Service().SetRoleFinder(nil)
|
|
|
|
return
|
|
}
|
|
|
|
func (app *App) Upgrade(ctx context.Context) (err error) {
|
|
db := factory.Database.MustGet(SERVICE, "default").With(ctx).Quiet()
|
|
err = migrate.Migrate(db, app.Log)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (app *App) Initialize(ctx context.Context) (err error) {
|
|
// Connects to all services it needs to
|
|
err = service.Initialize(ctx, app.Log, service.Config{
|
|
Storage: app.Opts.Storage,
|
|
})
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (app *App) Activate(ctx context.Context) (err error) {
|
|
if err = service.Activate(ctx); err != nil {
|
|
return
|
|
}
|
|
|
|
service.Watchers(ctx)
|
|
websocket.Watch(ctx)
|
|
|
|
return
|
|
}
|
|
|
|
func (app *App) Provision(ctx context.Context) (err error) {
|
|
ctx = auth.SetSuperUserContext(ctx)
|
|
|
|
if err = provisionConfig(ctx, app.Log); err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (app *App) MountApiRoutes(r chi.Router) {
|
|
rest.MountRoutes(r)
|
|
app.ws.ApiServerRoutes(r)
|
|
}
|
|
|
|
func (app *App) RegisterCliCommands(p *cobra.Command) {
|
|
p.AddCommand(
|
|
commands.Importer(),
|
|
commands.Exporter(),
|
|
)
|
|
}
|