mirror of
https://github.com/hahwul/WebHackersWeapons.git
synced 2024-10-01 01:25:58 -04:00
215 lines
6.6 KiB
Go
215 lines
6.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"golang.org/x/net/html"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
/*
|
|
template
|
|
| [WebHackersWeapons](https://github.com/hahwul/WebHackersWeapons) | template |  |  | <br> <br>  <br>  |
|
|
*/
|
|
|
|
type Tools struct {
|
|
Type, Data, Method, Description string
|
|
Install, Update map[string]string
|
|
}
|
|
|
|
func isTitleElement(n *html.Node) bool {
|
|
return n.Type == html.ElementNode && n.Data == "title"
|
|
}
|
|
|
|
func traverse(n *html.Node) (string, bool) {
|
|
if isTitleElement(n) {
|
|
return n.FirstChild.Data, true
|
|
}
|
|
|
|
for c := n.FirstChild; c != nil; c = c.NextSibling {
|
|
result, ok := traverse(c)
|
|
if ok {
|
|
return result, ok
|
|
}
|
|
}
|
|
|
|
return "", false
|
|
}
|
|
|
|
func GetHtmlTitle(r io.Reader) (string, bool) {
|
|
doc, err := html.Parse(r)
|
|
if err != nil {
|
|
panic("Fail to parse html")
|
|
}
|
|
|
|
return traverse(doc)
|
|
}
|
|
|
|
func writeJSON(category, name, method, data, udesc string) {
|
|
jsonFile, err := os.Open("data.json")
|
|
// if we os.Open returns an error then handle it
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
fmt.Println("Successfully Opened data.json")
|
|
// defer the closing of our jsonFile so that we can parse it later on
|
|
defer jsonFile.Close()
|
|
byteValue, _ := ioutil.ReadAll(jsonFile)
|
|
var result map[string]interface{}
|
|
install := map[string]string{
|
|
"MacOS":"",
|
|
"Linux":"",
|
|
"Windows":"",
|
|
}
|
|
update := map[string]string{
|
|
"MacOS":"",
|
|
"Linux":"",
|
|
"Windows":"",
|
|
}
|
|
json.Unmarshal([]byte(byteValue), &result)
|
|
tool := Tools{
|
|
Type: category,
|
|
Data: data,
|
|
Method: method,
|
|
Description: udesc,
|
|
Install: install,
|
|
Update: update,
|
|
}
|
|
if result[name] != nil {
|
|
resultData := result[name].(map[string]interface{})
|
|
resultInstall := resultData["Install"].(map[string]interface{})
|
|
resultUpdate := resultData["Update"].(map[string]interface{})
|
|
tool.Install["MacOS"] = resultInstall["MacOS"].(string)
|
|
tool.Install["Linux"] = resultInstall["Linux"].(string)
|
|
tool.Install["Windows"] = resultInstall["Windows"].(string)
|
|
tool.Update["MacOS"] = resultUpdate["MacOS"].(string)
|
|
tool.Update["Linux"] = resultUpdate["Linux"].(string)
|
|
tool.Update["Windows"] = resultUpdate["Windows"].(string)
|
|
}
|
|
result[name] = tool
|
|
file, _ := json.MarshalIndent(result, "", " ")
|
|
_ = ioutil.WriteFile("data.json", file, 0644)
|
|
}
|
|
|
|
func main() {
|
|
repourl := flag.String("url", "", "any url")
|
|
first := flag.Bool("isFirst", false, "if you add new type, it use")
|
|
flag.Parse()
|
|
if flag.NFlag() == 0 {
|
|
flag.Usage()
|
|
return
|
|
}
|
|
u, err := url.Parse(*repourl)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if u.Host == "github.com" {
|
|
//fmt.Println(u.Path)
|
|
name := strings.Split(u.Path, "/")[2]
|
|
//fmt.Println(name)
|
|
desc := "asdf"
|
|
resp, err := http.Get(*repourl)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if title, ok := GetHtmlTitle(resp.Body); ok {
|
|
if strings.Contains(string(title), ": ") {
|
|
desc = strings.Split(string(title), ": ")[1]
|
|
} else {
|
|
reader1 := bufio.NewReader(os.Stdin)
|
|
fmt.Println("[+] Don't have descriptions / input description")
|
|
a, _ := reader1.ReadString('\n')
|
|
desc = strings.TrimRight(a, "\r\n")
|
|
|
|
}
|
|
} else {
|
|
println("Fail to get HTML title")
|
|
}
|
|
|
|
typeFile, err := os.Open("type.lst")
|
|
// if we os.Open returns an error then handle it
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
fmt.Println("Successfully Opened type.lst")
|
|
// defer the closing of our jsonFile so that we can parse it later on
|
|
defer typeFile.Close()
|
|
index := 0
|
|
m := make(map[int]string)
|
|
reader := bufio.NewReader(typeFile)
|
|
for {
|
|
line, isPrefix, err := reader.ReadLine()
|
|
if isPrefix || err != nil {
|
|
break
|
|
}
|
|
m[index] = string(line)
|
|
fmt.Println("[" + strconv.Itoa(index) + "] " + string(line))
|
|
index = index + 1
|
|
}
|
|
var choicetype int
|
|
fmt.Println("[+] What is type?")
|
|
_, err = fmt.Scan(&choicetype)
|
|
fmt.Println(m[choicetype])
|
|
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]+"/"+method+" | ["+name+"]("+*repourl+") | "+desc+" | ![](https://img.shields.io/github/stars"+u.Path+") | ![](https://img.shields.io/github/languages/top"+u.Path+") |", desc)
|
|
} else {
|
|
reader := bufio.NewReader(os.Stdin)
|
|
fmt.Println("[+] What is name?")
|
|
name, _ := reader.ReadString('\n')
|
|
name = strings.TrimRight(name, "\r\n")
|
|
fmt.Println("[+] Input Description?")
|
|
udesc, _ := reader.ReadString('\n')
|
|
udesc = strings.TrimRight(udesc, "\r\n")
|
|
|
|
typeFile, err := os.Open("type.lst")
|
|
// if we os.Open returns an error then handle it
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
fmt.Println("Successfully Opened type.lst")
|
|
// defer the closing of our jsonFile so that we can parse it later on
|
|
defer typeFile.Close()
|
|
index := 0
|
|
m := make(map[int]string)
|
|
readerF := bufio.NewReader(typeFile)
|
|
for {
|
|
line, isPrefix, err := readerF.ReadLine()
|
|
if isPrefix || err != nil {
|
|
break
|
|
}
|
|
m[index] = string(line)
|
|
fmt.Println("[" + strconv.Itoa(index) + "] " + string(line))
|
|
index = index + 1
|
|
}
|
|
var choicetype int
|
|
fmt.Println("What is type?")
|
|
_, err = fmt.Scan(&choicetype)
|
|
fmt.Println(m[choicetype])
|
|
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]+"/"+method+" | ["+name+"]("+*repourl+") | "+udesc+"||",udesc)
|
|
}
|
|
|
|
if *first {
|
|
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 + ") |")
|
|
}
|