aboutsummaryrefslogtreecommitdiff
path: root/src/components/emoji-input/suggestor.js
blob: 40c7aa3d016d9b28c225bde84b48dd17fc9e2801 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
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 */
  }
}