data_source_obmcs_core_drg_attachments.go

// Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved.

package main

import (
	"time"

	"github.com/hashicorp/terraform/helper/schema"
	"github.com/oracle/bmcs-go-sdk"

	"github.com/oracle/terraform-provider-oci/crud"
	"github.com/oracle/terraform-provider-oci/options"
)

func DrgAttachmentDatasource() *schema.Resource {
	return &schema.Resource{
		Read: readDrgAttachments,
		Schema: map[string]*schema.Schema{
			"filter": dataSourceFiltersSchema(),
			"compartment_id": {
				Type:     schema.TypeString,
				Required: true,
			},
			"drg_id": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"limit": {
				Type:     schema.TypeInt,
				Optional: true,
			},
			"page": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"vcn_id": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"drg_attachments": {
				Type:     schema.TypeList,
				Computed: true,
				Elem:     DrgAttachmentResource(),
			},
		},
	}
}

func readDrgAttachments(d *schema.ResourceData, m interface{}) (e error) {
	client := m.(*OracleClients)
	sync := &DrgAttachmentDatasourceCrud{}
	sync.D = d
	sync.Client = client.client
	return crud.ReadResource(sync)
}

type DrgAttachmentDatasourceCrud struct {
	crud.BaseCrud
	Res *baremetal.ListDrgAttachments
}

func (s *DrgAttachmentDatasourceCrud) Get() (e error) {
	compartmentID := s.D.Get("compartment_id").(string)

	opts := &baremetal.ListDrgAttachmentsOptions{}
	options.SetListOptions(s.D, &opts.ListOptions)
	if val, ok := s.D.GetOk("drg_id"); ok {
		opts.DrgID = val.(string)
	}
	if val, ok := s.D.GetOk("vcn_id"); ok {
		opts.VcnID = val.(string)
	}

	s.Res = &baremetal.ListDrgAttachments{
		DrgAttachments: []baremetal.DrgAttachment{},
	}

	for {
		var list *baremetal.ListDrgAttachments
		if list, e = s.Client.ListDrgAttachments(compartmentID, opts); e != nil {
			break
		}

		s.Res.DrgAttachments = append(s.Res.DrgAttachments, list.DrgAttachments...)

		if hasNextPage := options.SetNextPageOption(list.NextPage, &opts.ListOptions.PageListOptions); !hasNextPage {
			break
		}
	}

	return
}

func (s *DrgAttachmentDatasourceCrud) SetData() {
	if s.Res == nil {
		return
	}
	// Important, if you don't have an ID, make one up for your datasource
	// or things will end in tears
	s.D.SetId(time.Now().UTC().String())
	resources := []map[string]interface{}{}
	for _, v := range s.Res.DrgAttachments {
		res := map[string]interface{}{
			"compartment_id": v.CompartmentID,
			"display_name":   v.DisplayName,
			"drg_id":         v.DrgID,
			"id":             v.ID,
			"state":          v.State,
			"time_created":   v.TimeCreated.String(),
			"vcn_id":         v.VcnID,
		}
		resources = append(resources, res)
	}

	if f, fOk := s.D.GetOk("filter"); fOk {
		resources = ApplyFilters(f.(*schema.Set), resources)
	}

	if err := s.D.Set("drg_attachments", resources); err != nil {
		panic(err)
	}

	return
}