Develop

src/main/java/amailp/intellij/robot/elements/RobotTokenTypes.java

@@ -17,6 +17,7 @@ public interface RobotTokenTypes {
     final IElementType KeywordsHeader = new RobotIElementType("KeywordsHeader");
     final IElementType VariablesHeader = new RobotIElementType("VariablesHeader");
     final IElementType TasksHeader = new RobotIElementType("TasksHeader");
+    final IElementType CommentsHeader = new RobotIElementType("CommentsHeader");
 
     final IElementType Ellipsis = new RobotIElementType("Ellipsis");
     final IElementType ScalarVariable = new RobotIElementType("ScalarVariable");
@@ -35,5 +36,5 @@ public interface RobotTokenTypes {
     final TokenSet WhitespacesTokens = TokenSet.create(IrrelevantSpaces, BlankLine);
     final TokenSet CommentsTokens = TokenSet.create(Comment);
     final TokenSet StringLiteralElements = TokenSet.EMPTY;
-    final TokenSet HeaderTokens = TokenSet.create(SettingsHeader, TestCasesHeader, KeywordsHeader, VariablesHeader, TasksHeader);
+    final TokenSet HeaderTokens = TokenSet.create(SettingsHeader, TestCasesHeader, KeywordsHeader, VariablesHeader, TasksHeader, CommentsHeader);
 }

src/main/resources/amailp/intellij/robot/lexer/Robot.flex

@@ -29,6 +29,7 @@ TestCasesHeader  = "*** Test Cases ***" | "*** Test Case ***"
 KeywordsHeader  = "*** Keywords ***" | "*** Keyword ***"
 VariablesHeader  = "*** Variables ***" | "*** Variable ***"
 TasksHeader  = "*** Tasks ***" | "*** Task ***"
+CommentsHeader = "*** Comments ***"
 
 
 WordChar = [^$@&%\ \t\f\r\n]
@@ -68,6 +69,7 @@ EnvironmentVariable = "%{" ~"}"
         {KeywordsHeader}                { yybegin(LINE); return RobotTokenTypes.KeywordsHeader; }
         {VariablesHeader}               { yybegin(LINE); return RobotTokenTypes.VariablesHeader; }
         {TasksHeader}                   { yybegin(LINE); return RobotTokenTypes.TasksHeader; }
+        {CommentsHeader}                { yybegin(LINE); return RobotTokenTypes.CommentsHeader;}
 
         {Ellipsis}                      { yybegin(LINE); return RobotTokenTypes.Ellipsis; }
         {ScalarVariable}                { yybegin(LINE); return RobotTokenTypes.ScalarVariable; }

src/main/scala/amailp/intellij/robot/ast/package.scala

@@ -6,6 +6,7 @@ import com.intellij.psi.tree.TokenSet
 package object ast {
   object Tables extends RobotIElementType("Tables")
 
+  object CommentsTable extends RobotIElementType("CommentsTable")
   object SettingsTable extends RobotIElementType("SettingsTable")
   object Setting extends RobotIElementType("Setting")
   object SettingName extends RobotIElementType("SettingName")
@@ -31,6 +32,5 @@ package object ast {
   object TableRow extends RobotIElementType("TableRow")
   object NonEmptyCell extends RobotIElementType("NonEmptyCell")
 
-  val tableElementTypes = TokenSet.create(SettingsTable, TestCasesTable, KeywordsTable, VariablesTable)
-
+  val tableElementTypes = TokenSet.create(SettingsTable, TestCasesTable, KeywordsTable, VariablesTable, CommentsTable)
 }

src/main/scala/amailp/intellij/robot/parser/RobotParser.scala

@@ -5,6 +5,7 @@ import com.intellij.psi.tree.IElementType
 import amailp.intellij.robot.elements.RobotTokenTypes._
 import amailp.intellij.robot.ast
 
+
 object RobotParser extends PsiParser {
   def parse(root: IElementType, builder: PsiBuilder): ASTNode = {
     val robotBuilder = new RobotPsiBuilder(builder)
@@ -17,6 +18,7 @@ object RobotParser extends PsiParser {
         case TestCasesHeader | TasksHeader => parseTableItemsWith(parseTestCaseDefinition); Some(ast.TestCasesTable)
         case KeywordsHeader => parseTableItemsWith(parseKeywordDefinition); Some(ast.KeywordsTable)
         case VariablesHeader => parseTableItemsWith(parseVariableDefinition); Some(ast.VariablesTable)
+        case CommentsHeader => parseTableItemsWith(parseCommentsTable); Some(ast.CommentsTable)
         case _ => None
       }
       tableType match {
@@ -25,6 +27,13 @@ object RobotParser extends PsiParser {
       }
     }
 
+    def parseCommentsTable(): Unit = {
+        while (!isHeader(currentType)) {
+          //val comments_line = currentText
+          advanceLexer()
+        }
+    }
+
     def parseHeaderRow(): IElementType = {
       val headerMark = mark
       val headerType = currentType

src/main/scala/amailp/intellij/robot/parser/package.scala

@@ -18,6 +18,12 @@ package object parser {
     def currentType = getTokenType
     def currentText = getTokenText
 
+    def parserCommentLine(includingTerminator: Boolean = true): Unit = {
+        while (currentType != SettingsHeader) {
+          parseRowContent()
+        }
+    }
+
     def parseRowContent(includingTerminator: Boolean = true) {
       if (!currentIsSeparator) parseCellOfType(ast.NonEmptyCell)
       parseRemainingCells()

src/test/resources/CommentsTableTest/CommentsTableKeywords.robot

@@ -0,0 +1,3 @@
+*** Comments ***
+Hello World
+*** Keywords ***

src/test/resources/CommentsTableTest/CommentsTableSettings.robot

@@ -0,0 +1,3 @@
+*** Comments ***
+Hello World
+*** Settings ***

src/test/resources/CommentsTableTest/CommentsTableTasks.robot

@@ -0,0 +1,3 @@
+*** Comments ***
+Hello World
+*** Tasks ***

src/test/resources/CommentsTableTest/CommentsTableTestCases.robot

@@ -0,0 +1,3 @@
+*** Comments ***
+Hello World
+*** Test Cases ***

src/test/resources/CommentsTableTest/CommentsTableVariables.robot

@@ -0,0 +1,3 @@
+*** Comments ***
+Hello World
+*** Variables ***

src/test/scala/amailp/intellij/robot/psi/tables/RobotCommentsTableTest.scala

@@ -0,0 +1,31 @@
+package amailp.intellij.robot.psi.tables
+
+import amailp.intellij.robot.psi.RobotPsiFile
+import amailp.intellij.robot.structureView.RobotTreeBasedStructureViewBuilder
+import amailp.intellij.robot.testFramework.RobotCodeInsightFixtureTestCase
+
+class RobotCommentsTableTest extends RobotCodeInsightFixtureTestCase {
+
+  def testCommentsSetting(): Unit = {
+    val files = myFixture
+      .configureByFiles(
+          "CommentsTableTest/CommentsTableSettings.robot"
+      )
+      .map(_.asInstanceOf[RobotPsiFile])
+
+    val oneComments = files.head
+
+    val structureViewModel =
+      new RobotTreeBasedStructureViewBuilder(oneComments).createStructureViewModel(myFixture.getEditor)
+
+    val root = structureViewModel.getRoot
+    root.getChildren should have size 1
+
+
+    val comments = root.getChildren()(0)
+    val settings = root.getChildren()(1)
+
+    comments.getPresentation.getPresentableText should equal("Comments")
+    settings.getPresentation.getPresentableText should equal("Settings")
+  }
+}

src/test/scala/amailp/intellij/robot/testFramework/RobotCodeInsightFixtureTestCase.scala

@@ -11,10 +11,7 @@ import org.scalatest.matchers.should.Matchers
 import org.scalatestplus.junit.AssertionsForJUnit
 
 @Ignore
-class RobotCodeInsightFixtureTestCase
-    extends BasePlatformTestCase
-    with Matchers
-    with AssertionsForJUnit {
+class RobotCodeInsightFixtureTestCase extends BasePlatformTestCase with Matchers with AssertionsForJUnit {
   override def getTestDataPath =
     new File(this.getClass.getClassLoader.getResource("complete.robot").toURI).getParent