discourse/lib/tasks/cdn.rake

# frozen_string_literal: true

# cdn related tasks
#
desc "pre-stage assets on cdn"
task "assets:prestage" => :environment do |t|
  require "net/https"
  require "uri"

  def get_assets(path)
    Dir
      .glob("#{Rails.root}/public/assets/#{path}*")
      .map { |f| "/assets/#{path}#{f.split("/")[-1]}" if f =~ /[a-f0-9]{16}\.(css|js)\z/ }
      .compact
  end

  # pre-stage css/js only for now
  assets = get_assets("locales/") + get_assets("")
  puts "pre staging: #{assets.join(" ")}"

  # makes testing simpler leaving this here
  config = YAML.safe_load(File.open("#{Rails.root}/config/cdn.yml"))

  start = Time.now

  uri = URI.parse("https://client.cdn77.com/api/prefetch")
  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true
  http.verify_mode = OpenSSL::SSL::VERIFY_NONE

  failed_assets = []
  request = Net::HTTP::Post.new(uri.request_uri)
  assets.each do |asset|
    request.set_form_data(
      "id" => config["id"],
      "login" => config["login"],
      "passwd" => config["password"],
      "json" => { "prefetch_paths" => asset }.to_json,
    )

    response = http.request(request)
    json = JSON.parse(response.body)
    failed_assets.push(asset) if json["status"] != "ok"
  end

  if failed_assets.length > 0
    raise "Failed to pre-stage #{failed_assets.length}/#{assets.length} files"
  end
  puts "Done (took: #{((Time.now - start) * 1000.0).to_i}ms)"
end