diff options
Diffstat (limited to 'modules/starturl/starturl.go')
| -rw-r--r-- | modules/starturl/starturl.go | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/modules/starturl/starturl.go b/modules/starturl/starturl.go index 109d28f..9e3ec31 100644 --- a/modules/starturl/starturl.go +++ b/modules/starturl/starturl.go @@ -9,15 +9,29 @@ import ( ) func init() { - flyscrape.RegisterModule(new(Module)) + flyscrape.RegisterModule(Module{}) } type Module struct { URL string `json:"url"` } -func (m *Module) OnLoad(v flyscrape.Visitor) { - v.Visit(m.URL) +func (Module) ModuleInfo() flyscrape.ModuleInfo { + return flyscrape.ModuleInfo{ + ID: "starturl", + New: func() flyscrape.Module { return new(Module) }, + } } -var _ flyscrape.OnLoad = (*Module)(nil) +func (m *Module) Provision(ctx flyscrape.Context) { + if m.disabled() { + return + } + ctx.Visit(m.URL) +} + +func (m *Module) disabled() bool { + return m.URL == "" +} + +var _ flyscrape.Provisioner = (*Module)(nil) |