plc4go/internal/bacnetip/bacgopes/constructeddata/constructeddata_Choice.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package constructeddata

import (
	"fmt"
	"io"
	"reflect"
	"strings"

	"github.com/pkg/errors"

	. "github.com/apache/plc4x/plc4go/internal/bacnetip/bacgopes/comp"
	. "github.com/apache/plc4x/plc4go/internal/bacnetip/bacgopes/debugging"
	. "github.com/apache/plc4x/plc4go/internal/bacnetip/bacgopes/primitivedata"
	readWriteModel "github.com/apache/plc4x/plc4go/protocols/bacnetip/readwrite/model"
)

// ChoiceContract provides a set of functions which can be overwritten by a sub struct
type ChoiceContract interface {
	GetChoiceElements() []Element
}

// ChoiceContractRequirement is needed when one want to extend using ChoiceContract
type ChoiceContractRequirement interface {
	ChoiceContract
	// SetChoice callback is needed as we work in the constructor already with the finished object // TODO: maybe we need to return as init again as it might not be finished constructing....
	SetChoice(s *Choice)
}

type Choice struct {
	_contract      ChoiceContract
	choiceElements []Element
	attr           map[string]any
}

func NewChoice(args Args, kwArgs KWArgs, options ...Option) (*Choice, error) {
	c := new(Choice)
	ApplyAppliers(options, c)
	if c._contract == nil {
		c._contract = c
	} else {
		c._contract.(ChoiceContractRequirement).SetChoice(c)
	}
	if _debug != nil {
		_debug("__init__ %r %r", args, kwArgs)
	}

	var myKWArgs = make(KWArgs)
	var otherKWArgs = make(KWArgs)
	for _, element := range c._contract.GetChoiceElements() {
		if a, ok := kwArgs[KnownKey(element.GetName())]; ok {
			myKWArgs[KnownKey(element.GetName())] = a
		}
	}
	for key, a := range kwArgs {
		if _, ok := myKWArgs[key]; !ok {
			otherKWArgs[key] = a
		}
	}

	if _debug != nil {
		_debug("    - my_kwargs: %r", myKWArgs)
	}
	if _debug != nil {
		_debug("    - other_kwargs: %r", otherKWArgs)
	}

	// set the attribute/property values for the ones provided
	for _, element := range c._contract.GetChoiceElements() {
		a, ok := myKWArgs[KnownKey(element.GetName())]
		if ok {
			c.attr[element.GetName()] = a
		}
	}
	return c, nil
}

func WithChoiceExtension(contract ChoiceContractRequirement) GenericApplier[*Choice] {
	return WrapGenericApplier(func(s *Choice) { s._contract = contract })
}

func (c *Choice) GetChoiceElements() []Element {
	return c.choiceElements
}

func (c *Choice) GetAttr(key string) (any, bool) {
	v, ok := c.attr[key]
	return v, ok
}

