change: add ks

This commit is contained in:
lilong.129
2023-05-10 16:23:47 +08:00
parent 41e4d89c95
commit 56e065c10b
6 changed files with 164 additions and 8 deletions

View File

@@ -199,12 +199,23 @@ type LiveCrawler struct {
}
func (l *LiveCrawler) checkLiveVideo(texts OCRTexts) (enterPoint PointF, yes bool) {
// 预览流入口
points, err := texts.FindTexts([]string{".?点击进入直播间", "直播中"}, WithRegex(true))
// 预览流入口DY/KS
points, err := texts.FindTexts([]string{".?点击进入直播间"}, WithRegex(true))
if err == nil {
return points[0].Center(), true
}
// 预览流入口KS
points, err = texts.FindTexts([]string{"和主播聊聊天.*"}, WithRegex(true))
if err == nil {
point := points[0].Center()
enterPoint = PointF{
X: point.X,
Y: point.Y - 100,
}
return enterPoint, true
}
// TODO: 头像入口
return PointF{}, false