Function OutFileCsv

result/result.go:105–262

导出csv

(out string)

// 导出csv
func OutFileCsv(out string) {
	//获取域名
	var host string
	re := regexp.MustCompile("([a-z0-9\\-]+\\.)*([a-z0-9\\-]+\\.[a-z0-9\\-]+)(:[0-9]+)?")
	hosts := re.FindAllString(cmd.U, 1)
	if len(hosts) == 0 {
		host = cmd.U
	} else {
		host = hosts[0]
	}

	//抓取的域名优先排序
	if cmd.S != "" {
		ResultUrl = util.SelectSort(ResultUrl)
		ResultJs = util.SelectSort(ResultJs)
	}
	ResultJsHost, ResultJsOther := util.UrlDispose(ResultJs, host, util.GetHost(cmd.U))
	ResultUrlHost, ResultUrlOther := util.UrlDispose(ResultUrl, host, util.GetHost(cmd.U))
	Domains = util.GetDomains(util.MergeArray(ResultJs, ResultUrl))
	var fileName string
	if out != "" {
		fileName = out
	} else {
		//输出到文件
		if strings.Contains(host, ":") {
			host = strings.Replace(host, ":", "：", -1)
		}
		//在当前文件夹创建文件夹
		err := os.MkdirAll(cmd.O+"/"+host, 0755)
		if err != nil {
			fmt.Printf(cmd.O+"/"+host+" 目录创建失败 ：%s", err)
			return
		}
		//多相同url处理
		fileName = cmd.O + "/" + host + "/" + host + ".csv"
		for fileNum := 1; util.Exists(fileName); fileNum++ {
			fileName = cmd.O + "/" + host + "/" + host + "(" + strconv.Itoa(fileNum) + ").csv"
		}
	}
	file, err := os.OpenFile(fileName, os.O_CREATE|os.O_WRONLY, 0644)

	resultWriter := csv.NewWriter(file)
	// 写数据到文件
	if err != nil {
		fmt.Println("open file error:", err)
		return
	}
	if cmd.S == "" {
		resultWriter.Write([]string{"url", "Source"})
	} else {
		resultWriter.Write([]string{"url", "Status", "Size", "Title", "Redirect", "Source"})
	}
	if cmd.D == "" {
		resultWriter.Write([]string{strconv.Itoa(len(ResultJsHost)) + " JS to " + util.GetHost(cmd.U)})
	} else {
		resultWriter.Write([]string{strconv.Itoa(len(ResultJsHost)+len(ResultJsOther)) + " JS to " + cmd.D})
	}