feat(websocket): handle websocket formatting (#66)

Author: gorillamoe

bun.lock

@@ -1,6 +1,6 @@
 {
   "name": "@mistweaverco/kulala-fmt",
-  "version": "2.10.2",
+  "version": "2.11.0",
   "type": "module",
   "scripts": {
     "build": "./node_modules/.bin/rollup -c",
@@ -17,7 +17,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@mistweaverco/tree-sitter-kulala": "1.9.0",
+    "@mistweaverco/tree-sitter-kulala": "1.11.1",
     "tree-sitter": "0.22.4",
     "prettier": "^3.5.2"
   },

package.json

@@ -1,5 +1,5 @@
 import * as prettier from "prettier";
-import type { Document, Block, Header } from "./DocumentParser";
+import type { Block, Document, Header } from "./DocumentParser";
 
 function headerToPascalCase(str: string) {
   return str
@@ -140,7 +140,16 @@ const build = async (
     }
     if (block.request) {
       const formatParser = getFormatParser(block);
-      output += `${block.request.method} ${block.request.url} ${block.request.httpVersion}\n`;
+      output += `${block.request.method} ${block.request.url}`;
+      // Only include HTTP version for non-websocket requests
+      if (
+        block.request.httpVersion !== "" &&
+        !["WS", "WSS"].includes(block.request.method.toUpperCase())
+      ) {
+        output += ` ${block.request.httpVersion}\n`;
+      } else {
+        output += `\n`;
+      }
       for (const header of block.request.headers) {
         let headerKey = header.key;
         switch (block.request.httpVersion) {

src/lib/parser/DocumentBuilder.ts

@@ -1,4 +1,4 @@
-import Parser, { SyntaxNode, type Language } from "tree-sitter";
+import Parser, { type Language, SyntaxNode } from "tree-sitter";
 import Kulala from "@mistweaverco/tree-sitter-kulala";
 import { configparser } from "./../configparser";
 
@@ -209,6 +209,7 @@ const parse = (content: string): Document | null => {
         node.children.forEach((child) => {
           let postRequestScript: PostRequestScript | null = null;
           let parts, key, value;
+
           switch (child.type) {
             case "method":
               method = child.text.toUpperCase();
@@ -262,7 +263,10 @@ const parse = (content: string): Document | null => {
           method = config.defaults.http_method;
         }
 
-        if (httpVersion === "") {
+        if (
+          httpVersion === "" &&
+          !["WS", "WSS"].includes(method.toUpperCase())
+        ) {
           httpVersion = config.defaults.http_version;
         }
       }