unsafe type data field

Author: jakebailey

internal/checker/checker.go

@@ -22948,14 +22948,14 @@ func isUnaryTupleTypeNode(node *ast.Node) bool {
 	return ast.IsTupleTypeNode(node) && len(node.AsTupleTypeNode().Elements.Nodes) == 1
 }
 
-func (c *Checker) newType(flags TypeFlags, objectFlags ObjectFlags, data TypeData) *Type {
+func (c *Checker) newType(flags TypeFlags, objectFlags ObjectFlags, data typeData) *Type {
 	c.TypeCount++
 	t := data.AsType()
 	t.flags = flags
 	t.objectFlags = objectFlags &^ (ObjectFlagsCouldContainTypeVariablesComputed | ObjectFlagsCouldContainTypeVariables | ObjectFlagsMembersResolved)
 	t.id = TypeId(c.TypeCount)
 	t.checker = c
-	t.data = data
+	t.setData(data)
 	return t
 }
 
@@ -23006,7 +23006,7 @@ func (c *Checker) newUniqueESSymbolType(symbol *ast.Symbol, name string) *Type {
 }
 
 func (c *Checker) newObjectType(objectFlags ObjectFlags, symbol *ast.Symbol) *Type {
-	var data TypeData
+	var data typeData
 	switch {
 	case objectFlags&ObjectFlagsClassOrInterface != 0:
 		data = &InterfaceType{}

internal/checker/type_safe.go

@@ -0,0 +1,15 @@
+//go:build !tsunsafe
+
+package checker
+
+type typeDataField struct {
+	_data typeData
+}
+
+func (t *typeDataField) data() typeData {
+	return t._data
+}
+
+func (t *typeDataField) setData(data typeData) {
+	t._data = data
+}

internal/checker/type_unsafe.go

@@ -0,0 +1,25 @@
+//go:build tsunsafe
+
+package checker
+
+import "unsafe"
+
+type typeDataField struct {
+	_dataType unsafe.Pointer
+}
+
+type interfaceValue struct {
+	typ   unsafe.Pointer
+	value unsafe.Pointer
+}
+
+func (t *typeDataField) data() typeData {
+	var data typeData
+	(*interfaceValue)(unsafe.Pointer(&data)).typ = t._dataType
+	(*interfaceValue)(unsafe.Pointer(&data)).value = unsafe.Pointer(t)
+	return data
+}
+
+func (t *typeDataField) setData(data typeData) {
+	t._dataType = (*interfaceValue)(unsafe.Pointer(&data)).typ
+}

internal/checker/types.go

@@ -528,41 +528,41 @@ type Type struct {
 	symbol      *ast.Symbol
 	alias       *TypeAlias
 	checker     *Checker
-	data        TypeData // Type specific data
+	typeDataField
 }
 
 // Casts for concrete struct types
 
-func (t *Type) AsIntrinsicType() *IntrinsicType             { return t.data.(*IntrinsicType) }
-func (t *Type) AsLiteralType() *LiteralType                 { return t.data.(*LiteralType) }
-func (t *Type) AsUniqueESSymbolType() *UniqueESSymbolType   { return t.data.(*UniqueESSymbolType) }
-func (t *Type) AsTupleType() *TupleType                     { return t.data.(*TupleType) }
-func (t *Type) AsSingleSignatureType() *SingleSignatureType { return t.data.(*SingleSignatureType) }
+func (t *Type) AsIntrinsicType() *IntrinsicType             { return t.data().(*IntrinsicType) }
+func (t *Type) AsLiteralType() *LiteralType                 { return t.data().(*LiteralType) }
+func (t *Type) AsUniqueESSymbolType() *UniqueESSymbolType   { return t.data().(*UniqueESSymbolType) }
+func (t *Type) AsTupleType() *TupleType                     { return t.data().(*TupleType) }
+func (t *Type) AsSingleSignatureType() *SingleSignatureType { return t.data().(*SingleSignatureType) }
 func (t *Type) AsInstantiationExpressionType() *InstantiationExpressionType {
-	return t.data.(*InstantiationExpressionType)
-}
-func (t *Type) AsMappedType() *MappedType                   { return t.data.(*MappedType) }
-func (t *Type) AsReverseMappedType() *ReverseMappedType     { return t.data.(*ReverseMappedType) }
-func (t *Type) AsEvolvingArrayType() *EvolvingArrayType     { return t.data.(*EvolvingArrayType) }
-func (t *Type) AsTypeParameter() *TypeParameter             { return t.data.(*TypeParameter) }
-func (t *Type) AsUnionType() *UnionType                     { return t.data.(*UnionType) }
-func (t *Type) AsIntersectionType() *IntersectionType       { return t.data.(*IntersectionType) }
-func (t *Type) AsIndexType() *IndexType                     { return t.data.(*IndexType) }
-func (t *Type) AsIndexedAccessType() *IndexedAccessType     { return t.data.(*IndexedAccessType) }
-func (t *Type) AsTemplateLiteralType() *TemplateLiteralType { return t.data.(*TemplateLiteralType) }
-func (t *Type) AsStringMappingType() *StringMappingType     { return t.data.(*StringMappingType) }
-func (t *Type) AsSubstitutionType() *SubstitutionType       { return t.data.(*SubstitutionType) }
-func (t *Type) AsConditionalType() *ConditionalType         { return t.data.(*ConditionalType) }
+	return t.data().(*InstantiationExpressionType)
+}
+func (t *Type) AsMappedType() *MappedType                   { return t.data().(*MappedType) }
+func (t *Type) AsReverseMappedType() *ReverseMappedType     { return t.data().(*ReverseMappedType) }
+func (t *Type) AsEvolvingArrayType() *EvolvingArrayType     { return t.data().(*EvolvingArrayType) }
+func (t *Type) AsTypeParameter() *TypeParameter             { return t.data().(*TypeParameter) }
+func (t *Type) AsUnionType() *UnionType                     { return t.data().(*UnionType) }
+func (t *Type) AsIntersectionType() *IntersectionType       { return t.data().(*IntersectionType) }
+func (t *Type) AsIndexType() *IndexType                     { return t.data().(*IndexType) }
+func (t *Type) AsIndexedAccessType() *IndexedAccessType     { return t.data().(*IndexedAccessType) }
+func (t *Type) AsTemplateLiteralType() *TemplateLiteralType { return t.data().(*TemplateLiteralType) }
+func (t *Type) AsStringMappingType() *StringMappingType     { return t.data().(*StringMappingType) }
+func (t *Type) AsSubstitutionType() *SubstitutionType       { return t.data().(*SubstitutionType) }
+func (t *Type) AsConditionalType() *ConditionalType         { return t.data().(*ConditionalType) }
 
 // Casts for embedded struct types
 
-func (t *Type) AsConstrainedType() *ConstrainedType { return t.data.AsConstrainedType() }
-func (t *Type) AsStructuredType() *StructuredType   { return t.data.AsStructuredType() }
-func (t *Type) AsObjectType() *ObjectType           { return t.data.AsObjectType() }
-func (t *Type) AsTypeReference() *TypeReference     { return t.data.AsTypeReference() }
-func (t *Type) AsInterfaceType() *InterfaceType     { return t.data.AsInterfaceType() }
+func (t *Type) AsConstrainedType() *ConstrainedType { return t.data().AsConstrainedType() }
+func (t *Type) AsStructuredType() *StructuredType   { return t.data().AsStructuredType() }
+func (t *Type) AsObjectType() *ObjectType           { return t.data().AsObjectType() }
+func (t *Type) AsTypeReference() *TypeReference     { return t.data().AsTypeReference() }
+func (t *Type) AsInterfaceType() *InterfaceType     { return t.data().AsInterfaceType() }
 func (t *Type) AsUnionOrIntersectionType() *UnionOrIntersectionType {
-	return t.data.AsUnionOrIntersectionType()
+	return t.data().AsUnionOrIntersectionType()
 }
 
 func (t *Type) Distributed() []*Type {
@@ -623,7 +623,7 @@ func (t *Type) TargetTupleType() *TupleType {
 
 // TypeData
 
-type TypeData interface {
+type typeData interface {
 	AsType() *Type
 	AsConstrainedType() *ConstrainedType
 	AsStructuredType() *StructuredType