Merge pull request #27 from crowdsecurity/add_custom_error
Adding custom error.
This commit is contained in:
commit
d9a37683e7
8
pkg/cwhub/errors.go
Normal file
8
pkg/cwhub/errors.go
Normal file
|
@ -0,0 +1,8 @@
|
|||
package cwhub
|
||||
|
||||
import (
|
||||
"errors"
|
||||
)
|
||||
|
||||
/*To be used when reference(s) (is/are) missing in a collection*/
|
||||
var ReferenceMissingError = errors.New("Reference(s) missing in collection")
|
|
@ -3,6 +3,7 @@ package cwhub
|
|||
import (
|
||||
"crypto/sha256"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
|
@ -375,8 +376,10 @@ func GetHubIdx() error {
|
|||
}
|
||||
ret, err := LoadPkgIndex(bidx)
|
||||
if err != nil {
|
||||
if !errors.Is(err, ReferenceMissingError) {
|
||||
log.Fatalf("Unable to load existing index : %v.", err)
|
||||
}
|
||||
}
|
||||
HubIdx = ret
|
||||
if err := LocalSync(); err != nil {
|
||||
log.Fatalf("Failed to sync Hub index with local deployment : %v", err)
|
||||
|
@ -391,8 +394,10 @@ func UpdateHubIdx() error {
|
|||
}
|
||||
ret, err := LoadPkgIndex(bidx)
|
||||
if err != nil {
|
||||
if !errors.Is(err, ReferenceMissingError) {
|
||||
log.Fatalf("Unable to load freshly downloaded index : %v.", err)
|
||||
}
|
||||
}
|
||||
HubIdx = ret
|
||||
if err := LocalSync(); err != nil {
|
||||
log.Fatalf("Failed to sync Hub index with local deployment : %v", err)
|
||||
|
@ -450,6 +455,7 @@ func DisplaySummary() {
|
|||
func LoadPkgIndex(buff []byte) (map[string]map[string]Item, error) {
|
||||
var err error
|
||||
var RawIndex map[string]map[string]Item
|
||||
var missingItems []string
|
||||
|
||||
if err = json.Unmarshal(buff, &RawIndex); err != nil {
|
||||
return nil, fmt.Errorf("failed to unmarshal index : %v", err)
|
||||
|
@ -473,12 +479,16 @@ func LoadPkgIndex(buff []byte) (map[string]map[string]Item, error) {
|
|||
for _, p := range ptr {
|
||||
if _, ok := RawIndex[ptrtype][p]; !ok {
|
||||
log.Errorf("Referred %s %s in collection %s doesn't exist.", ptrtype, p, item.Name)
|
||||
missingItems = append(missingItems, p)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if len(missingItems) > 0 {
|
||||
return RawIndex, fmt.Errorf("%q : %w", missingItems, ReferenceMissingError)
|
||||
}
|
||||
|
||||
return RawIndex, nil
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue