feat: add state management DTOs and update profiles controller

ggazzo · ggazzo · commit 396d77fe8494 · 2025-09-15T09:52:45.000-03:00
diff --git a/packages/federation-sdk/src/repositories/state.repository.ts b/packages/federation-sdk/src/repositories/state.repository.ts
@@ -12,6 +12,7 @@ import type { StateMapKey } from '@hs/room';
 import type { PersistentEventBase } from '@hs/room';
 
 export type StateStore = {
+	_id: ObjectId;
 	delta: {
 		identifier: StateMapKey;
 		eventId: string;
diff --git a/packages/homeserver/src/controllers/federation/profiles.controller.ts b/packages/homeserver/src/controllers/federation/profiles.controller.ts
@@ -11,6 +11,12 @@ import {
 	GetMissingEventsBodyDto,
 	GetMissingEventsParamsDto,
 	GetMissingEventsResponseDto,
+	GetStateIdsParamsDto,
+	GetStateIdsQueryDto,
+	GetStateIdsResponseDto,
+	GetStateParamsDto,
+	GetStateQueryDto,
+	GetStateResponseDto,
 	MakeJoinParamsDto,
 	MakeJoinQueryDto,
 	MakeJoinResponseDto,
@@ -76,7 +82,7 @@ export const profilesPlugin = (app: Elysia) => {
 
 				const { ver } = query;
 
-				return profilesService.makeJoin(roomId, userId, ver ?? ['1']) as any;
+				return profilesService.makeJoin(roomId, userId, ver ?? ['1']);
 			},
 			{
 				params: MakeJoinParamsDto,
diff --git a/packages/homeserver/src/dtos/federation/profiles.dto.ts b/packages/homeserver/src/dtos/federation/profiles.dto.ts
@@ -76,19 +76,26 @@ export const MakeJoinQueryDto = t.Object({
 	),
 });
 
+const MembershipDto = t.Union([
+	t.Literal('join'),
+	t.Literal('leave'),
+	t.Literal('invite'),
+	t.Literal('ban'),
+	t.Literal('knock'),
+]);
+
 export const MakeJoinResponseDto = t.Object({
 	room_version: t.String({ description: 'Room version' }),
 	event: t.Object({
 		content: t.Object({
-			membership: t.Literal('join'),
+			membership: MembershipDto,
 			join_authorised_via_users_server: t.Optional(t.String()),
 		}),
 		room_id: RoomIdDto,
 		sender: UsernameDto,
 		state_key: UsernameDto,
 		type: t.Literal('m.room.member'),
 		origin_server_ts: TimestampDto,
-		origin: ServerNameDto,
 		depth: t.Optional(
 			t.Number({ description: 'Depth of the event in the DAG' }),
 		),
diff --git a/packages/homeserver/src/dtos/federation/state-ids.dto.ts b/packages/homeserver/src/dtos/federation/state-ids.dto.ts
@@ -0,0 +1,21 @@
+import { type Static, t } from 'elysia';
+import { RoomIdDto } from '../common/validation.dto';
+
+export const GetStateIdsParamsDto = t.Object({
+	roomId: RoomIdDto,
+});
+
+export const GetStateIdsQueryDto = t.Object({
+	event_id: t.String({ description: 'Event ID to get state at' }),
+});
+
+export const GetStateIdsResponseDto = t.Object({
+	pdu_ids: t.Array(t.String(), { description: 'List of state event IDs' }),
+	auth_chain_ids: t.Array(t.String(), {
+		description: 'List of auth chain event IDs',
+	}),
+});
+
+export type GetStateIdsParams = Static<typeof GetStateIdsParamsDto>;
+export type GetStateIdsQuery = Static<typeof GetStateIdsQueryDto>;
+export type GetStateIdsResponse = Static<typeof GetStateIdsResponseDto>;
diff --git a/packages/homeserver/src/dtos/federation/state.dto.ts b/packages/homeserver/src/dtos/federation/state.dto.ts
@@ -0,0 +1,23 @@
+import { type Static, t } from 'elysia';
+import { RoomIdDto } from '../common/validation.dto';
+
+export const GetStateParamsDto = t.Object({
+	roomId: RoomIdDto,
+});
+
+export const GetStateQueryDto = t.Object({
+	event_id: t.String({ description: 'Event ID to get state at' }),
+});
+
+export const GetStateResponseDto = t.Object({
+	pdus: t.Array(t.Record(t.String(), t.Any()), {
+		description: 'List of state event objects',
+	}),
+	auth_chain: t.Array(t.Record(t.String(), t.Any()), {
+		description: 'List of auth chain event objects',
+	}),
+});
+
+export type GetStateParams = Static<typeof GetStateParamsDto>;
+export type GetStateQuery = Static<typeof GetStateQueryDto>;
+export type GetStateResponse = Static<typeof GetStateResponseDto>;
diff --git a/packages/homeserver/src/dtos/index.ts b/packages/homeserver/src/dtos/index.ts
@@ -7,6 +7,8 @@ export * from './common/validation.dto';
 export * from './federation/invite.dto';
 export * from './federation/profiles.dto';
 export * from './federation/send-join.dto';
+export * from './federation/state-ids.dto';
+export * from './federation/state.dto';
 export * from './federation/transactions.dto';
 export * from './federation/versions.dto';
 
