emoji.lua

local emoji = {}

local parse_codepoints = function (str)
  return hs.fnutils.map(
    hs.fnutils.split(str, "-"),
    function (hex)
      return tonumber(hex, 16)
    end
  )
end

local stringify_codepoints = function (codepoints)
  return hs.utf8.codepointToUTF8(table.unpack(codepoints))
end

local get_joypixels_table = function ()
  local json_file = "emoji-assets/emoji.json"
  if hs.fs.attributes(json_file, "mode") ~= "file" then
    hs.execute("git clone --depth=1 https://github.com/joypixels/emoji-assets.git")
  else
    hs.execute("cd emoji-assets && git pull")
  end
  local f = io.open(json_file, "r")
  local json = f:read("a")
  f:close()
  return hs.json.decode(json)
end

local get_slack_shortnames_table = function ()
  local json_file = "emoji-data/emoji.json"
  if hs.fs.attributes(json_file, "mode") ~= "directory" then
    hs.execute("git clone --depth=1 https://github.com/iamcal/emoji-data.git")
  else
    hs.execute("cd emoji-data && git pull")
  end
  local f = io.open(json_file, "r")
  local json = f:read("a")
  f:close()
  local shortnames_table = {}
  for _, emoji in ipairs(hs.json.decode(json)) do
    local shortnames = emoji.short_names
    local emojis = { emoji }
    if emoji.skin_variations ~= nil then
      for _, variation in pairs(emoji.skin_variations) do
        table.insert(emojis, variation)
      end
    end
    for _, variation in ipairs(emojis) do
      local moji = stringify_codepoints(parse_codepoints(variation.unified))
      shortnames_table[moji] = shortnames
    end
  end
  return shortnames_table
end

local categories = {
  "people",
  "nature",
  "food",
  "activity",
  "places",
  "travel",
  "objects",
  "symbols",
  "flags",
  "extras",
  "modifier",
}

local capitalize = function (c, r)
  return string.upper(c) .. r
end

local titlecase = function (text)
  return text:gsub("^(%l)(%w*)", capitalize):gsub("( %l)(%w*)", capitalize)
end

local underscore = function (text)
  return text:gsub("[%s%-]+", "_"):gsub("([^%w_])", function (c)
    if c == 'ō' then
      -- We just type gyoza for gyōza.
      return 'o'
    else
      return c
    end
  end)
end

local initializers = {
  choices = function ()
    local choices = {}
    local compare = function (a, b)
      return a.order < b.order
    end
    local log = hs.logger.new("emoji", "debug")
    local joypixels_table = get_joypixels_table()
    local shortnames_table = get_slack_shortnames_table()
    local image_dir = "emoji-assets/png/64"
    local apple_image_dir = "emoji-data/img-apple-64";
    local get_image = function (entry)
      local file = apple_image_dir .. "/" .. entry.code_points.fully_qualified .. ".png"
      if hs.fs.attributes(file, "mode") == "file" then
        return hs.image.imageFromPath(file)
      else
        return hs.image.imageFromPath(image_dir .. "/" .. entry.code_points.base .. ".png")
      end
    end
    for key in hs.fnutils.sortByKeyValues(joypixels_table, compare) do
      local entry = joypixels_table[key]
      local codepoints_string = entry.code_points.fully_qualified
      local codepoints = parse_codepoints(codepoints_string)
      local us = hs.fnutils.map(codepoints,
        function (cp)
          return ("U+%04X"):format(cp)
        end
      )
      local moji = stringify_codepoints(codepoints)
      local keywords = { entry.shortname }
      hs.fnutils.concat(keywords, entry.shortname_alternates)
      local add_keyword = function (new_keyword)
        if new_keyword == "" then
          print(hs.inspect(emoji))
        end
        if not hs.fnutils.some(keywords, function (keyword) return keyword == new_keyword end) then
          table.insert(keywords, new_keyword)
        end
      end
      local shortnames = shortnames_table[moji]
      if shortnames ~= nil then
        for _, name in ipairs(shortnames) do
          add_keyword(":" .. name .. ":")
        end
      end
      for _, name in ipairs(entry.keywords) do
        add_keyword(":" .. underscore(name) .. ":")
      end
      local text = titlecase(entry.name)
      local subText = "<" .. table.concat(us, " ") .. "> " .. table.concat(keywords, " ")
      table.insert(choices, {
        text = text,
        subText = subText,
        image = get_image(entry),
        chars = moji,
      })
    end

    return choices
  end
}

setmetatable(emoji, {
    __index = function (self, name)
      local fn = initializers[name]
      if fn then
        self[name] = fn()
        return self[name]
      end
    end
})

-- Creates a chooser with a given callback, which will be called with
-- a selected emoji, or nil if canceled.
emoji.chooser = function (fn)
  local chooser = hs.chooser.new(function (choice)
      if choice then
        fn(choice.chars)
      else
        fn(nil)
      end
  end)
  chooser:searchSubText(true)
  chooser:queryChangedCallback(
    function (query)
      local choices = emoji.choices
      if #query > 0 then
        local words = hs.fnutils.split(query:lower(), "%s+")
        choices = hs.fnutils.ifilter(choices, function (choice)
            return hs.fnutils.every(words, function (word)
                return choice.text:lower():find(word, 1, true) or choice.subText:lower():find(word, 1, true)
            end)
        end)
      end
      chooser:choices(choices)
    end
  )
  chooser:choices(emoji.choices)

  return chooser
end

-- Preloads the table and choices
emoji.preload = function ()
  local _ = emoji.choices
  return true
end

return emoji