71 lines
2.1 KiB
JavaScript
71 lines
2.1 KiB
JavaScript
export default function suggest (data) {
|
|
return input => {
|
|
const firstChar = input[0]
|
|
if (firstChar === ':' && data.emoji) {
|
|
return suggestEmoji(data.emoji)(input)
|
|
}
|
|
if (firstChar === '@' && data.users) {
|
|
return suggestUsers(data.users)(input)
|
|
}
|
|
return []
|
|
}
|
|
}
|
|
|
|
function suggestEmoji (emojis) {
|
|
return input => {
|
|
const noPrefix = input.toLowerCase().substr(1)
|
|
return emojis
|
|
.filter(({ displayText }) => displayText.toLowerCase().startsWith(noPrefix))
|
|
.sort((a, b) => {
|
|
let aScore = 0
|
|
let bScore = 0
|
|
|
|
// Make custom emojis a priority
|
|
aScore += Number(!!a.imageUrl) * 10
|
|
bScore += Number(!!b.imageUrl) * 10
|
|
|
|
// Sort alphabetically
|
|
const alphabetically = a.displayText > b.displayText ? 1 : -1
|
|
|
|
return bScore - aScore + alphabetically
|
|
})
|
|
}
|
|
}
|
|
|
|
function suggestUsers (users) {
|
|
return input => {
|
|
const noPrefix = input.toLowerCase().substr(1)
|
|
return users.filter(
|
|
user =>
|
|
user.screen_name.toLowerCase().startsWith(noPrefix) ||
|
|
user.name.toLowerCase().startsWith(noPrefix)
|
|
/* eslint-disable camelcase */
|
|
).slice(0, 20).sort((a, b) => {
|
|
let aScore = 0
|
|
let bScore = 0
|
|
|
|
// Matches on screen name (i.e. user@instance) makes a priority
|
|
aScore += a.screen_name.toLowerCase().startsWith(noPrefix) * 2
|
|
bScore += b.screen_name.toLowerCase().startsWith(noPrefix) * 2
|
|
|
|
// Matches on name takes second priority
|
|
aScore += a.name.toLowerCase().startsWith(noPrefix)
|
|
bScore += b.name.toLowerCase().startsWith(noPrefix)
|
|
|
|
const diff = bScore * 10 - aScore * 10
|
|
|
|
// Then sort alphabetically
|
|
const nameAlphabetically = a.name > b.name ? 1 : -1
|
|
const screenNameAlphabetically = a.screen_name > b.screen_name ? 1 : -1
|
|
|
|
return diff + nameAlphabetically + screenNameAlphabetically
|
|
}).map(({ screen_name, name, profile_image_url_original }) => ({
|
|
displayText: screen_name,
|
|
detailText: name,
|
|
imageUrl: profile_image_url_original,
|
|
replacement: '@' + screen_name
|
|
}))
|
|
/* eslint-enable camelcase */
|
|
}
|
|
}
|