Added tests with Rspec, slight cleanup
This commit is contained in:
parent
62170478ca
commit
2e9f25cfa1
35
lib/file_nav.rb
Normal file
35
lib/file_nav.rb
Normal file
|
@ -0,0 +1,35 @@
|
|||
# frozen_string_literal: true
|
||||
# TODO: add recursion support to directories
|
||||
class DirTermine
|
||||
|
||||
attr_accessor :save_dir, :source_dir
|
||||
|
||||
def initialize
|
||||
@save_dir = save_dir
|
||||
@source_dir = source_dir
|
||||
end
|
||||
|
||||
def default_dir
|
||||
File.expand_path('~/Pictures/stickerized')
|
||||
end
|
||||
|
||||
def create_dir(dir)
|
||||
Dir.mkdir(File.expand_path(dir)) unless File.directory?(File.expand_path(dir))
|
||||
end
|
||||
|
||||
def destination_dir
|
||||
puts 'Please enter desired save directory (Leave blank for ~/Pictures/stickerized)'
|
||||
dir = gets.chomp
|
||||
dir = default_dir if dir.empty? || dir.nil?
|
||||
create_dir(dir)
|
||||
File.expand_path(dir)
|
||||
end
|
||||
|
||||
def origin_dir
|
||||
puts 'Please enter directory containing pictures to stickerize (Leave blank for current directory)'
|
||||
dir = gets.chomp
|
||||
dir = Dir.getwd if dir.empty?
|
||||
File.expand_path(dir)
|
||||
end
|
||||
|
||||
end
|
7
lib/main.rb
Executable file
7
lib/main.rb
Executable file
|
@ -0,0 +1,7 @@
|
|||
#!/usr/bin/env ruby
|
||||
# frozen_string_literal: true
|
||||
|
||||
require_relative 'stickerizer'
|
||||
|
||||
sticker_machine = Stickerizer.new
|
||||
sticker_machine.convert_images
|
64
lib/stickerizer.rb
Normal file
64
lib/stickerizer.rb
Normal file
|
@ -0,0 +1,64 @@
|
|||
# frozen_string_literal: true
|
||||
|
||||
require 'mini_magick'
|
||||
require_relative 'file_nav'
|
||||
|
||||
class Stickerizer
|
||||
Extension = Struct.new(:extension)
|
||||
|
||||
def file_ext(filename)
|
||||
Extension.new(filename[-4..-1])
|
||||
end
|
||||
|
||||
ImageFormat = Struct.new(:size, :format)
|
||||
|
||||
def telegram_format
|
||||
ImageFormat.new('512x512', 'PNG')
|
||||
end
|
||||
|
||||
def initialize
|
||||
@convert = DirTermine.new
|
||||
end
|
||||
|
||||
def stickerize(pic_file, save_dir)
|
||||
sticker_format = telegram_format
|
||||
image = MiniMagick::Image.open(pic_file)
|
||||
image.resize sticker_format.size
|
||||
image.format sticker_format.format
|
||||
image.write(save_dir + '/' + stickername(pic_file))
|
||||
end
|
||||
|
||||
def stickername(filename)
|
||||
file_only = filename.sub %r{/.+/}, ''
|
||||
"#{file_only[0..-5]}_sticker.png"
|
||||
end
|
||||
|
||||
def origin_dir
|
||||
@convert.origin_dir
|
||||
end
|
||||
|
||||
def dest_dir
|
||||
@convert.destination_dir
|
||||
end
|
||||
|
||||
def valid?(file)
|
||||
# TODO: Change this to check file types instead of relying on filenames
|
||||
valid = false
|
||||
file_info = file_ext(file)
|
||||
valid = true if file_info.extension.downcase == '.png'
|
||||
valid = true if file_info.extension.downcase == '.jpg'
|
||||
valid = true if file_info.extension.downcase == '.gif'
|
||||
puts "File #{file}: valid: #{valid}"
|
||||
valid
|
||||
end
|
||||
|
||||
def convert_images
|
||||
origin = origin_dir
|
||||
destination = dest_dir
|
||||
Dir.children(origin).each do |unstickerized|
|
||||
next unless valid?(unstickerized)
|
||||
|
||||
stickerize("#{origin}/#{unstickerized}", destination)
|
||||
end
|
||||
end
|
||||
end
|
100
spec/spec_helper.rb
Normal file
100
spec/spec_helper.rb
Normal file
|
@ -0,0 +1,100 @@
|
|||
# This file was generated by the `rspec --init` command. Conventionally, all
|
||||
# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
|
||||
# The generated `.rspec` file contains `--require spec_helper` which will cause
|
||||
# this file to always be loaded, without a need to explicitly require it in any
|
||||
# files.
|
||||
#
|
||||
# Given that it is always loaded, you are encouraged to keep this file as
|
||||
# light-weight as possible. Requiring heavyweight dependencies from this file
|
||||
# will add to the boot time of your test suite on EVERY test run, even for an
|
||||
# individual file that may not need all of that loaded. Instead, consider making
|
||||
# a separate helper file that requires the additional dependencies and performs
|
||||
# the additional setup, and require it from the spec files that actually need
|
||||
# it.
|
||||
#
|
||||
# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
|
||||
RSpec.configure do |config|
|
||||
# rspec-expectations config goes here. You can use an alternate
|
||||
# assertion/expectation library such as wrong or the stdlib/minitest
|
||||
# assertions if you prefer.
|
||||
config.expect_with :rspec do |expectations|
|
||||
# This option will default to `true` in RSpec 4. It makes the `description`
|
||||
# and `failure_message` of custom matchers include text for helper methods
|
||||
# defined using `chain`, e.g.:
|
||||
# be_bigger_than(2).and_smaller_than(4).description
|
||||
# # => "be bigger than 2 and smaller than 4"
|
||||
# ...rather than:
|
||||
# # => "be bigger than 2"
|
||||
expectations.include_chain_clauses_in_custom_matcher_descriptions = true
|
||||
end
|
||||
|
||||
# rspec-mocks config goes here. You can use an alternate test double
|
||||
# library (such as bogus or mocha) by changing the `mock_with` option here.
|
||||
config.mock_with :rspec do |mocks|
|
||||
# Prevents you from mocking or stubbing a method that does not exist on
|
||||
# a real object. This is generally recommended, and will default to
|
||||
# `true` in RSpec 4.
|
||||
mocks.verify_partial_doubles = true
|
||||
end
|
||||
|
||||
# This option will default to `:apply_to_host_groups` in RSpec 4 (and will
|
||||
# have no way to turn it off -- the option exists only for backwards
|
||||
# compatibility in RSpec 3). It causes shared context metadata to be
|
||||
# inherited by the metadata hash of host groups and examples, rather than
|
||||
# triggering implicit auto-inclusion in groups with matching metadata.
|
||||
config.shared_context_metadata_behavior = :apply_to_host_groups
|
||||
|
||||
# The settings below are suggested to provide a good initial experience
|
||||
# with RSpec, but feel free to customize to your heart's content.
|
||||
=begin
|
||||
# This allows you to limit a spec run to individual examples or groups
|
||||
# you care about by tagging them with `:focus` metadata. When nothing
|
||||
# is tagged with `:focus`, all examples get run. RSpec also provides
|
||||
# aliases for `it`, `describe`, and `context` that include `:focus`
|
||||
# metadata: `fit`, `fdescribe` and `fcontext`, respectively.
|
||||
config.filter_run_when_matching :focus
|
||||
|
||||
# Allows RSpec to persist some state between runs in order to support
|
||||
# the `--only-failures` and `--next-failure` CLI options. We recommend
|
||||
# you configure your source control system to ignore this file.
|
||||
config.example_status_persistence_file_path = "spec/examples.txt"
|
||||
|
||||
# Limits the available syntax to the non-monkey patched syntax that is
|
||||
# recommended. For more details, see:
|
||||
# - http://rspec.info/blog/2012/06/rspecs-new-expectation-syntax/
|
||||
# - http://www.teaisaweso.me/blog/2013/05/27/rspecs-new-message-expectation-syntax/
|
||||
# - http://rspec.info/blog/2014/05/notable-changes-in-rspec-3/#zero-monkey-patching-mode
|
||||
config.disable_monkey_patching!
|
||||
|
||||
# This setting enables warnings. It's recommended, but in some cases may
|
||||
# be too noisy due to issues in dependencies.
|
||||
config.warnings = true
|
||||
|
||||
# Many RSpec users commonly either run the entire suite or an individual
|
||||
# file, and it's useful to allow more verbose output when running an
|
||||
# individual spec file.
|
||||
if config.files_to_run.one?
|
||||
# Use the documentation formatter for detailed output,
|
||||
# unless a formatter has already been configured
|
||||
# (e.g. via a command-line flag).
|
||||
config.default_formatter = "doc"
|
||||
end
|
||||
|
||||
# Print the 10 slowest examples and example groups at the
|
||||
# end of the spec run, to help surface which specs are running
|
||||
# particularly slow.
|
||||
config.profile_examples = 10
|
||||
|
||||
# Run specs in random order to surface order dependencies. If you find an
|
||||
# order dependency and want to debug it, you can fix the order by providing
|
||||
# the seed, which is printed after each run.
|
||||
# --seed 1234
|
||||
config.order = :random
|
||||
|
||||
# Seed global randomization in this process using the `--seed` CLI option.
|
||||
# Setting this allows you to use `--seed` to deterministically reproduce
|
||||
# test failures related to randomization by passing the same `--seed` value
|
||||
# as the one that triggered the failure.
|
||||
Kernel.srand config.seed
|
||||
=end
|
||||
end
|
23
spec/stickerizer_spec.rb
Normal file
23
spec/stickerizer_spec.rb
Normal file
|
@ -0,0 +1,23 @@
|
|||
require './lib/stickerizer'
|
||||
|
||||
RSpec.describe Stickerizer do
|
||||
describe '#file_ext' do
|
||||
|
||||
it 'returns the last four characters of a string' do
|
||||
stickerizer = Stickerizer.new
|
||||
expect(stickerizer.file_ext('bananas.png').extension).to eql('.png')
|
||||
end
|
||||
end
|
||||
describe '#valid?' do
|
||||
it 'returns false when given an unsupported file extension' do
|
||||
stickerizer = Stickerizer.new
|
||||
expect(stickerizer.valid?('pork.txt')).to eql(false)
|
||||
end
|
||||
|
||||
it 'returns true when given a supported image format' do
|
||||
stickerizer = Stickerizer.new
|
||||
expect(stickerizer.valid?('buenosdias.png')).to eql(true)
|
||||
end
|
||||
end
|
||||
|
||||
end
|
Loading…
Reference in a new issue