39 lines
911 B
Go
39 lines
911 B
Go
package common
|
|
|
|
import (
|
|
"github.com/snyk/driftctl/enumeration/resource"
|
|
)
|
|
|
|
type Enumerator interface {
|
|
SupportedType() resource.ResourceType
|
|
Enumerate() ([]*resource.Resource, error)
|
|
}
|
|
|
|
type RemoteLibrary struct {
|
|
enumerators []Enumerator
|
|
detailsFetchers map[resource.ResourceType]DetailsFetcher
|
|
}
|
|
|
|
func NewRemoteLibrary() *RemoteLibrary {
|
|
return &RemoteLibrary{
|
|
make([]Enumerator, 0),
|
|
make(map[resource.ResourceType]DetailsFetcher),
|
|
}
|
|
}
|
|
|
|
func (r *RemoteLibrary) AddEnumerator(enumerator Enumerator) {
|
|
r.enumerators = append(r.enumerators, enumerator)
|
|
}
|
|
|
|
func (r *RemoteLibrary) Enumerators() []Enumerator {
|
|
return r.enumerators
|
|
}
|
|
|
|
func (r *RemoteLibrary) AddDetailsFetcher(ty resource.ResourceType, detailsFetcher DetailsFetcher) {
|
|
r.detailsFetchers[ty] = detailsFetcher
|
|
}
|
|
|
|
func (r *RemoteLibrary) GetDetailsFetcher(ty resource.ResourceType) DetailsFetcher {
|
|
return r.detailsFetchers[ty]
|
|
}
|