avatar-o-matic 0.0.1

data/lib/avatar_o_matic.rb

@@ -0,0 +1,5 @@
+require_relative 'avatar_o_matic/config'
+require_relative 'avatar_o_matic/generator'
+
+module AvatarOMatic
+end

data/lib/avatar_o_matic/config.rb

@@ -0,0 +1,53 @@
+require 'find'
+
+module AvatarOMatic
+  class Config
+    class << self
+
+      attr_accessor :image_lib
+
+      def image_lib
+        @image_lib ||= File.expand_path( File.join( '..', '..', '..', 'data', 'images'), __FILE__ )
+      end
+
+      def properties
+        [:background, :face, :clothes, :head, :eye, :mouth]
+      end
+
+      def types
+        [:male, :female]
+      end
+
+      def options_for(type, prop)
+        image_data[type.to_sym][prop.to_sym]
+      end
+
+      def image_data
+        return @_data if @_data
+
+        @_data  = {}
+        re_file = /
+          \/
+            (#{self.types.join('|')})
+          \/
+            (#{self.properties.join('|')})
+            \d+
+            \.png
+        \Z/xo
+
+        Find.find(image_lib) do |path|
+          m = re_file.match(path) or next
+
+          type = m[1].to_sym
+          prop = m[2].to_sym
+
+          @_data[type] ||= {}
+          @_data[type][prop] ||= []
+          @_data[type][prop] << path
+        end
+
+        @_data
+      end
+    end
+  end
+end

data/lib/avatar_o_matic/generator.rb

@@ -0,0 +1,98 @@
+require 'mini_magick'
+
+#
+# generator = AvatarOMatic::Generator.new(...)
+# generator.generate!   Does the magick, returns itself
+# generator.save(path)  Saves the image to path
+# generator.image       Returns the underlying minimagick object
+#
+# AvatarOMatic::Generator.new.generate!.save(path)
+#
+
+
+module AvatarOMatic
+  class Generator
+
+    attr_accessor :size
+    attr_reader :image
+
+    def initialize(opts={})
+      super()
+
+      opts.keys.select {|k| (Config.properties + [:size]).include? k }.each do |opt|
+        self.send :"#{opt}=", opts[opt]
+      end
+
+      @size ||= 400
+    end
+
+    def type=(type)
+      unless Config.types.include? type.to_sym
+        raise InvalidPropertyError.new("#{type} is not a recognised type")
+      end
+      @type = type
+    end
+
+    def type
+      @type ||= Config.types.sample
+    end
+
+
+    Config.properties.each do |prop|
+      define_method :"#{prop}=" do |value|
+        allowed_size = Config.options_for(type, prop).size
+
+        if value > allowed_size
+          raise InvalidPropertyError.new "There are only #{allowed_size} options for #{prop} (#{type}), you wanted #{value}"
+        end
+
+        instance_variable_set :"@#{prop}", value
+      end
+
+      define_method :"#{prop}" do
+        value = instance_variable_get :"@#{prop}"
+
+        unless value
+          value = Random.new.rand Config.options_for(type, prop).size
+          instance_variable_set :"@#{prop}", value
+        end
+
+        value
+      end
+    end
+
+    def generate!
+      @image = MiniMagick::Image.open path_for :background
+
+      Config.properties.select {|p| p != :background }.each do |prop|
+        overlay = MiniMagick::Image.open path_for prop
+        @image = @image.composite(overlay) do |c|
+          c.compose 'Over'
+        end
+      end
+
+      @image.resize size if image.width != size
+
+      self
+    end
+
+    def save(path)
+      raise NoImageYetError.new if @image.nil?
+      @image.write path
+      path
+    end
+
+    private
+
+    def option(prop)
+      self.send prop.to_sym
+    end
+
+    def path_for(prop)
+      Config.options_for(type, prop)[option prop]
+    end
+  end
+
+  class InvalidPropertyError < StandardError; end
+  class NoImageYetError < StandardError; end
+end

data/spec/avatar_o_matic/config_spec.rb

@@ -0,0 +1,64 @@
+require 'spec_helper'
+
+describe(AvatarOMatic::Config) do
+
+  subject { described_class }
+
+  describe '.image_lib' do
+    it { is_expected.to respond_to :image_lib }
+
+    # FIXME: test setting image lib, and see it's being used
+  end
+
+  describe '.properties' do
+    it 'returns array of image properties' do
+      expect(subject.properties).to eq [:background, :face, :clothes, :head, :eye, :mouth]
+    end
+  end
+
+  describe '.types' do
+    it 'returns array of possible types' do
+      expect(subject.types).to eq [:male, :female]
+    end
+  end
+
+  describe '.options_for' do
+    described_class.properties.each do |p|
+      described_class.types.each do |g|
+        re_file = /\A
+            #{described_class.image_lib} # Library path
+          \/
+          (?:.+\/)?
+            #{p}    # Property
+            \d+     # Serial
+            \.png
+          \Z/x
+
+        it "returns list of images for #{g} #{p}" do
+          expect( subject.options_for(g, p) ).to all match re_file
+        end
+      end
+    end
+  end
+
+  context '.image_data' do
+    subject { described_class.image_data }
+
+    before(:each) {
+      described_class.instance_variable_set(:'@_data', nil)
+    }
+
+    it 'data is populated on first access' do
+      expect {
+        subject
+      }.to change {
+        described_class.instance_variable_get(:'@_data')
+      }.from nil
+    end
+
+    described_class.properties.each do |p|
+      it { is_expected.to include :male   => a_hash_including( { p => kind_of(Array) } ) }
+      it { is_expected.to include :female => a_hash_including( { p => kind_of(Array) } ) }
+    end
+  end
+end