139 lines
3.7 KiB
Go
139 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
|
|
context "github.com/SentimensRG/ctx"
|
|
"github.com/SentimensRG/ctx/sigctx"
|
|
"github.com/go-chi/chi"
|
|
_ "github.com/joho/godotenv/autoload"
|
|
"github.com/namsral/flag"
|
|
|
|
compose "github.com/crusttech/crust/compose"
|
|
"github.com/crusttech/crust/internal/auth"
|
|
messaging "github.com/crusttech/crust/messaging"
|
|
system "github.com/crusttech/crust/system"
|
|
|
|
"github.com/crusttech/crust/internal/config"
|
|
"github.com/crusttech/crust/internal/metrics"
|
|
"github.com/crusttech/crust/internal/middleware"
|
|
"github.com/crusttech/crust/internal/routes"
|
|
"github.com/crusttech/crust/internal/subscription"
|
|
"github.com/crusttech/crust/internal/version"
|
|
)
|
|
|
|
// Serves index.html in case the requested file isn't found (or some other os.Stat error)
|
|
func serveIndex(assetPath string, indexPath string, serve http.Handler) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
indexPage := path.Join(assetPath, indexPath)
|
|
requestedPage := path.Join(assetPath, r.URL.Path)
|
|
_, err := os.Stat(requestedPage)
|
|
if err != nil {
|
|
http.ServeFile(w, r, indexPage)
|
|
return
|
|
}
|
|
serve.ServeHTTP(w, r)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// log to stdout not stderr
|
|
log.SetOutput(os.Stdout)
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
log.Printf("Starting "+os.Args[0]+", version: %v, built on: %v", version.Version, version.BuildTime)
|
|
|
|
ctx := context.AsContext(sigctx.New())
|
|
|
|
var flags struct {
|
|
http *config.HTTP
|
|
monitor *config.Monitor
|
|
}
|
|
flags.http = new(config.HTTP).Init()
|
|
flags.monitor = new(config.Monitor).Init()
|
|
|
|
compose.Flags("compose")
|
|
messaging.Flags("messaging")
|
|
system.Flags("system")
|
|
|
|
authJwtFlags := new(config.JWT).Init()
|
|
|
|
subscription.Flags()
|
|
|
|
flag.Parse()
|
|
|
|
var command string
|
|
if len(os.Args) > 1 {
|
|
command = os.Args[1]
|
|
}
|
|
|
|
switch command {
|
|
case "help":
|
|
flag.PrintDefaults()
|
|
default:
|
|
// Initialize configuration of our services
|
|
if err := system.Init(ctx); err != nil {
|
|
log.Fatalf("Error initializing system: %+v", err)
|
|
}
|
|
if err := compose.Init(ctx); err != nil {
|
|
log.Fatalf("Error initializing compose: %+v", err)
|
|
}
|
|
if err := messaging.Init(ctx); err != nil {
|
|
log.Fatalf("Error initializing messaging: %+v", err)
|
|
}
|
|
// Checks subscription, will os.Exit(1) if there is an error
|
|
// Disabled for now, system service is the only one that validates subscription
|
|
// ctx = subscription.Monitor(ctx)
|
|
|
|
log.Println("Starting http server on address " + flags.http.Addr)
|
|
listener, err := net.Listen("tcp", flags.http.Addr)
|
|
if err != nil {
|
|
log.Fatalf("Can't listen on addr %s", flags.http.Addr)
|
|
}
|
|
|
|
if flags.monitor.Interval > 0 {
|
|
go metrics.NewMonitor(flags.monitor.Interval)
|
|
}
|
|
|
|
r := chi.NewRouter()
|
|
|
|
// logging, cors and such
|
|
middleware.Mount(ctx, r, flags.http)
|
|
|
|
// Use JWT secret for hmac signer for now
|
|
auth.DefaultSigner = auth.HmacSigner(authJwtFlags.Secret)
|
|
auth.DefaultJwtHandler, err = auth.JWT(authJwtFlags.Secret, authJwtFlags.Expiry)
|
|
if err != nil {
|
|
log.Fatalf("Error creating JWT Auth: %v", err)
|
|
}
|
|
|
|
r.Route("/api", func(r chi.Router) {
|
|
r.Route("/compose", func(r chi.Router) {
|
|
compose.MountRoutes(ctx, r)
|
|
})
|
|
r.Route("/messaging", func(r chi.Router) {
|
|
messaging.MountRoutes(ctx, r)
|
|
})
|
|
r.Route("/system", func(r chi.Router) {
|
|
system.MountRoutes(ctx, r)
|
|
})
|
|
middleware.MountSystemRoutes(ctx, r, flags.http)
|
|
})
|
|
|
|
fileserver := http.FileServer(http.Dir("webapp"))
|
|
|
|
for _, service := range []string{"admin", "system", "messaging", "compose"} {
|
|
r.HandleFunc("/"+service+"*", serveIndex("webapp", "compose/index.html", fileserver))
|
|
}
|
|
r.HandleFunc("/*", serveIndex("webapp", "index.html", fileserver))
|
|
|
|
routes.Print(r)
|
|
|
|
go http.Serve(listener, r)
|
|
<-ctx.Done()
|
|
}
|
|
}
|