define custom state with rod

task.go

func NewTask(b *rod.Browser, user User) *Task {
	page := b.Page("https://www.your_site.com/")
	return &Task{page, user}
}

type Task struct {
	page		*rod.Page
	user		User
	isLoggedIn	atomic.Bool
}

func (t *Task) DoSomething(ctx context.Context) error {
	page := t.page.Context(ctx)

	if !t.isLoggedIn.Load() {
		err := t.login(page) // <- login if not logged in
		if err != nil {
			return err
		}
	}

	err := t.do(page)
	if errors.Is(err, ErrNotLoggedIn) {
		isLoggedIn.Store(false)
	}
	return err
}

func (t *Task) do(page *rod.page) error {
	e, err := page.Element("button.only-for-logged-in-users")
	if errors.Is(err, rod.ErrElementNotFound) {// <- not found if not logged in
		return ErrNotLoggedIn
	}
	return e.Click()
}

func (t *Task) login(page *rod.page) error

// Don't forget to close the page
func (t *Task) Close() error {
	return t.page.Close()
}