dlp_notification_templates.go

package dlp_notification_templates

import (
	"errors"
	"fmt"
	"net/http"
	"strings"

	"github.com/zscaler/zscaler-sdk-go/v2/zia/services/common"
)

const (
	dlpNotificationTemplatesEndpoint = "/dlpNotificationTemplates"
)

type DlpNotificationTemplates struct {
	// The unique identifier for a DLP notification template.
	ID int `json:"id"`

	// The DLP notification template name.
	Name string `json:"name,omitempty"`

	// The Subject line that is displayed within the DLP notification email.
	Subject string `json:"subject,omitempty"`

	// If set to true, the content that is violation is attached to the DLP notification email.
	AttachContent bool `json:"attachContent,omitempty"`

	// The template for the plain text UTF-8 message body that must be displayed in the DLP notification email.
	PlainTextMessage string `json:"plainTextMessage,omitempty"`

	// The template for the HTML message body that must be displayed in the DLP notification email.
	HtmlMessage string `json:"htmlMessage,omitempty"`

	// The template for the HTML message body that must be displayed in the DLP notification email.
	TLSEnabled bool `json:"tlsEnabled,omitempty"`
}

func (service *Service) Get(dlpTemplateID int) (*DlpNotificationTemplates, error) {
	var dlpTemplates DlpNotificationTemplates
	err := service.Client.Read(fmt.Sprintf("%s/%d", dlpNotificationTemplatesEndpoint, dlpTemplateID), &dlpTemplates)
	if err != nil {
		return nil, err
	}

	service.Client.Logger.Printf("[DEBUG]Returning dlp notification template from Get: %d", dlpTemplates.ID)
	return &dlpTemplates, nil
}

func (service *Service) GetByName(templateName string) (*DlpNotificationTemplates, error) {
	var dlpTemplates []DlpNotificationTemplates
	err := common.ReadAllPages(service.Client, dlpNotificationTemplatesEndpoint, &dlpTemplates)
	if err != nil {
		return nil, err
	}
	for _, template := range dlpTemplates {
		if strings.EqualFold(template.Name, templateName) {
			return &template, nil
		}
	}
	return nil, fmt.Errorf("no dictionary found with name: %s", templateName)
}

func (service *Service) Create(dlpTemplateID *DlpNotificationTemplates) (*DlpNotificationTemplates, *http.Response, error) {
	resp, err := service.Client.Create(dlpNotificationTemplatesEndpoint, *dlpTemplateID)
	if err != nil {
		return nil, nil, err
	}

	createdDlpTemplate, ok := resp.(*DlpNotificationTemplates)
	if !ok {
		return nil, nil, errors.New("object returned from api was not a dlp dictionary pointer")
	}

	service.Client.Logger.Printf("[DEBUG]returning new dlp notification template from create: %d", createdDlpTemplate.ID)
	return createdDlpTemplate, nil, nil
}

func (service *Service) Update(dlpTemplateID int, dlpTemplates *DlpNotificationTemplates) (*DlpNotificationTemplates, *http.Response, error) {
	resp, err := service.Client.UpdateWithPut(fmt.Sprintf("%s/%d", dlpNotificationTemplatesEndpoint, dlpTemplateID), *dlpTemplates)
	if err != nil {
		return nil, nil, err
	}
	updatedDlpTemplate, _ := resp.(*DlpNotificationTemplates)

	service.Client.Logger.Printf("[DEBUG]returning updates from dlp notification template from update: %d", updatedDlpTemplate.ID)
	return updatedDlpTemplate, nil, nil
}

func (service *Service) Delete(dlpTemplateID int) (*http.Response, error) {
	err := service.Client.Delete(fmt.Sprintf("%s/%d", dlpNotificationTemplatesEndpoint, dlpTemplateID))
	if err != nil {
		return nil, err
	}

	return nil, nil
}

func (service *Service) GetAll() ([]DlpNotificationTemplates, error) {
	var dlpTemplates []DlpNotificationTemplates
	err := common.ReadAllPages(service.Client, dlpNotificationTemplatesEndpoint, &dlpTemplates)
	return dlpTemplates, err
}