55 "strconv"
66 "time"
77
8+ "github.com/hashicorp/terraform-plugin-framework/attr"
89 "github.com/hashicorp/terraform-plugin-framework/datasource"
910 "github.com/hashicorp/terraform-plugin-framework/datasource/schema"
1011 "github.com/hashicorp/terraform-plugin-framework/types"
@@ -39,6 +40,7 @@ type projectDataSourceModel struct {
3940 GrafanaFolderUID types.String `tfsdk:"grafana_folder_uid"`
4041 Created types.String `tfsdk:"created"`
4142 Updated types.String `tfsdk:"updated"`
43+ AllowedLoadZones types.List `tfsdk:"allowed_load_zones"`
4244}
4345
4446// projectDataSource is the data source implementation.
@@ -80,6 +82,11 @@ func (d *projectDataSource) Schema(_ context.Context, _ datasource.SchemaRequest
8082 Description : "The date when the project was last updated." ,
8183 Computed : true ,
8284 },
85+ "allowed_load_zones" : schema.ListAttribute {
86+ Description : "List of allowed k6 load zone IDs for this project." ,
87+ Computed : true ,
88+ ElementType : types .StringType ,
89+ },
8390 },
8491 }
8592}
@@ -122,6 +129,27 @@ func (d *projectDataSource) Read(ctx context.Context, req datasource.ReadRequest
122129 state .Created = types .StringValue (p .GetCreated ().Format (time .RFC3339Nano ))
123130 state .Updated = types .StringValue (p .GetUpdated ().Format (time .RFC3339Nano ))
124131
132+ // Get allowed load zones
133+ allowedZones , err := d .getAllowedLoadZones (ctx , p .GetId ())
134+ if err != nil {
135+ resp .Diagnostics .AddError (
136+ "Error reading allowed load zones" ,
137+ "Could not read allowed load zones for k6 project: " + err .Error (),
138+ )
139+ return
140+ }
141+
142+ // Convert to types.List
143+ var zoneValues []attr.Value
144+ for _ , zone := range allowedZones {
145+ zoneValues = append (zoneValues , types .StringValue (zone ))
146+ }
147+ state .AllowedLoadZones , diags = types .ListValue (types .StringType , zoneValues )
148+ resp .Diagnostics .Append (diags ... )
149+ if resp .Diagnostics .HasError () {
150+ return
151+ }
152+
125153 diags = resp .State .Set (ctx , & state )
126154 resp .Diagnostics .Append (diags ... )
127155}
@@ -132,3 +160,23 @@ func handleGrafanaFolderUID(grafanaFolderUID k6.NullableString) types.String {
132160 }
133161 return types .StringValue (* grafanaFolderUID .Get ())
134162}
163+
164+ // getAllowedLoadZones retrieves the allowed load zones for a project
165+ // Returns k6_load_zone_ids directly from the API response
166+ func (d * projectDataSource ) getAllowedLoadZones (ctx context.Context , projectID int32 ) ([]string , error ) {
167+ ctx = context .WithValue (ctx , k6 .ContextAccessToken , d .config .Token )
168+
169+ resp , _ , err := d .client .LoadZonesAPI .ProjectsAllowedLoadZonesRetrieve (ctx , projectID ).
170+ XStackId (d .config .StackID ).
171+ Execute ()
172+ if err != nil {
173+ return nil , err
174+ }
175+
176+ var k6LoadZoneIds []string
177+ for _ , zone := range resp .GetValue () {
178+ k6LoadZoneIds = append (k6LoadZoneIds , zone .GetK6LoadZoneId ())
179+ }
180+
181+ return k6LoadZoneIds , nil
182+ }
0 commit comments