blob: 5efedc87ecf73db6fc2af17c74d4362ec225f9aa (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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 depth
import (
"github.com/philippta/flyscrape"
)
func init() {
flyscrape.RegisterModule(new(Module))
}
type Module struct {
Depth int `json:"depth"`
}
func (m *Module) ID() string {
return "depth"
}
func (m *Module) CanRequest(url string, depth int) bool {
return depth <= m.Depth
}
var (
_ flyscrape.Module = (*Module)(nil)
_ flyscrape.CanRequest = (*Module)(nil)
)
|