[golang] implement automation loop and add contract accepting

4 files changed, 175 insertions, 55 deletions

diff --git a/golang/pkg/agent/contracting.go b/golang/pkg/agent/contracting.go
new file mode 100644
index 0000000..44d7753
--- /dev/null
+++ b/golang/pkg/agent/contracting.go
@@ -0,0 +1,47 @@
+package agent
+
+import (
+	"fmt"
+	"time"
+
+	"git.adyxax.org/adyxax/spacetraders/golang/pkg/model"
+)
+
+func (a *agent) autoContracting(ship *model.Ship) {
+	defer a.wg.Done()
+	contracts, err := a.client.MyContracts()
+	if err != nil {
+		a.sendShipError(fmt.Errorf("failed to get my contracts: %w", err), ship)
+		return
+	}
+	for _, contract := range contracts {
+		if contract.Fullfilled {
+			continue
+		}
+		now := time.Now()
+		if now.Before(contract.Terms.Deadline) {
+			if err := a.runContract(&contract, ship); err != nil {
+				a.sendShipError(fmt.Errorf("failed to run contracts: %w", err), ship)
+				return
+			}
+		}
+	}
+	// TODO
+	//for {
+	// negotiate
+	// runContract
+	//}
+}
+
+func (a *agent) runContract(contract *model.Contract, ship *model.Ship) error {
+	if err := a.client.Accept(contract, a.db); err != nil {
+		return fmt.Errorf("failed to run contract: %w", err)
+	}
+	switch contract.Type {
+	// TODO
+	//case "PROCUREMENT":
+	default:
+		return fmt.Errorf("failed to run contract: handling contracts of type %s is not implemented yet", contract.Type)
+	}
+	return nil
+}
diff --git a/golang/pkg/agent/error.go b/golang/pkg/agent/error.go
new file mode 100644
index 0000000..64e6b8d
--- /dev/null
+++ b/golang/pkg/agent/error.go
@@ -0,0 +1,15 @@
+package agent
+
+import "git.adyxax.org/adyxax/spacetraders/golang/pkg/model"
+
+type shipError struct {
+	err  error
+	ship *model.Ship
+}
+
+func (a *agent) sendShipError(err error, ship *model.Ship) {
+	a.channel <- shipError{
+		err:  err,
+		ship: ship,
+	}
+}
diff --git a/golang/pkg/agent/init.go b/golang/pkg/agent/init.go
new file mode 100644
index 0000000..30b18e9
--- /dev/null
+++ b/golang/pkg/agent/init.go
@@ -0,0 +1,60 @@
+package agent
+
+import (
+	"errors"
+	"fmt"
+	"log/slog"
+
+	"git.adyxax.org/adyxax/spacetraders/golang/pkg/api"
+	"git.adyxax.org/adyxax/spacetraders/golang/pkg/model"
+)
+
+func (a *agent) init() error {
+	token, err := a.db.GetToken()
+	if err == nil && token != "" {
+		a.client.SetToken(token)
+		var agent *model.Agent
+		agent, err = a.client.MyAgent() // we need err to carry over outside this block
+		if err == nil {
+			slog.Info("agent", "/my/agent", agent)
+		} else {
+			slog.Info("failed to get agent, handling server reset", "err", err)
+			a.db.Reset()
+		}
+	}
+	if err != nil || token == "" {
+		// No token or invalid token (since we carry over the error of MyAgent() in the previous if), we need to register
+		accountToken := a.getenv("SPACETRADERS_ACCOUNT_TOKEN")
+		if accountToken == "" {
+			return fmt.Errorf("the SPACETRADERS_ACCOUNT_TOKEN environment variable is not set")
+		}
+		a.client.SetToken(accountToken)
+		agent := a.getenv("SPACETRADERS_AGENT")
+		if agent == "" {
+			return fmt.Errorf("the SPACETRADERS_AGENT environment variable is not set")
+		}
+		faction := a.getenv("SPACETRADERS_FACTION")
+		if faction == "" {
+			return fmt.Errorf("the SPACETRADERS_FACTION environment variable is not set")
+		}
+		register, err := a.client.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
+					return fmt.Errorf("failed to register and failed to get a token from the database: someone stole our agent's callsign: %w", err)
+				default:
+					return fmt.Errorf("failed to register: %w", err)
+				}
+			} else {
+				return fmt.Errorf("failed to register with an invalid apiError: %w", err)
+			}
+		}
+		a.client.SetToken(register.Token)
+		if err := a.db.SaveToken(register.Token); err != nil {
+			return fmt.Errorf("failed to save token %s after a successful registration: %w", register.Token, err)
+		}
+	}
+	return nil
+}
diff --git a/golang/pkg/agent/run.go b/golang/pkg/agent/run.go
index db4c614..bc5254e 100644
--- a/golang/pkg/agent/run.go
+++ b/golang/pkg/agent/run.go
@@ -1,78 +1,76 @@
 package agent
 
 import (
-	"errors"
 	"fmt"
 	"log/slog"
+	"sync"
 
 	"git.adyxax.org/adyxax/spacetraders/golang/pkg/api"
 	"git.adyxax.org/adyxax/spacetraders/golang/pkg/database"
+	"git.adyxax.org/adyxax/spacetraders/golang/pkg/model"
+)
+
+type agent struct {
+	channel chan shipError
+	client  *api.Client
+	db      *database.DB
+	getenv  func(string) string
+	ships   []model.Ship
+	wg      sync.WaitGroup
+}
+
+type State int
+
+const (
+	start_running_contracts_with_the_command_ship = iota
 )
 
 func Run(
-	apiClient *api.Client,
+	client *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 := agent{
+		channel: make(chan shipError),
+		client:  client,
+		db:      db,
+		getenv:  getenv,
 	}
-	agent := getenv("SPACETRADERS_AGENT")
-	if agent == "" {
-		return fmt.Errorf("the SPACETRADERS_AGENT environment variable is not set")
+	err := agent.init()
+	if err != nil {
+		return fmt.Errorf("failed to init agent: %w", err)
 	}
-	faction := getenv("SPACETRADERS_FACTION")
-	if faction == "" {
-		return fmt.Errorf("the SPACETRADERS_FACTION environment variable is not set")
+
+	if agent.ships, err = client.MyShips(); err != nil {
+		return fmt.Errorf("failed to init the agent's ships: %w", err)
 	}
-	// ----- 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)
+	var state State = start_running_contracts_with_the_command_ship
+	agent.wg.Add(1)
+	go func() {
+		defer agent.wg.Done()
+		for {
+			switch state {
+			case start_running_contracts_with_the_command_ship:
+				agent.wg.Add(1)
+				go agent.autoContracting(&agent.ships[0])
+				state++
+				return
 			default:
-				return fmt.Errorf("failed to register: %w", err)
+				agent.sendShipError(fmt.Errorf("agent runner reach an unknown state: %d", state), nil)
+				return
 			}
-		} 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")
+	}()
+	var errWg sync.WaitGroup
+	errWg.Add(1)
+	go func() {
+		defer errWg.Done()
+		for shipErr := range agent.channel {
+			slog.Error("ship error", "err", shipErr.err, "ship", shipErr.ship.Symbol)
 		}
-	}
-	// ----- 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)
+	}()
+	agent.wg.Wait()
+	close(agent.channel)
+	errWg.Wait()
 	return nil
 }