50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
|
package chromedp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
. "github.com/knq/chromedp/cdp"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Action is a single atomic action.
|
||
|
|
type Action interface {
|
||
|
|
Do(context.Context, FrameHandler) error
|
||
|
|
}
|
||
|
|
|
||
|
|
// ActionFunc is a single action func.
|
||
|
|
type ActionFunc func(context.Context, FrameHandler) error
|
||
|
|
|
||
|
|
// Do executes the action using the provided context.
|
||
|
|
func (f ActionFunc) Do(ctxt context.Context, h FrameHandler) error {
|
||
|
|
return f(ctxt, h)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tasks is a list of Actions that can be used as a single Action.
|
||
|
|
type Tasks []Action
|
||
|
|
|
||
|
|
// Do executes the list of Tasks using the provided context.
|
||
|
|
func (t Tasks) Do(ctxt context.Context, h FrameHandler) error {
|
||
|
|
var err error
|
||
|
|
|
||
|
|
// TODO: put individual task timeouts from context here
|
||
|
|
for _, a := range t {
|
||
|
|
// ctxt, cancel = context.WithTimeout(ctxt, timeout)
|
||
|
|
// defer cancel()
|
||
|
|
err = a.Do(ctxt, h)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Sleep is an empty action that calls time.Sleep with the specified duration.
|
||
|
|
func Sleep(d time.Duration) Action {
|
||
|
|
return ActionFunc(func(context.Context, FrameHandler) error {
|
||
|
|
time.Sleep(d)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
}
|