field_type_elem.go

package pgs

// FieldTypeElem describes a component of a FieldType. This type only shows up
// in repeated and map FieldTypes.
type FieldTypeElem interface {
	// ParentType returns the parent FieldType that holds this element.
	ParentType() FieldType

	// ProtoType returns the ProtoType describing this component.
	ProtoType() ProtoType

	// IsEmbed returns true if the component is an embedded message.
	IsEmbed() bool

	// IsEnum returns true if the component is an enum value.
	IsEnum() bool

	// Imports includes all external Files required by this field.
	Imports() []File

	// Enum returns the Enum associated with this FieldTypeElem. If IsEnum
	// returns false, this value will be nil.
	Enum() Enum

	// Embed returns the embedded Message associated with this FieldTypeElem. If
	// IsEmbed returns false, this value will be nil.
	Embed() Message

	setType(t FieldType)
}

type scalarE struct {
	typ   FieldType
	ptype ProtoType
}

func (s *scalarE) ParentType() FieldType { return s.typ }
func (s *scalarE) ProtoType() ProtoType  { return s.ptype }
func (s *scalarE) IsEmbed() bool         { return false }
func (s *scalarE) IsEnum() bool          { return false }
func (s *scalarE) setType(t FieldType)   { s.typ = t }
func (s *scalarE) Imports() []File       { return nil }
func (s *scalarE) Enum() Enum            { return nil }
func (s *scalarE) Embed() Message        { return nil }

type enumE struct {
	*scalarE
	enum Enum
}

func (e *enumE) IsEnum() bool { return true }
func (e *enumE) Enum() Enum   { return e.enum }

func (e *enumE) Imports() []File {
	if f := e.enum.File(); f.Name() != e.ParentType().Field().File().Name() {
		return []File{f}
	}
	return nil
}

type embedE struct {
	*scalarE
	msg Message
}

func (e *embedE) IsEmbed() bool  { return true }
func (e *embedE) Embed() Message { return e.msg }

func (e *embedE) Imports() []File {
	if f := e.msg.File(); f.Name() != e.ParentType().Field().File().Name() {
		return []File{f}
	}
	return nil
}

var (
	_ FieldTypeElem = (*scalarE)(nil)
	_ FieldTypeElem = (*enumE)(nil)
	_ FieldTypeElem = (*embedE)(nil)
)