typehint.go

package binxml

import (
	"encoding/xml"
	"fmt"
)

const (
	HintBool     = "bool"
	HintBytes    = "b"
	HintChars    = "ch"
	HintConst    = "c"
	HintDateTime = "t"
	HintDecimal  = "dec"
	HintDouble   = "d"
	HintFloat    = "f"
	HintInt      = "i"
	HintQName    = "q"
	HintUint     = "u64"
	HintUUID     = "uid"
	HintTimeSpan = "s"
	HintUnicode  = "uch"

	HintList = "lst" // this needs to be handled differently (if at all)

	HintSeparator = ":"
)

const (
	HintXmlNamespace           = "wcfp"
	HintXmlAttrName            = "hint"
	HintXmlAttrValArrayPrefix  = "array"
	HintXmlAttrValueArrayStart = "arraystart"

	HintXmlAttrValueArrayBool     = HintXmlAttrValArrayPrefix + "B"
	HintXmlAttrValueArrayInt16    = HintXmlAttrValArrayPrefix + "I16"
	HintXmlAttrValueArrayInt32    = HintXmlAttrValArrayPrefix + "I32"
	HintXmlAttrValueArrayInt64    = HintXmlAttrValArrayPrefix + "I64"
	HintXmlAttrValueArrayFloat    = HintXmlAttrValArrayPrefix + "F"
	HintXmlAttrValueArrayDouble   = HintXmlAttrValArrayPrefix + "D"
	HintXmlAttrValueArrayDecimal  = HintXmlAttrValArrayPrefix + "Dec"
	HintXmlAttrValueArrayDateTime = HintXmlAttrValArrayPrefix + "T"
	HintXmlAttrValueArrayTimeSpan = HintXmlAttrValArrayPrefix + "S"
	HintXmlAttrValueArrayUUid     = HintXmlAttrValArrayPrefix + "UUID"
)

var (
	// maps only record types with end element since arrays must contain only these
	arrayTypeMarkerMap = map[byte]string{
		RTBoolTextWithEndElement:     HintXmlAttrValueArrayBool,
		RTInt16TextWithEndElement:    HintXmlAttrValueArrayInt16,
		RTInt32TextWithEndElement:    HintXmlAttrValueArrayInt32,
		RTInt64TextWithEndElement:    HintXmlAttrValueArrayInt64,
		RTFloatTextWithEndElement:    HintXmlAttrValueArrayFloat,
		RTDoubleTextWithEndElement:   HintXmlAttrValueArrayDouble,
		RTDecimalTextWithEndElement:  HintXmlAttrValueArrayDecimal,
		RTDateTimeTextWithEndElement: HintXmlAttrValueArrayDateTime,
		RTTimeSpanTextWithEndElement: HintXmlAttrValueArrayTimeSpan,
		RTUuidTextWithEndElement:     HintXmlAttrValueArrayUUid,
	}

	arrayTypeMarkerReverseMap = reverseMap(arrayTypeMarkerMap)
)

func TypeHint(hint string, apply bool) string {
	if apply {
		return hint + HintSeparator
	}

	return ""
}

func createArrayMarker(recordType byte) (xml.Attr, error) {
	typeStr, ok := arrayTypeMarkerMap[recordType]
	if !ok {
		return xml.Attr{}, fmt.Errorf("invalid array type %d", recordType)
	}

	attr := xml.Attr{
		Name: xml.Name{
			Space: HintXmlNamespace,
			Local: HintXmlAttrName,
		},
		Value: typeStr,
	}

	return attr, nil
}

func arrayTypeFromHint(typeStr string) (byte, error) {
	arrType, ok := arrayTypeMarkerReverseMap[typeStr]
	if !ok {
		return 0, fmt.Errorf("unknown array type hint: %s", typeStr)
	}

	return arrType, nil
}