fix(typescript): refactor JSDoc comment creation and insertion logic …

…to handle null values and exceptions more gracefully.
unit-mesh · Apr 10, 2024 · 30f301e · 30f301e
1 parent dfc3840
commit 30f301e
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/...ript/src/main/kotlin/cc/unitmesh/ide/javascript/provider/JavaScriptLivingDocumentation.kt b/...ript/src/main/kotlin/cc/unitmesh/ide/javascript/provider/JavaScriptLivingDocumentation.kt
@@ -51,14 +51,19 @@ class JavaScriptLivingDocumentation : LivingDocumentation {
                     val existingComment = JSDocumentationUtils.findOwnDocComment(target)
                         ?: findDocFallback(target)
 
-                    val createJSDocComment: PsiElement = JSPsiElementFactory.createJSDocComment(newDoc, target)
-
-                    if (existingComment != null) {
-                        existingComment.replace(createJSDocComment)
-                    } else {
-                        val parent = target.parent
-                        parent.addBefore(createJSDocComment, target)
-                        JSChangeUtil.addWs(parent.node, target.node, "\n")
+                    try {
+                        val createJSDocComment: PsiElement = JSPsiElementFactory.createJSDocComment(newDoc, target)
+
+                        if (existingComment != null) {
+                            existingComment.replace(createJSDocComment)
+                        } else {
+                            val parent = target.parent
+                            parent.addBefore(createJSDocComment, target)
+                            JSChangeUtil.addWs(parent.node, target.node, "\n")
+                        }
+                    } catch (e: Exception) {
+                        editor.document.insertString(startOffset, newDoc)
+                        codeStyleManager.reformatText(target.containingFile, startOffset, newEndOffset)
                     }
                 }