data_source_zpa_cloud_browser_isolation_banner.go

package zpa

import (
	"fmt"
	"log"

	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	"github.com/zscaler/zscaler-sdk-go/v2/zpa/services/cloudbrowserisolation/cbibannercontroller"
)

func dataSourceCBIBanners() *schema.Resource {
	return &schema.Resource{
		Read: dataSourceCBIBannersRead,
		Schema: map[string]*schema.Schema{
			"id": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"name": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"primary_color": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"text_color": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"notification_title": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"notification_text": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"logo": {
				Type:     schema.TypeString,
				Computed: true,
			},
			"banner": {
				Type:     schema.TypeBool,
				Computed: true,
			},
			"is_default": {
				Type:     schema.TypeBool,
				Computed: true,
			},
		},
	}
}

func dataSourceCBIBannersRead(d *schema.ResourceData, m interface{}) error {
	zClient := m.(*Client)

	var resp *cbibannercontroller.CBIBannerController
	id, ok := d.Get("id").(string)
	if ok && id != "" {
		log.Printf("[INFO] Getting data for cbi banner %s\n", id)
		res, _, err := zClient.cbibannercontroller.Get(id)
		if err != nil {
			return err
		}
		resp = res
	}
	name, ok := d.Get("name").(string)
	if id == "" && ok && name != "" {
		log.Printf("[INFO] Getting data cbi banner name %s\n", name)
		res, _, err := zClient.cbibannercontroller.GetByName(name)
		if err != nil {
			return err
		}
		resp = res
	}
	if resp != nil {
		d.SetId(resp.ID)
		_ = d.Set("name", resp.Name)
		_ = d.Set("primary_color", resp.PrimaryColor)
		_ = d.Set("text_color", resp.TextColor)
		_ = d.Set("notification_title", resp.NotificationTitle)
		_ = d.Set("notification_text", resp.NotificationText)
		_ = d.Set("logo", resp.Logo)
		_ = d.Set("banner", resp.Banner)
		_ = d.Set("is_default", resp.IsDefault)

	} else {
		return fmt.Errorf("couldn't find any cbi banner with name '%s' or id '%s'", name, id)
	}

	return nil
}