add support for public preview api to 'foasclo version' command

Author: andreaangiolillo

tools/cli/internal/apiversion/version.go

@@ -11,6 +11,7 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
+
 package apiversion
 
 import (
@@ -23,15 +24,18 @@ import (
 )
 
 type APIVersion struct {
-	version     string
-	versionDate time.Time
+	version          string
+	stabilityVersion string
+	versionDate      time.Time
 }
 
 const (
-	dateFormat = "2006-01-02"
+	dateFormat            = "2006-01-02"
+	StableStabilityLevel  = "STABLE"
+	PreviewStabilityLevel = "PREVIEW"
 )
 
-var ContentPattern = regexp.MustCompile(`application/vnd\.atlas\.(\d{4})-(\d{2})-(\d{2})\+(.+)`)
+var ContentPattern = regexp.MustCompile(`application/vnd\.atlas\.((\d{4})-(\d{2})-(\d{2})|preview)\+(.+)`)
 
 // Option is a function that sets a value on the APIVersion.
 type Option func(v *APIVersion) error
@@ -66,10 +70,31 @@ func WithDate(date time.Time) Option {
 	return func(v *APIVersion) error {
 		v.version = date.Format(dateFormat)
 		v.versionDate = date
+		v.stabilityVersion = StableStabilityLevel
+		return nil
+	}
+}
+
+// WithStabilityLevel sets the version and stability level on the APIVersion.
+func WithStabilityLevel(version, stabilityLevel string) Option {
+	return func(v *APIVersion) error {
+		v.stabilityVersion = stabilityLevel
+		v.version = version
+		if stabilityLevel != StableStabilityLevel {
+			return nil
+		}
+
+		versionDate, err := DateFromVersion(version)
+		if err != nil {
+			return err
+		}
+
+		v.versionDate = versionDate
 		return nil
 	}
 }
 
+// WithContent returns an Option to generate a new APIVersion given the contentType.
 func WithContent(contentType string) Option {
 	return func(v *APIVersion) error {
 		version, err := Parse(contentType)
@@ -78,6 +103,12 @@ func WithContent(contentType string) Option {
 		}
 
 		v.version = version
+		v.stabilityVersion = StableStabilityLevel
+		if version == PreviewStabilityLevel {
+			v.stabilityVersion = PreviewStabilityLevel
+			return nil
+		}
+
 		v.versionDate, err = DateFromVersion(version)
 		if err != nil {
 			return err
@@ -124,7 +155,12 @@ func Parse(contentType string) (string, error) {
 	if matches == nil {
 		return "", fmt.Errorf("invalid content type: %s", contentType)
 	}
-	return fmt.Sprintf("%s-%s-%s", matches[1], matches[2], matches[3]), nil
+
+	if len(matches) == 3 {
+		return fmt.Sprintf("%s-%s-%s", matches[1], matches[2], matches[3]), nil
+	}
+
+	return matches[1], nil
 }
 
 // FindLatestContentVersionMatched finds the latest content version that matches the requested version.

tools/cli/internal/cli/flag/flag.go

@@ -47,4 +47,6 @@ const (
 	ChannelIDShort           = "c"
 	From                     = "from"
 	To                       = "to"
+	StabilityLevel           = "stability-level"
+	StabilityLevelShort      = "sl"
 )

tools/cli/internal/cli/usage/usage.go

@@ -37,4 +37,5 @@ const (
 	SlackChannelID      = "Slack Channel ID."
 	From                = "Date in the format YYYY-MM-DD that indicates the start of a date range"
 	To                  = "Date in the format YYYY-MM-DD that indicates the end of a date range"
+	StabilityLevel      = "Stability level related to the API Version. Valid values: [STABLE, PREVIEW]"
 )

tools/cli/internal/cli/versions/versions.go

@@ -17,6 +17,7 @@ package versions
 import (
 	"encoding/json"
 	"fmt"
+	"github.com/mongodb/openapi/tools/cli/internal/apiversion"
 	"strings"
 
 	"github.com/mongodb/openapi/tools/cli/internal/cli/flag"
@@ -28,11 +29,12 @@ import (
 )
 
 type Opts struct {
-	fs         afero.Fs
-	basePath   string
-	outputPath string
-	format     string
-	env        string
+	fs             afero.Fs
+	basePath       string
+	outputPath     string
+	format         string
+	env            string
+	stabilityLevel string
 }
 
 func (o *Opts) Run() error {
@@ -43,11 +45,7 @@ func (o *Opts) Run() error {
 	}
 
 	var versions []string
-	if o.env == "" {
-		versions, err = openapi.ExtractVersions(specInfo.Spec)
-	} else {
-		versions, err = openapi.ExtractVersionsWithEnv(specInfo.Spec, o.env)
-	}
+	versions, err = openapi.ExtractVersionsWithEnv(specInfo.Spec, o.env)
 
 	if err != nil {
 		return err
@@ -57,7 +55,7 @@ func (o *Opts) Run() error {
 		return fmt.Errorf("no versions found in the OpenAPI specification")
 	}
 
-	bytes, err := o.getVersionBytes(versions)
+	bytes, err := o.versionsAsBytes(versions)
 	if err != nil {
 		return err
 	}
@@ -70,7 +68,7 @@ func (o *Opts) Run() error {
 	return nil
 }
 
-func (o *Opts) getVersionBytes(versions []string) ([]byte, error) {
+func (o *Opts) versionsAsBytes(versions []string) ([]byte, error) {
 	data, err := json.MarshalIndent(versions, "", "  ")
 	if err != nil {
 		return nil, err
@@ -94,16 +92,21 @@ func (o *Opts) getVersionBytes(versions []string) ([]byte, error) {
 }
 
 func (o *Opts) PreRunE(_ []string) error {
+	o.stabilityLevel = strings.ToUpper(o.stabilityLevel)
+	if o.stabilityLevel != "" && o.stabilityLevel != apiversion.PreviewStabilityLevel && o.stabilityLevel != apiversion.StableStabilityLevel {
+		return fmt.Errorf("stability level must be %q or %q, got %q", apiversion.PreviewStabilityLevel, apiversion.StableStabilityLevel, o.stabilityLevel)
+	}
+
 	if o.basePath == "" {
-		return fmt.Errorf("no OAS detected. Please, use the flag %s to include the base OAS", flag.Base)
+		return fmt.Errorf("no OAS detected. Please, use the flag %q to include the base OAS", flag.Base)
 	}
 
 	if o.outputPath != "" && !strings.Contains(o.outputPath, ".json") && !strings.Contains(o.outputPath, ".yaml") {
-		return fmt.Errorf("output file must be either a JSON or YAML file, got %s", o.outputPath)
+		return fmt.Errorf("output file must be either a JSON or YAML file, got %q", o.outputPath)
 	}
 
 	if o.format != "json" && o.format != "yaml" {
-		return fmt.Errorf("output format must be either 'json' or 'yaml', got %s", o.format)
+		return fmt.Errorf("output format must be either 'json' or 'yaml', got %q", o.format)
 	}
 
 	return nil
@@ -117,9 +120,10 @@ func Builder() *cobra.Command {
 	}
 
 	cmd := &cobra.Command{
-		Use:   "versions -s spec ",
-		Short: "Get a list of versions from an OpenAPI specification.",
-		Args:  cobra.NoArgs,
+		Use:     "versions -s spec ",
+		Aliases: []string{"versions list", "versions ls"},
+		Short:   "Get a list of versions from an OpenAPI specification.",
+		Args:    cobra.NoArgs,
 		PreRunE: func(_ *cobra.Command, args []string) error {
 			return opts.PreRunE(args)
 		},
@@ -130,7 +134,9 @@ func Builder() *cobra.Command {
 
 	cmd.Flags().StringVarP(&opts.basePath, flag.Spec, flag.SpecShort, "", usage.Spec)
 	cmd.Flags().StringVar(&opts.env, flag.Environment, "", usage.Environment)
+	cmd.Flags().StringVarP(&opts.stabilityLevel, flag.StabilityLevel, flag.StabilityLevelShort, "", usage.StabilityLevel)
 	cmd.Flags().StringVarP(&opts.outputPath, flag.Output, flag.OutputShort, "", usage.Output)
 	cmd.Flags().StringVarP(&opts.format, flag.Format, flag.FormatShort, "json", usage.Format)
+
 	return cmd
 }

tools/cli/internal/openapi/versions.go

@@ -22,18 +22,24 @@ import (
 	"github.com/mongodb/openapi/tools/cli/internal/openapi/filter"
 )
 
+// ExtractVersionsWithEnv extracts API version Content Type strings from the given OpenAPI specification and environment.
+// When env is not set, the function returns the API Versions from all the environments.
 func ExtractVersionsWithEnv(oas *openapi3.T, env string) ([]string, error) {
+	if env == "" {
+		return extractVersions(oas)
+	}
+
 	// We need to remove the version that are hidden for the given environment
 	doc, err := filter.ApplyFilters(oas, filter.NewMetadata(nil, env), filter.FiltersToGetVersions)
 	if err != nil {
-		return nil, nil
+		return nil, err
 	}
 
-	return ExtractVersions(doc)
+	return extractVersions(doc)
 }
 
-// ExtractVersions extracts version strings from an OpenAPI specification.
-func ExtractVersions(oas *openapi3.T) ([]string, error) {
+// extractVersions extracts version strings from an OpenAPI specification.
+func extractVersions(oas *openapi3.T) ([]string, error) {
 	versions := make(map[string]struct{})
 	for _, pathItem := range oas.Paths.Map() {
 		if pathItem == nil {