require 'bebox/provision'
require 'bebox/node'

module Bebox
  class ProvisionWizard
    include Bebox::Logger
    # Apply a step for the nodes in a environment
    def apply_step(project_root, environment, step)
      # Check if environment has configured the ssh keys
      if Bebox::Environment.check_environment_access(project_root, environment)
        # Check already in step nodes
        nodes_to_step = check_nodes_to_step(project_root, environment, step)
        # Output the nodes that are ready for provisioning step-N
        if nodes_to_step.count > 0
          title "Provisioning #{step} in nodes:"
          nodes_to_step.each{|node| msg(node.hostname)}
          linebreak
          # Apply the nodes provisioning for step-N
          nodes_to_step.each do |node|
            title "Applying #{step} in node #{node.hostname}:"
            role = Bebox::Provision.role_from_node(project_root, step, node.hostname)
            profiles = Bebox::Provision.profiles_from_role(project_root, role) unless role.nil?
            # Before apply generate the Puppetfile with modules from all associated profiles
            Bebox::Provision.generate_puppetfile(project_root, step, profiles) unless profiles.nil?
            # Before apply generate the roles and profiles modules structure for puppet step
            Bebox::Provision.generate_roles_and_profiles(project_root, step, role, profiles)
            provision = Bebox::Provision.new(project_root, environment, node, step)
            provision.apply.success? ? (ok "Node '#{node.hostname}' provisioned to #{step}.") : (error "An error ocurred in the provision of #{step} for node '#{node.hostname}'")
          end
        else
          warn "There are no nodes for provision in #{step}. No changes were made."
        end
      else
        warn "Please add a ssh key pair (id_rsa, id_rsa.pub) in config/keys/environments/#{environment} to do this step."
      end
    end

    # Check the nodes already in step and ask confirmation to re-do-it
    def check_nodes_to_step(project_root, environment, step)
      nodes_to_step = []
      nodes = Bebox::Node.nodes_in_environment(project_root, environment, previous_checkpoint(step))
      in_step_nodes = Bebox::Node.list(project_root, environment, "steps/#{step}")
      nodes.each do |node|
        if in_step_nodes.include?(node.hostname)
          nodes_to_step << node if confirm_node_step?(node, step)
        else
          nodes_to_step << node
        end
      end
      nodes_to_step
    end

    # Obtain the previous checkpoint (step/phase) for a node
    def previous_checkpoint(step)
      case step
        when 'prepared_nodes'
          'nodes'
        when 'step-0'
          'prepared_nodes'
        when 'step-1'
          'steps/step-0'
        when 'step-2'
          'steps/step-1'
        when 'step-3'
          'steps/step-2'
      end
    end

    # Ask for confirmation of node step
    def confirm_node_step?(node, step)
      node_type = "steps/#{step}"
      checkpoint_status = "(start: #{node.checkpoint_parameter_from_file(node_type, 'started_at')} - end: #{node.checkpoint_parameter_from_file(node_type, 'finished_at')})"
      quest "The node '#{node.hostname}' was already provisioned in #{step} #{checkpoint_status}.\nDo you want to re-provision it?"
      response =  ask(highline_quest('(y/n)')) do |q|
        q.default = "n"
      end
      return response == 'y' ? true : false
    end
  end
end