lib/esse/cli/generate.rb in esse-0.2.2 vs lib/esse/cli/generate.rb in esse-0.2.3

- old
+ new

@@ -14,23 +14,21 @@ end desc 'index NAME *TYPES', 'Creates a new index' option :settings, type: :boolean, default: false, desc: 'Generate settings' option :mappings, type: :boolean, default: false, desc: 'Generate mappings' - option :serializers, type: :boolean, default: false, desc: 'Generate serializers' + option :documents, type: :boolean, default: false, desc: 'Generate documents' option :collections, type: :boolean, default: false, desc: 'Generate collections' option :active_record, type: :boolean, default: false, desc: 'Generate ActiveRecord models' option :cluster_id, type: :string, desc: 'Elasticsearch cluster ID' - def index(name, *types) + def index(name, *repos) ns_path = name.split(NAMESPACE_PATTERN_RE).tap(&:pop) @index_name = Hstring.new(name.to_s).modulize.sub(/Index$/, '') + 'Index' @index_name = Hstring.new(@index_name) - @types = types.map { |type| Hstring.new(type) } + @repos = repos.map { |repo| Hstring.new(repo) } @base_class = base_index_class(*ns_path) - if options[:cluster_id] - @base_class += format('(:%s)', options[:cluster_id]) - end + @index_cluster_id = options[:cluster_id] @cli_options = options base_dir = Esse.config.indices_directory.join(*ns_path.map { |n| Hstring.new(n).underscore.to_s }) index_name = @index_name.demodulize.underscore.to_s template( @@ -50,37 +48,37 @@ 'templates/mappings.json', base_dir.join(index_name, 'templates', 'mappings.json'), ) end - if @types.empty? - if options[:serializers] + if @repos.empty? + if options[:documents] template( - 'templates/serializer.rb.erb', - base_dir.join(index_name, 'serializers', 'serializer.rb'), + 'templates/document.rb.erb', + base_dir.join(index_name, 'documents', 'document.rb'), ) end if options[:collections] && !options[:active_record] template( 'templates/collection.rb.erb', base_dir.join(index_name, 'collections', 'collection.rb'), ) end end - @types.each do |type| - @type = Hstring.new(type).underscore + @repos.each do |type| + @repo = Hstring.new(type).underscore - if options[:serializers] + if options[:documents] template( - 'templates/serializer.rb.erb', - base_dir.join(index_name, 'serializers', "#{@type}_serializer.rb"), + 'templates/document.rb.erb', + base_dir.join(index_name, 'documents', "#{@repo}_document.rb"), ) end if options[:collections] && !options[:active_record] template( 'templates/collection.rb.erb', - base_dir.join(index_name, 'collections', "#{@type}_collection.rb"), + base_dir.join(index_name, 'collections', "#{@repo}_collection.rb"), ) end end end