forked from kubernetes/kubernetes
/
main.go
83 lines (75 loc) · 1.64 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// Copyright (c) 2015, Daniel Martí <mvdan@mvdan.cc>
// See LICENSE for licensing information
package main
import (
"bufio"
"flag"
"fmt"
"os"
"regexp"
"github.com/mvdan/xurls"
)
var (
matching = flag.String("m", "", "")
relaxed = flag.Bool("r", false, "")
)
func init() {
flag.Usage = func() {
p := func(format string, a ...interface{}) {
fmt.Fprintf(os.Stderr, format, a...)
}
p("Usage: xurls [-h] [files]\n\n")
p("If no files are given, it reads from standard input.\n\n")
p(" -m <regexp> only match urls whose scheme matches a regexp\n")
p(" example: 'https?://|mailto:'\n")
p(" -r also match urls without a scheme (relaxed)\n")
}
}
func scanPath(re *regexp.Regexp, path string) error {
r := os.Stdin
if path != "-" {
f, err := os.Open(path)
if err != nil {
return err
}
defer f.Close()
r = f
}
scanner := bufio.NewScanner(r)
scanner.Split(bufio.ScanWords)
for scanner.Scan() {
word := scanner.Text()
for _, match := range re.FindAllString(word, -1) {
fmt.Println(match)
}
}
return scanner.Err()
}
func main() {
flag.Parse()
if *relaxed && *matching != "" {
errExit(fmt.Errorf("-r and -m at the same time don't make much sense"))
}
re := xurls.Strict
if *relaxed {
re = xurls.Relaxed
} else if *matching != "" {
var err error
if re, err = xurls.StrictMatchingScheme(*matching); err != nil {
errExit(err)
}
}
args := flag.Args()
if len(args) == 0 {
args = []string{"-"}
}
for _, path := range args {
if err := scanPath(re, path); err != nil {
errExit(err)
}
}
}
func errExit(err error) {
fmt.Fprintf(os.Stderr, "%v\n", err)
os.Exit(1)
}