oapi-codegen · jamietanna · Aug 10, 2022 · May 4, 2024
@@ -14,6 +14,7 @@
 package main
 
 import (
+	"context"
 	"flag"
 	"fmt"
 	"io/ioutil"
@@ -170,6 +171,11 @@ func main() {
 		errExit("error loading swagger spec in %s\n: %s", flag.Arg(0), err)
 	}
 
+	err = swagger.Validate(context.Background())
+	if err != nil {
+		errExit("the swagger spec in %s is not valid\n: %s", flag.Arg(0), err)
+	}
+
 	code, err := codegen.Generate(swagger, opts.Configuration)
 	if err != nil {
 		errExit("error generating code: %s\n", err)

@@ -0,0 +1,121 @@
+// Copyright 2022 Dave Shanley / Quobix
+// SPDX-License-Identifier: MIT
+
+package main
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/daveshanley/vacuum/model"
+	"github.com/daveshanley/vacuum/plugin"
+	"github.com/daveshanley/vacuum/rulesets"
+	"github.com/dustin/go-humanize"
+	"github.com/pb33f/libopenapi/index"
+	"github.com/pterm/pterm"
+)
+
+// BuildRuleSetFromUserSuppliedSet creates a ready to run ruleset, augmented or provided by a user
+// configured ruleset. This ruleset could be lifted directly from a Spectral configuration.
+func BuildRuleSetFromUserSuppliedSet(rsBytes []byte, rs rulesets.RuleSets) (*rulesets.RuleSet, error) {
+
+	// load in our user supplied ruleset and try to validate it.
+	userRS, userErr := rulesets.CreateRuleSetFromData(rsBytes)
+	if userErr != nil {
+		pterm.Error.Printf("Unable to parse ruleset file: %s\n", userErr.Error())
+		pterm.Println()
+		return nil, userErr
+
+	}
+	return rs.GenerateRuleSetFromSuppliedRuleSet(userRS), nil
+}
+
+// RenderTimeAndFiles  will render out the time taken to process a specification, and the size of the file in kb.
+// it will also render out how many files were processed.
+func RenderTimeAndFiles(timeFlag bool, duration time.Duration, fileSize int64, totalFiles int) {
+	if timeFlag {
+		pterm.Println()
+		l := "milliseconds"
+		d := fmt.Sprintf("%d", duration.Milliseconds())
+		if duration.Milliseconds() > 1000 {
+			l = "seconds"
+			d = humanize.FormatFloat("##.##", duration.Seconds())
+		}
+		pterm.Info.Println(fmt.Sprintf("vacuum took %s %s to lint %s across %d files", d, l,
+			index.HumanFileSize(float64(fileSize)), totalFiles))
+		pterm.Println()
+	}
+}
+
+// RenderTime will render out the time taken to process a specification, and the size of the file in kb.
+func RenderTime(timeFlag bool, duration time.Duration, fi int64) {
+	if timeFlag {
+		pterm.Println()
+		if (fi / 1000) <= 1024 {
+			pterm.Info.Println(fmt.Sprintf("vacuum took %d milliseconds to lint %dkb", duration.Milliseconds(), fi/1000))
+		} else {
+			pterm.Info.Println(fmt.Sprintf("vacuum took %d milliseconds to lint %dmb", duration.Milliseconds(), fi/1000000))
+		}
+		pterm.Println()
+	}
+}
+
+func PrintBanner() {
+	pterm.Println()
+
+	//_ = pterm.DefaultBigText.WithLetters(
+	//	putils.LettersFromString(pterm.LightMagenta("vacuum"))).Render()
+	banner := `
+██╗   ██╗ █████╗  ██████╗██╗   ██╗██╗   ██╗███╗   ███╗
+██║   ██║██╔══██╗██╔════╝██║   ██║██║   ██║████╗ ████║
+██║   ██║███████║██║     ██║   ██║██║   ██║██╔████╔██║
+╚██╗ ██╔╝██╔══██║██║     ██║   ██║██║   ██║██║╚██╔╝██║
+ ╚████╔╝ ██║  ██║╚██████╗╚██████╔╝╚██████╔╝██║ ╚═╝ ██║
+  ╚═══╝  ╚═╝  ╚═╝ ╚═════╝ ╚═════╝  ╚═════╝ ╚═╝     ╚═╝
+`
+
+	pterm.Println(pterm.LightMagenta(banner))
+	pterm.Println()
+	// pterm.Printf("version: %s | compiled: %s\n", pterm.LightGreen(Version), pterm.LightGreen(Date))
+	pterm.Println(pterm.Cyan("🔗 https://quobix.com/vacuum | https://github.com/daveshanley/vacuum"))
+	pterm.Println()
+	pterm.Println()
+}
+
+// LoadCustomFunctions will scan for (and load) custom functions defined as vacuum plugins.
+func LoadCustomFunctions(functionsFlag string, silence bool) (map[string]model.RuleFunction, error) {
+	// check custom functions
+	if functionsFlag != "" {
+		pm, err := plugin.LoadFunctions(functionsFlag, silence)
+		if err != nil {
+			pterm.Error.Printf("Unable to open custom functions: %v\n", err)
+			pterm.Println()
+			return nil, err
+		}
+		pterm.Info.Printf("Loaded %d custom function(s) successfully.\n", pm.LoadedFunctionCount())
+		return pm.GetCustomFunctions(), nil
+	}
+	return nil, nil
+}
+
+func CheckFailureSeverity(failSeverityFlag string, errors int, warnings int, informs int) error {
+	if failSeverityFlag != model.SeverityError {
+		switch failSeverityFlag {
+		case model.SeverityWarn:
+			if warnings > 0 || errors > 0 {
+				return fmt.Errorf("failed with %d errors and %d warnings", errors, warnings)
+			}
+		case model.SeverityInfo:
+			if informs > 0 || warnings > 0 || errors > 0 {
+				return fmt.Errorf("failed with %d errors, %d warnings and %d informs",
+					errors, warnings, informs)
+			}
+			return nil
+		}
+	} else {
+		if errors > 0 {
+			return fmt.Errorf("failed with %d errors", errors)
+		}
+	}
+	return nil
+}