implement support for type definitions

ast/src/main/ast/pork.yml

@@ -330,6 +330,13 @@ types:
       type: Symbol
     - name: definitions
       type: List<StringLiteral>
+  NativeTypeDescriptor:
+    parent: Node
+    values:
+    - name: form
+      type: Symbol
+    - name: definitions
+      type: List<StringLiteral>
   IndexedBy:
     parent: Expression
     values:
@@ -337,3 +344,13 @@ types:
       type: Expression
     - name: index
       type: Expression
+  TypeDefinition:
+    parent: Definition
+    namedElementValue: symbol
+    values:
+    - name: modifiers
+      type: DefinitionModifiers
+    - name: symbol
+      type: Symbol
+    - name: nativeTypeDescriptor
+      type: NativeTypeDescriptor?

ast/src/main/graph/types.dot

@@ -40,7 +40,9 @@ digraph A {
   type_Return [shape=box,label="Return"]
   type_NoneLiteral [shape=box,label="NoneLiteral"]
   type_NativeFunctionDescriptor [shape=box,label="NativeFunctionDescriptor"]
+  type_NativeTypeDescriptor [shape=box,label="NativeTypeDescriptor"]
   type_IndexedBy [shape=box,label="IndexedBy"]
+  type_TypeDefinition [shape=box,label="TypeDefinition"]
   type_Node -> type_Expression
   type_Node -> type_Symbol
   type_Node -> type_Declaration
@@ -52,6 +54,7 @@ digraph A {
   type_Node -> type_ImportPath
   type_Node -> type_ForInItem
   type_Node -> type_NativeFunctionDescriptor
+  type_Node -> type_NativeTypeDescriptor
   type_Expression -> type_LetAssignment
   type_Expression -> type_VarAssignment
   type_Expression -> type_SetAssignment
@@ -77,6 +80,7 @@ digraph A {
   type_Expression -> type_IndexedBy
   type_Definition -> type_FunctionDefinition
   type_Definition -> type_LetDefinition
+  type_Definition -> type_TypeDefinition
   type_Declaration -> type_ImportDeclaration
   type_Definition -> type_Symbol [style=dotted]
   type_Definition -> type_DefinitionModifiers [style=dotted]
@@ -129,5 +133,10 @@ digraph A {
   type_Return -> type_Expression [style=dotted]
   type_NativeFunctionDescriptor -> type_Symbol [style=dotted]
   type_NativeFunctionDescriptor -> type_StringLiteral [style=dotted]
+  type_NativeTypeDescriptor -> type_Symbol [style=dotted]
+  type_NativeTypeDescriptor -> type_StringLiteral [style=dotted]
   type_IndexedBy -> type_Expression [style=dotted]
+  type_TypeDefinition -> type_DefinitionModifiers [style=dotted]
+  type_TypeDefinition -> type_Symbol [style=dotted]
+  type_TypeDefinition -> type_NativeTypeDescriptor [style=dotted]
 }

ast/src/main/kotlin/gay/pizza/pork/ast/FunctionLevelVisitor.kt

@@ -23,6 +23,10 @@ abstract class FunctionLevelVisitor<T> : NodeVisitor<T> {
     topLevelUsedError("FunctionDefinition")
   }
 
+  override fun visitTypeDefinition(node: TypeDefinition): T {
+    topLevelUsedError("TypeDefinition")
+  }
+
   override fun visitImportDeclaration(node: ImportDeclaration): T {
     topLevelUsedError("ImportDeclaration")
   }
@@ -39,6 +43,10 @@ abstract class FunctionLevelVisitor<T> : NodeVisitor<T> {
     topLevelUsedError("NativeFunctionDescriptor")
   }
 
+  override fun visitNativeTypeDescriptor(node: NativeTypeDescriptor): T {
+    topLevelUsedError("NativeTypeDescriptor")
+  }
+
   private fun topLevelUsedError(name: String): Nothing {
     throw RuntimeException("$name cannot be visited in a FunctionVisitor.")
   }

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NativeTypeDescriptor.kt

@@ -0,0 +1,29 @@
+// GENERATED CODE FROM PORK AST CODEGEN
+package gay.pizza.pork.ast.gen
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+
+@Serializable
+@SerialName("nativeTypeDescriptor")
+class NativeTypeDescriptor(val form: Symbol, val definitions: List<StringLiteral>) : Node() {
+  override val type: NodeType = NodeType.NativeTypeDescriptor
+
+  override fun <T> visitChildren(visitor: NodeVisitor<T>): List<T> =
+    visitor.visitAll(listOf(form), definitions)
+
+  override fun <T> visit(visitor: NodeVisitor<T>): T =
+    visitor.visitNativeTypeDescriptor(this)
+
+  override fun equals(other: Any?): Boolean {
+    if (other !is NativeTypeDescriptor) return false
+    return other.form == form && other.definitions == definitions
+  }
+
+  override fun hashCode(): Int {
+    var result = form.hashCode()
+    result = 31 * result + definitions.hashCode()
+    result = 31 * result + type.hashCode()
+    return result
+  }
+}

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NodeCoalescer.kt

@@ -68,6 +68,9 @@ class NodeCoalescer(val followChildren: Boolean = true, val handler: (Node) -> U
   override fun visitNativeFunctionDescriptor(node: NativeFunctionDescriptor): Unit =
     handle(node)
 
+  override fun visitNativeTypeDescriptor(node: NativeTypeDescriptor): Unit =
+    handle(node)
+
   override fun visitNoneLiteral(node: NoneLiteral): Unit =
     handle(node)
 
@@ -95,6 +98,9 @@ class NodeCoalescer(val followChildren: Boolean = true, val handler: (Node) -> U
   override fun visitSymbolReference(node: SymbolReference): Unit =
     handle(node)
 
+  override fun visitTypeDefinition(node: TypeDefinition): Unit =
+    handle(node)
+
   override fun visitTypeSpec(node: TypeSpec): Unit =
     handle(node)
 

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NodeParser.kt

@@ -52,6 +52,8 @@ interface NodeParser {
 
   fun parseNativeFunctionDescriptor(): NativeFunctionDescriptor
 
+  fun parseNativeTypeDescriptor(): NativeTypeDescriptor
+
   fun parseNoneLiteral(): NoneLiteral
 
   fun parseParentheses(): Parentheses
@@ -70,6 +72,8 @@ interface NodeParser {
 
   fun parseSymbolReference(): SymbolReference
 
+  fun parseTypeDefinition(): TypeDefinition
+
   fun parseTypeSpec(): TypeSpec
 
   fun parseVarAssignment(): VarAssignment

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NodeParserExtensions.kt

@@ -39,6 +39,8 @@ fun NodeParser.parse(type: NodeType): Node =
     NodeType.Return -> parseReturn()
     NodeType.NoneLiteral -> parseNoneLiteral()
     NodeType.NativeFunctionDescriptor -> parseNativeFunctionDescriptor()
+    NodeType.NativeTypeDescriptor -> parseNativeTypeDescriptor()
     NodeType.IndexedBy -> parseIndexedBy()
+    NodeType.TypeDefinition -> parseTypeDefinition()
     else -> throw RuntimeException("Unable to automatically parse type: ${type.name}")
   }

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NodeType.kt

@@ -28,6 +28,7 @@ enum class NodeType(val parent: NodeType? = null) {
   ListLiteral(Expression),
   LongLiteral(Expression),
   NativeFunctionDescriptor(Node),
+  NativeTypeDescriptor(Node),
   NoneLiteral(Expression),
   Parentheses(Expression),
   PrefixOperation(Expression),
@@ -37,6 +38,7 @@ enum class NodeType(val parent: NodeType? = null) {
   SuffixOperation(Expression),
   Symbol(Node),
   SymbolReference(Expression),
+  TypeDefinition(Definition),
   TypeSpec(Node),
   VarAssignment(Expression),
   While(Expression)

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NodeVisitor.kt

@@ -46,6 +46,8 @@ interface NodeVisitor<T> {
 
   fun visitNativeFunctionDescriptor(node: NativeFunctionDescriptor): T
 
+  fun visitNativeTypeDescriptor(node: NativeTypeDescriptor): T
+
   fun visitNoneLiteral(node: NoneLiteral): T
 
   fun visitParentheses(node: Parentheses): T
@@ -64,6 +66,8 @@ interface NodeVisitor<T> {
 
   fun visitSymbolReference(node: SymbolReference): T
 
+  fun visitTypeDefinition(node: TypeDefinition): T
+
   fun visitTypeSpec(node: TypeSpec): T
 
   fun visitVarAssignment(node: VarAssignment): T

ast/src/main/kotlin/gay/pizza/pork/ast/gen/NodeVisitorExtensions.kt

@@ -36,7 +36,9 @@ fun <T> NodeVisitor<T>.visit(node: Node): T =
     is Return -> visitReturn(node)
     is NoneLiteral -> visitNoneLiteral(node)
     is NativeFunctionDescriptor -> visitNativeFunctionDescriptor(node)
+    is NativeTypeDescriptor -> visitNativeTypeDescriptor(node)
     is IndexedBy -> visitIndexedBy(node)
+    is TypeDefinition -> visitTypeDefinition(node)
   }
 
 fun <T> NodeVisitor<T>.visitNodes(vararg nodes: Node?): List<T> =

ast/src/main/kotlin/gay/pizza/pork/ast/gen/TypeDefinition.kt

@@ -0,0 +1,30 @@
+// GENERATED CODE FROM PORK AST CODEGEN
+package gay.pizza.pork.ast.gen
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+
+@Serializable
+@SerialName("typeDefinition")
+class TypeDefinition(override val modifiers: DefinitionModifiers, override val symbol: Symbol, val nativeTypeDescriptor: NativeTypeDescriptor?) : Definition() {
+  override val type: NodeType = NodeType.TypeDefinition
+
+  override fun <T> visitChildren(visitor: NodeVisitor<T>): List<T> =
+    visitor.visitNodes(symbol, nativeTypeDescriptor)
+
+  override fun <T> visit(visitor: NodeVisitor<T>): T =
+    visitor.visitTypeDefinition(this)
+
+  override fun equals(other: Any?): Boolean {
+    if (other !is TypeDefinition) return false
+    return other.modifiers == modifiers && other.symbol == symbol && other.nativeTypeDescriptor == nativeTypeDescriptor
+  }
+
+  override fun hashCode(): Int {
+    var result = modifiers.hashCode()
+    result = 31 * result + symbol.hashCode()
+    result = 31 * result + nativeTypeDescriptor.hashCode()
+    result = 31 * result + type.hashCode()
+    return result
+  }
+}