add user and pass for world to env variables
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
@@ -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")
|
||||
)
|
||||
|
||||
@@ -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 {
|
||||
|
||||
@@ -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 {
|
||||
|
||||
@@ -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])
|
||||
|
||||
|
||||
Reference in New Issue
Block a user