added a variable to the to the generator

Author: Joel Quenneville

lib/generators/ajaxdatatable/ajaxdatatable_generator.rb

@@ -1,7 +1,8 @@
 class AjaxdatatableGenerator < Rails::Generators::Base
   source_root File.expand_path('../templates', __FILE__)
+  argument :model, type: :string
 
   def generate_ajaxdatatable
-    template 'datatable.rb', File.join('app/datatables', "datatable.rb")
+    template 'datatable.rb', File.join('app/datatables', "#{model}.rb")
   end
 end

lib/generators/ajaxdatatable/templates/datatable.rb

@@ -1,5 +1,5 @@
-class <%#= class_name %>Datatable < AjaxDatatableRails
-  model_name <%#= class_name %>
+class <%= model.classify %>Datatable < AjaxDatatableRails
+  model_name <%= model.classify %>
   columns # insert array of column names here
   searchable_columns #insert array of columns that will be searched
 
@@ -9,8 +9,8 @@ def data
       # generate a 2-dimensional array that holds the data
     end
 
-    def <%#= class_name.downcase %>
-      @records ||= fetch_records
+    def <%= model.tableize %>
+      @<%= model.tableize %> ||= fetch_records
     end
 
     def get_raw_records