|
1 | 1 | package baremetal |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "context" |
| 5 | + "errors" |
| 6 | + "fmt" |
| 7 | + "strings" |
| 8 | + |
4 | 9 | "github.com/fatih/color" |
| 10 | + "github.com/scaleway/scaleway-cli/v2/core" |
5 | 11 | "github.com/scaleway/scaleway-cli/v2/core/human" |
6 | 12 | "github.com/scaleway/scaleway-sdk-go/api/baremetal/v1" |
| 13 | + product_catalog "github.com/scaleway/scaleway-sdk-go/api/product_catalog/v2alpha1" |
7 | 14 | ) |
8 | 15 |
|
9 | 16 | var offerAvailabilityMarshalSpecs = human.EnumMarshalSpecs{ |
@@ -53,3 +60,78 @@ func listOfferMarshalerFunc(i any, opt *human.MarshalOpt) (string, error) { |
53 | 60 |
|
54 | 61 | return str, nil |
55 | 62 | } |
| 63 | + |
| 64 | +type customOffer struct { |
| 65 | + baremetal.Offer |
| 66 | + KgCo2Equivalent *float32 `json:"kg_co2_equivalent"` |
| 67 | + M3WaterUsage *float32 `json:"m3_water_usage"` |
| 68 | +} |
| 69 | + |
| 70 | +func serverOfferListBuilder(c *core.Command) *core.Command { |
| 71 | + c.View = &core.View{ |
| 72 | + Fields: []*core.ViewField{ |
| 73 | + {Label: "Disks", FieldName: "Disks"}, |
| 74 | + {Label: "CPUs", FieldName: "CPUs"}, |
| 75 | + {Label: "Memories", FieldName: "Memories"}, |
| 76 | + {Label: "Options", FieldName: "Options"}, |
| 77 | + {Label: "Bandwidth", FieldName: "Bandwidth"}, |
| 78 | + {Label: "PrivateBandwidth", FieldName: "PrivateBandwidth"}, |
| 79 | + {Label: "CO2 (kg)", FieldName: "KgCo2Equivalent"}, |
| 80 | + {Label: "Water (m³)", FieldName: "M3WaterUsage"}, |
| 81 | + }, |
| 82 | + } |
| 83 | + |
| 84 | + c.Interceptor = func(ctx context.Context, argsI any, runner core.CommandRunner) (any, error) { |
| 85 | + rawResp, err := runner(ctx, argsI) |
| 86 | + if err != nil { |
| 87 | + return nil, err |
| 88 | + } |
| 89 | + |
| 90 | + offers, ok := rawResp.([]*baremetal.Offer) |
| 91 | + for _, offer := range offers { |
| 92 | + fmt.Printf("Print value of offer name: %s\n", offer.Name) |
| 93 | + } |
| 94 | + |
| 95 | + if !ok { |
| 96 | + return nil, errors.New("unexpected type for offer response") |
| 97 | + } |
| 98 | + |
| 99 | + client := core.ExtractClient(ctx) |
| 100 | + productAPI := product_catalog.NewPublicCatalogAPI(client) |
| 101 | + environmentalImpact, _ := productAPI.ListPublicCatalogProducts( |
| 102 | + &product_catalog.PublicCatalogAPIListPublicCatalogProductsRequest{ |
| 103 | + ProductTypes: []product_catalog.ListPublicCatalogProductsRequestProductType{ |
| 104 | + product_catalog.ListPublicCatalogProductsRequestProductTypeElasticMetal, |
| 105 | + }, |
| 106 | + }, |
| 107 | + ) |
| 108 | + |
| 109 | + impactMap := make(map[string]*product_catalog.PublicCatalogProduct) |
| 110 | + for _, impact := range environmentalImpact.Products { |
| 111 | + if impact != nil { |
| 112 | + key := strings.TrimSpace(strings.TrimPrefix(impact.Product, "Elastic Metal ")) |
| 113 | + impactMap[key] = impact |
| 114 | + } |
| 115 | + } |
| 116 | + |
| 117 | + var customOfferRes []customOffer |
| 118 | + for _, offer := range offers { |
| 119 | + fmt.Printf("Print value of offer: %s\n", offer.Name) |
| 120 | + impact, ok := impactMap[offer.Name] |
| 121 | + if !ok || impact == nil { |
| 122 | + fmt.Printf("No environmental impact data found for offer: %s\n", offer.Name) |
| 123 | + |
| 124 | + continue |
| 125 | + } |
| 126 | + customOfferRes = append(customOfferRes, customOffer{ |
| 127 | + Offer: *offer, |
| 128 | + KgCo2Equivalent: impact.EnvironmentalImpactEstimation.KgCo2Equivalent, |
| 129 | + M3WaterUsage: impact.EnvironmentalImpactEstimation.M3WaterUsage, |
| 130 | + }) |
| 131 | + } |
| 132 | + |
| 133 | + return customOfferRes, nil |
| 134 | + } |
| 135 | + |
| 136 | + return c |
| 137 | +} |
0 commit comments