Sha256: a7003e97a54db7bc431fd7407e96de0b5cf0a45252a3fe1c79b68eca9240ee85
Contents?: true
Size: 1.31 KB
Versions: 10
Compression:
Stored size: 1.31 KB
Contents
module KindleManager
class FileStore
TIME_FORMAT_FOR_FILENAME = '%Y%m%d%H%M%S'
attr_accessor :sub_dir, :dir_name, :session
def initialize(options = {})
@sub_dir = options.fetch(:sub_dir, 'books').to_s
@dir_name = options.fetch(:dir_name) do
tmp_dir_name = options[:create] ? nil : find_latest_dir_name
tmp_dir_name.presence || Time.current.strftime("%Y%m%d%H%M%S")
end
@session = options.fetch(:session, nil)
end
def target_dir
File.join(sub_dir, dir_name)
end
def list_work_dirs
Dir[File.join(Capybara.save_path, sub_dir,'*')].select{|f| File.directory? f }
end
def find_latest_dir_name
list_work_dirs.sort.last.to_s.split('/').last
end
def list_html_files
Dir[File.join(Capybara.save_path, target_dir,'*.html')].select{|f| File.file? f }
end
def html_path(time)
build_filepath(time, 'html')
end
def image_path(time)
build_filepath(time, 'png')
end
def record_page
time = Time.current
@session.save_page(html_path(time))
@session.save_screenshot(image_path(time))
end
private
def build_filepath(time, ext)
File.join(target_dir, "#{time.strftime(TIME_FORMAT_FOR_FILENAME)}#{(time.usec / 1000.0).round.to_s.rjust(3,'0')}.#{ext}")
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems