[golang] Bootstrap contracting and refactor the agent code
This commit is contained in:
parent
8fbd912414
commit
3dad3f60f2
8 changed files with 158 additions and 106 deletions
|
@ -1,26 +0,0 @@
|
|||
package agent
|
||||
|
||||
import (
|
||||
"sync"
|
||||
|
||||
"git.adyxax.org/adyxax/spacetraders/golang/pkg/model"
|
||||
)
|
||||
|
||||
type Agent struct {
|
||||
data *model.Agent
|
||||
mutex sync.Mutex
|
||||
}
|
||||
|
||||
var agent Agent
|
||||
|
||||
func GetAgent() *model.Agent {
|
||||
agent.mutex.Lock()
|
||||
defer agent.mutex.Unlock()
|
||||
return agent.data
|
||||
}
|
||||
|
||||
func SetAgent(data *model.Agent) {
|
||||
agent.mutex.Lock()
|
||||
defer agent.mutex.Unlock()
|
||||
agent.data = data
|
||||
}
|
78
golang/pkg/agent/run.go
Normal file
78
golang/pkg/agent/run.go
Normal file
|
@ -0,0 +1,78 @@
|
|||
package agent
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"log/slog"
|
||||
|
||||
"git.adyxax.org/adyxax/spacetraders/golang/pkg/api"
|
||||
"git.adyxax.org/adyxax/spacetraders/golang/pkg/database"
|
||||
)
|
||||
|
||||
func Run(
|
||||
apiClient *api.Client,
|
||||
db *database.DB,
|
||||
getenv func(string) string,
|
||||
) error {
|
||||
accountToken := getenv("SPACETRADERS_ACCOUNT_TOKEN")
|
||||
if accountToken == "" {
|
||||
return fmt.Errorf("the SPACETRADERS_ACCOUNT_TOKEN environment variable is not set")
|
||||
}
|
||||
agent := getenv("SPACETRADERS_AGENT")
|
||||
if agent == "" {
|
||||
return fmt.Errorf("the SPACETRADERS_AGENT environment variable is not set")
|
||||
}
|
||||
faction := getenv("SPACETRADERS_FACTION")
|
||||
if faction == "" {
|
||||
return fmt.Errorf("the SPACETRADERS_FACTION environment variable is not set")
|
||||
}
|
||||
// ----- Get token or register ---------------------------------------------
|
||||
apiClient.SetToken(accountToken)
|
||||
register, err := apiClient.Register(faction, agent)
|
||||
if err != nil {
|
||||
apiError := &api.APIError{}
|
||||
if errors.As(err, &apiError) {
|
||||
switch apiError.Code {
|
||||
case 4111: // Agent symbol has already been claimed
|
||||
token, err := db.GetToken()
|
||||
if err != nil || token == "" {
|
||||
return fmt.Errorf("failed to register and failed to get a token from the database: someone stole our agent's callsign: %w", err)
|
||||
}
|
||||
apiClient.SetToken(token)
|
||||
agent, err := apiClient.MyAgent()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get agent: %w", err)
|
||||
}
|
||||
slog.Info("agent", "/my/agent", agent)
|
||||
default:
|
||||
return fmt.Errorf("failed to register: %w", err)
|
||||
}
|
||||
} else {
|
||||
return fmt.Errorf("failed to register with an invalid apiError: %w", err)
|
||||
}
|
||||
} else {
|
||||
token, err := db.GetToken()
|
||||
if err != nil || token == "" {
|
||||
if err := db.AddToken(register.Token); err != nil {
|
||||
return fmt.Errorf("failed to save token: %w", err)
|
||||
}
|
||||
apiClient.SetToken(register.Token)
|
||||
} else {
|
||||
// We successfully registered but have a tainted database
|
||||
slog.Error("token", "token", register.Token)
|
||||
return fmt.Errorf("TODO server reset not implemented yet")
|
||||
}
|
||||
}
|
||||
// ----- run agent ---------------------------------------------------------
|
||||
contracts, err := apiClient.MyContracts()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
slog.Info("start", "contract", contracts[0], "err", err)
|
||||
ships, err := apiClient.MyShips()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
slog.Info("start", "ship", ships[0].Nav.Status, "err", err)
|
||||
return nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue