-
Notifications
You must be signed in to change notification settings - Fork 105
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
7 changed files
with
216 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,53 @@ | ||
| package openstack | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| "github.com/gophercloud/gophercloud" | ||
| "github.com/gophercloud/gophercloud/openstack/compute/v2/servers" | ||
| "github.com/gophercloud/gophercloud/pagination" | ||
| "github.com/projectdiscovery/cloudlist/pkg/schema" | ||
| "github.com/projectdiscovery/gologger" | ||
| ) | ||
|
|
||
| // instanceProvider is an instance provider for Hetzner Cloud API | ||
| type instanceProvider struct { | ||
| id string | ||
| client *gophercloud.ServiceClient | ||
| } | ||
|
|
||
| // GetResource returns all the instances in the store for a provider. | ||
| func (p *instanceProvider) GetResource(ctx context.Context) (*schema.Resources, error) { | ||
| list := schema.NewResources() | ||
|
|
||
| err := servers.List(p.client, nil).EachPage(func(page pagination.Page) (bool, error) { | ||
| s, err := servers.ExtractServers(page) | ||
| if err != nil { | ||
| return false, err | ||
| } | ||
|
|
||
| for _, server := range s { | ||
| for _, a := range server.Addresses { | ||
| for _, networkAddresses := range a.([]interface{}) { | ||
| address := networkAddresses.(map[string]interface{}) | ||
| if address["OS-EXT-IPS:type"] == "floating" { | ||
| list.Append(&schema.Resource{ | ||
| Provider: providerName, | ||
| ID: p.id, | ||
| PrivateIpv4: address["addr"].(string), | ||
| }) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return true, nil | ||
| }) | ||
|
|
||
| if err != nil { | ||
| gologger.Error().Msgf("Couldn't list Openstack servers: %s\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| return list, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,98 @@ | ||
| package openstack | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| "github.com/gophercloud/gophercloud" | ||
| "github.com/gophercloud/gophercloud/openstack" | ||
| "github.com/projectdiscovery/cloudlist/pkg/schema" | ||
| "github.com/projectdiscovery/gologger" | ||
| ) | ||
|
|
||
| const ( | ||
| id = `id` | ||
| identityEndpoint = `identity_endpoint` | ||
| domainName = `domain_name` | ||
| tenantName = `tenant_name` | ||
| username = `username` | ||
| password = `password` | ||
|
|
||
| providerName = "openstack" | ||
| ) | ||
|
|
||
| // Provider is a data provider for Openstack API | ||
| type Provider struct { | ||
| id string | ||
| client *gophercloud.ServiceClient | ||
| } | ||
|
|
||
| // New creates a new provider client for Openstack API | ||
| func New(options schema.OptionBlock) (*Provider, error) { | ||
| id, _ := options.GetMetadata(id) | ||
|
|
||
| identityEndpoint, ok := options.GetMetadata(identityEndpoint) | ||
| if !ok { | ||
| return nil, &schema.ErrNoSuchKey{Name: identityEndpoint} | ||
| } | ||
|
|
||
| domainName, ok := options.GetMetadata(domainName) | ||
| if !ok { | ||
| return nil, &schema.ErrNoSuchKey{Name: domainName} | ||
| } | ||
|
|
||
| tenantName, ok := options.GetMetadata(tenantName) | ||
| if !ok { | ||
| return nil, &schema.ErrNoSuchKey{Name: tenantName} | ||
| } | ||
|
|
||
| username, ok := options.GetMetadata(username) | ||
| if !ok { | ||
| return nil, &schema.ErrNoSuchKey{Name: username} | ||
| } | ||
|
|
||
| password, ok := options.GetMetadata(password) | ||
| if !ok { | ||
| return nil, &schema.ErrNoSuchKey{Name: password} | ||
| } | ||
|
|
||
| opts := gophercloud.AuthOptions{ | ||
| IdentityEndpoint: identityEndpoint, | ||
| DomainName: domainName, | ||
| TenantName: tenantName, | ||
| Username: username, | ||
| Password: password, | ||
| } | ||
|
|
||
| provider, err := openstack.AuthenticatedClient(opts) | ||
| if err != nil { | ||
| gologger.Error().Msgf("Couldn't connect using Openstack credentials: %s\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| client, err := openstack.NewComputeV2(provider, gophercloud.EndpointOpts{ | ||
| Region: "RegionOne", | ||
| }) | ||
|
|
||
| if err != nil { | ||
| gologger.Error().Msgf("Couldn't use Openstack region: %s\n", err) | ||
| return nil, err | ||
| } | ||
|
|
||
| return &Provider{id: id, client: client}, nil | ||
| } | ||
|
|
||
| // Name returns the name of the provider | ||
| func (p *Provider) Name() string { | ||
| return providerName | ||
| } | ||
|
|
||
| // ID returns the name of the provider id | ||
| func (p *Provider) ID() string { | ||
| return p.id | ||
| } | ||
|
|
||
| // Resources returns the provider for an resource | ||
| func (p *Provider) Resources(ctx context.Context) (*schema.Resources, error) { | ||
| provider := &instanceProvider{id: p.id, client: p.client} | ||
| return provider.GetResource(ctx) | ||
| } |