peto 0.1.3 → 0.2.0

data/README.rdoc

@@ -2,11 +2,10 @@
 
 == Usage
 
-installing
-  gem install peto
-
-contract file (foo.yml)
+installing:
+  % gem install peto
 
+contract file (foo.yml):
   name: foo
 
   types:
@@ -17,12 +16,13 @@ contract file (foo.yml)
     set_user:
       args: [user:user]
 
-invoke command
-  peto foo.yml > generated.rb
-
-use generated.rb
+invoke command:
+  % peto foo.yml
 
-  require "generated"
+use foo.rb:
+  require "foo"
+  require "animal"
+  require "user"
 
   cat = Peto::Animal.new(:name => "cat")
   dog = Peto::Animal.new(:name => "dog")

data/VERSION

@@ -1 +1 @@
-0.1.3
+0.2.0

data/bin/peto

@@ -4,7 +4,22 @@ $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 $LOAD_PATH.unshift(File.dirname(__FILE__))
 require "peto"
 
+require "optparse"
+options = {}
+OptionParser.new do |opt|
+  opt.on("-o output directory") {|value| options[:output] = value }
+  opt.parse!(ARGV)
+end
+input = ARGV.pop
+
+if input.nil?
+  puts "usage:\n  % peto contract.yml"
+  exit
+end
+
+options[:output] = File.dirname(input) if options[:output].nil?
+
 peto = Peto::Master.new
-peto.load(ARGV.first)
-puts peto.generate
+peto.load(input)
+peto.generate
 

data/lib/peto/generator.rb

@@ -23,12 +23,18 @@ module Peto
     def initialize(contract)
       @contract = contract
     end
+    attr_reader :contract
 
-    def generate(template_filename)
+    def generate_procedure(template_filename)
       erb = ERB.new(IO.read(template_filename), nil, "-")
-      erb.result(binding)
+      { "#{@contract["name"]}" => erb.result(binding) }
     end
 
+    def generate_class(template_filename, type)
+      erb = ERB.new(IO.read(template_filename), nil, "-")
+      @target = {:name => type.first, :args => args(type.second)}
+      { "#{@target[:name]}" => erb.result(binding) }
+    end
 
     def class_name
       @contract["name"].to_class_type

data/lib/peto/master.rb

@@ -3,15 +3,45 @@ require "yaml"
 require "peto/generator"
 
 module Peto
-  TEMPLATE_DIR = File::dirname( File::expand_path( __FILE__ ) ) + "/../templates"
+  TEMPLATE_DIR = File::dirname(File::expand_path( __FILE__ )) + "/../templates"
   class Master
     def load(filename)
+      @work_dir = File::dirname(filename)
       @contract = YAML.load(IO.read(Pathname(filename)))
     end
     attr_reader :contract
 
+    def parse
+      (@contract["types"]||{}).inject({}) {|result, type|
+        result.merge!(Generator.new(@contract).generate_class(TEMPLATE_DIR + "/rb_classes.erb", type))
+      }.merge!(Generator.new(@contract).generate_procedure(TEMPLATE_DIR + "/rb_procedures.erb"))
+    end
+
     def generate
-      Generator.new(@contract).generate(TEMPLATE_DIR + "/rb_procedures.erb")
+      parse.each do |name, content|
+        filepath = File.join(@work_dir, "#{name}.rb")
+        write(filepath, content)
+      end
+    end
+
+    def write(filepath, content)
+      print "  "
+      if File.exist?(filepath)
+        if File.read(filepath) == content
+          print "identical"
+        else
+          print "update   "
+        end
+      else
+        print "create   "
+      end
+      print "  "
+
+      open(filepath, "w") do |file|
+        file.write(content)
+      end
+
+      puts filepath
     end
   end
 end

data/lib/templates/rb_classes.erb

