Download emojis

lib/slack_mattermost_emoji/slack_client.rb

@@ -1,24 +1,30 @@
+require 'httparty'
+require 'json'
 require 'selenium-webdriver'
 
 module SlackMattermostEmoji
   class SlackClient
-    def initialize(username:, password:, domain:)
+    def initialize(username:, password:, domain:, download_path: nil, headless: true, log: true)
       @username = username
       @password = password
       @domain = domain
+      @download_path = File.expand_path(download_path || './slack-emojis')
+      @log = log
       @authenticated = false
 
+      Dir.mkdir(@download_path) unless Dir.exist?(@download_path)
+
       options = Selenium::WebDriver::Chrome::Options.new
-      options.headless!
+      options.headless! unless headless == false
       @driver = Selenium::WebDriver.for :chrome, options: options
 
-
       if [@username, @password, @domain].any? { |el| el.nil? || el.empty? }
         raise 'Slack username, password, and domain are all required'
       end
     end
 
     def authenticate
+      print "Logging into Slack at #{@domain} as #{@username}... " if @log
       @driver.navigate.to "https://#{@domain}"
       @driver.find_element(name: 'email').send_keys(@username)
       @driver.find_element(name: 'password').send_keys(@password)
@@ -31,9 +37,53 @@ module SlackMattermostEmoji
         nil
       end
 
+      puts 'success' if @log
+
       @authenticated = true
     end
 
+    def download_emojis
+      print "Finding emojis... " if @log
+
+      token = @driver.execute_script(<<~JS)
+        const config = JSON.parse(window.localStorage.localConfig_v2);
+        const teams = Object.values(config.teams);
+        const team = teams.find((team) => team.url.match('#{@domain}'));
+        if (!team) {
+          return null;
+        } else {
+          return team.token;
+        }
+      JS
+
+      @driver.navigate.to("https://#{@domain}/api/emoji.list?token=#{token}")
+      body = @driver.find_element(tag_name: 'body')
+      data = JSON.parse(body.text)
+      emoji_count = data['emoji'].size
+
+      puts "found #{emoji_count} emojis" if @log
+
+      data['emoji'].each_with_index do |emoji, index|
+        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)
+
+        File.open(file_path, "wb") do |f|
+          image = HTTParty.get(image_url)
+          f.write(image.body)
+        end
+      end
+
+      data['emoji']
+    end
+
     def authenticated?
       @authenticated
     end

spec/slack_client_spec.rb

@@ -6,7 +6,8 @@ RSpec.describe SlackMattermostEmoji::SlackClient do
     client = SlackMattermostEmoji::SlackClient.new(
       username: ENV['SLACK_USERNAME'],
       password: ENV['SLACK_PASSWORD'],
-      domain: ENV['SLACK_DOMAIN']
+      domain: ENV['SLACK_DOMAIN'],
+      log: false,
     )
     client.authenticate
 
@@ -17,8 +18,33 @@ RSpec.describe SlackMattermostEmoji::SlackClient do
     client = SlackMattermostEmoji::SlackClient.new(
       username: ENV['SLACK_USERNAME'],
       password: 'wrongpassword',
-      domain: ENV['SLACK_DOMAIN']
+      domain: ENV['SLACK_DOMAIN'],
+      log: false,
     )
     expect { client.authenticate }.to raise_error('Invalid credentials')
   end
+
+  it 'should download emojis from slack' do
+    download_path = File.expand_path('~/Desktop/slack-emoji')
+    FileUtils.rm_rf(download_path) if Dir.exist?(download_path)
+
+    client = SlackMattermostEmoji::SlackClient.new(
+      username: ENV['SLACK_USERNAME'],
+      password: ENV['SLACK_PASSWORD'],
+      domain: ENV['SLACK_DOMAIN'],
+      download_path: download_path,
+      log: false,
+    )
+    client.authenticate
+
+    emojis = client.download_emojis
+    emojis.each do |name, image_url|
+      next if URI(image_url).scheme == 'alias'
+
+      extension = File.extname(image_url)
+      file_name = "#{name}#{extension}"
+      file_path = File.join(download_path, file_name)
+      expect(File.exist?(file_path)).to be true
+    end
+  end
 end