44 lines
1.1 KiB
Ruby
44 lines
1.1 KiB
Ruby
|
|
module Webgalen
|
|
class Screenshot
|
|
def self.start(sitemap:, output_path: , device:, orientation:)
|
|
config = YAML.load File.open(sitemap)
|
|
urls = config['pages']
|
|
|
|
# Start workpools
|
|
cores = Celluloid.cores
|
|
screenshot_pool = ScreenshotActor.pool(
|
|
size: cores,
|
|
args: [ output_path ]
|
|
)
|
|
crop_pool = CropPngActor.pool(
|
|
size: cores,
|
|
args: [ output_path ]
|
|
)
|
|
|
|
futures =
|
|
urls.
|
|
map do |url|
|
|
Celluloid::Future.new { Work.new(input: {url: url}, id: url) }
|
|
end.
|
|
map do |future_work|
|
|
screenshot_pool.future.perform(future_work)
|
|
end.
|
|
map do |future_work|
|
|
crop_pool.future.perform(future_work)
|
|
end
|
|
|
|
puts '(main) Waiting for remaining jobs'
|
|
results = futures.map { |future_work| future_work.value }
|
|
|
|
puts '(main) Cleaning'
|
|
results
|
|
.map { |result| result.shift!.artefacts }
|
|
.flatten
|
|
#.select { |artefact| artefact[:type] == :file }
|
|
#.each { |file| FileUtils.rm_f file }
|
|
|
|
end
|
|
end
|
|
end
|