@@ -0,0 +1,37 @@
+require "peto/mixin/peto_class"
+
+<%- name = @target[:name].camelize -%>
+<%- args = @target[:args] -%>
+module Peto
+  class <%= name %>
+    include PetoClass
+    def initialize(args={})
+      <%- args.each do |arg| -%>
+        <%- if arg[:array_type] -%>
+      @<%= arg[:name] %> = [] # for <%= arg[:array_type] %>
+        <%- else -%>
+      @<%= arg[:name] %> = nil
+        <%- end -%>
+      <%- end -%>
+
+      set_by_hash(args)
+      raise_errors unless valid?
+    end
+
+    <%- args.each do |arg| -%>
+    attr_reader :<%= arg[:name] %>
+    <%- end -%>
+
+    def members
+      [<%= args.map{|arg| ":#{arg[:name]}"}.join(",") %>]
+    end
+
+    def types
+      {<%= args.map{|arg| ":#{arg[:name]} => #{arg[:type]}"}.join(",") %>}
+    end
+
+    def arrays
+      {<%= args.select{|arg|arg[:array_type]}.map{|arg| ":#{arg[:name]} => #{arg[:array_type]}"}.join(",") %>}
+    end
+  end
+end

data/lib/templates/rb_procedures.erb

@@ -1,42 +1,5 @@
-require "peto/mixin/peto_class"
 require "peto/mixin/peto_errorable"
 
-module Peto
-  <%- each_types do |name, args| -%>
-  class <%= name %>
-    include PetoClass
-    def initialize(args={})
-      <%- args.each do |arg| -%>
-        <%- if arg[:array_type] -%>
-      @<%= arg[:name] %> = [] # for <%= arg[:array_type] %>
-        <%- else -%>
-      @<%= arg[:name] %> = nil
-        <%- end -%>
-      <%- end -%>
-
-      set_by_hash(args)
-      raise_errors unless valid?
-    end
-
-    <%- args.each do |arg| -%>
-    attr_reader :<%= arg[:name] %>
-    <%- end -%>
-
-    def members
-      [<%= args.map{|arg| ":#{arg[:name]}"}.join(",") %>]
-    end
-
-    def types
-      {<%= args.map{|arg| ":#{arg[:name]} => #{arg[:type]}"}.join(",") %>}
-    end
-
-    def arrays
-      {<%= args.select{|arg|arg[:array_type]}.map{|arg| ":#{arg[:name]} => #{arg[:array_type]}"}.join(",") %>}
-    end
-  end
-  <%- end -%>
-end
-
 module Peto
   class <%= class_name %>
     extend PetoErrorable

data/test/test_peto.rb

@@ -9,21 +9,27 @@ class TestPeto < Test::Unit::TestCase
       @peto = Peto::Master.new
     end
 
-    context "load contract yaml file" do
+    context "loads contract yaml file" do
       setup do
         @peto.load("test/contracts/loading.yml")
       end
-      should ".contract returns loaded contract" do
-        assert_equal({ "name" => "loading" }, @peto.contract)
+      context ".contract" do
+        should "return loaded contract" do
+          assert_equal({ "name" => "loading" }, @peto.contract)
+        end
       end
     end
 
-    context "generate procedures" do
+    context "parse procedures" do
       setup do
         @peto.load("test/contracts/generating.yml")
       end
       should "returns string by loaded contract" do
-        assert_equal String, @peto.generate.class
+        assert_equal Hash, @peto.parse.class
+        @peto.parse.each do |name, content|
+          assert_equal String, name.class
+          assert_equal String, content.class
+        end
       end
     end
   end
@@ -32,10 +38,12 @@ class TestPeto < Test::Unit::TestCase
     setup do
       @peto = Peto::Master.new
       @peto.load("test/contracts/generating.yml")
-      @generated = @peto.generate
+      @generated = @peto.parse
     end
-    should "is readable as ruby" do
-      Closed.class_eval(@generated)
+    should "be readable as ruby" do
+      @generated.each do |filepath, content|
+        Closed.class_eval(content)
+      end
       assert_equal({
         :procedure => "do_a",
         :args => {

metadata

@@ -4,9 +4,9 @@ version: !ruby/object:Gem::Version
   prerelease: false
   segments: 
   - 0
-  - 1
-  - 3
-  version: 0.1.3
+  - 2
+  - 0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Toshiyuki Hirooka
@@ -53,6 +53,7 @@ files:
 - lib/peto/mixin/peto_class.rb
 - lib/peto/mixin/peto_errorable.rb
 - lib/peto/rails/helper.rb
+- lib/templates/rb_classes.erb
 - lib/templates/rb_procedures.erb
 - test/contracts/generating.yml
 - test/contracts/loading.yml