Add comments to utils

uqbar-project · Dec 28, 2024 · 7f930ec · 7f930ec
1 parent 1ffc325
commit 7f930ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/packages/client/src/highlighter.ts b/packages/client/src/highlighter.ts
@@ -9,7 +9,7 @@ const convertToVSCPosition = (position: WollokPosition) =>
   new vscode.Position(position.line, position.column)
 
 const convertToVSCTokens = (wollokNodesPlotter: WollokNodePlotter[]) =>
-  excludeNullish(wollokNodesPlotter)
+  wollokNodesPlotter
     .filter(wollokNodePlotter => {
       const { range } = wollokNodePlotter
       return !!range && range.start && range.end

diff --git a/packages/client/src/test/highlighter/utils.ts b/packages/client/src/test/highlighter/utils.ts
@@ -1,14 +1,14 @@
 import { excludeNullish, parse } from 'wollok-ts'
 import { readFileSync } from 'fs'
-import { processCode } from '../../highlighter/tokenProvider'
+import { processCode, processComments } from '../../highlighter/tokenProvider'
 import { WollokNodePlotter } from '../../highlighter/utils'
 
 export const readFileForTokenProvider = (filePath: string): WollokNodePlotter[] => {
   const parsedFile = parse.File(filePath)
   const docText = readFileSync(filePath, { encoding: 'utf-8' })
-  const tp = parsedFile.tryParse(docText)
+  const parsedPackage = parsedFile.tryParse(docText)
   const splittedLines = docText.split('\n')
-  return excludeNullish(processCode(tp.members[0], splittedLines))
+  return excludeNullish(processCode(parsedPackage.members[0], splittedLines)).concat(processComments(splittedLines))
 }
 
 export const processedByTokenType = (processed: WollokNodePlotter[], tokenType: string): IterableIterator<WollokNodePlotter> => processed.filter(token => token.tokenType === tokenType).values()