Implement oneof parsing in proto-parser.js

packages/test-client/proto-parser.js

@@ -110,23 +110,28 @@ function parseMethods(serviceBody) {
  */
 function parseMessages(content) {
   const messages = [];
-  const messageRegex = /message\s+(\w+)\s*\{([^}]+)\}/g;
+  const messageRegex = /message\s+(\w+)\s*\{/g;
   let messageMatch;
 
   while ((messageMatch = messageRegex.exec(content)) !== null) {
     const messageName = messageMatch[1];
-    const messageBody = messageMatch[2];
-
-    const message = {
-      name: messageName,
-      fields: parseFields(messageBody),
-      nestedMessages: [], // TODO: Handle nested messages
-      nestedEnums: [],    // TODO: Handle nested enums
-      oneofs: [],         // TODO: Handle oneofs
-      options: {}
-    };
-
-    messages.push(message);
+    const openBraceIndex = messageMatch.index + messageMatch[0].length - 1;
+
+    const [messageBody, endIndex] = extractBalancedBody(content, openBraceIndex);
+
+    if (messageBody !== null) {
+      const message = {
+        name: messageName,
+        fields: parseFields(messageBody),
+        nestedMessages: [], // TODO: Handle nested messages
+        nestedEnums: [],    // TODO: Handle nested enums
+        oneofs: parseOneofs(messageBody),
+        options: {}
+      };
+
+      messages.push(message);
+      messageRegex.lastIndex = endIndex;
+    }
   }
 
   return messages;
@@ -213,3 +218,57 @@ function parseEnumValues(enumBody) {
 module.exports = {
   parseProtoFile
 };
+
+/**
+ * Parse oneof definitions from message body
+ */
+function parseOneofs(messageBody) {
+  const oneofs = [];
+  const oneofRegex = /oneof\s+(\w+)\s*\{/g;
+  let oneofMatch;
+
+  while ((oneofMatch = oneofRegex.exec(messageBody)) !== null) {
+    const oneofName = oneofMatch[1];
+    const openBraceIndex = oneofMatch.index + oneofMatch[0].length - 1;
+
+    const [oneofBody, endIndex] = extractBalancedBody(messageBody, openBraceIndex);
+
+    if (oneofBody !== null) {
+      const oneof = {
+        name: oneofName,
+        fields: parseFields(oneofBody)
+      };
+
+      oneofs.push(oneof);
+      oneofRegex.lastIndex = endIndex;
+    }
+  }
+
+  return oneofs;
+}
+
+/**
+ * Helper to extract content inside balanced braces
+ * @param {string} content - Full content
+ * @param {number} startIndex - Index of the opening brace
+ * @returns {[string|null, number]} - [extracted content, end index]
+ */
+function extractBalancedBody(content, startIndex) {
+  let depth = 1;
+  let i = startIndex + 1;
+
+  while (i < content.length && depth > 0) {
+    if (content[i] === '{') {
+      depth++;
+    } else if (content[i] === '}') {
+      depth--;
+    }
+    i++;
+  }
+
+  if (depth === 0) {
+    return [content.substring(startIndex + 1, i - 1), i];
+  }
+
+  return [null, i];
+}

packages/test-client/src/service.proto

@@ -47,3 +47,11 @@ service UserService {
   // Bidirectional streaming RPC
   rpc Chat(stream StreamMessage) returns (stream StreamMessage);
 }
+
+message OneofMessage {
+  string id = 1;
+  oneof test_oneof {
+    string name = 2;
+    int32 age = 3;
+  }
+}

packages/test-client/test-parser.js

@@ -46,6 +46,13 @@ protoFile.messages.forEach(message => {
     const typeStr = field.repeated ? `repeated ${field.type}` : field.type;
     console.log(`     • ${field.name}: ${typeStr} = ${field.number}`);
   });
+  console.log(`   - oneofs: ${message.oneofs.length}`);
+  message.oneofs.forEach(oneof => {
+    console.log(`     • Oneof: ${oneof.name}`);
+    oneof.fields.forEach(field => {
+       console.log(`       - ${field.name}: ${field.type} = ${field.number}`);
+    });
+  });
 });
 console.log();
 
@@ -112,6 +119,18 @@ if (protoFile.messages.length > 0) {
       console.log(`   ❌ Field missing fields: ${missingFieldFields.join(', ')}`);
     }
   }
+
+  // Check oneof structure
+  if (message.oneofs.length > 0) {
+     const oneof = message.oneofs[0];
+     const oneofFields = ['name', 'fields'];
+     const missingOneofFields = oneofFields.filter(f => !(f in oneof));
+     if (missingOneofFields.length === 0) {
+        console.log('   ✅ Oneof structure valid');
+     } else {
+        console.log(`   ❌ Oneof missing fields: ${missingOneofFields.join(', ')}`);
+     }
+  }
 }
 
 console.log();