feat: add db.getCollectionInfos() operation support (#7)

Add support for the db.getCollectionInfos() command which returns
collection metadata including name, type, options, and info fields.

Supports optional filter argument:
- db.getCollectionInfos()
- db.getCollectionInfos({ name: "users" })

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>

Author: h3n4l

executor.go

@@ -71,6 +71,8 @@ func executeOperation(ctx context.Context, client *mongo.Client, database string
 		return executeShowCollections(ctx, client, database)
 	case opGetCollectionNames:
 		return executeGetCollectionNames(ctx, client, database)
+	case opGetCollectionInfos:
+		return executeGetCollectionInfos(ctx, client, database, op)
 	default:
 		return nil, &UnsupportedOperationError{
 			Operation: statement,
@@ -251,3 +253,40 @@ func executeShowCollections(ctx context.Context, client *mongo.Client, database
 func executeGetCollectionNames(ctx context.Context, client *mongo.Client, database string) (*Result, error) {
 	return executeShowCollections(ctx, client, database)
 }
+
+// executeGetCollectionInfos executes a db.getCollectionInfos() command.
+func executeGetCollectionInfos(ctx context.Context, client *mongo.Client, database string, op *mongoOperation) (*Result, error) {
+	filter := op.filter
+	if filter == nil {
+		filter = bson.D{}
+	}
+
+	cursor, err := client.Database(database).ListCollections(ctx, filter)
+	if err != nil {
+		return nil, fmt.Errorf("list collections failed: %w", err)
+	}
+	defer func() { _ = cursor.Close(ctx) }()
+
+	var rows []string
+	for cursor.Next(ctx) {
+		var doc bson.M
+		if err := cursor.Decode(&doc); err != nil {
+			return nil, fmt.Errorf("decode failed: %w", err)
+		}
+
+		jsonBytes, err := bson.MarshalExtJSONIndent(doc, false, false, "", "  ")
+		if err != nil {
+			return nil, fmt.Errorf("marshal failed: %w", err)
+		}
+		rows = append(rows, string(jsonBytes))
+	}
+
+	if err := cursor.Err(); err != nil {
+		return nil, fmt.Errorf("cursor error: %w", err)
+	}
+
+	return &Result{
+		Rows:     rows,
+		RowCount: len(rows),
+	}, nil
+}

executor_test.go

@@ -1362,3 +1362,75 @@ func TestAggregateMultiFieldJoin(t *testing.T) {
 	require.Contains(t, result.Rows[0], `"variation"`)
 	require.NotContains(t, result.Rows[0], `"_id"`)
 }
+
+func TestGetCollectionInfos(t *testing.T) {
+	client, cleanup := setupTestContainer(t)
+	defer cleanup()
+
+	ctx := context.Background()
+
+	// Create collections by inserting documents
+	_, err := client.Database("testdb").Collection("users").InsertOne(ctx, bson.M{"name": "alice"})
+	require.NoError(t, err)
+	_, err = client.Database("testdb").Collection("orders").InsertOne(ctx, bson.M{"item": "book"})
+	require.NoError(t, err)
+
+	gc := gomongo.NewClient(client)
+
+	// Test without filter - should return all collections
+	result, err := gc.Execute(ctx, "testdb", "db.getCollectionInfos()")
+	require.NoError(t, err)
+	require.NotNil(t, result)
+	require.Equal(t, 2, result.RowCount)
+
+	// Verify that results contain collection info structure
+	for _, row := range result.Rows {
+		require.Contains(t, row, `"name"`)
+		require.Contains(t, row, `"type"`)
+	}
+}
+
+func TestGetCollectionInfosWithFilter(t *testing.T) {
+	client, cleanup := setupTestContainer(t)
+	defer cleanup()
+
+	ctx := context.Background()
+
+	// Create collections by inserting documents
+	_, err := client.Database("testdb").Collection("users").InsertOne(ctx, bson.M{"name": "alice"})
+	require.NoError(t, err)
+	_, err = client.Database("testdb").Collection("orders").InsertOne(ctx, bson.M{"item": "book"})
+	require.NoError(t, err)
+
+	gc := gomongo.NewClient(client)
+
+	// Test with filter - should return only matching collection
+	result, err := gc.Execute(ctx, "testdb", `db.getCollectionInfos({ name: "users" })`)
+	require.NoError(t, err)
+	require.NotNil(t, result)
+	require.Equal(t, 1, result.RowCount)
+
+	// Verify that the returned collection is "users"
+	require.Contains(t, result.Rows[0], `"name": "users"`)
+	require.Contains(t, result.Rows[0], `"type": "collection"`)
+}
+
+func TestGetCollectionInfosEmptyResult(t *testing.T) {
+	client, cleanup := setupTestContainer(t)
+	defer cleanup()
+
+	ctx := context.Background()
+
+	// Create a collection
+	_, err := client.Database("testdb").Collection("users").InsertOne(ctx, bson.M{"name": "alice"})
+	require.NoError(t, err)
+
+	gc := gomongo.NewClient(client)
+
+	// Test with filter that matches no collections
+	result, err := gc.Execute(ctx, "testdb", `db.getCollectionInfos({ name: "nonexistent" })`)
+	require.NoError(t, err)
+	require.NotNil(t, result)
+	require.Equal(t, 0, result.RowCount)
+	require.Empty(t, result.Rows)
+}

go.mod

@@ -4,7 +4,7 @@ go 1.24.5
 
 require (
 	github.com/antlr4-go/antlr/v4 v4.13.1
-	github.com/bytebase/parser v0.0.0-20260113083029-8ac2a658441d
+	github.com/bytebase/parser v0.0.0-20260119035746-76308b5d11fd
 	github.com/google/uuid v1.6.0
 	github.com/stretchr/testify v1.11.1
 	github.com/testcontainers/testcontainers-go/modules/mongodb v0.40.0

go.sum

@@ -8,8 +8,8 @@ github.com/Microsoft/go-winio v0.6.2 h1:F2VQgta7ecxGYO8k3ZZz3RS8fVIXVxONVUPlNERo
 github.com/Microsoft/go-winio v0.6.2/go.mod h1:yd8OoFMLzJbo9gZq8j5qaps8bJ9aShtEA8Ipt1oGCvU=
 github.com/bytebase/antlr/v4 v4.0.0-20240827034948-8c385f108920 h1:IfmPt5o5R70NKtOrs+QHOoCgViYZelZysGxVBvV4ybA=
 github.com/bytebase/antlr/v4 v4.0.0-20240827034948-8c385f108920/go.mod h1:ykhjIPiv0IWpu3OGXCHdz2eUSe8UNGGD6baqjs8jSuU=
-github.com/bytebase/parser v0.0.0-20260113083029-8ac2a658441d h1:v6jYI26RpZgfLxlrRbGIjoMsrLtirZp/dQvPfZlA95E=
-github.com/bytebase/parser v0.0.0-20260113083029-8ac2a658441d/go.mod h1:jeak/EfutSOAuWKvrFIT2IZunhWprM7oTFBRgZ9RCxo=
+github.com/bytebase/parser v0.0.0-20260119035746-76308b5d11fd h1:JCEEza5T4CTNWZuwHe6/7mqG7Qg+q2ZiHP00UtW+NtQ=
+github.com/bytebase/parser v0.0.0-20260119035746-76308b5d11fd/go.mod h1:jeak/EfutSOAuWKvrFIT2IZunhWprM7oTFBRgZ9RCxo=
 github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
 github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
 github.com/containerd/errdefs v1.0.0 h1:tg5yIfIlQIrxYtu9ajqY42W3lpS19XqdxRQeEwYG8PI=

translator.go

@@ -20,6 +20,7 @@ const (
 	opShowDatabases
 	opShowCollections
 	opGetCollectionNames
+	opGetCollectionInfos
 )
 
 // mongoOperation represents a parsed MongoDB operation.
@@ -86,6 +87,9 @@ func (v *mongoShellVisitor) visitDbStatement(ctx mongodb.IDbStatementContext) {
 		v.visitCollectionOperation(c)
 	case *mongodb.GetCollectionNamesContext:
 		v.operation.opType = opGetCollectionNames
+	case *mongodb.GetCollectionInfosContext:
+		v.operation.opType = opGetCollectionInfos
+		v.extractGetCollectionInfosArgs(c)
 	}
 }
 
@@ -121,6 +125,53 @@ func (v *mongoShellVisitor) VisitGetCollectionNames(_ *mongodb.GetCollectionName
 	return nil
 }
 
+func (v *mongoShellVisitor) VisitGetCollectionInfos(ctx *mongodb.GetCollectionInfosContext) any {
+	v.operation.opType = opGetCollectionInfos
+	v.extractGetCollectionInfosArgs(ctx)
+	return nil
+}
+
+func (v *mongoShellVisitor) extractGetCollectionInfosArgs(ctx *mongodb.GetCollectionInfosContext) {
+	args := ctx.Arguments()
+	if args == nil {
+		return
+	}
+
+	argsCtx, ok := args.(*mongodb.ArgumentsContext)
+	if !ok {
+		return
+	}
+
+	allArgs := argsCtx.AllArgument()
+	if len(allArgs) == 0 {
+		return
+	}
+
+	// First argument is the filter (optional)
+	firstArg, ok := allArgs[0].(*mongodb.ArgumentContext)
+	if !ok {
+		return
+	}
+
+	valueCtx := firstArg.Value()
+	if valueCtx == nil {
+		return
+	}
+
+	docValue, ok := valueCtx.(*mongodb.DocumentValueContext)
+	if !ok {
+		v.err = fmt.Errorf("getCollectionInfos() filter must be a document")
+		return
+	}
+
+	filter, err := convertDocument(docValue.Document())
+	if err != nil {
+		v.err = fmt.Errorf("invalid filter: %w", err)
+		return
+	}
+	v.operation.filter = filter
+}
+
 func (v *mongoShellVisitor) extractCollectionName(ctx mongodb.ICollectionAccessContext) string {
 	switch c := ctx.(type) {
 	case *mongodb.DotAccessContext: