/
main.go
95 lines (81 loc) · 1.97 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
84
85
86
87
88
89
90
91
92
93
94
95
package main
import (
"flag"
"fmt"
"io/ioutil"
"net/http"
"strconv"
"strings"
"github.com/tidwall/gjson"
)
func getPRs(token string) []float64 {
// Get request
var result []float64
page := 1
for {
ghprs := "https://api.github.com/repos/projectdiscovery/nuclei-templates/pulls?state=open&per_page=100&page=" + strconv.Itoa(page)
req, _ := http.NewRequest("GET", ghprs, nil)
req.Header.Set("Host", "api.github.com")
if token != "" {
req.Header.Set("Authorization", "Bearer "+token)
}
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
panic(err)
}
defer resp.Body.Close()
body, _ := ioutil.ReadAll(resp.Body)
json := string(body[:])
if strings.TrimRight(json, "\n") == "[]" {
break
}
prs := gjson.Get(json, "#(title)#.number").Array()
for _, v := range prs {
result = append(result, v.Num)
}
page++
}
return result
}
func getFiles(prs float64, token string, cves bool) gjson.Result {
// Get request
prstr := fmt.Sprintf("%v", prs)
ghfiles := "https://api.github.com/repos/projectdiscovery/nuclei-templates/pulls/" + prstr + "/files"
req, _ := http.NewRequest("GET", ghfiles, nil)
req.Header.Set("Host", "api.github.com")
if token != "" {
req.Header.Set("Authorization", "Bearer "+token)
}
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
panic(err)
}
defer resp.Body.Close()
body, _ := ioutil.ReadAll(resp.Body)
json := string(body[:])
var files gjson.Result
if cves {
files = gjson.Get(json, "#(filename%\"*cves*\")#.raw_url")
} else {
files = gjson.Get(json, "#(filename)#.raw_url")
}
return files
}
func main() {
var token string
flag.StringVar(&token, "token", "", "- GitHub token to use.\n")
var cves bool
flag.BoolVar(&cves, "cves", false, "- Show only CVEs")
flag.Parse()
prs := getPRs(token)
var files gjson.Result
for i := range prs {
pr := prs[i]
files = getFiles(pr, token, cves)
for _, name := range files.Array() {
fmt.Println(name.String())
}
}
}