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/depth/depth.go | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) (limited to 'modules/depth/depth.go') diff --git a/modules/depth/depth.go b/modules/depth/depth.go index 0cfbc71..866f5ae 100644 --- a/modules/depth/depth.go +++ b/modules/depth/depth.go @@ -9,15 +9,22 @@ import ( ) func init() { - flyscrape.RegisterModule(new(Module)) + flyscrape.RegisterModule(Module{}) } type Module struct { Depth int `json:"depth"` } -func (m *Module) CanRequest(url string, depth int) bool { - return depth <= m.Depth +func (Module) ModuleInfo() flyscrape.ModuleInfo { + return flyscrape.ModuleInfo{ + ID: "depth", + New: func() flyscrape.Module { return new(Module) }, + } } -var _ flyscrape.CanRequest = (*Module)(nil) +func (m *Module) ValidateRequest(r *flyscrape.Request) bool { + return r.Depth <= m.Depth +} + +var _ flyscrape.RequestValidator = (*Module)(nil) -- cgit v1.2.3