Archive a discord DM and all attachments

archive_dm.rb

#!/usr/bin/env ruby
# encoding: utf-8
Encoding.default_external = "UTF-8"
Encoding.default_internal = "UTF-8"

require "open-uri"
require "json"
require "net/https"
require "uri"
require "date"
require "fileutils"

CHANNEL = "" # Channel ID of the DM
TOKEN   = "" # My API Token
COOKIE  = "__cfduid=" # My Cookie

# Get channel object
data = open("https://discordapp.com/api/v6/channels/#{CHANNEL}", 
  "Authorization" => TOKEN,
  "Cookie" => COOKIE).read
data = JSON.parse(data)

# Username 
DM = "#{data["recipients"][0]["username"]}\##{data["recipients"][0]["discriminator"]}"
puts "Downloading DM history with #{DM}"

# Make a directory for the attachments
CHAN_DIR = "#{File.expand_path(File.dirname(__FILE__))}/#{CHANNEL}__#{DM}"
FileUtils.mkpath(CHAN_DIR) 

# Initialize
@messages = []
last = data["last_message_id"]
direction = "before"

# Check if there is already a saved archive
if File.file?("#{CHAN_DIR}.json")
  data = File.read("#{CHAN_DIR}.json")
  @messages = JSON.parse(data)
  last = @messages.last["id"]
  direction = "after"
end

more = true
n = 0

while more == true do # Loop through search result pages
  begin
    data = open("https://discordapp.com/api/v6/channels/#{CHANNEL}/messages?#{direction}=#{last}&limit=100",
      "Authorization" => TOKEN,
      "Cookie" => COOKIE).read
    data = JSON.parse(data)
    @messages << data

    if direction == "before"
      last = data.last["id"]
    else
      last = data.first["id"]
    end

    more = false if data.length == 0
    n += 1

    print "\r"
    print "Pages downloaded: #{n}"
    $stdout.flush

    sleep 1 # Sleep to avoid Rate Limit
  rescue
    more = false
  end
end

# Clean up messages for saving
@messages.flatten!.uniq!{ |msg| msg["id"] } # delete duplicates
@messages.sort_by!{ |message| -message["timestamp"] } # Sort oldest to newest
# Save log
File.write("#{CHAN_DIR}.json", JSON.pretty_generate(@messages))


# Download attachments
puts ""
puts "Saved logs"
puts "Downloading attachments..."

n = 0

@messages.each do |message|
  begin
    next unless message["attachments"].length > 0

    message["attachments"].each do |attachment|
      name = "#{attachment["id"]}__#{attachment["filename"]}"
      fn = "#{CHAN_DIR}/#{name}"
      next if File.file?(fn) # Skip files we've downloaded already

      open(attachment["url"]) do |f|
        File.open(fn,"wb") do |file|
          file.puts f.read
        end
      end

      n += 1

      print "\r"
      print "Attachments downloaded: #{n}"
      $stdout.flush

      sleep 0.5 # Sleep to avoid Rate Limit
    end
  rescue Exception => e
    puts e.message
  end
end

puts ""
puts "Complete"