func (c *Choice) Encode(arg Arg) error {
	tagList, ok := arg.(*TagList)
	if !ok {
		return errors.New("arg is not a TagList")
	}
	if _debug != nil {
		_debug("encode %r", tagList)
	}

	for _, element := range c._contract.GetChoiceElements() {
		value, ok := c.attr[element.GetName()]
		if element.IsOptional() && !ok {
			continue
		}
		if !element.IsOptional() && !ok {
			return errors.Errorf("%s is a missing required element of %T", element.GetName(), c)
		}
		elementKlass, err := element.GetKlass()(Nothing())
		if err != nil {
			return errors.Wrap(err, "can't get zero object")
		}
		_, elementInSequenceOfClasses := _sequenceOfClasses[fmt.Sprintf("%T", elementKlass)]
		_, elementInListOfClasses := _listOfClasses[fmt.Sprintf("%T", elementKlass)]
		isAtomic := false
		switch elementKlass.(type) {
		case IsAtomic, IsAnyAtomic:
			isAtomic = true
		}
		isValue := reflect.TypeOf(value) == reflect.TypeOf(elementKlass)

		if elementInSequenceOfClasses || elementInListOfClasses {
			// might need to encode an opening tag
			if element.GetContext() != nil {
				openingTag, err := NewOpeningTag(*element.GetContext())
				if err != nil {
					return errors.Wrap(err, "error creating opening tag")
				}
				tagList.Append(openingTag)
			}

			helper, err := element.GetKlass()(NA(value), NoKWArgs())
			if err != nil {
				return errors.Wrap(err, "error klass element")
			}

			// encode the value
			if err := helper.Encode(tagList); err != nil {
				return errors.Wrap(err, "error encoding tag list")
			}

			// might need to encode a closing tag
			if element.GetContext() != nil {
				closingTag, err := NewClosingTag(*element.GetContext())
				if err != nil {
					return errors.Wrap(err, "error creating closing tag")
				}
				tagList.Append(closingTag)
			}
		} else if isAtomic {
			helper, err := element.GetKlass()(NA(value), NoKWArgs())
			if err != nil {
				return errors.Wrap(err, "error klass element")
			}

			// build a tag and encode the data into it
			tag, err := NewTag(nil)
			if err != nil {
				return errors.Wrap(err, "error creating tag")
			}
			// encode the value
			if err := helper.Encode(tag); err != nil {
				return errors.Wrap(err, "error encoding tag list")
			}

			// convert it to context encoding if necessary
			if element.GetContext() != nil {
				tag, err = tag.AppToContext(uint(*element.GetContext()))
				if err != nil {
					return errors.Wrap(err, "error converting tag to context")
				}
			}
			tagList.Append(tag)
		} else if isValue {
			// might need to encode an opening tag
			if element.GetContext() != nil {
				openingTag, err := NewOpeningTag(*element.GetContext())
				if err != nil {
					return errors.Wrap(err, "error creating opening tag")
				}
				tagList.Append(openingTag)
			}

			// encode the tag
			if err := value.(interface{ Encode(Arg) error }).Encode(tagList); err != nil { // TODO: ugly case, need a encode interface soon
				return errors.Wrap(err, "error encoding tag list")
			}

			// might need to encode a closing tag
			if element.GetContext() != nil {
				closingTag, err := NewClosingTag(*element.GetContext())
				if err != nil {
					return errors.Wrap(err, "error creating closing tag")
				}
				tagList.Append(closingTag)
			}
		}
	}
	return nil
}

