Update add-tool.go

pull/9/head
하훌 2020-04-07 23:52:40 +09:00 committed by GitHub
parent 293beec4e4
commit 9b290b543b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 8 deletions

View File

@ -21,7 +21,7 @@ template
*/ */
type Tools struct { type Tools struct {
Type, Data string Type, Data, Method string
} }
func isTitleElement(n *html.Node) bool { func isTitleElement(n *html.Node) bool {
@ -52,7 +52,7 @@ func GetHtmlTitle(r io.Reader) (string, bool) {
return traverse(doc) return traverse(doc)
} }
func writeJSON(category string, name string, data string) { func writeJSON(category string, name string, method string, data string) {
jsonFile, err := os.Open("data.json") jsonFile, err := os.Open("data.json")
// if we os.Open returns an error then handle it // if we os.Open returns an error then handle it
if err != nil { if err != nil {
@ -65,8 +65,9 @@ func writeJSON(category string, name string, data string) {
var result map[string]interface{} var result map[string]interface{}
json.Unmarshal([]byte(byteValue), &result) json.Unmarshal([]byte(byteValue), &result)
tool := Tools{ tool := Tools{
Type: category, Type: category,
Data: data, Data: data,
Method: method,
} }
result[name] = tool result[name] = tool
file, _ := json.MarshalIndent(result, "", " ") file, _ := json.MarshalIndent(result, "", " ")
@ -124,10 +125,14 @@ func main() {
index = index + 1 index = index + 1
} }
var choicetype int var choicetype int
fmt.Println("What is type?") fmt.Println("[+] What is type?")
_, err = fmt.Scan(&choicetype) _, err = fmt.Scan(&choicetype)
fmt.Println(m[choicetype]) fmt.Println(m[choicetype])
writeJSON(m[choicetype], name, "| ["+name+"]("+*repourl+") | "+desc+" | ![](https://img.shields.io/github/stars"+u.Path+") | ![](https://img.shields.io/github/languages/top"+u.Path+") | ![](https://img.shields.io/github/repo-size"+u.Path+")<br>![](https://img.shields.io/github/license"+u.Path+") <br> ![](https://img.shields.io/github/forks"+u.Path+") <br> ![](https://img.shields.io/github/watchers"+u.Path+") |") reader1 := bufio.NewReader(os.Stdin)
fmt.Println("[+] What is method(e.g XSS, WVS, SSL, ETC..)?")
method, _ := reader1.ReadString('\n')
method = strings.TrimRight(method, "\r\n")
writeJSON(m[choicetype], name, method, "| "+m[choicetype]+" | ["+name+"]("+*repourl+") | "+method+" | "+desc+" | ![](https://img.shields.io/github/stars"+u.Path+") | ![](https://img.shields.io/github/languages/top"+u.Path+") |")
} else { } else {
reader := bufio.NewReader(os.Stdin) reader := bufio.NewReader(os.Stdin)
fmt.Println("[+] What is name?") fmt.Println("[+] What is name?")
@ -161,11 +166,15 @@ func main() {
fmt.Println("What is type?") fmt.Println("What is type?")
_, err = fmt.Scan(&choicetype) _, err = fmt.Scan(&choicetype)
fmt.Println(m[choicetype]) fmt.Println(m[choicetype])
writeJSON(m[choicetype], name, "| ["+name+"]("+*repourl+") | "+udesc+"|it's | not | github:dog:|") reader1 := bufio.NewReader(os.Stdin)
fmt.Println("[+] What is method(e.g XSS, WVS, SSL, ETC..)?")
method, _ := reader1.ReadString('\n')
method = strings.TrimRight(method, "\r\n")
writeJSON(m[choicetype], name, method, "| "+m[choicetype]+" | ["+name+"]("+*repourl+") | "+method+" | "+udesc+"|it's not|github:dog:|")
} }
if *first { if *first {
fmt.Println("| Name | Description | Popularity | Language | Metadata |") fmt.Println("| Type | Name | Description | Popularity | Language |")
fmt.Println("| ---------- | :---------- | :----------: | :----------: | :----------: |") fmt.Println("| ---------- | :---------- | :----------: | :----------: | :----------: |")
} }
//fmt.Println("| [" + name + "](" + *repourl + ") | " + desc + " | ![](https://img.shields.io/github/stars" + u.Path + ") | ![](https://img.shields.io/github/languages/top" + u.Path + ") | ![](https://img.shields.io/github/repo-size" + u.Path + ")<br>![](https://img.shields.io/github/license" + u.Path + ") <br> ![](https://img.shields.io/github/forks" + u.Path + ") <br> ![](https://img.shields.io/github/watchers" + u.Path + ") |") //fmt.Println("| [" + name + "](" + *repourl + ") | " + desc + " | ![](https://img.shields.io/github/stars" + u.Path + ") | ![](https://img.shields.io/github/languages/top" + u.Path + ") | ![](https://img.shields.io/github/repo-size" + u.Path + ")<br>![](https://img.shields.io/github/license" + u.Path + ") <br> ![](https://img.shields.io/github/forks" + u.Path + ") <br> ![](https://img.shields.io/github/watchers" + u.Path + ") |")