module Lina
  module Generators
    class ApiGenerator < Rails::Generators::NamedBase # :nodoc:
      source_root File.expand_path('../templates', __FILE__)

      argument :actions, type: :array, default: [], banner: 'action1 action2'
      check_class_collision suffix: 'Controller'

      def create_controller_files
        template 'controller.rb', File.join('app/controllers', class_path, "#{file_name}_controller.rb")
      end

      def add_routes
        actions.reverse.each do |action|
          route generate_routing_code(action)
        end
      end

      hook_for :test_framework, as: :controller

      if Lina.generate_views
        hook_for :view, default: true
      end

      private

      # This method creates nested route entry for namespaced resources.
      # For eg. rails g controller foo/bar/baz index
      # Will generate -
      # namespace :foo do
      #   namespace :bar do
      #     get 'baz/index'
      #   end
      # end
      def generate_routing_code(action)
        depth = regular_class_path.length
        # Create 'namespace' ladder
        # namespace :foo do
        #   namespace :bar do
        namespace_ladder = regular_class_path.each_with_index.map do |ns, i|
          indent("namespace :#{ns} do\n", i * 2)
        end.join

        # Create route
        #     get 'baz/index'
        route = indent(%{get '#{file_name}/#{action}'\n}, depth * 2)

        # Create `end` ladder
        #   end
        # end
        end_ladder = (1..depth).reverse_each.map do |i|
          indent("end\n", i * 2)
        end.join

        # Combine the 3 parts to generate complete route entry
        namespace_ladder + route + end_ladder
      end
    end
  end
end