saruman 0.0.2 → 0.0.3

data/README.md

@@ -45,7 +45,7 @@ NOTE: I couldn't remember whether Magento even allows anything but 0.0.1 (3 digi
 
 ### saruman observer
 
-  saruman observer
+    saruman observer
   
 Will create a new observer file with an unlimited amount of events you wish to observe. I also parsed a document containing most of the observer events in magento version 1.5ish I believe, so there are about 300 events to choose from, no guarantees that they all work or are up to date. Just type the number of the observer event youd like to observer, rinse and repeat until you are finished (note that it will create all the observer events into one observer.php file so it will overwrite if you have an observer already.)
 

data/lib/saruman.rb

@@ -1,9 +1,11 @@
 require "saruman/version"
 require "highline"
 require "virtus"
+require "saruman/virtus/attributes"
 require "saruman"
 require "nokogiri"
 require 'active_support'
+
 module Saruman
   
   module Base
@@ -28,7 +30,6 @@ module Saruman
       name.downcase
     end
     
-    #alias for name
     def extension_name_lower
       name.downcase
     end
@@ -53,6 +54,10 @@ module Saruman
       "#{extension_base_path}Model/"
     end
     
+    def controller_path
+      "#{extension_base_path}controllers/"
+    end    
+    
     def helper_path
       "#{extension_base_path}Helper/"
     end
@@ -65,6 +70,10 @@ module Saruman
       "#{extension_base_path}sql/#{name_lower}_setup/"
     end
     
+    def block_klass_name
+      "#{combined_namespace}_Block"
+    end  
+    
     def model_klass_name
       "#{combined_namespace}_Model"
     end  
@@ -81,6 +90,40 @@ module Saruman
       "#{global_config_basepath}#{combined_namespace}.xml"
     end
     
+    def controller_block_path
+      "#{extension_base_path}Block/"
+    end
+    
+    def controller_block_file_path(controller_name)
+      "#{controller_block_path}#{controller_name.capitalize}.php"      
+    end  
+    
+    def app_design_frontend_base_path
+      newer_magento_base_path = "#{Dir.pwd}/app/design/frontend/base/default/"
+      older_magento_base_path = "#{Dir.pwd}/app/design/frontend/default/default/"
+      if File.directory?(newer_magento_base_path)
+        return newer_magento_base_path
+      else
+        return older_magento_base_path
+      end
+    end
+    
+    def app_design_frontend_base_layout_path
+      "#{app_design_frontend_base_path}layout/"
+    end
+    
+    def app_design_frontend_base_layout_local_xml_path
+      "#{app_design_frontend_base_path}layout/#{name_lower}.xml"
+    end
+    
+    def app_design_frontend_base_template_path
+      "#{app_design_frontend_base_path}template/"
+    end
+    
+    def app_design_frontend_base_template_namespace_path
+      "#{app_design_frontend_base_template_path}#{name_lower}/"
+    end  
+    
     def model?
       if arguments[:model] == true
         return true
@@ -117,6 +160,22 @@ module Saruman
       arguments[:observer_events]
     end
     
+    def controller?
+      if arguments[:controller] == true
+        return true
+      else
+        return false
+      end
+    end
+    
+    def controllers
+      arguments[:controllers]
+    end
+    
+    def controller_front_name
+      arguments[:controller_front_name]
+    end       
+    
     def helper?
       if arguments[:helper] == true
         return true
@@ -140,7 +199,7 @@ module Saruman
       file = File.open(extension_config_file_path,'w')
       file.puts @config.to_xml
       file.close
-    end  
+    end
     
     def extension_current_version
       @extension_current_version = @config.css("version").first.content
@@ -163,7 +222,6 @@ module Saruman
     
     def insert_tag_at_node(tag, tag_lookup)
       target_tag = @config.css(tag_lookup).first
-      puts "inserting #{tag} from looku #{tag_lookup}"
       unless target_tag.nil?
         new_tag = "<#{tag}></#{tag}>"
         target_tag.add_child(new_tag)
@@ -177,26 +235,30 @@ module Saruman
       end
     end  
     
-  end  
-  
+  end
   
-  module MenuBuilder
-    include Virtus
-    attribute :decisions, Array, :default => []
-
-    # def decisions
-    #   return decisions_made
-    # end  
+  module BuilderInstanceBase
     
