mirror of
https://github.com/nestriness/nestri.git
synced 2025-12-12 16:55:37 +02:00
⭐ feat(maitred): Update maitred - hookup to the API (#198)
## Description We are attempting to hookup maitred to the API Maitred duties will be: - [ ] Hookup to the API - [ ] Wait for signal (from the API) to start Steam - [ ] Stop signal to stop the gaming session, clean up Steam... and maybe do the backup ## Summary by CodeRabbit - **New Features** - Introduced Docker-based deployment configurations for both the main and relay applications. - Added new API endpoints enabling real-time machine messaging and enhanced IoT operations. - Expanded database schema and actor types to support improved machine tracking. - **Improvements** - Enhanced real-time communication and relay management with streamlined room handling. - Upgraded dependencies, logging, and error handling for greater stability and performance. <!-- end of auto-generated comment: release notes by coderabbit.ai --> --------- Co-authored-by: DatCaptainHorse <DatCaptainHorse@users.noreply.github.com> Co-authored-by: Kristian Ollikainen <14197772+DatCaptainHorse@users.noreply.github.com>
This commit is contained in:
@@ -1,21 +1,126 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"nestri/maitred/pkg/party"
|
||||
"context"
|
||||
"log/slog"
|
||||
"nestri/maitred/internal"
|
||||
"nestri/maitred/internal/containers"
|
||||
"nestri/maitred/internal/realtime"
|
||||
"nestri/maitred/internal/resource"
|
||||
"nestri/maitred/internal/system"
|
||||
"os"
|
||||
|
||||
"github.com/charmbracelet/log"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
)
|
||||
|
||||
func main() {
|
||||
var teamSlug string //FIXME: Switch to team-slug as they are more memorable but still unique
|
||||
// Setup main context and stopper
|
||||
mainCtx, mainStop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
|
||||
|
||||
if len(os.Args) > 1 {
|
||||
teamSlug = os.Args[1]
|
||||
} else {
|
||||
log.Fatal("Nestri needs a team slug to register this container to")
|
||||
// Get flags and log them
|
||||
internal.InitFlags()
|
||||
internal.GetFlags().DebugLog()
|
||||
|
||||
logLevel := slog.LevelInfo
|
||||
if internal.GetFlags().Verbose {
|
||||
logLevel = slog.LevelDebug
|
||||
}
|
||||
party.Run(teamSlug)
|
||||
|
||||
//TODO: On stop here, set the API as the instance is not running (stopped)
|
||||
// Create the base handler with debug level
|
||||
baseHandler := slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{
|
||||
Level: logLevel,
|
||||
})
|
||||
customHandler := &internal.CustomHandler{Handler: baseHandler}
|
||||
logger := slog.New(customHandler)
|
||||
slog.SetDefault(logger)
|
||||
|
||||
if !internal.GetFlags().NoMonitor {
|
||||
// Start system monitoring, fetch every 5 seconds
|
||||
system.StartMonitoring(mainCtx, 5*time.Second)
|
||||
}
|
||||
|
||||
// Get machine ID
|
||||
machineID, err := system.GetID()
|
||||
if err != nil {
|
||||
slog.Error("failed getting machine id", "err", machineID)
|
||||
}
|
||||
|
||||
slog.Info("Machine ID", "id", machineID)
|
||||
|
||||
// Initialize container engine
|
||||
ctrEngine, err := containers.NewContainerEngine()
|
||||
if err != nil {
|
||||
slog.Error("failed initializing container engine", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
defer func(ctrEngine containers.ContainerEngine) {
|
||||
// Stop our managed containers first, with a 30 second timeout
|
||||
cleanupCtx, cleanupCancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cleanupCancel()
|
||||
err = realtime.CleanupManaged(cleanupCtx, ctrEngine)
|
||||
if err != nil {
|
||||
slog.Error("failed cleaning up managed containers", "err", err)
|
||||
}
|
||||
|
||||
err = ctrEngine.Close()
|
||||
if err != nil {
|
||||
slog.Error("failed closing container engine", "err", err)
|
||||
}
|
||||
}(ctrEngine)
|
||||
|
||||
// Print engine info
|
||||
info, err := ctrEngine.Info(mainCtx)
|
||||
if err != nil {
|
||||
slog.Error("failed getting engine info", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
slog.Info("Container engine", "info", info)
|
||||
|
||||
if err = realtime.InitializeManager(mainCtx, ctrEngine); err != nil {
|
||||
slog.Error("failed initializing container manager", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
|
||||
// If in debug mode, skip running SST - MQTT connections
|
||||
if !internal.GetFlags().Debug {
|
||||
// Initialize SST resource
|
||||
res, err := resource.NewResource()
|
||||
if err != nil {
|
||||
slog.Error("failed getting resource", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
|
||||
// Run realtime
|
||||
err = realtime.Run(mainCtx, machineID, ctrEngine, res)
|
||||
if err != nil {
|
||||
slog.Error("failed running realtime", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
// Create relay container
|
||||
slog.Info("Creating default relay container")
|
||||
relayID, err := realtime.CreateRelay(mainCtx, ctrEngine)
|
||||
if err != nil {
|
||||
slog.Error("failed creating relay container", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
// Start relay container
|
||||
slog.Info("Starting default relay container", "id", relayID)
|
||||
if err = realtime.StartRelay(mainCtx, ctrEngine, relayID); err != nil {
|
||||
slog.Error("failed starting relay container", "err", err)
|
||||
mainStop()
|
||||
return
|
||||
}
|
||||
|
||||
// Wait for signal
|
||||
<-mainCtx.Done()
|
||||
slog.Info("Shutting down gracefully by signal..")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user