summaryrefslogtreecommitdiff
path: root/golang
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--golang/cmd/spacetraders/main.go2
-rw-r--r--golang/pkg/agent/contracting.go8
-rw-r--r--golang/pkg/agent/trading.go20
-rw-r--r--golang/pkg/agent/utils.go8
-rw-r--r--golang/pkg/agent/visit.go10
-rw-r--r--golang/pkg/api/client.go6
-rw-r--r--golang/pkg/api/contracts.go11
-rw-r--r--golang/pkg/api/ships.go19
-rw-r--r--golang/pkg/api/systems.go31
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)
}
}
generated by cgit v1.2.3 (git 2.39.1) at 2025-02-22 21:23:19 +0000