add user and pass for world to env variables

This commit is contained in:
lbenedar
2026-04-27 13:14:17 +03:00
parent fd65631be1
commit 4bb4b04004
11 changed files with 161 additions and 36 deletions

View File

@@ -10,30 +10,28 @@ import (
"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/types"
)
func (tr *FoundryTransport) FillDBWithFoundryData() {
func (tr *FoundryTransport) FillDBWithFoundryData() error {
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
return err
}
if !setupExist {
err := tr.InsertSetupToDB()
if err != nil {
tr.ReadChan.Err() <- &types.FoundryError{Direction: types.WriterCode, Err: err, Type: types.DbCode}
return
return err
}
}
err = tr.InitWorldsData()
if err != nil {
tr.ReadChan.Err() <- &types.FoundryError{Direction: types.WriterCode, Err: err, Type: types.DbCode}
return
return err
}
tr.Logger.Info("Initialization has been successful", "elapsed_time", time.Since(now).String())
return nil
}
func (tr *FoundryTransport) InsertSetupToDB() error {
@@ -66,8 +64,8 @@ func (tr *FoundryTransport) InitWorldsData() error {
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)
if !slices.Contains(worldNames, initWorld.Name) {
tr.Logger.Warn("World name does not exist in database or have been already inserted. Skip!", "world_name", initWorld.Name)
continue
}
@@ -84,14 +82,21 @@ func (tr *FoundryTransport) InitWorldsData() error {
return nil
}
func (tr *FoundryTransport) InitWorldData(worldName string) error {
tr.Logger.Debug("Launching world...", "world_name", worldName)
err := tr.LaunchWorld(worldName)
func (tr *FoundryTransport) InitWorldData(worldData types.WorldData) error {
tr.Logger.Debug("Launching world...", "world_name", worldData.Name)
err := tr.LaunchWorld(worldData.Name)
if err != nil {
return err
}
err = tr.ConnectToWorld()
if worldData.UserId == "" {
worldData.UserId, err = tr.GetUserId(worldData.Username)
if err != nil {
return err
}
}
err = tr.ConnectToWorld(worldData.UserId, worldData.UserPass)
if err != nil {
return err
}

View File

@@ -10,4 +10,5 @@ var (
ErrorAuthPassWrong = errors.New("Authentication password is wrong")
ErrorChannelIsClosed = errors.New("Channel has been closed")
ErrorUserIsNotConnected = errors.New("User is not connected")
ErrorUserDoesntExist = errors.New("User does not exist")
)

View File

@@ -85,7 +85,7 @@ func (tr *FoundryTransport) CheckAuthResponse(body io.Reader, respLength int) er
return nil
}
func (tr *FoundryTransport) LogInToWorld() error {
func (tr *FoundryTransport) LogInToWorld(userId string, userPass string) error {
if tr.Http.SessionID == nil {
err := tr.Http.GetSessionId()
if err != nil {
@@ -93,8 +93,7 @@ func (tr *FoundryTransport) LogInToWorld() error {
}
}
// TODO: make userid and password from tr object
resp, err := tr.Http.PostJoinWorld("YpMHZge0dxBZS5Cm", "")
resp, err := tr.Http.PostJoinWorld(userId, userPass)
if err != nil {
return err
}
@@ -106,8 +105,6 @@ func (tr *FoundryTransport) LogInToWorld() error {
}
return tr.CheckLoginResponse(resp.Body, contentLen)
// {"userid":"YpMHZge0dxBZS5Cm","password":"","action":"join"}
}
func (tr *FoundryTransport) CheckLoginResponse(body io.Reader, respLength int) error {

View File

@@ -29,7 +29,7 @@ type FoundryTransport struct {
// Models *db.Models
DB *sqlx.DB
IsDbInit bool
InitWorlds []string
InitWorlds []types.WorldData
Logger *slog.Logger
}
@@ -38,7 +38,7 @@ type FoundryTransportData struct {
DbConn *sqlx.DB
Logger *slog.Logger
HttpConfig *requests.FoundryHttpRequest
Worlds []string
Worlds []types.WorldData
}
func NewFoundryTransport(data *FoundryTransportData) *FoundryTransport {

View File

@@ -1,11 +1,14 @@
package transport
import (
"encoding/json"
"fmt"
"net/http"
"net/url"
"strings"
"time"
json_models "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/types"
"github.com/gorilla/websocket"
@@ -105,8 +108,8 @@ func (tr *FoundryTransport) ListenWebSocket() {
}
}
func (tr *FoundryTransport) ConnectToWorld() error {
err := tr.LogInToWorld()
func (tr *FoundryTransport) ConnectToWorld(userId string, userPass string) error {
err := tr.LogInToWorld(userId, userPass)
if err != nil {
return err
}
@@ -126,6 +129,58 @@ func (tr *FoundryTransport) ConnectToWorld() error {
return nil
}
func (tr *FoundryTransport) GetUserId(username string) (string, error) {
msgJson, err := tr.GetJsonDataByType(requests.JoinPath)
if err != nil {
return "", err
}
var modelUsers []struct {
Users []json_models.User `json:"users"`
}
err = json.Unmarshal(msgJson, &modelUsers)
if err != nil {
return "", err
}
tr.Logger.Debug("User data", "msg in json format", modelUsers)
users := modelUsers[0].Users
for i := range users {
if strings.EqualFold(users[i].Name, username) {
return users[i].ID, nil
}
}
return "", ErrorUserDoesntExist
}
func (tr *FoundryTransport) GetUserIdAndPass() (string, string, error) {
status, err := tr.Http.GetStatus()
if err != nil {
return "", "", err
}
var worldData *types.WorldData
for i := range tr.InitWorlds {
if strings.EqualFold(tr.InitWorlds[i].Name, status.World) {
worldData = &tr.InitWorlds[i]
break
}
}
if worldData == nil {
return "", "", err
}
userId, err := tr.GetUserId(worldData.Username)
if err != nil {
return "", "", err
}
return userId, worldData.UserPass, nil
}
func (tr *FoundryTransport) SendOnlyCodeRequest(code string) error {
tr.Logger.Debug("WS: Data has been send\n", "msg", types.CodesRespToReq[code])