#!/usr/bin/env ruby
# encoding: UTF-8

# (c) Copyright 2014 Hewlett-Packard Development Company, L.P.
#
#    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.

# ~ require 'rubygems'
require 'lorj'
require 'highline/import'
require 'cloud_connection.rb'

# ~ require 'security.rb'
# ~ include SecurityGroup

#
# ssh module
#
module Forj
  # This module provides the behavior to create a ssh connection
  module Ssh
    def self.connect(name, account)
      o_cloud = Forj::CloudConnection.connect(account)

      PrcLib.state(format("Getting information about forge '%s'", name))

      o_forge = o_cloud.get(:forge, name)

      if o_forge[:servers].count > 0
        if account[:box_ssh]
          o_server = validate_server_name(name, account, o_forge)

          if !o_server.nil?
            ssh_connection(account, o_cloud, name, o_server)
          else
            PrcLib.debug("server '%s.%s' was not found",
                         account[:box_ssh], name)
            PrcLib.high_level_msg("server '%s.%s' was not found.\n",
                                  account[:box_ssh], name)
          end
        else
          server = select_forge_server(o_forge)

          ssh_connection(account, o_cloud, name, server)
        end
      else
        PrcLib.high_level_msg("No server(s) found for instance name '%s' \n",
                              name)
      end
    end

    def self.select_forge_server(o_forge)
      # Ask the user to get server(s) to create ssh connection
      server_list = []
      servers = []
      s_default = nil

      o_forge[:servers].each do |server_type, server|
        server_list << server[:name]
        servers << server
        s_default = server[:name] if server_type == 'maestro'
      end

      say(format('Select box for ssh connection %s',
                 ((s_default.nil?) ? '' : "Default: #{s_default}")))
      value = choose do |q|
        q.choices(*server_list)
        q.default = s_default unless s_default.nil?
      end

      servers[server_list.index(value)]
    end

    def self.ssh_connection(_account, o_cloud, _name, server)
      o_cloud.create(:ssh, :server => server)
    end

    def self.validate_server_name(name, account, o_forge)
      box_ssh = account[:box_ssh]

      o_server = nil
      regex =  Regexp.new(format('%s\.%s$', box_ssh, name))

      o_forge[:servers].each do |server|
        o_name = server[:name]
        next if (regex =~ o_name).nil?
        o_server = server
        break
      end

      o_server
    end
  end
end