module IronWorkerNG
  module Code
    module Initializer
      module InstanceMethods
        def initialize_code(*args, &block)
          @name = nil
          @exec = nil

          if args.length == 1 && args[0].class == String
            merge_exec(args[0])
          elsif args.length == 1 && args[0].class == Hash
            @name = args[0][:name] || args[0]['name']

            exec = args[0][:exec] || args[0]['exec'] || args[0][:worker] || args[0]['worker']
            merge_exec(exec) unless exec.nil?
          end

          wfiles = []

          if args.length == 1 && args[0].class == Hash && (args[0][:workerfile] || args[0]['workerfile'])
            wfiles << args[0][:workerfile] || args[0]['workerfile']
          end

          unless name.nil?
            wfiles << name + '.worker'
            wfiles << name + '.workerfile'
          end

          wfiles << 'Workerfile'

          wfiles.each do |wfile|
            if File.exists?(wfile)
              eval(File.read(wfile))

              @base_dir = File.dirname(wfile) + '/'

              break
            end
          end

          unless block.nil?
            instance_eval(&block)
          end
        end
      end
    end
  end
end