fix nil check errors (#21)

projectdiscovery · Mar 10, 2023 · 9267fe5 · 9267fe5
1 parent 4ad5ff0
commit 9267fe5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cmd/docgen/dstdocgen/dst.go b/cmd/docgen/dstdocgen/dst.go
@@ -118,7 +118,9 @@ func process() error {
 			pkg:        pkg,
 			structName: *structure,
 		})
-		structures = append(structures, main)
+		if main != nil {
+			structures = append(structures, main)
+		}
 		structures = append(structures, extra...)
 	}
 	if len(structures) == 0 {
@@ -424,7 +426,7 @@ func collectFields(s *structType, collectOpts *collectStructOptions) (fields []*
 	var foundStructures []*structType
 
 	for _, f := range s.node.Fields.List {
-		if f.Tag == nil && len(f.Names) > 0 {
+		if f.Tag == nil || len(f.Names) < 1 {
 			continue
 		}
 		tag := reflect.StructTag(strings.Trim(f.Tag.Value, "`"))