-  end
+    def initialize(options)
+      @options = options
+    end
+    
+    def method_missing(meth, *args, &block)
+      if(@options.has_key?(meth.to_sym))
+        @options[meth.to_sym]
+      else
+        super
+      end
+    end
+  end  
   
   class ObserverMenuBuilder
     
-    # include Virtus
-    include MenuBuilder
-    # attribute :chosen_events, Array
+    include Virtus
+
     attribute :observer_events, Array, :default => []
-    # attribute :decisions, Array, :default => []
+    attribute :decisions, Array, :default => []
+
     BASE_DIR = File.dirname(__FILE__)
     OBSERVER_EVENTS_FILE_PATH = BASE_DIR+"/saruman/generators/observer/observer_events.csv"
     
@@ -230,17 +292,27 @@ module Saruman
     
   end
   
-  class ModelBuilder
-    def initialize
+  class ControllerBuilder
+    
+    def initialize(options)
+      @options = options
       ask_question
     end
     
     def ask_question
-      model_name = ask("Enter Name of model") { |q| q.default = "Post" }
-      model_table_name = ask("Enter Table Name of model") { |q| q.default = "blog_post" }
-      model_fields_input = ask("Enter Model Fields") { |q| q.default = "title:string active:boolean blog_id:integer:index" }
-      model_sql = parse_model_fields(model_fields_input, model_table_name)
-      @question_answer = {:model_name => model_name, :model_name_lower => model_name.downcase, :model_table_name => model_table_name, :sql => model_sql}
+      controller = Saruman::Controller.new(@options)
+      controller.name = ask("Enter Controller Name (will match www.yourmagentoinstall.com/frontname/controllername)") { |q| q.default = "index" }
+      controller_actions_input = ask("Enter list of actions for this controller, I.E., index show add") { |q| q.default = "index show add"}
+      controller.actions = parse_controller_actions(controller_actions_input)
+      
+      say("Would you like to create controller action views?")
+      choose do |menu|
+        menu.choice(:yes) { controller.create_views = true }
+        menu.choice(:no) { controller.create_views  = false }
+      end
+      
+      # @question_answer = {:controller_name => controller_name.capitalize, :controller_actions => controller_actions}
+      @question_answer = controller
       @question_answer
     end
     
@@ -248,6 +320,103 @@ module Saruman
       return @question_answer
     end
     
+    def parse_controller_actions(controller_actions_input)
+      actions = []
+      controller_actions_input.split(" ").each do |action|
+        actions << Saruman::ControllerBuilderAction.new(action)
+      end
+      actions
+    end
+    
+  end
+  
+  class Controller
+    include Virtus
+    include BuilderInstanceBase
+    attribute :actions, Array
+    attribute :name, Capitalize
+    attribute :create_views, Boolean
+    attribute :name_lower, String, :default => lambda { |model,attribute| model.name.downcase }
+    
+    def klass_name
+      "#{combined_namespace}_#{name}Controller"
+    end
+      
+  end  
+    
+  
+  class ControllerBuilderAction
+    include Virtus
+    
+    #designed for future stuff, mostly pointless/overkill for now
+    
+    attribute :visibility, String
+    attribute :respond_to, String
+    attribute :name, Downcase
+    attribute :method_name, String
+    attribute :segs, Array
+    
+    def initialize(action)
+      @segs = action.split(":")
+      case segs.length
+      when 1
+        @visibility = "public"
+        @respond_to = "html"
+        @name = segs.first
+        @method_name = "#{name}Action"
+      end
+    end
+    
+    def layout_handle(extension_name, controller_name)
+      @extension_name = extension_name
+      @controller_name = controller_name
+      "#{@extension_name}_#{@controller_name}_#{name}"
+    end
+    
+    def block_name
+      "#{@extension_name}#{@controller_name}#{name}"
+    end
+    
+    def block_as
+      "#{@extension_name}#{@controller_name}#{name}"
+    end
+    
+    def block_template
+      "#{@extension_name}/#{@controller_name}/#{name}.phtml"
+    end
+    
+  end  
+  
+  class ModelBuilder
+    
+    def initialize(options)
+      @options = options
+      ask_questions
+    end
+    
+    def ask_questions
+      model = Saruman::Model.new(@options)
+      
+      model.name = ask("Enter Name of model") { |q| q.default = "Post" }
+      model.table_name = ask("Enter Table Name of model") { |q| q.default = "blog_post" }
+      model_fields_input = ask("Enter Model Fields") { |q| q.default = "title:string active:boolean blog_id:integer:index" }
+
+      model.sql = parse_model_fields(model_fields_input, model.table_name)
+      # @question_answer = {:model_name => model_name, :model_name_lower => model_name.downcase, :model_table_name => model_table_name, :sql => model_sql}
+      
+      say("Would you like to create a collection model?")
+      choose do |menu|
+        menu.choice(:yes) { model.collection = true }
+        menu.choice(:no) { model.collection  = false }
+      end
+      
+      @question_answer = model
+    end
+    
+    def output
+      return @question_answer
+    end
+    
     def parse_model_fields(model_fields, model_table_name)
       sql = ""
       index_sql = ""
