Merge pull request #552 from tdakkota/fix/clean-ref-correctly

fix(gen): clean remote references correctly

gen/gen_responses.go

@@ -157,7 +157,7 @@ func (g *Generator) responseToIR(
 			return r, nil
 		}
 
-		n, err := pascal(strings.TrimPrefix(ref, "#/components/responses/"))
+		n, err := pascal(cleanRef(ref))
 		if err != nil {
 			return nil, errors.Wrapf(err, "response name: %q", ref)
 		}

gen/names.go

@@ -8,6 +8,33 @@ import (
 	"github.com/go-faster/errors"
 )
 
+func cleanRef(ref string) string {
+	_, result, ok := strings.Cut(ref, "#")
+	if !ok {
+		result = ref
+	}
+
+	for _, prefix := range []string{
+		"/components/schemas/",
+		"/components/responses/",
+		"/components/parameters/",
+		"/components/examples/",
+		"/components/requestBodies/",
+		"/components/headers/",
+		"/components/securitySchemes/",
+		"/components/links/",
+		"/components/callbacks/",
+		"/components/pathItems/",
+	} {
+		// Do not try to trim all prefixes, do it only once.
+		if strings.HasPrefix(result, prefix) {
+			result = strings.TrimPrefix(result, prefix)
+			break
+		}
+	}
+	return result
+}
+
 type nameGen struct {
 	parts []string
 	src   []rune

gen/schema_gen.go

@@ -29,7 +29,7 @@ func newSchemaGen(filename string, lookupRef func(ref string) (*ir.Type, bool))
 		localRefs: map[string]*ir.Type{},
 		lookupRef: lookupRef,
 		nameRef: func(ref string) (string, error) {
-			name, err := pascal(strings.TrimPrefix(ref, "#/components/schemas/"))
+			name, err := pascal(cleanRef(ref))
 			if err != nil {
 				return "", err
 			}