mirror of
https://github.com/elyby/chrly.git
synced 2024-08-25 22:30:01 +05:30
143 lines
4.6 KiB
Go
143 lines
4.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/mono83/slf/wd"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/elyby/chrly/bootstrap"
|
|
"github.com/elyby/chrly/db"
|
|
"github.com/elyby/chrly/eventsubscribers"
|
|
"github.com/elyby/chrly/http"
|
|
"github.com/elyby/chrly/mojangtextures"
|
|
)
|
|
|
|
var serveCmd = &cobra.Command{
|
|
Use: "serve",
|
|
Short: "Starts HTTP handler for the skins system",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
dispatcher := bootstrap.CreateEventDispatcher()
|
|
|
|
// TODO: this is a mess, need to organize this code somehow to make services initialization more compact
|
|
logger, err := bootstrap.CreateLogger(viper.GetString("sentry.dsn"))
|
|
if err != nil {
|
|
log.Fatalf("Cannot initialize logger: %v", err)
|
|
}
|
|
logger.Info("Logger successfully initialized")
|
|
|
|
(&eventsubscribers.Logger{Logger: logger}).ConfigureWithDispatcher(dispatcher)
|
|
|
|
statsdAddr := viper.GetString("statsd.addr")
|
|
if statsdAddr != "" {
|
|
statsdReporter, err := bootstrap.CreateStatsReceiver(statsdAddr)
|
|
if err != nil {
|
|
logger.Emergency("Invalid statsd configuration :err", wd.ErrParam(err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
(&eventsubscribers.StatsReporter{StatsReporter: statsdReporter}).ConfigureWithDispatcher(dispatcher)
|
|
}
|
|
|
|
storageFactory := db.StorageFactory{Config: viper.GetViper()}
|
|
|
|
logger.Info("Initializing skins repository")
|
|
redisFactory := storageFactory.CreateFactory("redis")
|
|
skinsRepo, err := redisFactory.CreateSkinsRepository()
|
|
if err != nil {
|
|
logger.Emergency("Error on creating skins repo: :err", wd.ErrParam(err))
|
|
os.Exit(1)
|
|
}
|
|
logger.Info("Skins repository successfully initialized")
|
|
|
|
logger.Info("Initializing capes repository")
|
|
filesystemFactory := storageFactory.CreateFactory("filesystem")
|
|
capesRepo, err := filesystemFactory.CreateCapesRepository()
|
|
if err != nil {
|
|
logger.Emergency("Error on creating capes repo: :err", wd.ErrParam(err))
|
|
os.Exit(1)
|
|
}
|
|
logger.Info("Capes repository successfully initialized")
|
|
|
|
var mojangTexturesProvider http.MojangTexturesProvider
|
|
if viper.GetBool("mojang_textures.enabled") {
|
|
logger.Info("Preparing Mojang's textures queue")
|
|
mojangUuidsRepository, err := redisFactory.CreateMojangUuidsRepository()
|
|
if err != nil {
|
|
logger.Emergency("Error on creating mojang uuids repo: :err", wd.ErrParam(err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
uuidsProvider, err := bootstrap.CreateMojangUUIDsProvider(dispatcher)
|
|
if err != nil {
|
|
logger.Emergency("Unable to create mojang uuids provider: :err", wd.ErrParam(err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
texturesStorage := mojangtextures.NewInMemoryTexturesStorage()
|
|
texturesStorage.Start()
|
|
mojangTexturesProvider = &mojangtextures.Provider{
|
|
Emitter: dispatcher,
|
|
UUIDsProvider: uuidsProvider,
|
|
TexturesProvider: &mojangtextures.MojangApiTexturesProvider{
|
|
Emitter: dispatcher,
|
|
},
|
|
Storage: &mojangtextures.SeparatedStorage{
|
|
UuidsStorage: mojangUuidsRepository,
|
|
TexturesStorage: texturesStorage,
|
|
},
|
|
}
|
|
logger.Info("Mojang's textures queue is successfully initialized")
|
|
} else {
|
|
logger.Info("Mojang's textures queue is disabled")
|
|
mojangTexturesProvider = &mojangtextures.NilProvider{}
|
|
}
|
|
|
|
address := fmt.Sprintf("%s:%d", viper.GetString("server.host"), viper.GetInt("server.port"))
|
|
handler := (&http.Skinsystem{
|
|
Emitter: dispatcher,
|
|
SkinsRepo: skinsRepo,
|
|
CapesRepo: capesRepo,
|
|
MojangTexturesProvider: mojangTexturesProvider,
|
|
Authenticator: &http.JwtAuth{
|
|
Key: []byte(viper.GetString("chrly.secret")),
|
|
Emitter: dispatcher,
|
|
},
|
|
TexturesExtraParamName: viper.GetString("textures.extra_param_name"),
|
|
TexturesExtraParamValue: viper.GetString("textures.extra_param_value"),
|
|
}).CreateHandler()
|
|
|
|
finishChan := make(chan bool)
|
|
go func() {
|
|
logger.Info("Starting the app, HTTP on: :addr", wd.StringParam("addr", address))
|
|
if err := http.Serve(address, handler); err != nil {
|
|
logger.Emergency("Error in main(): :err", wd.ErrParam(err))
|
|
finishChan <- true
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
s := waitForExitSignal()
|
|
logger.Info("Got signal: :signal, exiting", wd.StringParam("signal", s.String()))
|
|
finishChan <- true
|
|
}()
|
|
|
|
<-finishChan
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
RootCmd.AddCommand(serveCmd)
|
|
viper.SetDefault("server.host", "")
|
|
viper.SetDefault("server.port", 80)
|
|
viper.SetDefault("storage.redis.host", "localhost")
|
|
viper.SetDefault("storage.redis.port", 6379)
|
|
viper.SetDefault("storage.redis.poll", 10)
|
|
viper.SetDefault("storage.filesystem.basePath", "data")
|
|
viper.SetDefault("storage.filesystem.capesDirName", "capes")
|
|
viper.SetDefault("mojang_textures.enabled", true)
|
|
}
|