From 9b290b543b1954132190117718765f97b047e3fb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=ED=95=98=ED=9B=8C?= <hahwul@gmail.com>
Date: Tue, 7 Apr 2020 23:52:40 +0900
Subject: [PATCH] Update add-tool.go

---
 add-tool.go | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/add-tool.go b/add-tool.go
index 2051678..53e8ae5 100644
--- a/add-tool.go
+++ b/add-tool.go
@@ -21,7 +21,7 @@ template
 */
 
 type Tools struct {
-	Type, Data string
+	Type, Data, Method string
 }
 
 func isTitleElement(n *html.Node) bool {
@@ -52,7 +52,7 @@ func GetHtmlTitle(r io.Reader) (string, bool) {
 	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")
 	// if we os.Open returns an error then handle it
 	if err != nil {
@@ -65,8 +65,9 @@ func writeJSON(category string, name string, data string) {
 	var result map[string]interface{}
 	json.Unmarshal([]byte(byteValue), &result)
 	tool := Tools{
-		Type: category,
-		Data: data,
+		Type:   category,
+		Data:   data,
+		Method: method,
 	}
 	result[name] = tool
 	file, _ := json.MarshalIndent(result, "", " ")
@@ -124,10 +125,14 @@ func main() {
 			index = index + 1
 		}
 		var choicetype int
-		fmt.Println("What is type?")
+		fmt.Println("[+] What is type?")
 		_, err = fmt.Scan(&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 {
 		reader := bufio.NewReader(os.Stdin)
 		fmt.Println("[+] What is name?")
@@ -161,11 +166,15 @@ func main() {
 		fmt.Println("What is type?")
 		_, err = fmt.Scan(&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 {
-		fmt.Println("| Name | Description | Popularity | Language | Metadata |")
+		fmt.Println("| Type | Name | Description | Popularity | Language |")
 		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 + ") |")