diff --git a/.DS_Store b/.DS_Store index 37cf3e3..4c3e721 100644 Binary files a/.DS_Store and b/.DS_Store differ diff --git a/README.md b/README.md index 9ceba5b..71761ff 100644 --- a/README.md +++ b/README.md @@ -46,3 +46,8 @@ ![WX20230903-6](./image/WX20230903-6.png) 大家在使用过程中遇到什么问题或者有什么建议的,欢迎大家提issues + +### 更新记录 + +2023.09.14:增加拍照截图计数展示,显示整体拍照进度,最终结果计数显示拍照成功的数量 + diff --git a/core/working.go b/core/working.go index 8a7ac1a..fa5fda1 100755 --- a/core/working.go +++ b/core/working.go @@ -10,7 +10,7 @@ import ( ) // 当前版本信息 -var version = "1.0.0" +var version = "1.0.1" // logo var slogan = ` @@ -29,9 +29,11 @@ var slogan = ` ` var ( - txtfilepath string - resultmap map[string]map[string][]string - arraymap map[string][]string + txtfilepath string + resultmap map[string]map[string][]string + arraymap map[string][]string + count = 0 // 用于统计当前截图的顺序 + count_result = 0 // 用于统计计数当前拍摄成功的条数 ) func Flag() { @@ -62,7 +64,8 @@ func run(path string) { urls := scripts.FindTextUrl(path) // 获取TXT里url列表 log.Debug(fmt.Sprint("以获取本地TXT文本里URL列表:", urls)) resultname := fmt.Sprintf("result_%s", scripts.GetTimeStrin()) - log.Common("以获取本地URL列表信息数据,正在截图拍照中...") + total := len(urls) // 获取本地列表中的数据长度 + log.Common(fmt.Sprintf("以获取本地URL列表信息数据,共涉及%v条,正在截图拍照中...", total)) arraymap = make(map[string][]string) // 初始化arraymap空间地址 resultmap = make(map[string]map[string][]string) // 初始化resultmap空间地址 // 格式参考样例 arraymap["Website URL Address"] = []string{"TItle Name", "status", "网站截图路径"} @@ -71,18 +74,19 @@ func run(path string) { dire.Dir_mk(fmt.Sprintf("./result/%s/%s", resultname, "data")) // 创建本次扫描图片存放目录 for _, url := range urls { urlresultlist := scripts.ChromeScreenshot(url, resultname) - + count = count + 1 // 用于展示当前截图拍照的顺序位置 // 判断是否拍照成功,并根据拍照结果进行对应的结果填充 if len(urlresultlist) == 0 { - log.Common(fmt.Sprintf("访问 %s 地址超时,无法进行首页截图拍照!", url)) + log.Common(fmt.Sprintf("[%v/%v] 访问 %s 地址超时,无法进行首页截图拍照!", count, total, url)) arraymap[url] = []string{"x_x!", "连接失败", "data/"} } else { - log.Common(fmt.Sprintf("已完成对 %s 地址的首页截图拍照!", url)) + count_result = count_result + 1 + log.Common(fmt.Sprintf("[%v/%v] 已完成对 %s 地址的首页截图拍照!", count, total, url)) arraymap[urlresultlist[0]] = []string{urlresultlist[1], urlresultlist[2], urlresultlist[3]} } } resultmap[resultname] = arraymap - log.Common(fmt.Sprintf("已完成所有URL地址的截图拍照,共拍摄 %d 条", len(arraymap))) + log.Common(fmt.Sprintf("已完成所有URL地址的截图拍照,网站首页拍摄成功数量总数为 %v 条", count_result)) log.Common("正在生成最终结果报告文件...") scripts.CreateHtml(resultmap) } diff --git a/url.txt b/url.txt index 604b4c3..5e123e7 100755 --- a/url.txt +++ b/url.txt @@ -1,2 +1,5 @@ -https://github.com -https://sh1yan.top +https://www.ctftools.com/down/ +https://ha1c9on.top/ +https://learnku.com/docs/the-way-to-go +http://wechat.doonsec.com/ +http://www.fooying.com/