@@ -263,6 +432,29 @@ module Saruman
     
   end
   
+  class Model
+    include Virtus
+    include BuilderInstanceBase
+    attribute :name, Capitalize
+    attribute :name_lower, String, :default => lambda { |model,attribute| model.name.downcase }
+    attribute :table_name, Downcase
+    attribute :sql, String
+    attribute :collection, Boolean
+    
+    def klass_name
+      "#{combined_namespace}_Model_#{name}"
+    end
+    
+    def resource_model_klass_name
+      "#{combined_namespace}_Model_Mysql4_#{name}"
+    end
+    
+    def collection_model_klass_name
+      "#{combined_namespace}_Model_Mysql4_#{name}_Collection"
+    end  
+
+  end
+  
   class ModelBuilderField
     
     SQL_TYPE_MAPPINGS = {:string => "varchar(255)", :text => "text NOT NULL", :boolean => "tinyint(1) NOT NULL default 0", :date => "DATETIME default NULL", :int => "int(10)", :integer => "int(10)" }
@@ -305,8 +497,8 @@ module Saruman
       else
         super
       end
-    end    
-  end  
+    end
+  end
   
   class ModelXmlConfigBuilder
     include Virtus
@@ -329,7 +521,7 @@ module Saruman
     def set_config_models_resource_entities_xml
       xml = ""
       models.each do |model|
-        xml << "<#{model[:model_name_lower]}><table>#{model[:model_table_name]}</table></#{model[:model_name_lower]}>\n"
+        xml << "<#{model.name_lower}><table>#{model.table_name}</table></#{model.name_lower}>\n"
       end
       return xml
     end
@@ -414,6 +606,61 @@ module Saruman
       return xml
     end
     
+  end
+  
+  class ControllerXmlConfigBuilder
+    include Virtus
+    include XmlBuilderBase
+    attribute :controllers, Array
+    attribute :config_frontend_routers_name_xml, String
+    attribute :config_frontend_layout_xml, String
+    attribute :config_global_blocks_xml, String
+    
+    def initialize(controllers, generator)
+      @controllers = controllers
+      @generator = generator
+      @config_frontend_routers_name_xml = set_config_frontend_routers_name_xml
+      @config_frontend_layout_xml = set_config_frontend_layout_xml
+      @config_global_blocks_xml = set_config_global_blocks_xml
+    end
+    
+    def set_config_frontend_routers_name_xml
+      xml="
+      <#{name_lower}>
+        <use>standard</use>
+        <args>
+          <module>#{combined_namespace}</module>
+          <frontName>#{controller_front_name}</frontName>
+        </args>  
+      </#{name_lower}>
+    "
+      return xml
+    end
+    
+    def set_config_frontend_layout_xml
+      xml="
+      <layout>
+        <updates>
+          <#{name_lower}>
+            <file>#{name_lower}.xml</file>
+          </#{name_lower}>
+        </updates>
+      </layout>
+    "
+      return xml
+    end
+    
+    def set_config_global_blocks_xml
+      xml="
+      <blocks>
+          <#{name_lower}>
+            <class>#{block_klass_name}</class>
+          </#{name_lower}>
+      </blocks>
+    "
+      return xml
+    end
+    
   end  
   
 end

data/lib/saruman/cli.rb

@@ -13,8 +13,30 @@ module Saruman
       options[:name] = ask("Enter extension name:") { |q| q.default = "Wizard" }
       options[:author] = ask("Author of extension:") { |q| q.default = "Jason Ayre www.bravenewwebdesign.com" }
       options[:version] = ask("Version number (Format - 0.0.1):") { |q| q.default = "0.0.1" }
