small bugfix

main.go

@@ -49,7 +49,7 @@ var (
 	lics     []license
 	errFound = errors.New("file found")
 	licre    = regexp.MustCompile("([Ll][Ii][Cc][Ee][Nn][Ss][Ee].*)")
-	pathre   = regexp.MustCompile("vendor/(.*/(.*/.*))/")
+	pathre   = regexp.MustCompile("vendor/(.*/(.*/.*))/|vendor/(.*/.*/)")
 )
 
 func init() {
@@ -120,11 +120,21 @@ func walker(path string, info os.FileInfo, err error) error {
 		return err
 	}
 	match := pathre.FindAllStringSubmatch(path, -1)
-	lics = append(lics, license{
-		Path: match[0][2],
-		URL:  match[0][1],
-		Text: string(text),
-	})
+	if len(match) >= 1 {
+		if len(match[0]) >= 4 {
+			if match[0][1] == "" {
+				match[0][1] = match[0][3]
+				match[0][2] = match[0][3]
+			}
+			lics = append(lics, license{
+				Path: match[0][2],
+				URL:  match[0][1],
+				Text: string(text),
+			})
+		}
+	} else {
+		return nil
+	}
 	return errFound
 }