move dataset to types
This commit is contained in:
parent
907674f10c
commit
c02cd538b6
2 changed files with 5 additions and 4 deletions
|
@ -16,6 +16,7 @@ import (
|
|||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/crowdsecurity/crowdsec/pkg/types"
|
||||
"github.com/enescakir/emoji"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"gopkg.in/yaml.v2"
|
||||
|
@ -751,7 +752,7 @@ func DownloadItem(target Item, tdir string, overwrite bool, dataFolder string) (
|
|||
log.Infof("Installing : %+v \n", string(body))
|
||||
dec := yaml.NewDecoder(bytes.NewReader(body))
|
||||
for {
|
||||
data := &dataSet{}
|
||||
data := &types.DataSet{}
|
||||
err = dec.Decode(data)
|
||||
if err != nil {
|
||||
if err == io.EOF {
|
||||
|
@ -760,7 +761,7 @@ func DownloadItem(target Item, tdir string, overwrite bool, dataFolder string) (
|
|||
return target, fmt.Errorf("unable to read file %s data: %s", tdir+"/"+target.RemotePath, err)
|
||||
}
|
||||
}
|
||||
err = getData(data.Data, dataFolder)
|
||||
err = types.getData(data.Data, dataFolder)
|
||||
if err != nil {
|
||||
return target, fmt.Errorf("unable to get data: %s", err)
|
||||
}
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
package cwhub
|
||||
package types
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
@ -15,7 +15,7 @@ type dataSource struct {
|
|||
DestPath string `yaml:"dest_file"`
|
||||
}
|
||||
|
||||
type dataSet struct {
|
||||
type DataSet struct {
|
||||
Data []*dataSource `yaml:"data",omitempty`
|
||||
}
|
||||
|
Loading…
Add table
Reference in a new issue