Rewrite word split imperatively for control

This commit is contained in:
Shpuld Shpuldson 2020-08-28 12:02:52 +03:00
parent f5e4ad601a
commit 0347d79bda
2 changed files with 54 additions and 42 deletions

View file

@ -4,15 +4,13 @@ export const replaceWord = (str, toReplace, replacement) => {
return str.slice(0, toReplace.start) + replacement + str.slice(toReplace.end)
}
// This seems to work fine
export const wordAtPosition = (str, pos) => {
const words = splitIntoWords(str)
const words = splitByWhitespaceBoundary(str)
const wordsWithPosition = addPositionToWords(words)
return find(wordsWithPosition, ({ start, end }) => start <= pos && end > pos)
}
// This works fine
export const addPositionToWords = (words) => {
return reduce(words, (result, word) => {
const data = {
@ -36,37 +34,36 @@ export const addPositionToWords = (words) => {
}, [])
}
// This needs to be altered, split words at space
export const splitIntoWords = (str) => {
// Split at word boundaries
const regex = /\b/
const triggers = /[@#:]+$/
let split = str.split(regex)
// Add trailing @ and # to the following word.
const words = reduce(split, (result, word) => {
if (result.length > 0) {
let previous = result.pop()
const matches = previous.match(triggers)
if (matches) {
previous = previous.replace(triggers, '')
word = matches[0] + word
}
result.push(previous)
export const splitByWhitespaceBoundary = (str) => {
let result = []
let currentWord = ''
for (let i = 0; i < str.length; i++) {
const currentChar = str[i]
// Starting a new word
if (!currentWord) {
currentWord = currentChar
continue
}
result.push(word)
return result
}, [])
return words
// current character is whitespace while word isn't, or vice versa:
// add our current word to results, start over the current word.
if (!!currentChar.trim() !== !!currentWord.trim()) {
result.push(currentWord)
currentWord = currentChar
continue
}
currentWord += currentChar
}
// Add the last word we were working on
if (currentWord) {
result.push(currentWord)
}
return result
}
const completion = {
wordAtPosition,
addPositionToWords,
splitIntoWords,
splitByWhitespaceBoundary,
replaceWord
}

View file

@ -1,8 +1,8 @@
import { replaceWord, addPositionToWords, wordAtPosition, splitIntoWords } from '../../../../../src/services/completion/completion.js'
import { replaceWord, addPositionToWords, wordAtPosition, splitByWhitespaceBoundary } from '../../../../../src/services/completion/completion.js'
describe('addPositiontoWords', () => {
it('adds the position to a word list', () => {
const words = ['hey', 'this', 'is', 'fun']
const words = ['hey', ' ', 'this', ' ', 'is', ' ', 'fun']
const expected = [
{
@ -11,19 +11,34 @@ describe('addPositiontoWords', () => {
end: 3
},
{
word: 'this',
word: ' ',
start: 3,
end: 7
end: 4
},
{
word: 'is',
start: 7,
word: 'this',
start: 4,
end: 8
},
{
word: ' ',
start: 8,
end: 9
},
{
word: 'fun',
word: 'is',
start: 9,
end: 11
},
{
word: ' ',
start: 11,
end: 12
},
{
word: 'fun',
start: 12,
end: 15
}
]
@ -33,11 +48,11 @@ describe('addPositiontoWords', () => {
})
})
describe('splitIntoWords', () => {
describe('splitByWhitespaceBoundary', () => {
it('splits at whitespace boundaries', () => {
const str = 'This is a #nice @test for you, @idiot.'
const expected = ['This', ' ', 'is', ' ', 'a', ' ', '#nice', ' ', '@test', ' ', 'for', ' ', 'you', ', ', '@idiot', '.']
const res = splitIntoWords(str)
const str = 'This is a #nice @test for you, @idiot@idiot.com'
const expected = ['This', ' ', 'is', ' ', 'a', ' ', '#nice', ' ', '@test', ' ', 'for', ' ', 'you,', ' ', '@idiot@idiot.com']
const res = splitByWhitespaceBoundary(str)
expect(res).to.eql(expected)
})
@ -57,13 +72,13 @@ describe('wordAtPosition', () => {
describe('replaceWord', () => {
it('replaces a word (with start and end) with another word in a given string', () => {
const str = 'hey @take, how are you'
const wordsWithPosition = addPositionToWords(splitIntoWords(str))
const str = 'hey @take , how are you'
const wordsWithPosition = addPositionToWords(splitByWhitespaceBoundary(str))
const toReplace = wordsWithPosition[2]
expect(toReplace.word).to.eql('@take')
const expected = 'hey @takeshitakenji, how are you'
const expected = 'hey @takeshitakenji , how are you'
const res = replaceWord(str, toReplace, '@takeshitakenji')
expect(res).to.eql(expected)
})