mapping.go

package normal

import (
	"encoding/json"

	"github.com/fxamacker/cbor/v2"
)

//
// Mapping
//

type Mapping struct {
	NodeTemplate *NodeTemplate
	Relationship *Relationship
	TargetType   string
	Target       string
	Value        Constrainable
}

func NewMapping(targetType string, target string) *Mapping {
	return &Mapping{
		TargetType: targetType,
		Target:     target,
	}
}

func NewMappingValue(targetType string, value Constrainable) *Mapping {
	return &Mapping{
		TargetType: targetType,
		Value:      value,
	}
}

func (self *NodeTemplate) NewMapping(targetType string, target string) *Mapping {
	return &Mapping{
		NodeTemplate: self,
		TargetType:   targetType,
		Target:       target,
	}
}

func (self *Relationship) NewMapping(targetType string, target string) *Mapping {
	return &Mapping{
		Relationship: self,
		TargetType:   targetType,
		Target:       target,
	}
}

type MarshalableMapping struct {
	NodeTemplateName string        `json:"nodeTemplateName,omitempty" yaml:"nodeTemplateName,omitempty"`
	TargetType       string        `json:"targetType" yaml:"targetType"`
	Target           string        `json:"target,omitempty" yaml:"target,omitempty"`
	Value            Constrainable `json:"value,omitempty" yaml:"value,omitempty"`
}

func (self *Mapping) Marshalable() any {
	if self.NodeTemplate != nil {
		return &MarshalableMapping{
			NodeTemplateName: self.NodeTemplate.Name,
			TargetType:       self.TargetType,
			Target:           self.Target,
			Value:            self.Value,
		}
	} else {
		return &MarshalableMapping{
			TargetType: self.TargetType,
			Target:     self.Target,
			Value:      self.Value,
		}
	}
}

// json.Marshaler interface
func (self *Mapping) MarshalJSON() ([]byte, error) {
	return json.Marshal(self.Marshalable())
}

// yaml.Marshaler interface
func (self *Mapping) MarshalYAML() (any, error) {
	return self.Marshalable(), nil
}

// cbor.Marshaler interface
func (self *Mapping) MarshalCBOR() ([]byte, error) {
	return cbor.Marshal(self.Marshalable())
}

//
// Mappings
//

type Mappings map[string]*Mapping