################################################################################
#
#      Author: Zachary Patten <zachary@jovelabs.com>
#   Copyright: Copyright (c) 2011-2013 Atalanta Systems Ltd
#     License: Apache License, Version 2.0
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
################################################################################

module Cucumber
  module Chef
    class Provider

      class VagrantError < Error; end

      class Vagrant
        # attr_accessor :env, :vm

        INVALID_STATES = %w(not_created aborted unknown).map(&:to_sym)
        RUNNING_STATES =  %w(running).map(&:to_sym)
        SHUTDOWN_STATES = %w(paused saved poweroff).map(&:to_sym)
        VALID_STATES = RUNNING_STATES+SHUTDOWN_STATES

################################################################################

        def initialize(ui=ZTK::UI.new)
          @ui = ui
        end

################################################################################
# CREATE
################################################################################

        def create
          ZTK::Benchmark.bench(:message => "Creating #{Cucumber::Chef::Config.provider.upcase} instance", :mark => "completed in %0.4f seconds.", :stdout => @stdout) do

            context = {
              :ip => Cucumber::Chef.lab_ip,
              :cpus => Cucumber::Chef::Config.vagrant[:cpus],
              :memory => Cucumber::Chef::Config.vagrant[:memory]
            }

            vagrantfile_template = File.join(Cucumber::Chef.root_dir, "lib", "cucumber", "chef", "templates", "cucumber-chef", "Vagrantfile.erb")

            vagrantfile = File.join(Cucumber::Chef.chef_repo, "Vagrantfile")

            !File.exists?(vagrantfile) and IO.write(vagrantfile, ::ZTK::Template.render(vagrantfile_template, context))

            self.vagrant_cli("up", id)
            ZTK::TCPSocketCheck.new(:host => self.ip, :port => self.port, :wait => 120).wait
          end

          self

        rescue Exception => e
          @ui.logger.fatal { e.message }
          @ui.logger.fatal { e.backtrace.join("\n") }
          raise VagrantError, e.message
        end

################################################################################
# DESTROY
################################################################################

        def destroy
          if exists?
            self.vagrant_cli("destroy", "--force", id)
          else
            raise VagrantError, "We could not find a test lab!"
          end

        rescue Exception => e
          @ui.logger.fatal { e.message }
          @ui.logger.fatal { e.backtrace.join("\n") }
          raise VagrantError, e.message
        end

################################################################################
# UP
################################################################################

        def up
          if (exists? && dead?)
            self.vagrant_cli("up", id)
            ZTK::TCPSocketCheck.new(:host => self.ip, :port => self.port, :wait => 120).wait
          else
            raise VagrantError, "We could not find a powered off test lab."
          end

        rescue Exception => e
          @ui.logger.fatal { e.message }
          @ui.logger.fatal { e.backtrace.join("\n") }
          raise VagrantError, e.message
        end

################################################################################
# HALT
################################################################################

        def down
          if (exists? && alive?)
            self.vagrant_cli("halt", id)
          else
            raise AWSError, "We could not find a running test lab."
          end

        rescue Exception => e
          @ui.logger.fatal { e.message }
          @ui.logger.fatal { e.backtrace.join("\n") }
          raise VagrantError, e.message
        end

################################################################################
# RELOAD
################################################################################

        def reload
          if (exists? && alive?)
            self.vagrant_cli("reload", id)
          else
            raise AWSError, "We could not find a running test lab."
          end

        rescue Exception => e
          @ui.logger.fatal { e.message }
          @ui.logger.fatal { e.backtrace.join("\n") }
          raise VagrantError, e.message
        end

################################################################################

        def exists?
          (self.state != :not_created)
        end

        def alive?
          (RUNNING_STATES.include?(self.state) rescue false)
        end

        def dead?
          (SHUTDOWN_STATES.include?(self.state) rescue true)
        end

################################################################################

        def id
          "test-lab-#{ENV['USER']}".downcase
        end

        def state
          output = self.vagrant_cli("status | grep '#{id}'").output
          result = :unknown
          (VALID_STATES+INVALID_STATES).each do |state|
            if output =~ /#{state}/
              result = state
              break
            end
          end
          result.to_sym
        end

        def username
          "vagrant"
        end

        def ip
          "192.168.33.10"
        end

        def port
          22
        end

################################################################################

        def vagrant_cli(*args)
          command = Cucumber::Chef.build_command("vagrant", *args)
          ZTK::Command.new.exec(command, :silence => true)
        end

################################################################################

      end

    end
  end
end

################################################################################