João Salavisa
112 lines
4.1 KiB
TypeScript
112 lines
4.1 KiB
TypeScript
import type { StoreApi } from 'zustand';
|
|
import type { AppState } from '@/lib/state/store';
|
|
import type { WsHandlers } from '@/lib/ws/handlers/types';
|
|
import type {
|
|
GitEventPayload,
|
|
GitStatusUpdateEvent,
|
|
GitCommitCreatedEvent,
|
|
GitCommitsResetEvent,
|
|
GitSnapshotCreatedEvent,
|
|
} from '@/lib/types/git-events';
|
|
import { invalidateCumulativeDiffCache } from '@/hooks/domains/session/use-cumulative-diff';
|
|
|
|
// Handler functions for each event type
|
|
type GitEventHandlers = {
|
|
status_update: (store: StoreApi<AppState>, event: GitStatusUpdateEvent) => void;
|
|
commit_created: (store: StoreApi<AppState>, event: GitCommitCreatedEvent) => void;
|
|
commits_reset: (store: StoreApi<AppState>, event: GitCommitsResetEvent) => void;
|
|
snapshot_created: (store: StoreApi<AppState>, event: GitSnapshotCreatedEvent) => void;
|
|
};
|
|
|
|
const gitEventHandlers: GitEventHandlers = {
|
|
status_update: (store, event) => {
|
|
store.getState().setGitStatus(event.session_id, {
|
|
branch: event.status.branch,
|
|
remote_branch: event.status.remote_branch,
|
|
modified: event.status.modified,
|
|
added: event.status.added,
|
|
deleted: event.status.deleted,
|
|
untracked: event.status.untracked,
|
|
renamed: event.status.renamed,
|
|
ahead: event.status.ahead,
|
|
behind: event.status.behind,
|
|
files: event.status.files,
|
|
timestamp: event.timestamp,
|
|
});
|
|
// Invalidate cumulative diff cache when files change
|
|
invalidateCumulativeDiffCache(event.session_id);
|
|
},
|
|
|
|
commit_created: (store, event) => {
|
|
store.getState().addSessionCommit(event.session_id, {
|
|
id: event.commit.id,
|
|
session_id: event.session_id,
|
|
commit_sha: event.commit.commit_sha,
|
|
parent_sha: event.commit.parent_sha,
|
|
commit_message: event.commit.commit_message,
|
|
author_name: event.commit.author_name,
|
|
author_email: event.commit.author_email,
|
|
files_changed: event.commit.files_changed,
|
|
insertions: event.commit.insertions,
|
|
deletions: event.commit.deletions,
|
|
committed_at: event.commit.committed_at,
|
|
created_at: event.commit.created_at ?? event.timestamp,
|
|
});
|
|
// Invalidate cumulative diff cache when new commit is created
|
|
invalidateCumulativeDiffCache(event.session_id);
|
|
},
|
|
|
|
commits_reset: (store, event) => {
|
|
// Clear commits to trigger refetch in useSessionCommits hook
|
|
store.getState().clearSessionCommits(event.session_id);
|
|
// Invalidate cumulative diff cache when commits are reset
|
|
invalidateCumulativeDiffCache(event.session_id);
|
|
},
|
|
|
|
snapshot_created: (store, event) => {
|
|
store.getState().addGitSnapshot(event.session_id, {
|
|
id: event.snapshot.id,
|
|
session_id: event.snapshot.session_id,
|
|
snapshot_type: event.snapshot.snapshot_type as 'status_update' | 'pre_commit' | 'post_commit' | 'pre_stage' | 'post_stage',
|
|
branch: event.snapshot.branch,
|
|
remote_branch: event.snapshot.remote_branch,
|
|
head_commit: event.snapshot.head_commit,
|
|
base_commit: event.snapshot.base_commit,
|
|
ahead: event.snapshot.ahead,
|
|
behind: event.snapshot.behind,
|
|
files: event.snapshot.files,
|
|
triggered_by: event.snapshot.triggered_by,
|
|
created_at: event.snapshot.created_at,
|
|
});
|
|
// Invalidate cumulative diff cache when snapshot is created
|
|
invalidateCumulativeDiffCache(event.session_id);
|
|
},
|
|
};
|
|
|
|
export function registerGitStatusHandlers(store: StoreApi<AppState>): WsHandlers {
|
|
return {
|
|
'session.git.event': (message) => {
|
|
const payload = message.payload as GitEventPayload;
|
|
if (!payload.session_id || !payload.type) {
|
|
return;
|
|
}
|
|
|
|
// Use switch for proper type narrowing
|
|
switch (payload.type) {
|
|
case 'status_update':
|
|
gitEventHandlers.status_update(store, payload);
|
|
break;
|
|
case 'commit_created':
|
|
gitEventHandlers.commit_created(store, payload);
|
|
break;
|
|
case 'commits_reset':
|
|
gitEventHandlers.commits_reset(store, payload);
|
|
break;
|
|
case 'snapshot_created':
|
|
gitEventHandlers.snapshot_created(store, payload);
|
|
break;
|
|
}
|
|
},
|
|
};
|
|
}
|