feat: Implement detailed tracing

package.json

@@ -31,7 +31,7 @@
 			"pre-commit": "bun test"
 		}
 	},
-	"packageManager": "bun@1.1.10",
+	"packageManager": "bun@1.2.23",
 	"lint-staged": {
 		"**.{js|ts|cjs|mjs|d.cts|d.mts|jsx|tsx|json|jsonc}": [
 			"biome check --files-ignore-unknown=true --diagnostic-level=error",

packages/federation-sdk/package.json

@@ -16,6 +16,7 @@
 		"test": "bun test"
 	},
 	"dependencies": {
+		"@opentelemetry/api": "^1.9.0",
 		"@rocket.chat/emitter": "^0.31.25",
 		"@rocket.chat/federation-core": "workspace:*",
 		"@rocket.chat/federation-crypto": "workspace:*",

packages/federation-sdk/src/index.ts

@@ -68,6 +68,16 @@ export {
 	FederationValidationError,
 } from './services/federation-validation.service';
 
+// Tracing utilities - compatible with @rocket.chat/tracing
+export {
+	addSpanAttributes,
+	traced,
+	tracedClass,
+	tracerActiveSpan,
+	hasActiveSpan,
+	type ITracedClassOptions,
+} from './utils/tracing';
+
 export type HomeserverEventSignatures = {
 	'homeserver.ping': {
 		message: string;

packages/federation-sdk/src/repositories/event-staging.repository.ts

@@ -1,9 +1,15 @@
 import { generateId } from '@rocket.chat/federation-core';
 import type { EventStagingStore } from '@rocket.chat/federation-core';
-import { type EventID, Pdu, RoomID } from '@rocket.chat/federation-room';
+import {
+	type EventID,
+	type Pdu,
+	type RoomID,
+} from '@rocket.chat/federation-room';
 import type { Collection, DeleteResult, UpdateResult } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
+@tracedClass({ type: 'repository', className: 'EventStagingRepository' })
 @singleton()
 export class EventStagingRepository {
 	constructor(

packages/federation-sdk/src/repositories/event.repository.ts

@@ -2,12 +2,12 @@ import { generateId } from '@rocket.chat/federation-core';
 import type { EventStore } from '@rocket.chat/federation-core';
 import {
 	type EventID,
-	Pdu,
-	PduForType,
-	PduType,
+	type Pdu,
+	type PduForType,
+	type PduType,
 	RejectCode,
-	RoomID,
-	StateID,
+	type RoomID,
+	type StateID,
 } from '@rocket.chat/federation-room';
 import type {
 	Collection,
@@ -18,7 +18,9 @@ import type {
 	WithId,
 } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
+@tracedClass({ type: 'repository', className: 'EventRepository' })
 @singleton()
 export class EventRepository {
 	constructor(

packages/federation-sdk/src/repositories/key.repository.ts

@@ -1,5 +1,6 @@
-import { Collection } from 'mongodb';
+import type { Collection } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
 export type Key = {
 	origin: string;
@@ -8,6 +9,7 @@ export type Key = {
 	valid_until: Date;
 };
 
+@tracedClass({ type: 'repository', className: 'KeyRepository' })
 @singleton()
 export class KeyRepository {
 	constructor(

packages/federation-sdk/src/repositories/lock.repository.ts

@@ -1,12 +1,14 @@
-import { Collection } from 'mongodb';
+import type { Collection } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
 export type Lock = {
 	roomId: string;
 	instanceId: string;
 	lockedAt: Date;
 };
 
+@tracedClass({ type: 'repository', className: 'LockRepository' })
 @singleton()
 export class LockRepository {
 	constructor(

packages/federation-sdk/src/repositories/room.repository.ts

@@ -1,7 +1,8 @@
 import type { EventBase } from '@rocket.chat/federation-core';
 import type { EventID } from '@rocket.chat/federation-room';
-import { Collection } from 'mongodb';
+import type { Collection } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
 export type Room = {
 	_id: string;
@@ -16,6 +17,7 @@ export type Room = {
 	};
 };
 
+@tracedClass({ type: 'repository', className: 'RoomRepository' })
 @singleton()
 export class RoomRepository {
 	constructor(

packages/federation-sdk/src/repositories/server.repository.ts

@@ -1,5 +1,6 @@
-import { Collection } from 'mongodb';
+import type { Collection } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
 export type Server = {
 	name: string;
@@ -11,6 +12,7 @@ export type Server = {
 	};
 };
 
+@tracedClass({ type: 'repository', className: 'ServerRepository' })
 @singleton()
 export class ServerRepository {
 	constructor(

packages/federation-sdk/src/repositories/state-graph.repository.ts

@@ -8,6 +8,7 @@ import {
 } from '@rocket.chat/federation-room';
 import { type Collection, ObjectId } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
 export type StateGraphStore = {
 	_id: StateID;
@@ -25,6 +26,7 @@ export type StateGraphStore = {
 	partial: boolean;
 };
 
+@tracedClass({ type: 'repository', className: 'StateGraphRepository' })
 @singleton()
 export class StateGraphRepository {
 	constructor(

packages/federation-sdk/src/repositories/upload.repository.ts

@@ -1,6 +1,7 @@
-import { RoomID } from '@rocket.chat/federation-room';
-import { Collection } from 'mongodb';
+import type { RoomID } from '@rocket.chat/federation-room';
+import type { Collection } from 'mongodb';
 import { inject, singleton } from 'tsyringe';
+import { tracedClass } from '../utils/tracing';
 
 export type Upload = {
 	rid: string;
@@ -12,6 +13,7 @@ export type Upload = {
 	};
 };
 
+@tracedClass({ type: 'repository', className: 'UploadRepository' })
 @singleton()
 export class UploadRepository {
 	constructor(

packages/federation-sdk/src/services/edu.service.ts

@@ -4,12 +4,14 @@ import {
 	createTypingEDU,
 } from '@rocket.chat/federation-core';
 import { createLogger } from '@rocket.chat/federation-core';
-import { RoomID } from '@rocket.chat/federation-room';
+import { type RoomID } from '@rocket.chat/federation-room';
 import { singleton } from 'tsyringe';
+import { traced, tracedClass } from '../utils/tracing';
 import { ConfigService } from './config.service';
 import { FederationService } from './federation.service';
 import { StateService } from './state.service';
 
+@tracedClass({ type: 'service', className: 'EduService' })
 @singleton()
 export class EduService {
 	private readonly logger = createLogger('EduService');
@@ -20,6 +22,11 @@ export class EduService {
 		private readonly stateService: StateService,
 	) {}
 
+	@traced((roomId: RoomID, userId: string, typing: boolean) => ({
+		roomId,
+		userId,
+		typing,
+	}))
 	async sendTypingNotification(
 		roomId: RoomID,
 		userId: string,
@@ -50,6 +57,10 @@ export class EduService {
 		}
 	}
 
+	@traced((presenceUpdates: PresenceUpdate[], roomIds: RoomID[]) => ({
+		presenceUpdateCount: presenceUpdates?.length,
+		roomCount: roomIds?.length,
+	}))
 	async sendPresenceUpdateToRooms(
 		presenceUpdates: PresenceUpdate[],
 		roomIds: RoomID[],