Skip to content

Commit

Permalink
Use public models for config
Browse files Browse the repository at this point in the history
  • Loading branch information
koplas committed Jul 9, 2024
1 parent 191b013 commit c71a235
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 9 deletions.
2 changes: 1 addition & 1 deletion cmd/csaf_downloader/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -233,7 +233,7 @@ func (cfg *config) GetDownloadConfig() (*downloader.Config, error) {
ClientPassphrase: cfg.ClientPassphrase,
Rate: cfg.Rate,
Worker: cfg.Worker,
Range: (*[2]time.Time)(cfg.Range),
Range: cfg.Range,
IgnorePattern: cfg.ignorePattern,
ExtraHeader: cfg.ExtraHeader,

Expand Down
8 changes: 4 additions & 4 deletions lib/downloader/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@ package downloader
import (
"crypto/tls"
"fmt"
"github.com/csaf-poc/csaf_distribution/v3/csaf/filter"
"github.com/csaf-poc/csaf_distribution/v3/csaf/models"
"log/slog"
"net/http"
"regexp"
"time"
)

// ValidationMode specifies the strict the validation is.
Expand All @@ -36,8 +36,8 @@ type Config struct {
ClientPassphrase *string
Rate *float64
Worker int
Range *[2]time.Time
IgnorePattern []*regexp.Regexp
Range *models.TimeRange
IgnorePattern filter.PatternMatcher
ExtraHeader http.Header

RemoteValidator string
Expand Down
6 changes: 2 additions & 4 deletions lib/downloader/downloader.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@ import (
"encoding/json"
"errors"
"fmt"
"github.com/csaf-poc/csaf_distribution/v3/csaf/models"
"hash"
"io"
"log/slog"
Expand Down Expand Up @@ -241,10 +240,9 @@ func (d *Downloader) download(ctx context.Context, domain string) error {

// Do we need time range based filtering?
if d.cfg.Range != nil {
timeRange := models.NewTimeInterval(d.cfg.Range[0], d.cfg.Range[1])
d.cfg.Logger.Debug("Setting up filter to accept advisories within",
"timerange", timeRange)
afp.AgeAccept = timeRange.Contains
"timerange", d.cfg.Range)
afp.AgeAccept = d.cfg.Range.Contains
}

return afp.Process(func(label csaf.TLPLabel, files []csaf.AdvisoryFile) error {
Expand Down

0 comments on commit c71a235

Please sign in to comment.