adyxax/spacetraders
1
0
Fork 0
Code Issues Activity

[golang] Bootstrap contracting and refactor the agent code

Browse source
This commit is contained in:
Julien Dessaux 2025-02-13 21:35:32 +01:00
parent 8fbd912414
commit 3dad3f60f2
Signed by: adyxax
GPG key ID: F92E51B86E07177E
8 changed files with 158 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

77
golang/cmd/spacetraders/main.go
View file

@ -2,12 +2,12 @@ package main
import (
"context"
"errors"
"fmt"
"log/slog"
"os"
"os/signal"
"git.adyxax.org/adyxax/spacetraders/golang/pkg/agent"
"git.adyxax.org/adyxax/spacetraders/golang/pkg/api"
"git.adyxax.org/adyxax/spacetraders/golang/pkg/database"
)
@ -37,7 +37,7 @@ func main() {
apiClient := api.NewClient(ctx)
defer apiClient.Close()
if err := run(
if err := agent.Run(
apiClient,
db,
os.Getenv,
@ -49,76 +49,3 @@ func main() {
os.Exit(2)
}
}
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 ---------------------------------------------------------
ships, err := apiClient.MyShips()
if err != nil {
return err
}
slog.Info("start", "ship", ships[0].Nav.Status, "err", err)
err = apiClient.Orbit(&ships[0])
if err != nil {
return err
}
slog.Info("orbit", "ship", ships[0].Nav.Status, "err", err)
err = apiClient.Dock(&ships[0])
if err != nil {
return err
}
slog.Info("dock", "ship", ships[0].Nav.Status, "err", err)
return nil
}