Files
omnivore/packages/web/lib/highlights/normalizeHighlightRange.ts
2022-02-11 09:24:33 -08:00

53 lines
1.1 KiB
TypeScript

export function extendRangeToWordBoundaries(range: Range): void {
const { nodeValue: startValue } = range.startContainer
const { nodeValue: endValue } = range.endContainer
const startOffset = findNextWord(range.startOffset, false, startValue)
const endOffset = findNextWord(range.endOffset, true, endValue)
try {
range.setStart(range.startContainer, startOffset)
range.setEnd(range.endContainer, endOffset)
} catch {
console.warn('Unable to snap selection to the next word')
}
}
const isWhitespace = (c?: string): boolean => !!c && /\s/.test(c)
function findNextWord(
start: number,
forward: boolean,
str?: string | null
): number {
if (!str) {
return start
}
const chars = str.split('')
let i = start
if (forward) {
if (isWhitespace(chars[i - 1])) {
return i - 1
}
while (i < str.length) {
if (isWhitespace(chars[i])) {
return i
}
i++
}
} else {
if (isWhitespace(chars[i])) {
return i + 1
}
while (i > 0) {
if (isWhitespace(chars[i - 1])) {
return i
}
i--
}
}
return i
}