require 'tmpdir' require 'zip/zip' require_relative '../feature/base' require_relative '../feature/common/merge_file' require_relative '../feature/common/merge_dir' module IronWorkerNG module Code class Base attr_accessor :name attr_reader :features @@registered_types = [] def self.registered_types @@registered_types end def self.register_type(type) @@registered_types << type end @@registered_features = [] def self.registered_features @@registered_features end def self.register_feature(feature) @@registered_features << feature end include IronWorkerNG::Feature::Common::MergeFile::InstanceMethods include IronWorkerNG::Feature::Common::MergeDir::InstanceMethods def initialize(*args) @name = nil @features = [] if args.length == 1 && args[0].class == String && File.exists?(args[0]) merge_worker(args[0]) elsif args.length == 1 && args.class == String @name = args[0] elsif args.length == 1 && args.class == Hash @name = args[0][:name] merge_worker(args[0][:worker]) unless args[0][:worker].nil? end end def fixate IronWorkerNG::Code::Base.registered_features.each do |rf| if rf[:for_klass] == self.class && respond_to?(rf[:name] + '_fixate') send(rf[:name] + '_fixate') end end end def hash_string fixate Digest::MD5.hexdigest(@features.map { |f| f.hash_string }.join) end def bundle(zip) @features.each do |feature| feature.bundle(zip) end end def create_zip unless @worker IronWorkerNG::Logger.error 'No worker specified' raise 'No worker specified' end fixate zip_name = Dir.tmpdir + '/' + Dir::Tmpname.make_tmpname("iron-worker-ng-", "code.zip") IronWorkerNG::Logger.debug "Creating code zip '#{zip_name}'" Zip::ZipFile.open(zip_name, Zip::ZipFile::CREATE) do |zip| bundle(zip) create_runner(zip) end zip_name end def create_runner(zip) end def runtime nil end def runner nil end end end end