finish world initialization, set up init on startup and on shutdown, move all models to main directory

internal/foundry/transport/db.go

@@ -1,37 +1,39 @@
 package transport
 
 import (
+	"slices"
+	"time"
+
+	"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/models/db"
+	"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/models/json"
 	"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/requests"
-	"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/temp_models/db"
-	"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/temp_models/json"
 	"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/types"
 )
 
 func (tr *FoundryTransport) FillDBWithFoundryData() {
-	err := tr.InitSetupData()
+	now := time.Now()
+
+	setupExist, err := db.HasSetup(tr.DB)
 	if err != nil {
 		tr.ReadChan.Err() <- &types.FoundryError{Direction: types.WriterCode, Err: err, Type: types.DbCode}
 		return
 	}
 
-	// idState, err := tr.Models.FoundryState.GetIdByType(db.SetupState)
-	// if err != nil {
-	// 	tr.ReadChan.Err() <- &types.FoundryError{Direction: types.WriterCode, Err: err, Type: types.DbCode}
-	// 	return
-	// }
+	if !setupExist {
+		err := tr.InsertSetupToDB()
+		if err != nil {
+			tr.ReadChan.Err() <- &types.FoundryError{Direction: types.WriterCode, Err: err, Type: types.DbCode}
+			return
+		}
+	}
 
 	err = tr.InitWorldsData()
 	if err != nil {
 		tr.ReadChan.Err() <- &types.FoundryError{Direction: types.WriterCode, Err: err, Type: types.DbCode}
 		return
 	}
-}
 
-func (tr *FoundryTransport) InitSetupData() error {
-	db.DeleteAll(tr.DB)
-	db.DeleteAllSeq(tr.DB)
-
-	return tr.InsertSetupToDB()
+	tr.Logger.Info("Initialization has been successful", "elapsed_time", time.Since(now).String())
 }
 
 func (tr *FoundryTransport) InsertSetupToDB() error {
@@ -57,11 +59,23 @@ func (tr *FoundryTransport) InsertSetupToDB() error {
 }
 
 func (tr *FoundryTransport) InitWorldsData() error {
-	worlds := []string{"kingmaker"} //TODO: get world name from database
+	worldNames, err := db.GetNotStartedWorlds(tr.DB)
+	if err != nil {
+		return err
+	}
+
+	for i := range tr.InitWorlds {
+		initWorld := tr.InitWorlds[i]
+		if !slices.Contains(worldNames, initWorld) {
+			tr.Logger.Warn("World name does not exist in database or have been already inserted. Skip!", "world_name", initWorld)
+			continue
+		}
+
+		err = tr.InitWorldData(initWorld)
+		if err != nil {
+			return err
+		}
 
-	for i := range worlds {
-		worldName := worlds[i]
-		tr.InitWorldData(worldName)
 		_, err := tr.Http.PostReturnToSetup()
 		if err != nil {
 			return err
@@ -71,34 +85,48 @@ func (tr *FoundryTransport) InitWorldsData() error {
 }
 
 func (tr *FoundryTransport) InitWorldData(worldName string) error {
-	tr.Logger.Debug("World name", "name", worldName)
+	tr.Logger.Debug("Launching world...", "world_name", worldName)
 	err := tr.LaunchWorld(worldName)
 	if err != nil {
 		return err
 	}
+
+	err = tr.ConnectToWorld()
+	if err != nil {
+		return err
+	}
 	return tr.InsertGameToDB()
 }
 
 func (tr *FoundryTransport) InsertGameToDB() error {
-	err := tr.ConnectToWorld()
-	if err != nil {
-		return err
-	}
-
 	msgJson, err := tr.GetJsonData("world")
 	if err != nil {
 		return err
 	}
 
-	tr.Logger.Info("Game data received")
-
-	gameJson, err := json.ParseGame(msgJson)
+	game, err := json.ParseGame(msgJson)
 	if err != nil {
 		return err
 	}
 
-	var foundryStateDb db.Game
-	gameJson.ToDB(&foundryStateDb)
+	var gameDb db.Game
+	game.ToDB(&gameDb)
 
-	return foundryStateDb.Insert(tr.DB)
+	isInserted, err := db.IsWorldInserted(tr.DB, game.World.ID)
+	if err != nil {
+		return err
+	}
+	if isInserted {
+		return nil
+	}
+
+	tr.Logger.Info("Game data received", "world_name", game.World.ID)
+
+	err = gameDb.Insert(tr.DB)
+	if err != nil {
+		return err
+	}
+
+	tr.Logger.Info("Game data inserted to DB", "world_name", game.World.ID)
+	return nil
 }