122 lines
2.9 KiB
Go
122 lines
2.9 KiB
Go
package digitalocean
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/digitalocean/godo"
|
|
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
|
|
)
|
|
|
|
func projectSchema() map[string]*schema.Schema {
|
|
return map[string]*schema.Schema{
|
|
"id": {
|
|
Type: schema.TypeString,
|
|
},
|
|
"name": {
|
|
Type: schema.TypeString,
|
|
},
|
|
"description": {
|
|
Type: schema.TypeString,
|
|
},
|
|
"purpose": {
|
|
Type: schema.TypeString,
|
|
},
|
|
"environment": {
|
|
Type: schema.TypeString,
|
|
},
|
|
"owner_uuid": {
|
|
Type: schema.TypeString,
|
|
},
|
|
"owner_id": {
|
|
Type: schema.TypeInt,
|
|
},
|
|
"is_default": {
|
|
Type: schema.TypeBool,
|
|
},
|
|
"created_at": {
|
|
Type: schema.TypeString,
|
|
Description: "the date and time when the project was created, (ISO8601)",
|
|
},
|
|
"updated_at": {
|
|
Type: schema.TypeString,
|
|
Description: "the date and time when the project was last updated, (ISO8601)",
|
|
},
|
|
"resources": {
|
|
Type: schema.TypeSet,
|
|
Elem: &schema.Schema{Type: schema.TypeString},
|
|
},
|
|
}
|
|
}
|
|
|
|
func getDigitalOceanProjects(meta interface{}) ([]interface{}, error) {
|
|
client := meta.(*CombinedConfig).godoClient()
|
|
|
|
var allProjects []interface{}
|
|
|
|
opts := &godo.ListOptions{
|
|
Page: 1,
|
|
PerPage: 200,
|
|
}
|
|
|
|
for {
|
|
projects, resp, err := client.Projects.List(context.Background(), opts)
|
|
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Error retrieving projects: %s", err)
|
|
}
|
|
|
|
for _, project := range projects {
|
|
allProjects = append(allProjects, project)
|
|
}
|
|
|
|
if resp.Links == nil || resp.Links.IsLastPage() {
|
|
break
|
|
}
|
|
|
|
page, err := resp.Links.CurrentPage()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Error retrieving projects: %s", err)
|
|
}
|
|
|
|
opts.Page = page + 1
|
|
}
|
|
|
|
return allProjects, nil
|
|
}
|
|
|
|
func flattenDigitalOceanProject(rawProject interface{}, meta interface{}) (map[string]interface{}, error) {
|
|
client := meta.(*CombinedConfig).godoClient()
|
|
|
|
project, ok := rawProject.(godo.Project)
|
|
if !ok {
|
|
return nil, fmt.Errorf("Unable to convert to godo.Project")
|
|
}
|
|
|
|
flattenedProject := map[string]interface{}{}
|
|
flattenedProject["id"] = project.ID
|
|
flattenedProject["name"] = project.Name
|
|
flattenedProject["purpose"] = strings.TrimPrefix(project.Purpose, "Other: ")
|
|
flattenedProject["description"] = project.Description
|
|
flattenedProject["environment"] = project.Environment
|
|
flattenedProject["owner_uuid"] = project.OwnerUUID
|
|
flattenedProject["owner_id"] = project.OwnerID
|
|
flattenedProject["is_default"] = project.IsDefault
|
|
flattenedProject["created_at"] = project.CreatedAt
|
|
flattenedProject["updated_at"] = project.UpdatedAt
|
|
|
|
urns, err := loadResourceURNs(client, project.ID)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Error loading project resource URNs for project ID %s: %s", project.ID, err)
|
|
}
|
|
|
|
flattenedURNS := schema.NewSet(schema.HashString, []interface{}{})
|
|
for _, urn := range *urns {
|
|
flattenedURNS.Add(urn)
|
|
}
|
|
flattenedProject["resources"] = flattenedURNS
|
|
|
|
return flattenedProject, nil
|
|
}
|