sendtables: make property type public via Property.Type()

Author: markus-wa

pkg/demoinfocs/sendtables/entity.go

@@ -314,11 +314,16 @@ func (pe *property) Name() string {
 	return pe.entry.name
 }
 
-// Value returns current value of the property.
+// Value returns the current value of the property.
 func (pe *property) Value() PropertyValue {
 	return pe.value
 }
 
+// Type returns the data type of the property.
+func (pe *property) Type() PropertyType {
+	return PropertyType(pe.entry.prop.rawType)
+}
+
 // PropertyValueType specifies the type of a PropertyValue
 type PropertyValueType int
 

pkg/demoinfocs/sendtables/entity_test.go

@@ -68,3 +68,9 @@ func TestProperty_Name(t *testing.T) {
 
 	assert.Equal(t, "test", prop.Name())
 }
+
+func TestProperty_Type(t *testing.T) {
+	prop := property{entry: &flattenedPropEntry{prop: &sendTableProperty{rawType: 1}}}
+
+	assert.Equal(t, PropTypeFloat, prop.Type())
+}

pkg/demoinfocs/sendtables/fake/property.go

@@ -23,6 +23,11 @@ func (p *Property) Value() st.PropertyValue {
 	return p.Called().Get(0).(st.PropertyValue)
 }
 
+// Type is a mock-implementation of Property.Type().
+func (p *Property) Type() st.PropertyType {
+	return p.Called().Get(0).(st.PropertyType)
+}
+
 // OnUpdate is a mock-implementation of Property.OnUpdate().
 func (p *Property) OnUpdate(handler st.PropertyUpdateHandler) {
 	p.Called(handler)

pkg/demoinfocs/sendtables/propdecoder.go

@@ -21,6 +21,22 @@ const (
 	propTypeInt64
 )
 
+// PropertyType identifies the data type of a property.
+type PropertyType int
+
+// PropType constants are duplicated as publics so we don't have to convert ints to PropertyType for normal use.
+
+const (
+	PropTypeInt PropertyType = iota
+	PropTypeFloat
+	PropTypeVector
+	PropTypeVectorXY
+	PropTypeString
+	PropTypeArray
+	PropTypeDataTable
+	PropTypeInt64
+)
+
 const (
 	coordFractionalBitsMp             = 5
 	coordFractionalBitsMpLowPrecision = 3

pkg/demoinfocs/sendtables/property_interface.go

@@ -8,8 +8,10 @@ package sendtables
 type Property interface {
 	// Name returns the property's name.
 	Name() string
-	// Value returns current value of the property.
+	// Value returns the current value of the property.
 	Value() PropertyValue
+	// Type returns the data type of the property.
+	Type() PropertyType
 	// OnUpdate registers a handler for updates of the property's value.
 	//
 	// The handler will be called with the current value upon registration.