+      options[:combined_namespace] = "#{options[:namespace]}_#{options[:name]}"
+            
+      say("Would you like to create a controller?")
+      choose do |menu|
+        menu.choice(:yes) { options[:controller] = true }
+        menu.choice(:no) { options[:controller] = false }
+      end
+
+      if(options[:controller])
+        
+        options[:controller_front_name] = ask("Enter Front Name of controller (will match www.yourmagentoinstall.com/frontname)") { |q| q.default = "extension_name" }
+        
+        if(options[:controllers]).nil?
+          options[:controllers] = Array.new
+        end
+        
+        begin
+          question = Saruman::ControllerBuilder.new(options)
+          options[:controllers] << question.output
+        end while agree("Create another controller?")
+        
+      end
       
-      say("Would you like me to create an observer?")
+      say("Would you like to create an observer?")
       choose do |menu|
         menu.choice(:yes) { options[:observer] = true }
         menu.choice(:no) { options[:observer] = false }
@@ -42,7 +64,7 @@ module Saruman
         
       end
       
-      say("Would you like me to create a model?")
+      say("Would you like to create a model?")
       choose do |menu|
         menu.choice(:yes) { options[:model] = true }
         menu.choice(:no) { options[:model] = false }
@@ -55,13 +77,13 @@ module Saruman
         end
         
         begin
-          question = Saruman::ModelBuilder.new
+          question = Saruman::ModelBuilder.new(options)
           options[:models] << question.output
         end while agree("Create another model?")
         
       end
       
-      say("Would you like me to create a helper?")
+      say("Would you like to create a helper?")
       choose do |menu|
         menu.choice(:yes) { options[:helper] = true }
         menu.choice(:no) { options[:helper] = false }
@@ -71,7 +93,11 @@ module Saruman
       
       if options[:model]
         Saruman::Generators::Model.start([options])
-      end  
+      end
+      
+      if options[:controller]
+        Saruman::Generators::Controller.start([options])
+      end        
       
     end
     
@@ -81,13 +107,13 @@ module Saruman
       options[:command] = __method__
       options[:namespace] = ask("Enter extension namespace:") { |q| q.default = "Saruman" }
       options[:name] = ask("Enter extension name:") { |q| q.default = "Wizard" }
-      
+      options[:combined_namespace] = "#{options[:namespace]}_#{options[:name]}"
       if(options[:models]).nil?
         options[:models] = Array.new
       end
       
       begin
-        question = Saruman::ModelBuilder.new
+        question = Saruman::ModelBuilder.new(options)
         options[:models] << question.output
       end while agree("Create another model?")
 
@@ -124,6 +150,29 @@ module Saruman
       
     end
     
+    desc "controller", "Creates a new magento controller"
+    def controller
+      options = Hash.new
+      options[:command] = __method__
+      options[:namespace] = ask("Enter extension namespace:") { |q| q.default = "Saruman" }
+      options[:name] = ask("Enter extension name:") { |q| q.default = "Wizard" }
+      options[:controller_front_name] = ask("Enter Front Name of controller (will match www.yourmagentoinstall.com/frontname)") { |q| q.default = "extension_name" }
+      
+      options[:combined_namespace] = "#{options[:namespace]}_#{options[:name]}"
+            
+      if(options[:controllers]).nil?
+        options[:controllers] = Array.new
+      end
+      
+      begin
+        question = Saruman::ControllerBuilder.new(options)
+        options[:controllers] << question.output
+      end while agree("Create another Controller?")
+
+      Saruman::Generators::Controller.start([options])
+      
+    end
+    
   end
   
 end

data/lib/saruman/generators/controller.rb

