Compare commits
No commits in common. "main" and "summary" have entirely different histories.
@ -4,7 +4,6 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -245,33 +244,7 @@ func (b playWrightBrowser) Open(ctx context.Context, url string, opts OpenPageOp
|
|||||||
|
|
||||||
func (b playWrightBrowser) Close() error {
|
func (b playWrightBrowser) Close() error {
|
||||||
return errors.Join(
|
return errors.Join(
|
||||||
b.browser.Close(),
|
|
||||||
b.ctx.Close(),
|
b.ctx.Close(),
|
||||||
b.pw.Stop(),
|
b.browser.Close(),
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
func deferClose(cl io.Closer) {
|
|
||||||
_ = cl.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
func Screenshot(ctx context.Context, target string, timeout time.Duration) ([]byte, error) {
|
|
||||||
browser, err := NewPlayWrightBrowser(PlayWrightBrowserOptions{
|
|
||||||
Timeout: &timeout,
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("error creating browser: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer deferClose(browser)
|
|
||||||
|
|
||||||
doc, err := browser.Open(ctx, target, OpenPageOptions{})
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("error opening page: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer deferClose(doc)
|
|
||||||
|
|
||||||
return doc.Screenshot()
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user