dsawardekar/speckle

module Speckle
  module CLI

    require 'rake'
    require 'speckle/list/builder'
    require 'speckle/list/absolute_path_transformer'
    require 'speckle/list/dir_expander'
    require 'speckle/list/extension_transformer'
    require 'speckle/list/file_content_filter'
    require 'speckle/list/pattern_filter'

    class RakeApp
      def initialize(options)
        @options = options
      end

      def inputs
        @options.inputs
      end

      def verbose
        @options.verbose
      end

      def debug
        @options.debug
      end

      def rake
        if @rake_app
          return @rake_app
        end

        configure_rake
        Dir.chdir @options.root_dir

        @rake_app = Rake.application
        @rake_app.init
        @rake_app.load_rakefile

        Dir.chdir @options.cwd
        @rake_app
      end

      def invoke_task(name)
        #puts "invoke_task: #{name}"
        #rake
        rake.invoke_task("speckle:#{name.to_s}")
      end

      def rake_env(key, value)
        unless value.nil?
          ENV[key] = if value.is_a?(Array) then value.join(';') else value end
          puts "rake_env: #{key} = #{ENV[key]}" if debug
        end
      end

      def configure_rake
        rake_env('TEST_SOURCES', test_sources)
        rake_env('TEST_LIBS', test_libs)
        rake_env('BUILD_DIR', test_build_dir)
        rake_env('TEST_COMPILED', test_compiled)
        rake_env('TEST_VIM', @options.vim)
        rake_env('TEST_REPORTER', @options.reporter)
        rake_env('SLOW_THRESHOLD', @options.slow_threshold.to_s)
        rake_env('SKIP_VIMRC', to_int(@options.skip_vimrc))
        rake_env('COLORIZE', to_int(@options.colorize))
        rake_env('BAIL', to_int(@options.bail))
        rake_env('TAG', @options.tag)

        if @options.profile
          rake_env('PROFILE', 'yes')
        end

        if @options.verbose
          rake_env('VERBOSE', 'yes')
        end

        if @options.debug
          rake_env('DEBUG', 'yes')
        end
      end

      def to_int(option)
        option ? '1' : '0'
      end

      def test_build_dir
        "#{@options.cwd}/build"
      end

      def test_sources
        get_builder.build
      end

      def test_compiled
        sources = test_sources
        sources.collect do |source|
          source.ext('vim')
        end
      end

      def get_builder
        if @source_builder.nil?
          @source_builder = Speckle::List::Builder.new
          @options.inputs.each do |input|
            @source_builder.add_source input
          end

          @source_builder.add_filter Speckle::List::DirExpander.new('**/*_spec.riml')

          unless @options.tag.nil?
            @source_builder.add_filter Speckle::List::FileContentFilter.new(@options.tag, false)
          end

          unless @options.grep_pattern.nil?
            @source_builder.add_filter Speckle::List::PatternFilter.new(@options.grep_pattern, @options.grep_invert)
          end

          # Not using this at the moment, as it's simpler to not rebuild the source paths for test_compiled
          #@source_builder.add_filter Speckle::List::AbsolutePathTransformer.new
          #@source_builder.add_filter Speckle::List::ExtensionTransformer.new('vim')
        end

        @source_builder
      end

      def test_libs
        input_libs = @options.libs
        return nil if input_libs.nil?

        input_libs = input_libs.split(':')
        input_libs << 'spec'
        if File.directory?(@options.speckle_lib_dir)
          input_libs << @options.speckle_lib_dir
        end

        libs = []
        input_libs.each do |lib|
          libs << File.absolute_path(lib)
        end

        libs.join(':')
      end

    end
  end
end