Upload emoji to Mattermost

lib/slack_mattermost_emoji/mattermost_client.rb

@@ -4,12 +4,14 @@ require 'json'
 
 module SlackMattermostEmoji
   class MattermostClient
-    def initialize(username:, password:, domain:, download_path: nil, headless: true, log: true)
+    def initialize(username:, password:, domain:, emoji_path: nil, headless: true, log: true)
       @username = username
       @password = password
       @domain = domain
-      @download_path = File.expand_path(download_path || './slack-emojis')
+      @emoji_path = File.expand_path(emoji_path || './slack-emojis')
+      @user_id = nil
       @token = nil
+      @log = log
 
       if [@username, @password, @domain].any? { |el| el.nil? || el.empty? }
         raise 'Mattermost username, password, and domain are all required'
@@ -24,17 +26,49 @@ module SlackMattermostEmoji
           password: @password,
         }),
         headers: {
-          'Content-Type' => 'application/json'
+          'Content-Type' => 'application/json',
         }
       })
 
-      if response.code == 401
-        raise 'Invalid Mattermost credentials'
-      end
+      raise 'Invalid Mattermost credentials' if response.code == 401
 
+      user = JSON.parse(response.body)
+      @user_id = user['id']
       @token = response.headers['token']
     end
 
+    def upload_emoji
+      emoji = Dir["#{@emoji_path}/*"]
+      emoji_count = emoji.size
+      digits = emoji_count.to_s.size
+
+      puts "Uploading #{emoji_count} emoji... to #{@domain}"
+
+      emoji.each_with_index do |file_path, index|
+        emoji_name = File.basename(file_path, '.*')
+        index = (index + 1).to_s.rjust(digits, ' ')
+        print "#{index}/#{emoji_count} :#{emoji_name}:... " if @log
+
+        url = "https://#{@domain}/api/v4/emoji"
+        HTTParty.post(url, {
+          multipart: true,
+          body: {
+            emoji: JSON.generate({
+              creator_id: @user_id,
+              name: emoji_name,
+            }),
+            image: File.open(file_path),
+          },
+          headers: {
+            'Content-Type' => 'multipart/form-data',
+            'Authorization': "Bearer #{@token}",
+          }
+        })
+
+        puts 'success' if @log
+      end
+    end
+
     def authenticated?
       !@token.nil?
     end

lib/slack_mattermost_emoji/slack_client.rb

@@ -60,17 +60,17 @@ module SlackMattermostEmoji
       body = @driver.find_element(tag_name: 'body')
       data = JSON.parse(body.text)
       emoji_count = data['emoji'].size
+      digits = emoji_count.to_s.size
 
       puts "found #{emoji_count} emojis" if @log
 
       data['emoji'].each_with_index do |emoji, index|
+        next if URI(image_url).scheme == 'alias'
+
         name, image_url = emoji
 
-        digits = emoji_count.to_s.size
         puts "#{(index + 1).to_s.rjust(digits, ' ')}/#{emoji_count} :#{name}:" if @log
 
-        next if URI(image_url).scheme == 'alias'
-
         extension = File.extname(image_url)
         file_name = "#{name}#{extension}"
         file_path = File.join(@download_path, file_name)

spec/mattermost_client_spec.rb

@@ -31,4 +31,25 @@ RSpec.describe SlackMattermostEmoji::MattermostClient do
     )
     expect { client.authenticate }.to raise_error('Invalid Mattermost credentials')
   end
+
+  it 'should upload emojis to mattermost' do
+    client = SlackMattermostEmoji::MattermostClient.new(
+      username: ENV['MATTERMOST_USERNAME'],
+      password: ENV['MATTERMOST_PASSWORD'],
+      domain: ENV['MATTERMOST_DOMAIN'],
+      emoji_path: '~/Desktop/slack-emoji',
+    )
+    token = client.authenticate
+    client.upload_emoji
+    url = "https://#{ENV['MATTERMOST_DOMAIN']}/api/v4/emoji"
+    response = HTTParty.get(url, {
+      headers: {
+        'Authorization': "Bearer #{token}",
+      }
+    })
+
+    emoji = JSON.parse(response.body).map { |e| e['name'] }
+    emoji_on_disk = Dir[File.expand_path('~/Desktop/slack-emoji/*')].map { |e| File.basename(e, '.*') }
+    expect(emoji).to eq(emoji_on_disk)
+  end
 end