restore: @name parsing behavior

swaggo · Dec 17, 2024 · 41a57e9 · 41a57e9
1 parent 04b5bc8
commit 41a57e9
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/types.go b/types.go
@@ -48,6 +48,20 @@ func (t *TypeSpecDef) Name() string {
 func (t *TypeSpecDef) TypeName() string {
 	if ignoreNameOverride(t.TypeSpec.Name.Name) {
 		return t.TypeSpec.Name.Name[1:]
+	} else if t.TypeSpec.Comment != nil {
+		// get alias from comment '// @name '
+		const regexCaseInsensitive = "(?i)"
+		reTypeName, err := regexp.Compile(regexCaseInsensitive + `^@name\s+(\S+)`)
+		if err != nil {
+			panic(err)
+		}
+		for _, comment := range t.TypeSpec.Comment.List {
+			trimmedComment := strings.TrimSpace(strings.TrimLeft(comment.Text, "/"))
+			texts := reTypeName.FindStringSubmatch(trimmedComment)
+			if len(texts) > 1 {
+				return texts[1]
+			}
+		}
 	}
 
 	var names []string