diff options
Diffstat (limited to 'flyscrape.go')
| -rw-r--r-- | flyscrape.go | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/flyscrape.go b/flyscrape.go index bb4ee30..81c9346 100644 --- a/flyscrape.go +++ b/flyscrape.go @@ -25,16 +25,17 @@ func Run(file string) error { client := &http.Client{} - script, err := Compile(string(src), nil) + exports, err := Compile(string(src), NewJSLibrary(client)) if err != nil { return fmt.Errorf("failed to compile script: %w", err) } scraper := NewScraper() - scraper.ScrapeFunc = script.Scrape + scraper.ScrapeFunc = exports.Scrape + scraper.LoginFunc = exports.Login scraper.Script = file scraper.Client = client - scraper.Modules = LoadModules(script.Config()) + scraper.Modules = LoadModules(exports.Config()) scraper.Run() return nil @@ -53,18 +54,19 @@ func Dev(file string) error { fn := func(s string) error { client := &http.Client{} - script, err := Compile(s, nil) + exports, err := Compile(s, NewJSLibrary(client)) if err != nil { printCompileErr(file, err) return nil } - cfg := script.Config() + cfg := exports.Config() cfg = updateCfg(cfg, "depth", 0) cfg = updateCfg(cfg, "cache", "file:"+cachefile) scraper := NewScraper() - scraper.ScrapeFunc = script.Scrape + scraper.ScrapeFunc = exports.Scrape + scraper.LoginFunc = exports.Login scraper.Script = file scraper.Client = client scraper.Modules = LoadModules(cfg) |