type_attribute.go.tmpl

// code generated by go generate - look at type_attribute.go.tmpl for source file
package superschema

import (
	"context"

	schemaD "github.com/hashicorp/terraform-plugin-framework/datasource/schema"
	schemaR "github.com/hashicorp/terraform-plugin-framework/resource/schema"
)

var _ Attribute = {{ .TypeName }}Attribute{}

type {{ .TypeName }}Attribute struct {
	Deprecated *Deprecated
	Common     *schemaR.{{ .TypeName }}Attribute
	Resource   *schemaR.{{ .TypeName }}Attribute
	DataSource *schemaD.{{ .TypeName }}Attribute
  {{- if or (eq .TypeName "ListNested") (eq .TypeName "SetNested") (eq .TypeName "SingleNested") (eq .TypeName "MapNested") }}
	Attributes Attributes
  {{- end }}
}

// IsResource returns true if the attribute is a resource attribute.
func (s {{ .TypeName }}Attribute) IsResource() bool {
	return s.Resource != nil || s.Common != nil
}

// IsDataSource returns true if the attribute is a data source attribute.
func (s {{ .TypeName }}Attribute) IsDataSource() bool {
	return s.DataSource != nil || s.Common != nil
}

//nolint:dupl
func (s {{ .TypeName }}Attribute) GetResource(ctx context.Context) schemaR.Attribute {
	var (
		common   schemaR.{{ .TypeName }}Attribute
		resource schemaR.{{ .TypeName }}Attribute
	)

	if s.Common != nil {
		common = *s.Common
	}

	if s.Resource != nil {
		resource = *s.Resource
	}

	a := schemaR.{{ .TypeName }}Attribute{
		Required:            computeIsRequired(common, resource),
		Optional:            computeIsOptional(common, resource),
		Computed:            computeIsComputed(common, resource),
		Sensitive:           computeIsSensitive(common, resource),
		MarkdownDescription: computeMarkdownDescription(common, resource),
		Description:         computeDescription(common, resource),
		DeprecationMessage:  computeDeprecationMessage(common, resource),
    {{- if or (eq .TypeName "List") (eq .TypeName "Set") (eq .TypeName "Map") }}
		ElementType:         common.ElementType,
    {{- end }}
    {{- if or (eq .TypeName "ListNested") (eq .TypeName "SetNested") (eq .TypeName "MapNested") }}
		NestedObject: schemaR.NestedAttributeObject{
			Attributes: s.Attributes.process(ctx, resourceT).(map[string]schemaR.Attribute),
		},
    {{- end }}
	{{- if (eq .TypeName "SingleNested") }}
		Attributes: s.Attributes.process(ctx, resourceT).(map[string]schemaR.Attribute),
    {{- end }}
	}

  {{- if or (eq .TypeName "List") (eq .TypeName "Set") (eq .TypeName "Map") }}

  if s.Resource != nil {
    if s.Resource.ElementType != nil {
      a.ElementType = s.Resource.ElementType
    }
  }
  {{- end }}

	a.Validators = append(a.Validators, common.Validators...)
	a.Validators = append(a.Validators, resource.Validators...)
	a.PlanModifiers = append(a.PlanModifiers, common.PlanModifiers...)
	a.PlanModifiers = append(a.PlanModifiers, resource.PlanModifiers...)

	defaultVDescription := ""
	if s.Resource != nil {
		if s.Resource.Default != nil {
			a.Default = s.Resource.Default
			defaultVDescription = s.Resource.Default.MarkdownDescription(ctx)
		}
	}

	deprecationMessage := ""
	if s.Deprecated != nil {
		a.DeprecationMessage = s.Deprecated.DeprecationMessage
		deprecationMessage = s.Deprecated.computeDeprecatedDocumentation()
	}

	a.MarkdownDescription = genResourceAttrDescription(ctx, a.MarkdownDescription, defaultVDescription, deprecationMessage, a.Validators, a.PlanModifiers)
	return a
}

//nolint:dupl
func (s {{ .TypeName }}Attribute) GetDataSource(ctx context.Context) schemaD.Attribute {
	var (
		common     schemaR.{{ .TypeName }}Attribute
		dataSource schemaD.{{ .TypeName }}Attribute
	)

	if s.Common != nil {
		common = *s.Common
	}

	if s.DataSource != nil {
		dataSource = *s.DataSource
	}

	a := schemaD.{{ .TypeName }}Attribute{
		Required:            computeIsRequired(common, dataSource),
		Optional:            computeIsOptional(common, dataSource),
		Computed:            computeIsComputed(common, dataSource),
		Sensitive:           computeIsSensitive(common, dataSource),
		MarkdownDescription: computeMarkdownDescription(common, dataSource),
		Description:         computeDescription(common, dataSource),
		DeprecationMessage:  computeDeprecationMessage(common, dataSource),
    {{- if or (eq .TypeName "List") (eq .TypeName "Set") (eq .TypeName "Map") }}
		ElementType:         common.ElementType,
    {{- end }}
    {{- if or (eq .TypeName "ListNested") (eq .TypeName "SetNested") (eq .TypeName "MapNested") }}
		NestedObject: schemaD.NestedAttributeObject{
			Attributes: s.Attributes.process(ctx, dataSourceT).(map[string]schemaD.Attribute),
		},
    {{- end }}
	{{- if (eq .TypeName "SingleNested") }}
		Attributes: s.Attributes.process(ctx, dataSourceT).(map[string]schemaD.Attribute),
    {{- end }}
	}

  {{- if or (eq .TypeName "List") (eq .TypeName "Set") (eq .TypeName "Map") }}
  if s.DataSource != nil {
    if s.DataSource.ElementType != nil {
      a.ElementType = s.DataSource.ElementType
    }
  }
  {{- end }}

	a.Validators = append(a.Validators, common.Validators...)
	a.Validators = append(a.Validators, dataSource.Validators...)

	deprecationMessage := ""
	if s.Deprecated != nil {
		a.DeprecationMessage = s.Deprecated.DeprecationMessage
		deprecationMessage = s.Deprecated.computeDeprecatedDocumentation()
	}

	a.MarkdownDescription = genDataSourceAttrDescription(ctx, a.MarkdownDescription, deprecationMessage, a.Validators)
	return a
}