package requests import ( "errors" "fmt" "net/http" "strings" "gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/models/db" ) var ( ErrorSidWrongFormat = errors.New("Session id wrong format") ErrorSidNotFound = errors.New("Session id didn't find in response header") ) const ( AuthPath = "/auth" LicensePath = "/license" JoinPath = "/join" PlayersPath = "/players" SetupPath = "/setup" UpdatePath = "/update" SocketPath = "/socket.io" ) var WsTypeData = map[string]string{ AuthPath: "getAuthData", LicensePath: "getAuthData", JoinPath: "getJoinData", PlayersPath: "getPlayersData", SetupPath: "getSetupData", UpdatePath: "getUpdateData", } var PathToSetupState = map[string]db.StateType{ SetupPath: db.SetupState, // AuthPath: db.AuthState, // UpdatePath: db.UpdateState, // LicensePath: db.LicenseState, } var PathToWorldState = map[string]db.StateType{ JoinPath: db.JoinState, PlayersPath: db.PlayersState, } type Config struct { Host string Password string SessionID string } func (conf *Config) SetSessionTokenFromHeader(resp http.Header) (bool, error) { setCookieHeader := resp.Get("Set-Cookie") if setCookieHeader == "" { return false, nil } for value := range strings.SplitSeq(setCookieHeader, ";") { if strings.Contains(value, "session") { sessionCookie := strings.Split(value, "=") if len(sessionCookie) != 2 { return false, ErrorSidWrongFormat } conf.SessionID = sessionCookie[1] return true, nil } } return false, nil } func (conf *Config) GetRequestHeader() *http.Header { header := http.Header{} header.Set("Cookie", fmt.Sprintf("session=%s", conf.SessionID)) header.Set("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8") header.Set("Connection", "keep-alive") header.Set("Content-Type", "application/x-www-form-urlencoded") header.Set("Host", conf.Host) header.Set("Origin", fmt.Sprintf("http://%s", conf.Host)) header.Set("Referer", fmt.Sprintf("http://%s%s", conf.Host, AuthPath)) return &header }