feat(documentation): add documentation for traits and contracts

Fixes #7

Author: i582

src/main/kotlin/org/tonstudio/tact/ide/documentation/DocumentationUtils.kt

@@ -34,6 +34,7 @@ object DocumentationUtils {
     val asStruct = loadKey(TactColor.STRUCT.textAttributesKey)
     val asMessage = loadKey(TactColor.MESSAGE.textAttributesKey)
     val asTrait = loadKey(TactColor.TRAIT.textAttributesKey)
+    val asContract = loadKey(TactColor.CONTRACT.textAttributesKey)
     val asPrimitive = loadKey(TactColor.PRIMITIVE.textAttributesKey)
     val asConst = loadKey(TactColor.CONSTANT.textAttributesKey)
     val asAsmInstruction = loadKey(TactColor.ASM_INSTRUCTION.textAttributesKey)

src/main/kotlin/org/tonstudio/tact/ide/documentation/TactDocumentationProvider.kt

@@ -17,6 +17,8 @@ class TactDocumentationProvider : AbstractDocumentationProvider() {
         is TactNativeFunctionDeclaration -> element.generateDoc()
         is TactStructDeclaration         -> element.generateDoc()
         is TactMessageDeclaration        -> element.generateDoc()
+        is TactTraitDeclaration          -> element.generateDoc()
+        is TactContractDeclaration       -> element.generateDoc()
         is TactPrimitiveDeclaration      -> element.generateDoc()
         is TactConstDefinition           -> element.generateDoc()
         is TactVarDefinition             -> element.generateDoc()

src/main/kotlin/org/tonstudio/tact/ide/documentation/documentation.kt

@@ -24,6 +24,7 @@ import org.tonstudio.tact.ide.documentation.DocumentationUtils.asBraces
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asBuiltin
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asComment
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asConst
+import org.tonstudio.tact.ide.documentation.DocumentationUtils.asContract
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asField
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asFunction
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asIdentifier
@@ -37,6 +38,7 @@ import org.tonstudio.tact.ide.documentation.DocumentationUtils.asStruct
 import org.tonstudio.tact.ide.documentation.DocumentationUtils.asTrait
 import org.tonstudio.tact.lang.TactSyntaxHighlighter
 import org.tonstudio.tact.lang.doc.psi.TactDocComment
+import org.tonstudio.tact.lang.psi.impl.TactPsiImplUtil.getInheritedTraitsBase
 
 private fun StringBuilder.generateOwnerDpc(element: TactNamedElement) {
     val owner = element.getOwner() ?: return
@@ -325,6 +327,49 @@ fun TactFieldDefinition.generateDoc(): String {
     }
 }
 
+fun TactContractDeclaration.generateDoc(): String {
+    return buildString {
+        append(DocumentationMarkup.DEFINITION_START)
+
+        part("contract", asKeyword)
+        part(name, asContract)
+
+        val inheritedTraits = contractType.getWithClause()?.typeList ?: emptyList()
+        formatInheritTraits(inheritedTraits)
+
+        append(DocumentationMarkup.DEFINITION_END)
+        generateCommentsPart(this@generateDoc)
+    }
+}
+
+fun TactTraitDeclaration.generateDoc(): String {
+    return buildString {
+        append(DocumentationMarkup.DEFINITION_START)
+
+        part("trait", asKeyword)
+        part(name, asTrait)
+
+        val inheritedTraits = traitType.getWithClause()?.typeList ?: emptyList()
+        formatInheritTraits(inheritedTraits)
+
+        append(DocumentationMarkup.DEFINITION_END)
+        generateCommentsPart(this@generateDoc)
+    }
+}
+
+private fun StringBuilder.formatInheritTraits(inheritedTraits: List<TactType>) {
+    val filtered = inheritedTraits.filter { it.text != "BaseTrait" }
+    if (filtered.isEmpty()) return
+
+    colorize("with ", asKeyword)
+    filtered.forEachIndexed { index, trait ->
+        colorize(trait.text, asTrait)
+        if (index != filtered.size - 1) {
+            append(", ")
+        }
+    }
+}
+
 fun TactPrimitiveDeclaration.generateDoc(): String {
     return buildString {
         append(DocumentationMarkup.DEFINITION_START)