diff options
author | Julien Dessaux | 2025-02-20 08:12:57 +0100 |
---|---|---|
committer | Julien Dessaux | 2025-02-20 08:12:57 +0100 |
commit | 40c4a8df155f7be75e34498ec538ac2eca5097bd (patch) | |
tree | 755a37b0f1563f4fb0d848236c1c8c57e18e3514 /golang | |
parent | [golang] implement trading and contracting (diff) | |
download | spacetraders-40c4a8df155f7be75e34498ec538ac2eca5097bd.tar.gz spacetraders-40c4a8df155f7be75e34498ec538ac2eca5097bd.tar.bz2 spacetraders-40c4a8df155f7be75e34498ec538ac2eca5097bd.zip |
Diffstat (limited to 'golang')
-rw-r--r-- | golang/cmd/spacetraders/main.go | 2 | ||||
-rw-r--r-- | golang/pkg/agent/contracting.go | 8 | ||||
-rw-r--r-- | golang/pkg/agent/trading.go | 20 | ||||
-rw-r--r-- | golang/pkg/agent/utils.go | 8 | ||||
-rw-r--r-- | golang/pkg/agent/visit.go | 10 | ||||
-rw-r--r-- | golang/pkg/api/client.go | 6 | ||||
-rw-r--r-- | golang/pkg/api/contracts.go | 11 | ||||
-rw-r--r-- | golang/pkg/api/ships.go | 19 | ||||
-rw-r--r-- | golang/pkg/api/systems.go | 31 |
9 files changed, 58 insertions, 57 deletions
diff --git a/golang/cmd/spacetraders/main.go b/golang/cmd/spacetraders/main.go index 5353c68..0e50497 100644 --- a/golang/cmd/spacetraders/main.go +++ b/golang/cmd/spacetraders/main.go @@ -36,7 +36,7 @@ func main() { os.Exit(1) } - apiClient := api.NewClient(ctx) + apiClient := api.NewClient(ctx, db) defer apiClient.Close() if err := agent.Run( apiClient, diff --git a/golang/pkg/agent/contracting.go b/golang/pkg/agent/contracting.go index ecd896a..4e476a6 100644 --- a/golang/pkg/agent/contracting.go +++ b/golang/pkg/agent/contracting.go @@ -41,7 +41,7 @@ func (a *agent) autoContracting(ship *model.Ship) { } func (a *agent) runContract(contract *model.Contract, ship *model.Ship) error { - if err := a.client.Accept(contract, a.db); err != nil { + if err := a.client.Accept(contract); err != nil { return fmt.Errorf("failed to accept contract: %w", err) } //slog.Info("running contract", "contract", contract, "ship", ship.Symbol) @@ -72,15 +72,15 @@ func (a *agent) runProcurement(contract *model.Contract, ship *model.Ship) error } } // deliver the goods - if err := a.client.Navigate(ship, deliver.DestinationSymbol, a.db); err != nil { + if err := a.client.Navigate(ship, deliver.DestinationSymbol); err != nil { return fmt.Errorf("failed to navigate to %s: %w", deliver.DestinationSymbol, err) } - if err := a.client.Deliver(contract, ship, a.db); err != nil { + if err := a.client.Deliver(contract, ship); err != nil { return fmt.Errorf("failed to deliver: %w", err) } deliver = contract.Terms.Deliver[0] if deliver.UnitsRequired == deliver.UnitsFulfilled { - if err := a.client.Fulfill(contract, a.db); err != nil { + if err := a.client.Fulfill(contract); err != nil { return fmt.Errorf("failed to fulfill: %w", err) } return nil diff --git a/golang/pkg/agent/trading.go b/golang/pkg/agent/trading.go index e84cef4..b900f7a 100644 --- a/golang/pkg/agent/trading.go +++ b/golang/pkg/agent/trading.go @@ -34,13 +34,13 @@ func (a *agent) buyTradeGood(ship *model.Ship, tradeGoodToBuy string) error { return &TradeGoodNotFoundError{} } // find the closest place to buy TODO - waypoint, err := a.client.GetWaypoint(ship.Nav.WaypointSymbol, a.db) + waypoint, err := a.client.GetWaypoint(ship.Nav.WaypointSymbol) if err != nil { return fmt.Errorf("failed to get nav waypoint %s: %w", ship.Nav.WaypointSymbol, err) } waypoints := make([]model.Waypoint, 0) for i := range markets { - waypoint, err := a.client.GetWaypoint(markets[i].Symbol, a.db) + waypoint, err := a.client.GetWaypoint(markets[i].Symbol) if err != nil { return fmt.Errorf("failed to get waypoint %s: %w", markets[i].Symbol, err) } @@ -48,10 +48,10 @@ func (a *agent) buyTradeGood(ship *model.Ship, tradeGoodToBuy string) error { } sortByDistanceFrom(waypoint, waypoints) // Go there and refresh our market data - if err := a.client.Navigate(ship, waypoints[0].Symbol, a.db); err != nil { + if err := a.client.Navigate(ship, waypoints[0].Symbol); err != nil { return fmt.Errorf("failed to navigate to %s: %w", waypoints[0].Symbol, err) } - market, err := a.client.GetMarket(waypoints[0].Symbol, a.db) + market, err := a.client.GetMarket(waypoints[0].Symbol) if err != nil { return fmt.Errorf("failed to get market %s: %w", waypoints[0].Symbol, err) } @@ -60,7 +60,7 @@ func (a *agent) buyTradeGood(ship *model.Ship, tradeGoodToBuy string) error { if tradeGood.Type == "EXPORT" && tradeGood.Symbol == tradeGoodToBuy { for ship.Cargo.Units < ship.Cargo.Capacity { increment := min(ship.Cargo.Capacity-ship.Cargo.Units, tradeGood.TradeVolume) - if err := a.client.Purchase(ship, tradeGoodToBuy, increment, a.db); err != nil { + if err := a.client.Purchase(ship, tradeGoodToBuy, increment); err != nil { return fmt.Errorf("failed to purchase %d units of %s: %w", increment, tradeGoodToBuy, err) } } @@ -98,13 +98,13 @@ func (a *agent) sellEverythingExcept(ship *model.Ship, keep string) error { return nil } // find the closest place to sell something TODO - waypoint, err := a.client.GetWaypoint(ship.Nav.WaypointSymbol, a.db) + waypoint, err := a.client.GetWaypoint(ship.Nav.WaypointSymbol) if err != nil { return fmt.Errorf("failed to get nav waypoint %s: %w", ship.Nav.WaypointSymbol, err) } waypoints := make([]model.Waypoint, 0) for i := range markets { - waypoint, err := a.client.GetWaypoint(markets[i].Symbol, a.db) + waypoint, err := a.client.GetWaypoint(markets[i].Symbol) if err != nil { return fmt.Errorf("failed to get waypoint %s: %w", markets[i].Symbol, err) } @@ -112,10 +112,10 @@ func (a *agent) sellEverythingExcept(ship *model.Ship, keep string) error { } sortByDistanceFrom(waypoint, waypoints) // Go there and refresh our market data - if err := a.client.Navigate(ship, waypoints[0].Symbol, a.db); err != nil { + if err := a.client.Navigate(ship, waypoints[0].Symbol); err != nil { return fmt.Errorf("failed to navigate to %s: %w", waypoints[0].Symbol, err) } - market, err := a.client.GetMarket(waypoints[0].Symbol, a.db) + market, err := a.client.GetMarket(waypoints[0].Symbol) if err != nil { return fmt.Errorf("failed to get market %s: %w", waypoints[0].Symbol, err) } @@ -126,7 +126,7 @@ func (a *agent) sellEverythingExcept(ship *model.Ship, keep string) error { if tradeGood.Type == "IMPORT" && tradeGood.Symbol == cargoItem.Symbol { for units > 0 { increment := min(units, tradeGood.TradeVolume) - if err := a.client.Sell(ship, cargoItem.Symbol, increment, a.db); err != nil { + if err := a.client.Sell(ship, cargoItem.Symbol, increment); err != nil { return fmt.Errorf("failed to sell %d units of %s: %w", units, cargoItem.Symbol, err) } units = units - increment diff --git a/golang/pkg/agent/utils.go b/golang/pkg/agent/utils.go index b274f43..7aa9c7b 100644 --- a/golang/pkg/agent/utils.go +++ b/golang/pkg/agent/utils.go @@ -25,7 +25,7 @@ func (a *agent) isThereAShipAtWaypoint(waypointSymbol string) bool { } func (a *agent) listWaypointsInSystemWithTrait(systemSymbol string, trait string) ([]model.Waypoint, error) { - waypoints, err := a.client.ListWaypointsInSystem(systemSymbol, a.db) + waypoints, err := a.client.ListWaypointsInSystem(systemSymbol) if err != nil { return nil, fmt.Errorf("failed to list waypoints: %w", err) } @@ -47,7 +47,7 @@ func (a *agent) listMarketsInSystem(systemSymbol string) ([]model.Market, error) } var markets []model.Market for i := range waypoints { - market, err := a.client.GetMarket(waypoints[i].Symbol, a.db) + market, err := a.client.GetMarket(waypoints[i].Symbol) if err != nil { return nil, fmt.Errorf("failed to get market %s: %w", waypoints[i].Symbol, err) } @@ -63,7 +63,7 @@ func (a *agent) listShipyardsInSystem(systemSymbol string) ([]model.Shipyard, er } var shipyards []model.Shipyard for i := range waypoints { - shipyard, err := a.client.GetShipyard(waypoints[i].Symbol, a.db) + shipyard, err := a.client.GetShipyard(waypoints[i].Symbol) if err != nil { return nil, fmt.Errorf("failed to get shipyard %s: %w", waypoints[i].Symbol, err) } @@ -104,7 +104,7 @@ func (a *agent) sendShipToShipyardThatSells(ship *model.Ship, shipType string) e } return cmp.Compare(aPrice, bPrice) }) - if err := a.client.Navigate(ship, shipyards[0].Symbol, a.db); err != nil { + if err := a.client.Navigate(ship, shipyards[0].Symbol); err != nil { return fmt.Errorf("failed to navigate to %s: %w", shipyards[0].Symbol, err) } return nil diff --git a/golang/pkg/agent/visit.go b/golang/pkg/agent/visit.go index 3004553..e420186 100644 --- a/golang/pkg/agent/visit.go +++ b/golang/pkg/agent/visit.go @@ -23,29 +23,29 @@ func (a *agent) visitAllShipyards(ship *model.Ship) error { if len(shipyards) == 0 { return nil } - waypoint, err := a.client.GetWaypoint(ship.Nav.WaypointSymbol, a.db) + waypoint, err := a.client.GetWaypoint(ship.Nav.WaypointSymbol) if err != nil { return fmt.Errorf("failed to get nav waypoint %s: %w", ship.Nav.WaypointSymbol, err) } waypoints := make([]model.Waypoint, 0) for i := range shipyards { - waypoint, err := a.client.GetWaypoint(shipyards[i].Symbol, a.db) + waypoint, err := a.client.GetWaypoint(shipyards[i].Symbol) if err != nil { return fmt.Errorf("failed to get waypoint %s: %w", shipyards[i].Symbol, err) } waypoints = append(waypoints, *waypoint) } sortByDistanceFrom(waypoint, waypoints) - if err := a.client.Navigate(ship, waypoints[0].Symbol, a.db); err != nil { + if err := a.client.Navigate(ship, waypoints[0].Symbol); err != nil { return fmt.Errorf("failed to navigate to %s: %w", waypoints[0].Symbol, err) } - if _, err := a.client.GetShipyard(waypoints[0].Symbol, a.db); err != nil { + if _, err := a.client.GetShipyard(waypoints[0].Symbol); err != nil { return fmt.Errorf("failed to get shipyard %s: %w", waypoints[0].Symbol, err) } // If this waypoint is also a marketplace, get its data for _, trait := range waypoints[0].Traits { if trait.Symbol == "MARKETPLACE" { - if _, err := a.client.GetMarket(waypoints[0].Symbol, a.db); err != nil { + if _, err := a.client.GetMarket(waypoints[0].Symbol); err != nil { return fmt.Errorf("failed to get market %s: %w", waypoints[0].Symbol, err) } break diff --git a/golang/pkg/api/client.go b/golang/pkg/api/client.go index 2ea555e..e6a1883 100644 --- a/golang/pkg/api/client.go +++ b/golang/pkg/api/client.go @@ -6,10 +6,13 @@ import ( "net/http" "net/url" "time" + + "git.adyxax.org/adyxax/spacetraders/golang/pkg/database" ) type Client struct { baseURI *url.URL + db *database.DB requestsChannel chan *Request ctx context.Context headers *http.Header @@ -17,7 +20,7 @@ type Client struct { pq *PriorityQueue } -func NewClient(ctx context.Context) *Client { +func NewClient(ctx context.Context, db *database.DB) *Client { baseURI, err := url.Parse("https://api.spacetraders.io/v2/") if err != nil { panic("baseURI failed to parse") @@ -26,6 +29,7 @@ func NewClient(ctx context.Context) *Client { heap.Init(&pq) client := &Client{ baseURI: baseURI, + db: db, requestsChannel: make(chan *Request), ctx: ctx, headers: &http.Header{ diff --git a/golang/pkg/api/contracts.go b/golang/pkg/api/contracts.go index 6a9a28a..fd20cce 100644 --- a/golang/pkg/api/contracts.go +++ b/golang/pkg/api/contracts.go @@ -5,11 +5,10 @@ import ( "net/url" "path" - "git.adyxax.org/adyxax/spacetraders/golang/pkg/database" "git.adyxax.org/adyxax/spacetraders/golang/pkg/model" ) -func (c *Client) Accept(contract *model.Contract, db *database.DB) error { +func (c *Client) Accept(contract *model.Contract) error { if contract.Accepted { return nil } @@ -22,7 +21,7 @@ func (c *Client) Accept(contract *model.Contract, db *database.DB) error { if err := c.Send("POST", &uriRef, nil, &response); err != nil { return fmt.Errorf("failed API request: %w", err) } - if err := db.SaveAgent(response.Agent); err != nil { + if err := c.db.SaveAgent(response.Agent); err != nil { return fmt.Errorf("failed to save agent: %w", err) } contract.Accepted = response.Contract.Accepted @@ -30,7 +29,7 @@ func (c *Client) Accept(contract *model.Contract, db *database.DB) error { return nil } -func (c *Client) Deliver(contract *model.Contract, ship *model.Ship, db *database.DB) error { +func (c *Client) Deliver(contract *model.Contract, ship *model.Ship) error { deliver := contract.Terms.Deliver[0] var units int for _, cargoItem := range ship.Cargo.Inventory { @@ -58,7 +57,7 @@ func (c *Client) Deliver(contract *model.Contract, ship *model.Ship, db *databas return nil } -func (c *Client) Fulfill(contract *model.Contract, db *database.DB) error { +func (c *Client) Fulfill(contract *model.Contract) error { if contract.Fulfilled { return nil } @@ -71,7 +70,7 @@ func (c *Client) Fulfill(contract *model.Contract, db *database.DB) error { if err := c.Send("POST", &uriRef, nil, &response); err != nil { return fmt.Errorf("failed API request: %w", err) } - if err := db.SaveAgent(response.Agent); err != nil { + if err := c.db.SaveAgent(response.Agent); err != nil { return fmt.Errorf("failed to save agent: %w", err) } contract.Fulfilled = response.Contract.Fulfilled diff --git a/golang/pkg/api/ships.go b/golang/pkg/api/ships.go index 21676d1..c3b7074 100644 --- a/golang/pkg/api/ships.go +++ b/golang/pkg/api/ships.go @@ -6,7 +6,6 @@ import ( "path" "time" - "git.adyxax.org/adyxax/spacetraders/golang/pkg/database" "git.adyxax.org/adyxax/spacetraders/golang/pkg/model" ) @@ -35,7 +34,7 @@ func (c *Client) MyShips() ([]model.Ship, error) { return ships, nil } -func (c *Client) Navigate(s *model.Ship, waypointSymbol string, db *database.DB) error { +func (c *Client) Navigate(s *model.Ship, waypointSymbol string) error { if s.Nav.WaypointSymbol == waypointSymbol { return nil } @@ -96,7 +95,7 @@ func (c *Client) orbit(s *model.Ship) error { return nil } -func (c *Client) Purchase(s *model.Ship, cargoItem string, units int, db *database.DB) error { +func (c *Client) Purchase(s *model.Ship, cargoItem string, units int) error { if err := c.dock(s); err != nil { return fmt.Errorf("failed to dock: %w", err) } @@ -114,11 +113,11 @@ func (c *Client) Purchase(s *model.Ship, cargoItem string, units int, db *databa if err := c.Send("POST", &uriRef, purchaseRequest{cargoItem, units}, &response); err != nil { return fmt.Errorf("failed API request: %w", err) } - if err := db.SaveAgent(response.Agent); err != nil { + if err := c.db.SaveAgent(response.Agent); err != nil { return fmt.Errorf("failed to save agent: %w", err) } s.Cargo = response.Cargo - if err := db.AppendTransaction(response.Transaction); err != nil { + if err := c.db.AppendTransaction(response.Transaction); err != nil { return fmt.Errorf("failed to append transaction: %w", err) } return nil @@ -141,17 +140,17 @@ func (c *Client) refuel(s *model.Ship, db *database.DB) error { if err := c.Send("POST", &uriRef, nil, &response); err != nil { return fmt.Errorf("failed API request: %w", err) } - if err := db.SaveAgent(response.Agent); err != nil { + if err := c.db.SaveAgent(response.Agent); err != nil { return fmt.Errorf("failed to save agent: %w", err) } s.Fuel = response.Fuel - if err := db.AppendTransaction(response.Transaction); err != nil { + if err := c.db.AppendTransaction(response.Transaction); err != nil { return fmt.Errorf("failed to append transaction: %w", err) } return nil } -func (c *Client) Sell(s *model.Ship, cargoItem string, units int, db *database.DB) error { +func (c *Client) Sell(s *model.Ship, cargoItem string, units int) error { if err := c.dock(s); err != nil { return fmt.Errorf("failed to dock: %w", err) } @@ -169,11 +168,11 @@ func (c *Client) Sell(s *model.Ship, cargoItem string, units int, db *database.D if err := c.Send("POST", &uriRef, sellRequest{cargoItem, units}, &response); err != nil { return fmt.Errorf("failed API request: %w", err) } - if err := db.SaveAgent(response.Agent); err != nil { + if err := c.db.SaveAgent(response.Agent); err != nil { return fmt.Errorf("failed to save agent: %w", err) } s.Cargo = response.Cargo - if err := db.AppendTransaction(response.Transaction); err != nil { + if err := c.db.AppendTransaction(response.Transaction); err != nil { return fmt.Errorf("failed to append transaction: %w", err) } return nil diff --git a/golang/pkg/api/systems.go b/golang/pkg/api/systems.go index 58770cb..2cdfeca 100644 --- a/golang/pkg/api/systems.go +++ b/golang/pkg/api/systems.go @@ -5,12 +5,11 @@ import ( "net/url" "path" - "git.adyxax.org/adyxax/spacetraders/golang/pkg/database" "git.adyxax.org/adyxax/spacetraders/golang/pkg/model" ) -func (c *Client) GetMarket(waypointSymbol string, db *database.DB) (*model.Market, error) { - if market, err := db.LoadMarket(waypointSymbol); err == nil && market != nil { +func (c *Client) GetMarket(waypointSymbol string) (*model.Market, error) { + if market, err := c.db.LoadMarket(waypointSymbol); err == nil && market != nil { // TODO check last updated time return market, nil } @@ -20,14 +19,14 @@ func (c *Client) GetMarket(waypointSymbol string, db *database.DB) (*model.Marke if err := c.Send("GET", &uriRef, nil, &market); err != nil { return nil, fmt.Errorf("failed API request: %w", err) } - if err := db.SaveMarket(&market); err != nil { + if err := c.db.SaveMarket(&market); err != nil { return nil, fmt.Errorf("failed to save market %s: %w", market.Symbol, err) } return &market, nil } -func (c *Client) GetShipyard(waypointSymbol string, db *database.DB) (*model.Shipyard, error) { - if shipyard, err := db.LoadShipyard(waypointSymbol); err == nil && shipyard != nil && +func (c *Client) GetShipyard(waypointSymbol string) (*model.Shipyard, error) { + if shipyard, err := c.db.LoadShipyard(waypointSymbol); err == nil && shipyard != nil && (shipyard.Ships != nil) { // TODO || !IsThereAShipAtWaypoint(waypoint)) { // TODO check last updated time return shipyard, nil @@ -38,14 +37,14 @@ func (c *Client) GetShipyard(waypointSymbol string, db *database.DB) (*model.Shi if err := c.Send("GET", &uriRef, nil, &shipyard); err != nil { return nil, fmt.Errorf("failed API request: %w", err) } - if err := db.SaveShipyard(&shipyard); err != nil { + if err := c.db.SaveShipyard(&shipyard); err != nil { return nil, fmt.Errorf("failed to save shipyard %s: %w", shipyard.Symbol, err) } return &shipyard, nil } -func (c *Client) GetSystem(systemSymbol string, db *database.DB) (*model.System, error) { - if system, err := db.LoadSystem(systemSymbol); err == nil && system != nil { +func (c *Client) GetSystem(systemSymbol string) (*model.System, error) { + if system, err := c.db.LoadSystem(systemSymbol); err == nil && system != nil { return system, nil } uriRef := url.URL{Path: path.Join("systems", systemSymbol)} @@ -53,14 +52,14 @@ func (c *Client) GetSystem(systemSymbol string, db *database.DB) (*model.System, if err := c.Send("GET", &uriRef, nil, &system); err != nil { return nil, fmt.Errorf("failed API request: %w", err) } - if err := db.SaveSystem(&system); err != nil { + if err := c.db.SaveSystem(&system); err != nil { return nil, fmt.Errorf("failed to save system %s: %w", system.Symbol, err) } return &system, nil } -func (c *Client) GetWaypoint(waypointSymbol string, db *database.DB) (*model.Waypoint, error) { - if waypoint, err := db.LoadWaypoint(waypointSymbol); err == nil && waypoint != nil { +func (c *Client) GetWaypoint(waypointSymbol string) (*model.Waypoint, error) { + if waypoint, err := c.db.LoadWaypoint(waypointSymbol); err == nil && waypoint != nil { // TODO check last updated time return waypoint, nil } @@ -70,14 +69,14 @@ func (c *Client) GetWaypoint(waypointSymbol string, db *database.DB) (*model.Way if err := c.Send("GET", &uriRef, nil, &waypoint); err != nil { return nil, fmt.Errorf("failed API request: %w", err) } - if err := db.SaveWaypoint(&waypoint); err != nil { + if err := c.db.SaveWaypoint(&waypoint); err != nil { return nil, fmt.Errorf("failed to save waypoint %s: %w", waypoint.Symbol, err) } return &waypoint, nil } -func (c *Client) ListWaypointsInSystem(systemSymbol string, db *database.DB) ([]model.Waypoint, error) { - if waypoints, err := db.LoadWaypointsInSystem(systemSymbol); err == nil && waypoints != nil { +func (c *Client) ListWaypointsInSystem(systemSymbol string) ([]model.Waypoint, error) { + if waypoints, err := c.db.LoadWaypointsInSystem(systemSymbol); err == nil && waypoints != nil { // TODO check last updated time return waypoints, nil } @@ -87,7 +86,7 @@ func (c *Client) ListWaypointsInSystem(systemSymbol string, db *database.DB) ([] return nil, fmt.Errorf("failed API request: %w", err) } for _, waypoint := range waypoints { - if err := db.SaveWaypoint(&waypoint); err != nil { + if err := c.db.SaveWaypoint(&waypoint); err != nil { return nil, fmt.Errorf("failed to save waypoint %s: %w", waypoint.Symbol, err) } } |