Replace markdown contents with plaintext in ls request.

stratisdevex · Feb 5, 2024 · 0e2d09f · 0e2d09f
1 parent b4f5d2b
commit 0e2d09f
Showing 1 changed file with 49 additions and 1 deletion.
diff --git a/src/Stratis.VS.StratisEVM/SolidityLanguageClient.cs b/src/Stratis.VS.StratisEVM/SolidityLanguageClient.cs
@@ -264,7 +264,55 @@ public async Task HandleNotificationAsync(string methodName, JToken methodParam,
             public async Task<JToken> HandleRequestAsync(string methodName, JToken methodParam, Func<JToken, Task<JToken>> sendRequest)
             {
                 var resp = await sendRequest(methodParam);
-                Info("Request {req} {param}: {resp}", methodName, methodParam.ToString(), resp?.ToString() ?? "(null)");
+                try
+                {
+                    Info("Request {req} {param}: {resp}", methodName, methodParam.ToString(), resp?.ToString() ?? "(null)");
+                    if (resp != null)
+                    {
+                        if (methodName == "textDocument/hover")
+                        {
+                            if (resp.Root != null && resp.Root["contents"] != null && resp.Root["contents"]["kind"] != null && resp.Root["contents"]["kind"].Value<string>() == "markdown")
+                            {
+                                Info("Replace hover markup contents with plaintext.");
+                                resp.Root["contents"]["kind"] = JValue.CreateString("plaintext");
+                                if (resp.Root["contents"]["value"] != null)
+                                {
+                                    resp.Root["contents"]["value"] = JValue.CreateString(resp.Root["contents"]["value"].Value<string>().Replace("### ", "").Replace("#", ""));
+                                }
+                                else
+                                {
+                                    resp.Root["contents"]["value"] = "";
+                                }
+                            }
+                        }
+                        else if (methodName == "textDocument/completion")
+                        {
+                            if (resp.Root.Type == JTokenType.Array && resp.Root.HasValues)
+                            {
+                                Info("array detected");
+                                foreach (var f in resp.Root)
+                                {
+                                    if (f != null && f["documentation"] != null && f["documentation"]["kind"] != null && f["documentation"]["value"] != null && f["documentation"]["kind"].Value<string>() == "markdown")
+                                    {
+                                        Info("Replace completion markup contents with plaintext.");
+                                        f["documentation"]["kind"] = JValue.CreateString("plaintext");
+                                        f["documentation"]["value"] = JValue.CreateString(f["documentation"]["value"].Value<string>().Replace("### ", "").Replace("#", ""));
+                                    }
+                                }
+                            }
+
+                        }
+                    }
+                    else
+                    {
+                        Info("resp is null");
+                    }
+                }
+                catch(Exception ex) 
+                {
+                    Error(ex, "error reading response");
+                }
+
                 return resp;
             }
         }