seamless-streaming-v0 / streaming-react-app /src /sliceTranslationSentencesUtils.ts
Anna Sun
squash for release
2485dd8
raw
history blame
1.21 kB
import {TranslationSentences} from './types/StreamingTypes';
export function getTotalSentencesLength(
translatedSentences: TranslationSentences,
) {
return translatedSentences.reduce((acc, curr) => acc + curr.length, 0);
}
/**
* @returns A new array of strings where the total length of the strings === targetIndex,
* aka it's as if we joined all the strings together, called joined.slice(0, targetIndex), and then
* split the string back into an array of strings.
*/
export function sliceTranslationSentencesUpToIndex(
translatedSentences: TranslationSentences,
targetIndex: number,
): TranslationSentences {
return translatedSentences.reduce<TranslationSentences>((acc, sentence) => {
const accTotalLength = getTotalSentencesLength(acc);
if (accTotalLength === targetIndex) {
return acc;
}
// If adding the current sentence does not exceed the targetIndex, then add the whole sentence
if (accTotalLength + sentence.length <= targetIndex) {
return [...acc, sentence];
}
// If adding the current sentence DOES exceed the targetIndex, then slice the sentence and add it
return [...acc, sentence.slice(0, targetIndex - accTotalLength)];
}, []);
}