diff options
Diffstat (limited to 'modules/depth/depth.go')
| -rw-r--r-- | modules/depth/depth.go | 15 |
1 files changed, 11 insertions, 4 deletions
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) |