forked from notahat/machinist
-
Notifications
You must be signed in to change notification settings - Fork 0
/
machinist.rb
71 lines (62 loc) · 2.04 KB
/
machinist.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
require 'active_support'
require 'active_record'
require 'sham'
require 'machinist/active_record'
module Machinist
# A Lathe is used to execute the blueprint and construct an object.
#
# The blueprint is instance_eval'd against the Lathe.
class Lathe
def self.run(object, *args)
blueprint = object.class.blueprint
named_blueprint = object.class.blueprint(args.shift) if args.first.is_a?(Symbol)
attributes = args.pop || {}
raise "No blueprint for class #{object.class}" if blueprint.nil?
returning self.new(object, attributes) do |lathe|
lathe.instance_eval(&named_blueprint) if named_blueprint
lathe.instance_eval(&blueprint)
end
end
def initialize(object, attributes = {})
@object = object
@assigned_attributes = {}
attributes.each do |key, value|
@object.send("#{key}=", value)
@assigned_attributes[key.to_sym] = value
end
end
# Undef a couple of methods that are common ActiveRecord attributes.
# (Both of these are deprecated in Ruby 1.8 anyway.)
undef_method :id if respond_to?(:id)
undef_method :type if respond_to?(:type)
def object
yield @object if block_given?
@object
end
attr_reader :assigned_attributes
def method_missing(symbol, *args, &block)
if @assigned_attributes.has_key?(symbol)
@object.send(symbol)
elsif @object.class.reflect_on_association(symbol) && !@object.send(symbol).nil?
@object.send(symbol)
else
@object.send("#{symbol}=", generate_attribute(symbol, args, &block))
end
end
def generate_attribute(attribute, args)
value = if block_given?
yield
elsif args.empty?
association = @object.class.reflect_on_association(attribute)
if association
association.class_name.constantize.make(args.first || {})
else
Sham.send(attribute)
end
else
args.first
end
@assigned_attributes[attribute] = value
end
end
end