1
0
Fork 0
musicbrainz/lib/musicbrainz/models/base_model.rb

68 lines
1.5 KiB
Ruby

module MusicBrainz
class BaseModel
def self.inherited(klass)
klass.send(:include, InstanceMethods)
klass.send(:extend, ClassMethods)
end
module ClassMethods
def field(name, type)
fields[name] = type
attr_reader name
define_method("#{name}=") do |val|
instance_variable_set("@#{name}", validate_type(val, type))
end
end
def fields
instance_variable_set(:@fields, {}) unless instance_variable_defined?(:@fields)
instance_variable_get(:@fields)
end
def client
MusicBrainz.client
end
end
module InstanceMethods
def initialize(params = {})
params.each do |field, value|
self.send(:"#{field}=", value)
end
end
def client
MusicBrainz.client
end
private
def validate_type(val, type)
if type == Integer
val.to_i
elsif type == Float
val.to_f
elsif type == String
val.to_s
elsif type == Date
val = if val.nil? or val == ""
[2030, 12, 31]
elsif val.split("-").length == 1
[val.split("-").first.to_i, 12, 31]
elsif val.split("-").length == 2
val = val.split("-").map(&:to_i)
[val.first, val.last, -1]
else
val.split("-").map(&:to_i)
end
Date.new(*val)
else
val
end
end
end
end
end