diff options
Diffstat (limited to 'modules/followlinks/followlinks.go')
| -rw-r--r-- | modules/followlinks/followlinks.go | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/modules/followlinks/followlinks.go b/modules/followlinks/followlinks.go new file mode 100644 index 0000000..dde0e90 --- /dev/null +++ b/modules/followlinks/followlinks.go @@ -0,0 +1,30 @@ +// This Source Code Form is subject to the terms of the Mozilla Public +// License, v. 2.0. If a copy of the MPL was not distributed with this +// file, You can obtain one at http://mozilla.org/MPL/2.0/. + +package followlinks + +import ( + "github.com/philippta/flyscrape" +) + +func init() { + flyscrape.RegisterModule(new(Module)) +} + +type Module struct{} + +func (m *Module) ID() string { + return "followlinks" +} + +func (m *Module) OnResponse(resp *flyscrape.Response) { + for _, link := range flyscrape.ParseLinks(resp.HTML, resp.URL) { + resp.Visit(link) + } +} + +var ( + _ flyscrape.Module = (*Module)(nil) + _ flyscrape.OnResponse = (*Module)(nil) +) |