add support for team

Author: andreaangiolillo

tools/cli/internal/openapi/sunset.go

@@ -22,13 +22,15 @@ import (
 const (
 	sunsetExtensionName     = "x-sunset"
 	apiVersionExtensionName = "x-xgen-version"
+	teamExtensionName       = "x-xgen-owner-team"
 )
 
 type Sunset struct {
 	Operation  string `json:"http_method" yaml:"http_method"`
 	Path       string `json:"path" yaml:"path"`
 	Version    string `json:"version" yaml:"version"`
 	SunsetDate string `json:"sunset_date" yaml:"sunset_date"`
+	Team       string `json:"team" yaml:"team"`
 }
 
 func NewSunsetListFromSpec(spec *load.SpecInfo) []*Sunset {
@@ -37,6 +39,7 @@ func NewSunsetListFromSpec(spec *load.SpecInfo) []*Sunset {
 
 	for path, pathBody := range paths.Map() {
 		for operationName, operationBody := range pathBody.Operations() {
+			teamName := newTeamNameFromOperation(operationBody)
 			extensions := newExtensionsFrom2xxResponse(operationBody.Responses.Map())
 			if extensions == nil {
 				continue
@@ -57,6 +60,7 @@ func NewSunsetListFromSpec(spec *load.SpecInfo) []*Sunset {
 				Path:       path,
 				SunsetDate: sunsetExt.(string),
 				Version:    apiVersion.(string),
+				Team:       teamName,
 			}
 
 			sunsets = append(sunsets, &sunset)
@@ -66,6 +70,13 @@ func NewSunsetListFromSpec(spec *load.SpecInfo) []*Sunset {
 	return sunsets
 }
 
+func newTeamNameFromOperation(op *openapi3.Operation) string {
+	if value, ok := op.Extensions[teamExtensionName]; ok {
+		return value.(string)
+	}
+	return ""
+}
+
 func newExtensionsFrom2xxResponse(responsesMap map[string]*openapi3.ResponseRef) map[string]any {
 	if val, ok := responsesMap["200"]; ok {
 		return newExtensionsFromContent(val.Value.Content)

tools/cli/internal/openapi/sunset_test.go

@@ -74,13 +74,13 @@ func TestNewSunsetListFromSpec(t *testing.T) {
 			expected: nil,
 		},
 		{
-			name: "Multiple operations with extensions",
+			name: "201 operations with extensions",
 			specInfo: &load.SpecInfo{
 				Spec: &openapi3.T{
 					Paths: openapi3.NewPaths(
 						openapi3.WithPath("/example1", &openapi3.PathItem{
 							Get: &openapi3.Operation{
-								Responses: openapi3.NewResponses(openapi3.WithName("200", &openapi3.Response{
+								Responses: openapi3.NewResponses(openapi3.WithName("201", &openapi3.Response{
 									Content: openapi3.Content{
 										"application/json": &openapi3.MediaType{
 											Extensions: map[string]any{
@@ -92,20 +92,6 @@ func TestNewSunsetListFromSpec(t *testing.T) {
 								})),
 							},
 						}),
-						openapi3.WithPath("/example2", &openapi3.PathItem{
-							Post: &openapi3.Operation{
-								Responses: openapi3.NewResponses(openapi3.WithName("201", &openapi3.Response{
-									Content: openapi3.Content{
-										"application/json": &openapi3.MediaType{
-											Extensions: map[string]any{
-												sunsetExtensionName:     "2025-01-01",
-												apiVersionExtensionName: "v1.5",
-											},
-										},
-									},
-								})),
-							},
-						}),
 					),
 				},
 			},
@@ -116,12 +102,6 @@ func TestNewSunsetListFromSpec(t *testing.T) {
 					Version:    "v2.0",
 					SunsetDate: "2024-06-15",
 				},
-				{
-					Operation:  "POST",
-					Path:       "/example2",
-					Version:    "v1.5",
-					SunsetDate: "2025-01-01",
-				},
 			},
 		},
 	}