File size: 2,093 Bytes
867b17d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
'use client';
import { useEffect, useRef } from 'react';
import { artifactDefinitions } from './artifact';
import { initialArtifactData, useArtifact } from '@/hooks/use-artifact';
import { useDataStream } from './data-stream-provider';
export function DataStreamHandler() {
const { dataStream } = useDataStream();
const { artifact, setArtifact, setMetadata } = useArtifact();
const lastProcessedIndex = useRef(-1);
useEffect(() => {
if (!dataStream?.length) return;
const newDeltas = dataStream.slice(lastProcessedIndex.current + 1);
lastProcessedIndex.current = dataStream.length - 1;
newDeltas.forEach((delta) => {
const artifactDefinition = artifactDefinitions.find(
(artifactDefinition) => artifactDefinition.kind === artifact.kind,
);
if (artifactDefinition?.onStreamPart) {
artifactDefinition.onStreamPart({
streamPart: delta,
setArtifact,
setMetadata,
});
}
setArtifact((draftArtifact) => {
if (!draftArtifact) {
return { ...initialArtifactData, status: 'streaming' };
}
switch (delta.type) {
case 'data-id':
return {
...draftArtifact,
documentId: delta.data,
status: 'streaming',
};
case 'data-title':
return {
...draftArtifact,
title: delta.data,
status: 'streaming',
};
case 'data-kind':
return {
...draftArtifact,
kind: delta.data,
status: 'streaming',
};
case 'data-clear':
return {
...draftArtifact,
content: '',
status: 'streaming',
};
case 'data-finish':
return {
...draftArtifact,
status: 'idle',
};
default:
return draftArtifact;
}
});
});
}, [dataStream, setArtifact, setMetadata, artifact]);
return null;
}
|