@@ -0,0 +1,103 @@
+require 'thor/group'
+require 'nokogiri'
+module Saruman
+  module Generators
+    class Controller < Thor::Group
+
+      include Thor::Actions
+      include Saruman::Base
+      
+      argument :arguments, :type => :hash
+      
+      def self.source_root
+        File.dirname(__FILE__) + "/controller/templates"
+      end
+      
+      def validate_extension
+        unless File.directory?("#{extension_base_path}")
+          @error = "error"
+        end
+
+        unless File.exist?("#{extension_config_path}/config.xml")
+          @error = "error"
+        end
+      end
+      
+      def load_builders
+        @controller_xml_config_builder = Saruman::ControllerXmlConfigBuilder.new(controllers, self)
+      end
+      
+      def create_controller_directory
+        empty_directory(controller_path) unless File.directory?(controller_path)
+      end
+      
+      def modify_config
+        
+        @config = read_extension_config
+        
+        unless config_has_tag?("config frontend")
+          insert_tag_at_node("frontend", "config")
+        end
+        
+        unless config_has_tag?("config frontend routers")
+          insert_tag_at_node("routers", "config frontend")
+        end
+        
+        unless config_has_tag?("config frontend routers #{name_lower}")
+          insert_xml_at_node(@controller_xml_config_builder.config_frontend_routers_name_xml, "config frontend routers")
+        end
+        
+        @controllers_with_views = controllers.select { |controller| controller.create_views == true }
+
+        if @controllers_with_views.length > 0
+          unless config_has_tag?("config frontend layout")
+            insert_xml_at_node(@controller_xml_config_builder.config_frontend_layout_xml, "config frontend")
+          end
+          
+          unless config_has_tag?("config global blocks")
+            insert_xml_at_node(@controller_xml_config_builder.config_global_blocks_xml, "config global")            
+          end
+          
+          template("local.xml", app_design_frontend_base_layout_local_xml_path)
+          
+        end
+        
+      end
+      
+      def create_controllers
+        controllers.each do |controller|
+          @controller = controller
+          template("Controller.php", "#{controller_path}#{@controller.name}Controller.php")
+          
+          if controller.create_views
+            empty_directory(controller_view_path(@controller.name_lower)) unless File.directory?(controller_view_path(@controller.name_lower))
+            empty_directory(controller_block_path) unless File.directory?(controller_block_path)
+            @controller_block_klass_name = "#{combined_namespace}_Block_#{@controller.name.capitalize}"
+            template("Block.php", controller_block_file_path(@controller.name))
+            controller.actions.each do |action|
+              @action = action
+              @kontroller_front_name = controller_front_name
+              template("view.phtml", controller_view_action_path(@controller.name_lower, @action.name))
+            end
+            app_design_frontend_base_template_namespace_path
+          end  
+        end       
+      end
+      
+      def write_changes
+        write_extension_config
+      end  
+      
+      private
+      
+      def controller_view_path(controller_name)
+        "#{app_design_frontend_base_template_namespace_path}#{controller_name}/"
+      end
+      
+      def controller_view_action_path(controller_name, action_name)
+        "#{controller_view_path(controller_name)}#{action_name}.phtml"
+      end  
+      
+    end
+  end
+end

data/lib/saruman/generators/controller/templates/Block.php

@@ -0,0 +1,8 @@
+<?php
+class <%= @controller_block_klass_name %> extends Mage_Core_Block_Template
+{
+  public function __construct()
+  {
+    parent::__construct();
+  }
+}

data/lib/saruman/generators/controller/templates/Controller.php

@@ -0,0 +1,21 @@
+<?php
+
+class <%= @controller.klass_name %> extends Mage_Core_Controller_Front_Action 
+{
+  <% @controller.actions.each do |action| %>
+  <%= action.visibility %> function <%= action.method_name %>()
+  {
+    <% if @controller.create_views %>
+    
+    $params = $this->getRequest()->getParams();
+    $this->loadLayout();    
+    $this->renderLayout();
+
+    <% else %>
+      echo "Hello world.";
+      print_r($params);    
+    <% end %>
+  }
+  <% end %>
+  
+}

data/lib/saruman/generators/controller/templates/local.xml

@@ -0,0 +1,11 @@
+<layout version="0.1.0">
+  <% @controllers_with_views.each do |controller| %>
+  <% controller.actions.each do |action| %>
+    <<%= action.layout_handle(name_lower, controller.name_lower) %>>
+      <reference name="content">
+        <block type="<%= name_lower %>/<%= controller.name_lower %>" name="<%= action.block_name %>" as="<%= action.block_as %>" template="<%= action.block_template %>" />
+      </reference>
+    </<%= action.layout_handle(name_lower, controller.name_lower) %>>
+  <% end %>
+  <% end %>
+</layout>

data/lib/saruman/generators/controller/templates/view.phtml

@@ -0,0 +1 @@
+<h1><%= @kontroller_front_name.capitalize %> <%= @controller.name %> <%= @action.name %></h1>

data/lib/saruman/generators/extension.rb

@@ -36,113 +36,6 @@ module Saruman
         end  
       end
       
