fix matcher

minoritea · Aug 20, 2021 · 71260a8 · 71260a8
1 parent 9708a1c
commit 71260a8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -39,16 +39,16 @@ func run() error {
 	return nil
 }
 
-var matcher = regexp.MustCompile(`^(FROM|from) \w+ (AS|as) (\w+)$`)
+var matcher = regexp.MustCompile(`^(FROM|from) ([\w:.-]+) (AS|as) (\w+)$`)
 
 func parse(r io.Reader) ([]string, error) {
 	scanner := bufio.NewScanner(r)
 	var result []string
 	for scanner.Scan() {
 		line := scanner.Bytes()
 		m := matcher.FindAllSubmatch(line, -1)
-		if len(m) > 0 && len(m[0]) > 3 {
-			result = append(result, string(m[0][3]))
+		if len(m) > 0 && len(m[0]) > 4 {
+			result = append(result, string(m[0][4]))
 		}
 	}
 	return result, scanner.Err()