httprunner/hrp/pkg/uixt/video_crawler.go

package uixt

import (
	"time"

	"github.com/rs/zerolog/log"
)

type VideoCrawlerConfigs struct {
	AppPackageName string `json:"app_package_name"`

	TargetFeedCount int `json:"target_feed_count"`
	TargetLiveCount int `json:"target_live_count"`
}

func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
	// launch app
	if configs.AppPackageName != "" {
		if err = dExt.Driver.AppLaunch(configs.AppPackageName); err != nil {
			return err
		}
		time.Sleep(5 * time.Second)
	}

	// loop until target count achieved
	for {
		// take screenshot and get screen texts by OCR
		_, err := dExt.GetScreenTextsByOCR()
		if err != nil {
			log.Error().Err(err).Msg("OCR GetTexts failed")
			return err
		}

		// TODO: check if text popup exists

		// TODO: check if live video

		// assert feed video type

		// swipe to next video
		if err = dExt.SwipeUp(); err != nil {
			log.Error().Err(err).Msg("swipe up failed")
			return err
		}

		time.Sleep(5 * time.Second)
	}

	// return nil
}