zirconium-go/core/app_context.go

package core

import (
	"context"
	"fmt"
	"github.com/google/logger"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"go.mongodb.org/mongo-driver/mongo/readpref"
)

type AppContext struct {
	router            *Router
	authManager       *AuthManager
	connectionHandler *ConnectionHandler
	websocketServer   *WebsocketServer
	cfg               *Config
	database          *mongo.Database
	userManager       *UserManager
}

func NewAppContext(cfg *Config) *AppContext {
	var err error
	appContext := &AppContext{}
	appContext.cfg = cfg

	appContext.connectToDatabase()

	router, err := NewRouter(appContext)
	if err != nil {
		logger.Fatalf("Unable to initialize router: %s", err.Error())
	}
	appContext.router = router

	authManager, err := NewAuthManager()
	if err != nil {
		logger.Fatalf("Unable to initialize authentication manager: %s", err.Error())
	}
	appContext.authManager = authManager

	um := NewUserManager(appContext.database)
	appContext.userManager = um

	connHandler := NewConnectionHandler(router)
	appContext.connectionHandler = connHandler

	wss := NewWebsocketServer(cfg, connHandler)
	appContext.websocketServer = wss

	return appContext
}

func (ac *AppContext) connectToDatabase() {
	ctx := context.TODO()
	dbUri := fmt.Sprintf("mongodb://%s:%s@%s:%d", ac.cfg.Mongo.User, ac.cfg.Mongo.Password, ac.cfg.Mongo.Host, ac.cfg.Mongo.Port)
	opts := options.Client().ApplyURI(dbUri)
	err := opts.Validate()
	if err != nil {
		logger.Fatalf("invalid database config: %s", err)
	}
	mongoClient, err := mongo.Connect(ctx, opts)
	if err != nil {
		logger.Fatalf("cannot connect to mongo database: %s", err)
	}
	err = mongoClient.Ping(ctx, readpref.Primary())
	if err != nil {
		logger.Fatalf("cannot connect to mongo database: %s", err)
	}

	db := mongoClient.Database(ac.cfg.Mongo.Database)
	ac.database = db
}

func (ac *AppContext) Run() error {
	// TODO
	return nil
}