s0rg/crawley

package crawler

import (
    "time"
)

// Option is a configuration func.
type Option func(*config)

// WithUserAgent sets User-Agent string.
func WithUserAgent(v string) Option {
    return func(c *config) {
        c.Client.UserAgent = v
    }
}

// WithDelay sets crawl delay.
func WithDelay(v time.Duration) Option {
    return func(c *config) {
        c.Delay = v
    }
}

// WithMaxCrawlDepth sets maximum depth to crawl.
func WithMaxCrawlDepth(v int) Option {
    return func(c *config) {
        c.Depth = v
    }
}

// WithWorkersCount sets maximum workers.
func WithWorkersCount(v int) Option {
    return func(c *config) {
        c.Client.Workers = v
    }
}

// WithRobotsPolicy sets RobotsPolicy for crawler.
func WithRobotsPolicy(v RobotsPolicy) Option {
    return func(c *config) {
        c.Robots = v
    }
}

// WithDirsPolicy sets DirsPolicy for crawler.
func WithDirsPolicy(v DirsPolicy) Option {
    return func(c *config) {
        c.Dirs = v
    }
}

// WithSkipSSL tells crawley to skip any ssl handshake errors.
func WithSkipSSL(v bool) Option {
    return func(c *config) {
        c.Client.SkipSSL = v
    }
}

// WithBruteMode enables "brute-mode" - html comments scan.
func WithBruteMode(v bool) Option {
    return func(c *config) {
        c.Brute = v
    }
}

// WithoutHeads disables pre-flight HEAD requests.
func WithoutHeads(v bool) Option {
    return func(c *config) {
        c.NoHEAD = v
    }
}

// WithExtraHeaders add extra HTTP headers to requests.
func WithExtraHeaders(v []string) Option {
    return func(c *config) {
        c.Client.Headers = v
    }
}

// WithExtraCookies add cookies to requests.
func WithExtraCookies(v []string) Option {
    return func(c *config) {
        c.Client.Cookies = v
    }
}

// WithTagsFilter apply tag filter for crawler.
func WithTagsFilter(v []string) Option {
    return func(c *config) {
        c.AlowedTags = append(c.AlowedTags, v...)
    }
}

// WithSkipPatterns apply URL skip filter for crawler.
func WithIgnored(v []string) Option {
    return func(c *config) {
        c.Ignored = append(c.Ignored, v...)
    }
}

// WithScanJS enables js scanning.
func WithScanJS(v bool) Option {
    return func(c *config) {
        c.ScanJS = v
    }
}

// WithProxyAuth enables proxy credentials.
func WithProxyAuth(v string) Option {
    return func(c *config) {
        if v != "" {
            c.Client.Headers = append(c.Client.Headers, proxyAuthHeader(v))
        }
    }
}

// WithTimeout sets request timeout.
func WithTimeout(v time.Duration) Option {
    return func(c *config) {
        c.Client.Timeout = v
    }
}

// WithScanCSS enables css scanning.
func WithScanCSS(v bool) Option {
    return func(c *config) {
        c.ScanCSS = v
    }
}

// WithSubdomins enables subdomains scanning.
func WithSubdomains(v bool) Option {
    return func(c *config) {
        c.Subdomains = v
    }
}