mirror of
https://github.com/MontFerret/ferret.git
synced 2024-12-23 01:24:30 +02:00
416 lines
9.2 KiB
Go
416 lines
9.2 KiB
Go
package input
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/mafredri/cdp"
|
|
"github.com/mafredri/cdp/protocol/dom"
|
|
"github.com/mafredri/cdp/protocol/runtime"
|
|
|
|
"github.com/MontFerret/ferret/pkg/drivers"
|
|
"github.com/MontFerret/ferret/pkg/drivers/cdp/eval"
|
|
"github.com/MontFerret/ferret/pkg/drivers/cdp/templates"
|
|
"github.com/MontFerret/ferret/pkg/runtime/core"
|
|
"github.com/MontFerret/ferret/pkg/runtime/values"
|
|
)
|
|
|
|
type (
|
|
TypeParams struct {
|
|
Text string
|
|
Clear bool
|
|
Delay time.Duration
|
|
}
|
|
|
|
Manager struct {
|
|
client *cdp.Client
|
|
exec *eval.ExecutionContext
|
|
keyboard *Keyboard
|
|
mouse *Mouse
|
|
}
|
|
)
|
|
|
|
func NewManager(
|
|
client *cdp.Client,
|
|
exec *eval.ExecutionContext,
|
|
keyboard *Keyboard,
|
|
mouse *Mouse,
|
|
) *Manager {
|
|
return &Manager{
|
|
client,
|
|
exec,
|
|
keyboard,
|
|
mouse,
|
|
}
|
|
}
|
|
|
|
func (m *Manager) Keyboard() *Keyboard {
|
|
return m.keyboard
|
|
}
|
|
|
|
func (m *Manager) Mouse() *Mouse {
|
|
return m.mouse
|
|
}
|
|
|
|
func (m *Manager) ScrollTop(ctx context.Context) error {
|
|
return m.exec.Eval(ctx, templates.ScrollTop())
|
|
}
|
|
|
|
func (m *Manager) ScrollBottom(ctx context.Context) error {
|
|
return m.exec.Eval(ctx, templates.ScrollBottom())
|
|
}
|
|
|
|
func (m *Manager) ScrollIntoView(ctx context.Context, objectID runtime.RemoteObjectID) error {
|
|
return m.exec.EvalWithArguments(
|
|
ctx,
|
|
templates.ScrollIntoView(),
|
|
runtime.CallArgument{
|
|
ObjectID: &objectID,
|
|
},
|
|
)
|
|
}
|
|
|
|
func (m *Manager) ScrollIntoViewBySelector(ctx context.Context, selector string) error {
|
|
return m.exec.Eval(ctx, templates.ScrollIntoViewBySelector(selector))
|
|
}
|
|
|
|
func (m *Manager) ScrollByXY(ctx context.Context, x, y float64) error {
|
|
return m.exec.Eval(
|
|
ctx,
|
|
templates.Scroll(eval.ParamFloat(x), eval.ParamFloat(y)),
|
|
)
|
|
}
|
|
|
|
func (m *Manager) Focus(ctx context.Context, objectID runtime.RemoteObjectID) error {
|
|
err := m.ScrollIntoView(ctx, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.client.DOM.Focus(ctx, dom.NewFocusArgs().SetObjectID(objectID))
|
|
}
|
|
|
|
func (m *Manager) FocusBySelector(ctx context.Context, parentNodeID dom.NodeID, selector string) error {
|
|
err := m.ScrollIntoViewBySelector(ctx, selector)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
found, err := m.client.DOM.QuerySelector(ctx, dom.NewQuerySelectorArgs(parentNodeID, selector))
|
|
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
return m.client.DOM.Focus(ctx, dom.NewFocusArgs().SetNodeID(found.NodeID))
|
|
}
|
|
|
|
func (m *Manager) Blur(ctx context.Context, objectID runtime.RemoteObjectID) error {
|
|
return m.exec.EvalWithArguments(ctx, templates.Blur(), runtime.CallArgument{
|
|
ObjectID: &objectID,
|
|
})
|
|
}
|
|
|
|
func (m *Manager) BlurBySelector(ctx context.Context, parentObjectID runtime.RemoteObjectID, selector string) error {
|
|
return m.exec.EvalWithArguments(ctx, templates.BlurBySelector(selector), runtime.CallArgument{
|
|
ObjectID: &parentObjectID,
|
|
})
|
|
}
|
|
|
|
func (m *Manager) MoveMouse(ctx context.Context, objectID runtime.RemoteObjectID) error {
|
|
if err := m.ScrollIntoView(ctx, objectID); err != nil {
|
|
return err
|
|
}
|
|
|
|
q, err := GetClickablePointByObjectID(ctx, m.client, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.mouse.Move(ctx, q.X, q.Y)
|
|
}
|
|
|
|
func (m *Manager) MoveMouseBySelector(ctx context.Context, parentNodeID dom.NodeID, selector string) error {
|
|
if err := m.ScrollIntoViewBySelector(ctx, selector); err != nil {
|
|
return err
|
|
}
|
|
|
|
found, err := m.client.DOM.QuerySelector(ctx, dom.NewQuerySelectorArgs(parentNodeID, selector))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
q, err := GetClickablePointByNodeID(ctx, m.client, found.NodeID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.mouse.Move(ctx, q.X, q.Y)
|
|
}
|
|
|
|
func (m *Manager) MoveMouseByXY(ctx context.Context, x, y float64) error {
|
|
if err := m.ScrollByXY(ctx, x, y); err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.mouse.Move(ctx, x, y)
|
|
}
|
|
|
|
func (m *Manager) Click(ctx context.Context, objectID runtime.RemoteObjectID, count int) error {
|
|
if err := m.ScrollIntoView(ctx, objectID); err != nil {
|
|
return err
|
|
}
|
|
|
|
points, err := GetClickablePointByObjectID(ctx, m.client, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
delay := time.Duration(drivers.DefaultMouseDelay) * time.Millisecond
|
|
|
|
if err := m.mouse.ClickWithCount(ctx, points.X, points.Y, delay, count); err != nil {
|
|
return nil
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Manager) ClickBySelector(ctx context.Context, parentNodeID dom.NodeID, selector string, count int) error {
|
|
if err := m.ScrollIntoViewBySelector(ctx, selector); err != nil {
|
|
return err
|
|
}
|
|
|
|
found, err := m.client.DOM.QuerySelector(ctx, dom.NewQuerySelectorArgs(parentNodeID, selector))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
points, err := GetClickablePointByNodeID(ctx, m.client, found.NodeID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
delay := time.Duration(drivers.DefaultMouseDelay) * time.Millisecond
|
|
|
|
if err := m.mouse.ClickWithCount(ctx, points.X, points.Y, delay, count); err != nil {
|
|
return nil
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Manager) ClickBySelectorAll(ctx context.Context, parentNodeID dom.NodeID, selector string, count int) error {
|
|
if err := m.ScrollIntoViewBySelector(ctx, selector); err != nil {
|
|
return err
|
|
}
|
|
|
|
found, err := m.client.DOM.QuerySelectorAll(ctx, dom.NewQuerySelectorAllArgs(parentNodeID, selector))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, nodeID := range found.NodeIDs {
|
|
beforeTypeDelay := time.Duration(core.NumberLowerBoundary(drivers.DefaultMouseDelay*10)) * time.Millisecond
|
|
|
|
time.Sleep(beforeTypeDelay)
|
|
|
|
points, err := GetClickablePointByNodeID(ctx, m.client, nodeID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
delay := time.Duration(drivers.DefaultMouseDelay) * time.Millisecond
|
|
|
|
if err := m.mouse.ClickWithCount(ctx, points.X, points.Y, delay, count); err != nil {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Manager) Type(ctx context.Context, objectID runtime.RemoteObjectID, params TypeParams) error {
|
|
err := m.ScrollIntoView(ctx, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = m.client.DOM.Focus(ctx, dom.NewFocusArgs().SetObjectID(objectID))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if params.Clear {
|
|
points, err := GetClickablePointByObjectID(ctx, m.client, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := m.ClearByXY(ctx, points); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
d := core.NumberLowerBoundary(float64(params.Delay))
|
|
beforeTypeDelay := time.Duration(d)
|
|
|
|
time.Sleep(beforeTypeDelay)
|
|
|
|
return m.keyboard.Type(ctx, params.Text, params.Delay)
|
|
}
|
|
|
|
func (m *Manager) TypeBySelector(ctx context.Context, parentNodeID dom.NodeID, selector string, params TypeParams) error {
|
|
err := m.ScrollIntoViewBySelector(ctx, selector)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
found, err := m.client.DOM.QuerySelector(ctx, dom.NewQuerySelectorArgs(parentNodeID, selector))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = m.client.DOM.Focus(ctx, dom.NewFocusArgs().SetNodeID(found.NodeID))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if params.Clear {
|
|
points, err := GetClickablePointByNodeID(ctx, m.client, found.NodeID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := m.ClearByXY(ctx, points); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
d := core.NumberLowerBoundary(float64(params.Delay))
|
|
beforeTypeDelay := time.Duration(d)
|
|
|
|
time.Sleep(beforeTypeDelay)
|
|
|
|
return m.keyboard.Type(ctx, params.Text, params.Delay)
|
|
}
|
|
|
|
func (m *Manager) Clear(ctx context.Context, objectID runtime.RemoteObjectID) error {
|
|
err := m.ScrollIntoView(ctx, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
points, err := GetClickablePointByObjectID(ctx, m.client, objectID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = m.client.DOM.Focus(ctx, dom.NewFocusArgs().SetObjectID(objectID))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.ClearByXY(ctx, points)
|
|
}
|
|
|
|
func (m *Manager) ClearBySelector(ctx context.Context, parentNodeID dom.NodeID, selector string) error {
|
|
err := m.ScrollIntoViewBySelector(ctx, selector)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
found, err := m.client.DOM.QuerySelector(ctx, dom.NewQuerySelectorArgs(parentNodeID, selector))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
points, err := GetClickablePointByNodeID(ctx, m.client, found.NodeID)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = m.client.DOM.Focus(ctx, dom.NewFocusArgs().SetNodeID(found.NodeID))
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.ClearByXY(ctx, points)
|
|
}
|
|
|
|
func (m *Manager) ClearByXY(ctx context.Context, points Quad) error {
|
|
delay := time.Duration(drivers.DefaultMouseDelay) * time.Millisecond
|
|
err := m.mouse.ClickWithCount(ctx, points.X, points.Y, delay, 2)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.keyboard.Press(ctx, "Backspace")
|
|
}
|
|
|
|
func (m *Manager) Select(ctx context.Context, objectID runtime.RemoteObjectID, value *values.Array) (*values.Array, error) {
|
|
if err := m.Focus(ctx, objectID); err != nil {
|
|
return values.NewArray(0), err
|
|
}
|
|
|
|
val, err := m.exec.EvalWithArgumentsAndReturnValue(ctx, templates.Select(value.String()), runtime.CallArgument{
|
|
ObjectID: &objectID,
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
arr, ok := val.(*values.Array)
|
|
|
|
if !ok {
|
|
return values.NewArray(0), core.ErrUnexpected
|
|
}
|
|
|
|
return arr, nil
|
|
}
|
|
|
|
func (m *Manager) SelectBySelector(ctx context.Context, parentNodeID dom.NodeID, selector string, value *values.Array) (*values.Array, error) {
|
|
if err := m.FocusBySelector(ctx, parentNodeID, selector); err != nil {
|
|
return values.NewArray(0), err
|
|
}
|
|
|
|
res, err := m.exec.EvalWithReturnValue(ctx, templates.SelectBySelector(selector, value.String()))
|
|
|
|
if err != nil {
|
|
return values.NewArray(0), err
|
|
}
|
|
|
|
arr, ok := res.(*values.Array)
|
|
|
|
if !ok {
|
|
return values.NewArray(0), core.ErrUnexpected
|
|
}
|
|
|
|
return arr, nil
|
|
}
|