Added capacity field to resources and effects query (#8)

cardinal/component/effect.go

@@ -13,8 +13,9 @@ func GetAllEffectTypes() []EffectType {
 }
 
 type Effect struct {
-	Type   EffectType `json:"type"`
-	Amount int        `json:"amount"`
+	Type     EffectType `json:"type"`
+	Amount   int        `json:"amount"`
+	Capacity int        `json:"capacity,omitempty"`
 }
 
 func (Effect) Name() string {

cardinal/component/farming.go

@@ -1,8 +1,9 @@
 package component
 
 type Farming struct {
-	Type  ResourceType `json:"type"`
-	Speed float32      `json:"speed"`
+	Type     ResourceType `json:"type"`
+	Speed    float32      `json:"speed"`
+	Capacity int          `json:"capacity,omitempty"`
 }
 
 func (Farming) Name() string {

cardinal/component/player_resources.go

@@ -3,7 +3,6 @@ package component
 type PlayerResources struct {
 	Resources []Resource `json:"resources"`
 	Effects   []Effect   `json:"effects"`
-	// Farming   []Farming  `json:"farming"`
 }
 
 func (PlayerResources) Name() string {

cardinal/component/resource.go

@@ -21,8 +21,9 @@ func GetAllResourceTypes() []ResourceType {
 }
 
 type Resource struct {
-	Type   ResourceType `json:"type"`
-	Amount float32      `json:"amount"`
+	Type     ResourceType `json:"type"`
+	Amount   float32      `json:"amount"`
+	Capacity int          `json:"capacity,omitempty"`
 }
 
 func (Resource) Name() string {

cardinal/query/building_info.go

@@ -10,25 +10,34 @@ type BuildingsInfoRequest struct {
 }
 
 type BuildingInfoResponse struct {
-	Building  comp.BuildingType `json:"building"`
-	TileType  comp.TileType     `json:"tileType"`
-	Resources []comp.Resource   `json:"resources"`
-	Effect    *comp.Effect      `json:"effect,omitempty"`
-	Farming   *comp.Farming     `json:"farming,omitempty"`
+	Building        comp.BuildingType `json:"building"`
+	TileType        comp.TileType     `json:"tileType"`
+	Resources       []comp.Resource   `json:"resources"`
+	Effect          *comp.Effect      `json:"effect,omitempty"`
+	Farming         *comp.Farming     `json:"farming,omitempty"`
+	UnitLimit       int               `json:"unitLimit,omitempty"`
+	StorageCapacity int               `json:"storageCapacity,omitempty"`
 }
 
 func AllBuildings(_ cardinal.WorldContext, _ *BuildingsInfoRequest) (*[]BuildingInfoResponse, error) {
+	return GetAllBuildings(), nil
+}
+
+func GetAllBuildings() *[]BuildingInfoResponse {
 	buildings := make([]BuildingInfoResponse, len(comp.GetAllBuildingTypes()))
 	for i, buildingType := range comp.GetAllBuildingTypes() {
 		buildingConf := comp.BuildingConfigs[buildingType]
 
 		buildings[i] = BuildingInfoResponse{
-			Building:  buildingType,
-			TileType:  buildingConf.TileType,
-			Resources: buildingConf.Resources,
-			Effect:    buildingConf.Effect,
-			Farming:   buildingConf.Farming,
+			Building:        buildingType,
+			TileType:        buildingConf.TileType,
+			Resources:       buildingConf.Resources,
+			Effect:          buildingConf.Effect,
+			Farming:         buildingConf.Farming,
+			UnitLimit:       buildingConf.UnitLimit,
+			StorageCapacity: buildingConf.StorageCapacity,
 		}
 	}
-	return &buildings, nil
+
+	return &buildings
 }

cardinal/query/player_resources.go

@@ -40,7 +40,6 @@ func PlayerResources(world cardinal.WorldContext, req *PlayerResourcesRequest) (
 	}
 
 	aggregatedfarmingSlice := make([]comp.Farming, 0, len(aggregatedFarmingMap))
-
 	for resourceType, speed := range aggregatedFarmingMap {
 		aggregatedfarmingSlice = append(aggregatedfarmingSlice, comp.Farming{
 			Type:  resourceType,
@@ -52,9 +51,31 @@ func PlayerResources(world cardinal.WorldContext, req *PlayerResourcesRequest) (
 		return nil, fmt.Errorf("error querying player %s resources", req.Nickname)
 	}
 
+	effectsMap := make(map[comp.EffectType]int)
+	capacity := 0
+	for _, building := range *GetAllBuildings() {
+		if building.Effect != nil {
+			effectsMap[building.Effect.Type] += building.Effect.Amount
+		}
+
+		capacity += building.StorageCapacity
+	}
+
+	resourcesResponse := make([]comp.Resource, 0, len(playerResources.Resources))
+	for _, resource := range playerResources.Resources {
+		resource.Capacity = capacity
+		resourcesResponse = append(resourcesResponse, resource)
+	}
+
+	effectsResponse := make([]comp.Effect, 0, len(playerResources.Effects))
+	for _, effect := range playerResources.Effects {
+		effect.Capacity = effectsMap[effect.Type]
+		effectsResponse = append(effectsResponse, effect)
+	}
+
 	return &PlayerResourcesResponse{
-		Resources: playerResources.Resources,
-		Effects:   playerResources.Effects,
+		Resources: resourcesResponse,
+		Effects:   effectsResponse,
 		Farming:   aggregatedfarmingSlice,
 	}, err
 }