feat: stateid for all events and state endpoints #171

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

ggazzo merged 27 commits into main from chore/state-ids-for-all-events

Sep 15, 2025

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

-Original file line number
+Diff line change
@@ Expand Up / @@ -22,6 +22,7 @@ import { @@
     	type PduForType,
     	type PduType,
     	PersistentEventFactory,
+    	getAuthChain,
     } from '@hs/room';
     import { singleton } from 'tsyringe';
     import type { z } from 'zod';
@@ Expand Down Expand Up / @@ -679,4 +680,129 @@ export class EventService { @@
     			await new Promise((resolve) => setTimeout(resolve, 5000));
     		}
     	}
+    	async getStateIds(
+    		roomId: string,
+    		eventId: string,
+    	): Promise<{ pdu_ids: string[]; auth_chain_ids: string[] }> {
+    		try {
+    			// Ensure the event exists and belongs to the requested room
+    			const persisted = await this.eventRepository.findById(eventId);
+    			if (!persisted || persisted.event.room_id !== roomId) {
+    				throw new Error('M_NOT_FOUND');
+    			}
+    			const state = await this.stateService.findStateAtEvent(eventId);
+    			const pduIds: string[] = [];
+    			const authChainIds = new Set<string>();
+    			// Get room version for the store
+    			const roomVersion = await this.stateService.getRoomVersion(roomId);
+    			if (!roomVersion) {
+    				throw new Error('Room version not found');
+    			}
+    			// Get the event store
+    			const store = this.stateService._getStore(roomVersion);
+    			// Extract state event IDs and collect auth chain IDs
+    			for (const [, event] of state.entries()) {
+    				// PersistentEventBase has an eventId getter
+    				pduIds.push(event.eventId);
+    				// Get the complete auth chain for this event
+    				try {
+    					const authChain = await getAuthChain(event, store);
+    					for (const authEventId of authChain) {
+    						authChainIds.add(authEventId);
+    					}
+    				} catch (error) {
+    					this.logger.warn(
+    						`Failed to get auth chain for event ${event.eventId}:`,
+    						error,
+    					);
+    				}
+    			}
+    			return {
+    				pdu_ids: pduIds,
+    				auth_chain_ids: Array.from(authChainIds),
+    			};
+    		} catch (error) {
+    			this.logger.error(`Failed to get state IDs for room ${roomId}:`, error);
+    			throw error;
+    		}
+    	}
+    	async getState(
+    		roomId: string,
+    		eventId: string,
+    	): Promise<{
+    		pdus: Record<string, unknown>[];
+    		auth_chain: Record<string, unknown>[];
+    	}> {
+    		try {
+    			// Ensure the event exists and belongs to the requested room
+    			const persisted = await this.eventRepository.findById(eventId);
+    			if (!persisted || persisted.event.room_id !== roomId) {
+    				throw new Error('M_NOT_FOUND');
+    			}
+    			let state: Map<string, any>;
+    			// Get state at a specific event
+    			state = await this.stateService.findStateAtEvent(eventId);
+    			const pdus: Record<string, unknown>[] = [];
+    			const authChainIds = new Set<string>();
+    			// Get room version for the store
+    			const roomVersion = await this.stateService.getRoomVersion(roomId);
+    			if (!roomVersion) {
+    				throw new Error('Room version not found');
+    			}
+    			// Get the event store
+    			const store = this.stateService._getStore(roomVersion);
+    			// Extract state event objects and collect auth chain IDs
+    			for (const [, event] of state.entries()) {
+    				// PersistentEventBase has an event getter that contains the actual event data
+    				pdus.push(event.event);
+    				// Get the complete auth chain for this event
+    				try {
+    					const authChain = await getAuthChain(event, store);
+    					for (const authEventId of authChain) {
+    						authChainIds.add(authEventId);
+    					}
+    				} catch (error) {
+    					this.logger.warn(
+    						`Failed to get auth chain for event ${event.eventId}:`,
+    						error,
+    					);
+    				}
+    			}
+    			// Fetch the actual auth event objects
+    			const authChain: Record<string, unknown>[] = [];
+    			if (authChainIds.size > 0) {
+    				try {
+    					const authEvents = await store.getEvents(Array.from(authChainIds));
+    					for (const authEvent of authEvents) {
+    						authChain.push(authEvent.event);
+    					}
+    				} catch (error) {
+    					this.logger.warn('Failed to fetch auth event objects:', error);
+    				}
+    			}
+    			return {
+    				pdus: pdus,
+    				auth_chain: authChain,
+    			};
+    		} catch (error) {
+    			this.logger.error(`Failed to get state for room ${roomId}:`, error);
+    			throw error;
+    		}
+    	}
     }

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

feat: stateid for all events and state endpoints #171

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!