From 1fc497fbdc79a43c62ac2e8eaf4827752dbeef8e Mon Sep 17 00:00:00 2001 From: Philipp Tanlak Date: Thu, 5 Oct 2023 14:53:37 +0200 Subject: Refactor codebase into modules --- modules/starturl/starturl.go | 22 ++++++++++++++++++---- 1 file changed, 18 insertions(+), 4 deletions(-) (limited to 'modules/starturl/starturl.go') 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) -- cgit v1.2.3