yandere_fe/src/components/emoji_input/suggestor.js

95 lines
3.0 KiB
JavaScript
Raw Normal View History

2019-07-17 20:40:02 -07:00
import { debounce } from 'lodash'
/**
* suggest - generates a suggestor function to be used by emoji-input
* data: object providing source information for specific types of suggestions:
* data.emoji - optional, an array of all emoji available i.e.
* (state.instance.emoji + state.instance.customEmoji)
* data.users - optional, an array of all known users
2019-07-17 20:40:02 -07:00
* updateUsersList - optional, a function to search and append to users
*
* Depending on data present one or both (or none) can be present, so if field
* doesn't support user linking you can just provide only emoji.
*/
2019-07-17 20:40:02 -07:00
const debounceUserSearch = debounce((data, input) => {
data.updateUsersList(input)
}, 500, { leading: true, trailing: false })
2019-07-17 20:40:02 -07:00
2019-06-18 12:13:03 -07:00
export default data => input => {
const firstChar = input[0]
if (firstChar === ':' && data.emoji) {
return suggestEmoji(data.emoji)(input)
}
2019-06-18 12:13:03 -07:00
if (firstChar === '@' && data.users) {
2019-07-17 20:40:02 -07:00
return suggestUsers(data)(input)
2019-06-18 12:13:03 -07:00
}
return []
}
2019-06-18 12:13:03 -07:00
export const suggestEmoji = emojis => input => {
2019-06-18 11:30:35 -07:00
const noPrefix = input.toLowerCase().substr(1)
return emojis
.filter(({ displayText }) => displayText.toLowerCase().startsWith(noPrefix))
.sort((a, b) => {
let aScore = 0
let bScore = 0
2019-06-18 11:30:35 -07:00
// Make custom emojis a priority
aScore += a.imageUrl ? 10 : 0
bScore += b.imageUrl ? 10 : 0
2019-06-18 11:30:35 -07:00
// Sort alphabetically
const alphabetically = a.displayText > b.displayText ? 1 : -1
2019-06-18 11:30:35 -07:00
return bScore - aScore + alphabetically
})
}
2019-07-17 20:40:02 -07:00
export const suggestUsers = data => input => {
2019-06-18 11:30:35 -07:00
const noPrefix = input.toLowerCase().substr(1)
2019-07-17 20:40:02 -07:00
const users = data.users
const newUsers = users.filter(
2019-06-18 11:30:35 -07:00
user =>
user.screen_name.toLowerCase().startsWith(noPrefix) ||
user.name.toLowerCase().startsWith(noPrefix)
2019-06-18 11:30:35 -07:00
/* taking only 20 results so that sorting is a bit cheaper, we display
* only 5 anyway. could be inaccurate, but we ideally we should query
* backend anyway
*/
).slice(0, 20).sort((a, b) => {
let aScore = 0
let bScore = 0
2019-06-08 07:15:48 -07:00
2019-06-18 11:30:35 -07:00
// Matches on screen name (i.e. user@instance) makes a priority
aScore += a.screen_name.toLowerCase().startsWith(noPrefix) ? 2 : 0
bScore += b.screen_name.toLowerCase().startsWith(noPrefix) ? 2 : 0
2019-06-18 11:30:35 -07:00
// Matches on name takes second priority
aScore += a.name.toLowerCase().startsWith(noPrefix) ? 1 : 0
bScore += b.name.toLowerCase().startsWith(noPrefix) ? 1 : 0
2019-06-08 07:15:48 -07:00
2019-06-18 11:30:35 -07:00
const diff = (bScore - aScore) * 10
2019-06-18 11:30:35 -07:00
// Then sort alphabetically
const nameAlphabetically = a.name > b.name ? 1 : -1
const screenNameAlphabetically = a.screen_name > b.screen_name ? 1 : -1
2019-06-08 07:15:48 -07:00
2019-06-18 11:30:35 -07:00
return diff + nameAlphabetically + screenNameAlphabetically
/* eslint-disable camelcase */
}).map(({ screen_name, name, profile_image_url_original }) => ({
displayText: screen_name,
detailText: name,
imageUrl: profile_image_url_original,
replacement: '@' + screen_name + ' '
}))
2019-07-17 20:40:02 -07:00
// BE search users if there are no matches
if (newUsers.length === 0 && data.updateUsersList) {
debounceUserSearch(data, noPrefix)
}
return newUsers
2019-06-18 11:30:35 -07:00
/* eslint-enable camelcase */
}