func (c *Choice) Decode(arg Arg) error {
	tagList, ok := arg.(*TagList)
	if !ok {
		return errors.New("arg is not a TagList")
	}
	if _debug != nil {
		_debug("decode %r", tagList)
	}

	for _, element := range c._contract.GetChoiceElements() {
		tag := tagList.Peek()
		if _debug != nil {
			_debug("    - element, tag: %r, %r", element, tag)
		}

		elementKlass, err := element.GetKlass()(Nothing())
		if err != nil {
			return errors.Wrap(err, "can't get zero object")
		}
		_, elementInSequenceOfClasses := _sequenceOfClasses[elementKlass]
		_, elementInListOfClasses := _listOfClasses[elementKlass]
		isAtomic := false
		isAnyAtomic := false
		switch elementKlass.(type) {
		case IsAtomic:
			isAtomic = true
		case IsAnyAtomic:
			isAnyAtomic = true
		}
		// no more elements
		if tag == nil {
			if element.IsOptional() {
				// ommited optional element
				c.attr[element.GetName()] = nil
			} else if elementInSequenceOfClasses || elementInListOfClasses {
				// empty list
				//a.attr[element.GetName()] = nil // TODO: what to do???
			} else {
				return errors.Errorf("%s is a missing required element of %T", element.GetName(), c)
			}
		} else if tag.GetTagClass() == TagClosingTagClass {
			if !element.IsOptional() {
				return errors.Errorf("%s is a missing required element of %T", element.GetName(), c)
			}

			// ommited optional element
			// a.attr[element.GetName()] = nil // TODO: don't set it for now as we use _,ok:=
		} else if elementInSequenceOfClasses {
			// check for context encoding
			panic("finish me") // TODO: finish me
		} else if isAnyAtomic {
			// convert it to application encoding
			panic("finish me") // TODO: finish me
		} else if isAtomic {
			// convert it to application encoding
			if context := element.GetContext(); context != nil {
				if tag.GetTagClass() != readWriteModel.TagClass_CONTEXT_SPECIFIC_TAGS && tag.GetTagNumber() != uint(*context) {
					if !element.IsOptional() {
						return errors.Errorf("%s expected context tag %d", element.GetName(), *context)
					} else {
						// TODO: we don't do this
						//a.attr[element.GetName()] = nil
						continue
					}
				}
				atomicTag := tag.(interface {
					GetAppTag() readWriteModel.BACnetDataType
				})
				tag, err = tag.ContextToApp(uint(atomicTag.GetAppTag()))
				if err != nil {
					return errors.Wrap(err, "error converting tag")
				}
			} else {
				if tag.GetTagClass() != readWriteModel.TagClass_APPLICATION_TAGS || tag.GetTagNumber() != uint(elementKlass.GetAppTag()) {
					if !element.IsOptional() {
						return errors.Errorf("%s expected context tag %d", element.GetName(), context)
					} else {
						// TODO: we don't do this
						//a.attr[element.GetName()] = nil
						continue
					}
				}
			}

			// consume the tag
			tagList.Pop()

			// a helper cooperates between the atomic value and the tag
			helper, err := element.GetKlass()(NA(tag), NoKWArgs())
			if err != nil {
				return errors.Wrap(err, "error creating helper")
			}
			c.attr[element.GetName()] = helper
		} else if isAnyAtomic { // TODO: what is upstream doing here??? how???
			// convert it to application encoding
			panic("finish me") // TODO: finish me
		} else {
			panic("finish me") // TODO: finish me
		}
	}
	return nil
}

func (c *Choice) PrintDebugContents(indent int, file io.Writer, _ids []uintptr) {
	for _, element := range c.choiceElements {
		value, ok := c.GetAttr(element.GetName())
		if element.IsOptional() && !ok {
			continue
		} else if !element.IsOptional() && !ok {
			_, _ = fmt.Fprintf(file, "%s%s is a missing required element of %s\n", strings.Repeat("    ", indent), element.GetName(), StructName())
			continue
		}

		elementKlass, err := element.GetKlass()(Nothing())
		if err != nil {
			if _debug != nil {
				_debug("can't get zero object. %w", err)
			}
			return
		}
		_, elementInSequenceOfClasses := _sequenceOfClasses[elementKlass]
		_, elementInListOfClasses := _listOfClasses[elementKlass]
		isAtomic := false
		isAnyAtomic := false
		switch elementKlass.(type) {
		case IsAtomic:
			isAtomic = true
		case IsAnyAtomic:
			isAnyAtomic = true
		}
		isValueSubOfElementKlass := false // TODO: how to figure that out??
		if elementInSequenceOfClasses || elementInListOfClasses {
			_, _ = fmt.Fprintf(file, "%s%s\n", strings.Repeat("    ", indent), element.GetName())
			helper, err := element.GetKlass()(NA(value), NoKWArgs())
			if err != nil {
				if _debug != nil {
					_debug("    - helper class %s, err: %v", element.GetName(), err)
				}
				return
			}
			helper.(DebugContentPrinter).PrintDebugContents(indent+1, file, _ids)
		} else if isAtomic || isAnyAtomic {
			printVerb := VerbForType(value, 'r')
			_, _ = fmt.Fprintf(file, "%s%s = "+string(printVerb)+"\n", strings.Repeat("    ", indent), element.GetName(), value)
		} else if isValueSubOfElementKlass {
			value.(DebugContentPrinter).PrintDebugContents(indent+1, file, _ids)
		} else {
			_, _ = fmt.Fprintf(file, "%s%s must be a %T\n", strings.Repeat("    ", indent), element.GetName(), element.GetKlass())
		}
	}
}