-      private
-      
-      # def namespace
-      #   arguments[:namespace]
-      # end
-      # 
-      # def name
-      #   arguments[:name]
-      # end
-      # 
-      # def combined_namespace
-      #   "#{arguments[:namespace]}_#{arguments[:name]}"
-      # end
-      # 
-      # def namespace_lower
-      #   namespace.downcase
-      # end
-      # 
-      # def name_lower
-      #   name.downcase
-      # end
-      # 
-      # #alias for name
-      # def extension_name_lower
-      #   name.downcase
-      # end
-      # 
-      # def author
-      #   arguments[:author]
-      # end
-      # 
-      # def version
-      #   arguments[:version]
-      # end  
-      # 
-      # def global_config_basepath
-      #   "app/etc/modules/"
-      # end
-      # 
-      # def extension_base_path
-      #   "app/code/local/#{namespace}/#{name}/"
-      # end
-      # 
-      # def extension_config_path
-      #   "#{extension_base_path}etc/"
-      # end
-      # 
-      # def model_path
-      #   "#{extension_base_path}Model/"
-      # end
-      # 
-      # def resource_model_path
-      #   "#{model_path}Mysql4/"
-      # end  
-      # 
-      # def helper_path
-      #   "#{extension_base_path}Helper/"
-      # end
-      # 
-      # def setup_base_path
-      #   "#{extension_base_path}/sql/"
-      # end
-      # 
-      # def model_klass_name
-      #   "#{combined_namespace}_Model"
-      # end
-      # 
-      # def resource_model_name_lower
-      #   "#{name_lower}_mysql4"
-      # end
-      # 
-      # def resource_model_klass_name
-      #   "#{combined_namespace}_Model_Mysql4"
-      # end  
-      # 
-      # def observer?
-      #   if arguments[:observer] == true
-      #     return true
-      #   else
-      #     return false
-      #   end
-      # end
-      # 
-      # def observers
-      #   arguments[:observer_events]
-      # end
-      # 
-      # def helper?
-      #   if arguments[:helper] == true
-      #     return true
-      #   else
-      #     return false
-      #   end
-      # end
-      # 
-      # def model?
-      #   if arguments[:model] == true
-      #     return true
-      #   else
-      #     return false
-      #   end
-      # end
-      # 
-      # def models
-      #   arguments[:models]
-      # end
-      
     end
   end
 end

data/lib/saruman/generators/model.rb

@@ -60,14 +60,13 @@ module Saruman
       
       def create_models
         models.each do |model|
-          @model_name = model[:model_name]
-          @model_klass_name = "#{namespace}_#{name}_Model_#{@model_name}"
-          @model_name_lower = model[:model_name_lower]
-
-          @resource_model_klass_name = "#{namespace}_#{name}_Model_Mysql4_#{@model_name}"
-          @table_name = model[:model_table_name]
-          template("Model.php", "#{model_path}#{@model_name}.php")
-          template("Resource_Model.php", "#{resource_model_path}#{@model_name}.php")
+          @model = model
+          template("Model.php", "#{model_path}#{@model.name}.php")
+          template("Resource_Model.php", "#{resource_model_path}#{@model.name}.php")
+          if model.collection?
+            empty_directory("#{resource_model_path}#{@model.name}/")
+            template("Collection.php", "#{resource_model_path}#{@model.name}/Collection.php")
+          end  
         end       
       end  
       

data/lib/saruman/generators/model/templates/Collection.php

@@ -0,0 +1,10 @@
+<?php
+
+class <%= @model.collection_model_klass_name %> extends Mage_Core_Model_Mysql4_Collection_Abstract
+{
+    public function _construct()
+    {
+        parent::_construct();
+        $this->_init('<%= name_lower %>/<%= @model.name_lower %>');
+    }
+}

data/lib/saruman/generators/model/templates/Model.php

@@ -1,10 +1,10 @@
 <?php
 
-class <%= @model_klass_name %> extends Mage_Core_Model_Abstract
+class <%= @model.klass_name %> extends Mage_Core_Model_Abstract
 {
     public function _construct()
     {
         parent::_construct();
-        $this->_init('<%= name_lower %>/<%= @model_name_lower %>');
+        $this->_init('<%= name_lower %>/<%= @model.name_lower %>');
     }
 }

data/lib/saruman/generators/model/templates/Resource_Model.php

@@ -1,10 +1,10 @@
 <?php
 
