biu

Author: 0xbug

README.MD

@@ -0,0 +1,24 @@
+## 构建
+```shell
+go mod download
+go build -o biu-cli cmd/biu-cli.go
+```
+
+## 初始化配置
+访问个人中心->安全设置 生成新的API密钥
+
+```shell
+biu-cli -ak API密钥 -host biu地址(https://x.x.x.x)
+
+```
+
+## 使用
+ - 查看已有项目
+```shell
+biu-cli
+```
+ - 添加目标到已有项目中
+```shell
+cat ip.txt|biu-cli -pid 项目ID
+cat domain.txt|biu-cli -pid 项目ID
+```

bin/.gitkeep

@@ -0,0 +1,107 @@
+package main
+
+import (
+	"crypto/tls"
+	"flag"
+	"fmt"
+	"github.com/go-resty/resty/v2"
+	"github.com/joho/godotenv"
+	"github.com/tidwall/gjson"
+	"golang.org/x/crypto/ssh/terminal"
+	"io/ioutil"
+	"log"
+	"os"
+	"strings"
+)
+
+var (
+	biu      string
+	ak       string
+	pid      string
+	pageSize int
+	client   = resty.New()
+)
+
+func addTargetToProject(target string) {
+	if target != "" {
+		resp, err := client.R().
+			SetHeader("Content-Type", "application/json").
+			SetHeader("Biu-Api-Key", ak).
+			SetBody(`{"asset": "` + target + `" }`).
+			Patch(fmt.Sprintf("%s/api/project/optimize?project_id=%s", biu, pid))
+		if err != nil {
+			fmt.Print(err)
+		}
+		if resp.StatusCode() == 200 {
+			fmt.Println(fmt.Sprintf("添加成功 %s", target))
+
+		}
+	}
+}
+func listProjects() {
+	resp, err := client.R().
+		SetHeader("Biu-Api-Key", ak).
+		Get(fmt.Sprintf("%s/api/project?limit=%d&from=1&public=false", biu, pageSize))
+	if err != nil {
+		fmt.Print(err)
+	}
+	if resp.StatusCode() == 200 {
+		fmt.Println(fmt.Sprintf("编号\t项目ID                            \t名称"))
+		value := gjson.Get(string(resp.Body()), "result")
+		for index, result := range value.Array() {
+			fmt.Println(fmt.Sprintf("%d\t%s\t%s", index, result.Get("md5"), result.Get("name")))
+		}
+
+	}
+}
+
+func initEnv() {
+	homeDir, _ := os.UserHomeDir()
+	envPath := fmt.Sprintf("%s/.biu.env", homeDir)
+	err := godotenv.Load(fmt.Sprintf(envPath))
+	if err != nil {
+		if ak != "" && biu != "" {
+			var DefaultServerOptions = map[string]string{
+				"BIU_AK":   ak,
+				"BIU_HOST": biu,
+			}
+			err := godotenv.Write(DefaultServerOptions, envPath)
+			if err != nil {
+				fmt.Println("配置初始化成功")
+			}
+		} else {
+			log.Fatal("请初始化配置: biu-cli -ak xxx -host https://x.x.x.x")
+
+		}
+	} else {
+		ak = os.Getenv("BIU_AK")
+		biu = os.Getenv("BIU_HOST")
+
+	}
+
+}
+func main() {
+	client.SetTLSClientConfig(&tls.Config{InsecureSkipVerify: true})
+	flag.StringVar(&ak, "ak", "", "biu api key")
+	flag.StringVar(&biu, "host", "", "biu host url: https://x.x.x.x")
+	flag.StringVar(&pid, "pid", "", "biu project id")
+	flag.IntVar(&pageSize, "l", 20, "pageSize")
+	flag.Parse()
+	initEnv()
+	if pid == "" {
+		listProjects()
+	} else {
+		data := flag.Args()
+		if !terminal.IsTerminal(0) {
+			b, err := ioutil.ReadAll(os.Stdin)
+			if err == nil {
+				data = append(data, string(b))
+			}
+		}
+		targets := strings.Split(strings.Join(data, " "), "\n")
+		for _, target := range targets {
+			go addTargetToProject(target)
+		}
+	}
+
+}

cmd/biu-cli.go

@@ -0,0 +1,10 @@
+module biu-cli
+
+go 1.15
+
+require (
+	github.com/go-resty/resty/v2 v2.6.0
+	github.com/joho/godotenv v1.3.0
+	github.com/tidwall/gjson v1.8.0
+	golang.org/x/crypto v0.0.0-20210513164829-c07d793c2f9a
+)

go.mod

@@ -0,0 +1,3 @@
+CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags '-w -s' -gcflags '-N -l' -o bin/biu-cli_linux_amd64 cmd/biu-cli.go
+CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -ldflags '-w -s' -gcflags '-N -l' -o bin/biu-cli_win_amd64.exe cmd/biu-cli.go
+CGO_ENABLED=0 GOOS=darwin GOARCH=amd64 go build -ldflags '-w -s' -gcflags '-N -l' -o bin/biu-cli_darwin_amd64 cmd/biu-cli.go