Contents

#!/usr/bin/env ruby

require 'screenshots'

input_file_path = ARGV[0] || ''
unless File.file?(input_file_path)
  puts "The input file is invalid."
  puts "screenshots my-input-file.txt"
  puts "screenshots my-input-file.txt my/output/folder"
  exit -1
end

output_path = ARGV[1] || Dir.pwd
unless File.directory?(output_path)
  puts "The output folder is invalid."
  puts "screenshots my-input-file.txt"
  puts "screenshots my-input-file.txt my/output/folder"
  exit -1
end

# Load input file
urls = []
File.open(input_file_path, 'r') do |file|
  file.each_line do |line|
    urls << line.chomp if line.start_with?('http')
  end
end

if urls.empty?
  puts "No URL found."
  exit -1
end

# Generate the HTML
html = Screenshots::Processor.generate_list(urls)

# Print output on disk
output_file = File.join(output_path, 'html_for_blog.txt')
if File.file?(output_file)
  File.delete(output_file)
end

File.open(output_file, 'w') do |file|
  file.write(html)
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
screenshots-0.0.1	bin/screenshots