require 'rails/generators/generated_attribute'

module AppThemer
  class ViewsGenerator < Rails::Generators::Base
    source_root File.expand_path('../templates', __FILE__)

    argument :controller_path,  :type => :string
    argument :model_name,       :type => :string, :required => false

    class_option :layout,         :type => :string,   :desc => 'Specify the layout name'
    class_option :engine,         :type => :string,   :default => 'erb', :desc => 'Specify the template engine'
    class_option :will_paginate,  :type => :boolean,  :default => false, :desc => 'Specify if you use will_paginate'
    class_option :view_type,      :type => :string,   :default => 'crud', :desc => 'Specify the type of view(s): crud, text or login.'

    def initialize(args, *options)
      super(args, *options)
      initialize_views_variables
    end

    def copy_views
      generate_views
      unless options.layout.blank?
        gsub_file(File.join('app/views/layouts', "#{options[:layout]}.html.#{options.engine}"), /\<nav\s+id=\"main-navigation\">.*\<\/ul\>/mi) do |match|
          match.gsub!(/\<\/ul\>/, "")
          if @engine.to_s =~ /haml/
            %|#{match}
          %li{:class => controller.controller_path == '#{@controller_file_path}' ? 'active' : '' }
            %a{:href => #{controller_routing_path}_path} #{plural_model_name}
          </ul>|
          else
            %|#{match} <li class="<%= controller.controller_path == '#{@controller_file_path}' ? 'active' : '' %>"><a href="<%= #{controller_routing_path}_path %>">#{plural_model_name}</a></li></ul>|
          end
        end
      end
    end

  protected

    def initialize_views_variables
      @base_name, @controller_class_path, @controller_file_path, @controller_class_nesting, @controller_class_nesting_depth = extract_modules(controller_path)
      @controller_routing_path = @controller_file_path.gsub(/\//, '_')
      @model_name = @base_name.singularize unless @model_name
      @model_name = @model_name.camelize
    end

    def controller_routing_path
      @controller_routing_path
    end

    def singular_controller_routing_path
      @controller_routing_path.singularize
    end

    def model_name
      @model_name
    end

    def plural_model_name
      @model_name.pluralize
    end

    def resource_name
      @model_name.underscore
    end

    def plural_resource_name
      resource_name.pluralize
    end

    def columns
      excluded_column_names = %w[id created_at updated_at]
      Kernel.const_get(@model_name).columns.reject{|c| excluded_column_names.include?(c.name) }.collect{|c| Rails::Generators::GeneratedAttribute.new(c.name, c.type)}
    end

    def extract_modules(name)
      modules = name.include?('/') ? name.split('/') : name.split('::')
      name    = modules.pop
      path    = modules.map { |m| m.underscore }
      file_path = (path + [name.underscore]).join('/')
      nesting = modules.map { |m| m.camelize }.join('::')
      [name, path, file_path, nesting, modules.size]
    end

    def generate_views
      views = {
        'crud' => {
          'tables.html.erb'  => File.join('app/views', @controller_file_path, "index.html.#{options.engine}"),
          'new.html.erb'     => File.join('app/views', @controller_file_path, "new.html.#{options.engine}"),
          'edit.html.erb'    => File.join('app/views', @controller_file_path, "edit.html.#{options.engine}"),
          '_form.html.erb'    => File.join('app/views', @controller_file_path, "_form.html.#{options.engine}"),
          'show.html.erb'    => File.join('app/views', @controller_file_path, "show.html.#{options.engine}"),
          '_sidebar.html.erb' => File.join('app/views', @controller_file_path, "_sidebar.html.#{options.engine}")
        },
        'text' => {
          'text.html.erb'    => File.join('app/views', @controller_file_path, "show.html.#{options.engine}"),
          '_sidebar.html.erb' => File.join('app/views', @controller_file_path, "_sidebar.html.#{options.engine}")
        },
        'login' => {
          'login.html.erb'    => File.join('app/views', @controller_file_path, "new.html.#{options.engine}")
        }
      }
      selected_views = views[options.view_type]
      options.engine == 'haml' ? generate_haml_views(selected_views) : generate_erb_views(selected_views)
    end

    def generate_erb_views(views)
      views.each do |template_name, output_path|
        template template_name, output_path
      end
    end

    def generate_haml_views(views)
      require 'haml'
      Dir.mktmpdir('app-themer-haml') do |haml_root|
        views.each do |template_name, output_path|
          tmp_html_path = "#{haml_root}/#{template_name}"
          tmp_haml_path = "#{haml_root}/#{template_name}.haml"
          template template_name, tmp_html_path, :verbose => false
          `html2haml --erb --xhtml #{tmp_html_path} #{tmp_haml_path}`
          copy_file tmp_haml_path, output_path
        end
      end

    rescue LoadError
      say "HAML is not installed, or it is not specified in your Gemfile."
      exit
    end
  end
end