require 'fileutils'
module Noe
  class Main
    #
    # Instantiate a project template using a .noespec file
    #
    # SYNOPSIS
    #   #{program_name} #{command_name} [--dry-run] [--force|--interactive|--add-only|--safe-override] [SPEC_FILE]
    #
    # OPTIONS
    # #{summarized_options}
    #
    # DESCRIPTION
    #   This command instantiate a project template using a .noespec file 
    #   given as first argument. If no spec file is specified, Noe expects 
    #   one .noespec file to be present in the current directory and uses it.
    # 
    #   This command is generally used immediately after invoking 'prepare',
    #   on an almost empty directory. By default it safely fails if any file
    #   would be overriden by the instantiation process. This safe behavior 
    #   can be bypassed through the --force, --add-only, --interactive and
    #   --safe-override options.
    #
    # TYPICAL USAGE
    #
    #   When a fresh new project is created, this command is typically used
    #   within the following scenario
    #
    #     noe prepare --template=ruby hello_world
    #     cd hello_world
    #     edit hello_world.noespec
    #     noe go
    #
    #   If you modify your .noespec file and want to force overriding of all 
    #   files:
    #     
    #     noe go --force
    #
    #   If you want to regenerate some files only (README and gemspec, for 
    #   example):
    #
    #     rm README.md hello_world.gemspec
    #     noe go --add-only
    #
    #   If you want to regenerate some files by controlling what will be 
    #   overriden:
    #
    #     noe go --interactive
    #
    #   If you want to regenerate some files according to the template 
    #   manifest:
    #
    #     noe go --safe-override
    #
    class Go < Quickl::Command(__FILE__, __LINE__)
      include Noe::Commons

      attr_reader :dry_run
      
      attr_reader :force
      alias :force? :force

      attr_reader :adds_only
      alias :adds_only? :adds_only

      attr_reader :safe_override
      alias :safe_override? :safe_override

      # Install options
      options do |opt|
        @dry_run = false
        opt.on('--dry-run', '-d',
               "Say what would be done but don't do it"){ 
          @dry_run = true
        }
        opt.separator ""
        opt.separator "File overriding control: "
        @force = false
        opt.on('--force', '-f',
               "Force overriding on all existing files"){ 
          @force = true
        }
        @interactive = false
        opt.on('--interactive', '-i',
               "Request the user to take a decision"){ 
          @interactive = true
          @highline = HighLine.new
        }
        @adds_only = false
        opt.on('--add-only', '-a',
               "Only make additions, do not override any existing file"){ 
          @adds_only = true
        }
        @safe_override = false
        opt.on('--safe-override', '-s',
               "Follow safe-override information provided by the manifest"){ 
          @safe_override = true
        }
        Commons.add_common_options(opt)
      end
      
      # Checks if the interactive mode is enabled. If yes, highline is prepared
      # as a side effect.
      def interactive?
        if @interactive and not(@highline)
          begin
            require "highline"
            @highline = HighLine.new
          rescue LoadError
            raise Quickl::Exit.new(1), "Highline is required for interactive mode, try 'gem install highline'"
          end
        else
          @interactive
        end
      end
      
      def say(what, color)
        if @highline
          @highline.say(@highline.color(what, color))
        else
          puts what
        end
      end
      
      def choose(&block)
        @highline.choose(&block)
      end
      
      # Checks if one is a file and the other a directory or the inverse
      def kind_clash?(entry, relocated)
        (entry.file? and relocated.directory?) or
        (entry.directory? and relocated.file?)
      end
      
      def build_one_directory(entry, variables)
        return if entry.ignore?
        relocated = entry.relocate(variables)
        todo = []
        
        skipped = false
        if relocated.exists?
          # file exists already exists, check what can be done!
          if kind_clash?(entry, relocated)
            if interactive?
              say("File #{relocated} conflicts with directory to create", :red)
              choose do |menu|
                menu.prompt = "What do we do?"
                menu.index = :letter
                menu.choice(:abord) { raise Quickl::Exit.new(1), "Noe aborted!" }
                menu.choice(:remove){ todo << Rm.new(entry, variables)          }
              end 
            elsif force?
              todo << Rm.new(entry, variables)
            else
              raise Quickl::Exit.new(2), "Noe aborted: file #{relocated} already exists.\n"\
                                         "Use --force to override or --interactive for more options."
            end
          else
            # file exists and is already a folder; we simply do nothing 
            # because there is no dangerous action here
            skipped = true
          end
        end 
        
        # Create the directory unless it has been explicitely skipped
        unless skipped
          todo << MkDir.new(entry, variables)
        end
        
        todo
      end

      def build_one_file(entry, variables)
        return if entry.ignore?
        relocated = entry.relocate(variables)
        todo = []
        
        skipped = false
        if relocated.exists?
          # name clash, the file exists
          if adds_only?
            # file exists and we are only allowed to add new things
            # we just do nothing
            skipped = true
          elsif interactive? and kind_clash?(entry, relocated)
            say("Directory #{relocated} conflicts with file to create", :red)
            choose do |menu|
              menu.prompt = "What do we do?"
              menu.index = :letter
              menu.choice(:abord) { raise Quickl::Exit.new(1), "Noe aborted!" }
              menu.choice(:remove){ todo << Rm.new(entry, variables)          }
              menu.choice(:skip)  { skipped = true                            }
            end 
          elsif interactive?
            say("File #{relocated} already exists", :red)
            choose do |menu|
              would = entry.safe_override? ? :override : :skip
              menu.prompt = "What do we do? (safe-override would #{would})"
              menu.index = :letter
              menu.choice(:abord)   { raise Quickl::Exit.new(1), "Noe aborted!" }
              menu.choice(:override){ todo << Rm.new(entry, variables)          }
              menu.choice(:skip)    { skipped = true                            }
            end
          elsif safe_override?
            if entry.safe_override?
              todo << Rm.new(entry, variables)
            else
              skipped = true
            end
          elsif force?
            todo << Rm.new(entry, variables)
          else
            raise Quickl::Exit.new(2), "Noe aborted: file #{relocated} already exists.\n"\
                                       "Use --force to override or --interactive for more options."
          end
        end
        
        # Add file instantiation unless skipped
        unless skipped
          todo << FileInstantiate.new(entry, variables)
        end
        
        todo
      end
      
      def execute(args)
        raise Quickl::Help if args.size > 1
        
        # Load spec now
        spec_file = find_noespec_file(args)
        spec = YAML::load(spec_file.read)
        template = template(spec['template-info']['name'])
        template.merge_spec(spec)
        variables = template.variables
        
        # Build what has to be done
        commands = template.collect{|entry|
          if entry.file?
            build_one_file(entry, variables)
          elsif entry.directory?
            build_one_directory(entry, variables)
          end
        }.flatten.compact
        
        # let's go now
        if dry_run
          commands.each{|c| puts c}
        else
          commands.each{|c| c.run}
        end
        
      end
      
      class DoSomething

        attr_reader :entry
        attr_reader :variables

        def initialize(entry, variables)
          @entry, @variables = entry, variables
        end
        
        def template
          entry.template
        end
        
        def relocated
          entry.relocate(variables)
        end

      end
        
      class MkDir < DoSomething

        def run
          relocated.mkdir
        end
        
        def to_s
          "mkdir #{relocated}"
        end
        
      end # class MkDir
      
      class Rm < DoSomething

        def run
          relocated.rm_rf
        end
        
        def to_s
          "rm -rf #{relocated}"
        end
        
      end # class Rm
      
      class FileInstantiate < DoSomething
        
        def run
          source = entry.realpath.to_s
          target = relocated
          if dialect = entry.wlang_dialect
            target.open('w') do |out|
              braces = entry.wlang_braces
              variables.methodize!
              out << WLang::file_instantiate(source, variables, dialect, braces)
            end
          else
            FileUtils.cp(source, target)
          end
          Path(target).chmod(Path(source).stat.mode)
        end
        
        def to_s
          "wlang #{entry.path} > #{relocated}"
        end
        
      end

    end # class Go
  end # class Main
end # module Noe