@@ -65,6 +65,12 @@ module Gantty return task end + def self.create_resource(name, &block) + resource = Resource.new(name) + resource.instance_eval(&block) + return resource + end + class GanttChart attr_accessor :resources, :tasks @@ -85,6 +91,12 @@ module Gantty @tasks << task end + def resource(name, &block) + resource = Resource.new(name) + resource.instance_eval(&block) + @resources << resource + end + def parse_gantt depends = { 1 => [], 2 => [], 3 => [], 4 => [] } @gantt.xpath('//resource').each { |r| @resources << Resource.new(r) } @@ -472,6 +484,16 @@ module Gantty @id = r.attributes["id"].to_i end end + + %w/ name email phone /.each do |attribute| + define_method(attribute.to_sym) do |*args| + set = *args + set ||= nil + return self.instance_variable_get("@" + attribute) unless set + self.send("#{attribute}=".to_sym, set) + end + end + def to_xml x = Builder::XmlMarkup.new :indent => 2 lib/gantty.rb @@ -20,5 +20,31 @@ describe "Resource" do it "should return itself in correct XML" it "should require a name" + + it "can be created with new syntax" do + me = Gantty::create_resource "Adam Alpha" do + email "adam.alpha@gmail.com" + phone "555-555-1234" + end + + me.name.should == "Adam Alpha" + me.email.should == "adam.alpha@gmail.com" + me.phone.should == "555-555-1234" + end + + it "can be created within a Gantty block" do + g = Gantty::create do + resource "Adam Alpha" do + email "adam.alpha@gmail.com" + phone "555-555-2345" + end + end + + me = g.resources.first + me.name.should == "Adam Alpha" + me.email.should == "adam.alpha@gmail.com" + me.phone.should == "555-555-2345" + + end end spec/resource_spec.rb e3b839acf5202d8a122a91729efd7573450c24f1 Ardekantur greystone@ardekantur.com http://github.com/ardekantur/gantty/commit/262da8034f3686adac77d04e22e7b480fcbdd893 262da8034f3686adac77d04e22e7b480fcbdd893 2008-11-30T20:07:52-08:00 2008-11-30T20:07:52-08:00 resources use new syntax outside and inside of Gantty::create blocks 4f3fe2f50c3c764933a58b28dac880556b80a2a4 Ardekantur greystone@ardekantur.com