Skip to content

Commit

Permalink
fix: repair hunter example
Browse files Browse the repository at this point in the history
  • Loading branch information
brittonhayes committed Mar 2, 2022
1 parent efbc44d commit 6ef9570
Showing 1 changed file with 13 additions and 20 deletions.
33 changes: 13 additions & 20 deletions _examples/hunter/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,40 +3,33 @@ package main
import (
"os"

"github.com/brittonhayes/pillager"
"github.com/brittonhayes/pillager/pkg/format"
"github.com/brittonhayes/pillager/pkg/hunter"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)

func main() {
err := example()
if err != nil {
panic(err)
}
}

func example() error {
opts := []pillager.ConfigOption{
pillager.WithLogLevel(zerolog.DebugLevel),
}

// Create a new hunter config
p, err := hunter.New(opts...)
h, err := hunter.New(
hunter.WithScanPath("."),
hunter.WithWorkers(2),
hunter.WithFormat(format.Simple{}),
hunter.WithLogLevel(zerolog.DebugLevel.String()),
)
if err != nil {
return err
log.Fatal().Err(err).Send()
}

// Start hunting
results, err := p.Hunt()
results, err := h.Hunt()
if err != nil {
return err
log.Fatal().Err(err).Send()
}

// Report results
err = p.Report(os.Stdout, results)
err = h.Report(os.Stdout, results)
if err != nil {
return err
log.Fatal().Err(err).Send()
}

return nil
}

0 comments on commit 6ef9570

Please sign in to comment.