require 'selenium-webdriver'
require 'RMagick'

class ScreenshotWorker
  include Sidekiq::Worker
  sidekiq_options queue: :screenshots, retry: true, backtrace: true

  def perform(username)
    screenshot = Tempfile.new 'neocities_screenshot'
    screenshot.close

    caps = Selenium::WebDriver::Remote::Capabilities.htmlunit javascript_enabled: true, takesScreenshot: true

    driver = Selenium::WebDriver.for :remote, url: $config['phantomjs_url'][rand($config['phantomjs_url'].length)], desired_capabilities: caps
    driver.manage.window.resize_to 1280, 720

    wait = Selenium::WebDriver::Wait.new(timeout: 30) # seconds
    wait.until {
      driver.navigate.to "http://#{username}.neocities.org"
      driver.save_screenshot screenshot.path
    }

    driver.quit

    img_list = Magick::ImageList.new
    img_list.read screenshot.path
    screenshot.unlink
    img_list.new_image(img_list.first.columns, img_list.first.rows) { self.background_color = "white" }
    img = img_list.reverse.flatten_images
    img.crop!(0, 0, 1280, 720)
    img.resize! 500, 300
    img.write File.join(DIR_ROOT, 'public', 'site_screenshots', "#{username}.jpg")
  end
end