v0.5 创建项目更便捷

Author: 0xbug

README.MD

@@ -42,9 +42,16 @@ cat domain.txt|biu-cli -pid 项目ID
 
 - 查询备案并添加到已有项目中
 ```shell
-biu-cli -icp xx公司|biu-cli -pid 项目ID
+biu-cli -icp xx公司 -pid 项目ID
 ```
 
+
+- 查询备案并创建项目
+```shell
+biu-cli -icp xx公司 -pnew 项目名称
+```
+
+
  - 配合https://opendata.rapid7.com/sonar.https/使用
 
 ```shell

cmd/biu-cli.go

@@ -26,7 +26,7 @@ var (
 	ip       string
 	pageSize int
 	client   = resty.New()
-	version  = "v0.4"
+	version  = "v0.5"
 )
 
 func biuPrint(header []string, data [][]string) {
@@ -63,9 +63,19 @@ func icpSearch() {
 		reg, err := regexp.Compile(`<span class="ranking-ym" rel="nofollow">([a-z0-9-\.]+)`)
 		if err == nil {
 			match := reg.FindAllString(resp.String(), -1)
-			for _, domain := range match {
-				fmt.Println(strings.Split(domain, ">")[1])
+			if len(match) > 0 {
+				if pnew != "" {
+					addProject()
+				}
+				for _, domain := range match {
+					if pid != "" {
+						addTargetToProject(strings.Split(domain, ">")[1])
+					} else {
+						fmt.Println(strings.Split(domain, ">")[1])
+					}
+				}
 			}
+
 		}
 	}
 }
@@ -113,6 +123,7 @@ func addProject() {
 		msg := gjson.Get(string(resp.Body()), "msg").Value()
 		fmt.Println(msg)
 		fmt.Println(result.Get("project_id").Value())
+		pid = result.Get("project_id").Str
 	}
 }