-class <%= @resource_model_klass_name %> extends Mage_Core_Model_Mysql4_Abstract
+class <%= @model.resource_model_klass_name %> extends Mage_Core_Model_Mysql4_Abstract
 {
   public function _construct()
   {   
-    $this->_init('<%= name.downcase %>/<%= @model_name.downcase %>', 'id');
+    $this->_init('<%= name.downcase %>/<%= @model.name_lower %>', 'id');
   }
     
 }    

data/lib/saruman/generators/model/templates/mysql4-install.php

@@ -6,10 +6,10 @@ $installer->startSetup();
 
 $installer->run("
 <% models.each do |model| %>
-DROP TABLE IF EXISTS {$this->getTable('<%= model[:model_table_name] %>')};
-CREATE TABLE {$this->getTable('<%= model[:model_table_name] %>')} (
+DROP TABLE IF EXISTS {$this->getTable('<%= model.table_name %>')};
+CREATE TABLE {$this->getTable('<%= model.table_name %>')} (
   `id` int(11) unsigned NOT NULL auto_increment,
-<%= model[:sql] %>
+<%= model.sql %>
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
 <% end %>
 ");

data/lib/saruman/generators/model/templates/resource_model_config_block.xml

@@ -2,7 +2,7 @@
   <class><%= resource_model_klass_name %></class>
   <entities>
     <% models.each do |model| %>
-    <<%= model[:model_name_lower] %>><table><%= model[:model_table_name] %></table></<%= model[:model_name_lower] %>>
+    <<%= model.name_lower %>><table><%= model.table_name %></table></<%= model.name_lower %>>
     <% end %>
     
   </entities>

data/lib/saruman/generators/model/templates/resource_model_entities_models_block.xml

@@ -1,5 +1,5 @@
 <% models.each do |model| %>
-<<%= model[:model_name_lower] %>><table><%= model[:model_table_name] %></table></<%= model[:model_name_lower] %>>
+<<%= model.name_lower %>><table><%= model.table_name %></table></<%= model.name_lower %>>
 
 <% end %>
 

data/lib/saruman/generators/saruman.rb

@@ -3,4 +3,5 @@ require 'thor/group'
 require "saruman/generators/extension"
 require "saruman/generators/model"
 require "saruman/generators/observer"
+require "saruman/generators/controller"
 require "saruman"

data/lib/saruman/version.rb

@@ -1,3 +1,3 @@
 module Saruman
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/saruman/virtus/attributes.rb

@@ -0,0 +1,25 @@
+class Capitalize < Virtus::Attribute::Object
+  primitive String
+  coercion_method :capitalize
+end
+
+class Downcase < Virtus::Attribute::Object
+  primitive String
+  coercion_method :downcase
+end
+
+module Virtus
+  class Coercion
+    class String < Virtus::Coercion::Object
+      
+      def self.capitalize(value)
+        value.capitalize
+      end
+      
+      def self.downcase(value)
+        value.downcase
+      end
+      
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: saruman
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-06-17 00:00:00.000000000 Z
+date: 2012-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -156,12 +156,18 @@ files:
 - doc_assets/saruman_extension_example.jpg
 - lib/saruman.rb
 - lib/saruman/cli.rb
+- lib/saruman/generators/controller.rb
+- lib/saruman/generators/controller/templates/Block.php
+- lib/saruman/generators/controller/templates/Controller.php
+- lib/saruman/generators/controller/templates/local.xml
+- lib/saruman/generators/controller/templates/view.phtml
 - lib/saruman/generators/extension.rb
 - lib/saruman/generators/extension/templates/Helper.php
 - lib/saruman/generators/extension/templates/Observer.php
 - lib/saruman/generators/extension/templates/extension_config.xml
 - lib/saruman/generators/extension/templates/module.xml
 - lib/saruman/generators/model.rb
+- lib/saruman/generators/model/templates/Collection.php
 - lib/saruman/generators/model/templates/Model.php
 - lib/saruman/generators/model/templates/Resource_Model.php
 - lib/saruman/generators/model/templates/model_config_block.xml
@@ -173,6 +179,7 @@ files:
 - lib/saruman/generators/observer/templates/Observer.php
 - lib/saruman/generators/saruman.rb
 - lib/saruman/version.rb
+- lib/saruman/virtus/attributes.rb
 - saruman.gemspec
 - spec/saruman_spec.rb
 homepage: https://github.com/jasonayre/saruman.git