[golang] implement automation loop and add contract accepting
This commit is contained in:
parent
3dad3f60f2
commit
d97985a694
GPG key ID:
F92E51B86E07177E
9 changed files with 223 additions and 64 deletions
47
golang/pkg/agent/contracting.go
Normal file
47
golang/pkg/agent/contracting.go
Normal file
|
|
@ -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
|
||||
}
|
||||
15
golang/pkg/agent/error.go
Normal file
15
golang/pkg/agent/error.go
Normal file
|
|
@ -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,
|
||||
}
|
||||
}
|
||||
60
golang/pkg/agent/init.go
Normal file
60
golang/pkg/agent/init.go
Normal file
|
|
@ -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
|
||||
}
|
||||
|
|
@ -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")
|
||||
}
|
||||
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)
|
||||
err := agent.init()
|
||||
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)
|
||||
return fmt.Errorf("failed to init agent: %w", err)
|
||||
}
|
||||
|
||||
if agent.ships, err = client.MyShips(); err != nil {
|
||||
return fmt.Errorf("failed to init the agent's ships: %w", err)
|
||||
}
|
||||
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
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue