#!/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_relative 'network.rb'
include Network
require_relative 'yaml_parse.rb'
include YamlParse
require_relative 'security.rb'
include SecurityGroup
require_relative 'repositories.rb'
include Repositories
#require_relative 'log.rb'
#include Logging
require_relative 'helpers.rb'
include Helpers


#
# Boot module
#
module Boot
  def boot(blueprint, name, build, boothook, box_name, oConfig)
    begin

      Logging.fatal(1, 'FORJ account not specified. Did you used `forj setup`, before?') if not oConfig.get(:account_name)

      # Check options and set data
      cloud_provider = oConfig.get(:provider_name, 'hpcloud')

      if cloud_provider != 'hpcloud'
         Logging.fatal(1, "forj setup support only hpcloud. '%s' is currently not supported." % cloud_provider)
      end

      initial_msg = "booting %s on %s\nCheck log with `tail -f ~/.forj/forj.log`.\nUse --verbose or --debug for more boot status on screen" % [blueprint , cloud_provider]
      Logging.high_level_msg(initial_msg) #################

      # Initialize defaults
      maestro_url =  oConfig.get(:maestro_url)

      infra_dir = File.expand_path(oConfig.get(:infra_repo))

      # Ask information if needed.
      if not Dir.exist?(File.expand_path(infra_dir))
         Logging.warning(<<-END
Your infra workspace directory is missing.

Forj uses an infra workspace directory to store any kind of data that are private to you.
We provides ways to send those data securily to your new Forge instance, as metadata.
In production case, we suggest you to keep it safe in your SCM preferred database.

If you already have an existing infra workspace, use 'forj set infra_repo=<PathToYourRepo>' to set it and restart.

Otherwise, we will build a new one with some predefined data, you can review and update later.
         END
         )
         sAsk = "Do you want to create a new one from Maestro (yes/no)?" % [infra_dir]
         bBuildInfra=agree(sAsk)
         if not bBuildInfra
            puts 'Process aborted on your demand.'
            exit 0
         end
      else
         # Check about infra repo compatibility with forj cli
         bBuildInfra = Repositories.infra_rebuild_required?(oConfig, infra_dir)
      end

      # Get FORJ DNS setting
      yDNS = rhGet(oConfig.hAccountData, :dns)
      Logging.fatal(1, "DNS or domain name are missing. Please execute forj setup %s" % oConfig.getAccountData(:account, 'name')) if not yDNS

      branch = oConfig.get(:branch, 'master')

      # Step Maestro Clone
      if not oConfig.get(:maestro_repo)
         Logging.state("cloning maestro repo from '%s' - branch '%s'" % [maestro_url, branch])
         Repositories.clone_repo(maestro_url, oConfig)
         maestro_repo=File.expand_path('~/.forj/maestro')
      else
         maestro_repo=File.expand_path(oConfig.get(:maestro_repo))
         if not File.exists?('%s/templates/infra/maestro.box.master.env' % [maestro_repo])
            msg = <<-END
'#{maestro_repo}' is not a recognized Maestro repository or is too old. forj cli searched for templates/infra/maestro.box.master.env

Suggestion:
1. Clone #{maestro_url} to a different location.
   $ mkdir -p ~/src/forj-oss
   $ cd ~/src/forj-oss
   $ git clone #{maestro_url}
2. Use this master branch of maestro repository with forj
   $ forj set maestro_repo=~/src/forj-oss/maestro

then retry your boot.
END
            Logging.fatal(1, msg)
         end
         Logging.info("Using your maestro cloned repo '%s'" % maestro_repo)
      end

      bBuildInfra = Repositories.infra_rebuild(oConfig, infra_dir) unless bBuildInfra

      if bBuildInfra
         Logging.info("Building your infra... in '%s'" % [infra_dir])
         Repositories.create_infra(maestro_repo, branch)
      else
         Logging.info("Re-using your infra... in '%s'" % [infra_dir]) if not bBuildInfra
      end

      Repositories.ensure_build_env_file(maestro_url, maestro_repo, branch)

      # Connect to services
      oFC=ForjConnection.new(oConfig)

      Logging.info('Configuring network \'%s\'' % [oConfig.get('network_name')])
      begin
        network = Network.get_or_create_network(oFC, oConfig.get('network_name'))
        subnet = Network.get_or_create_subnet(oFC, network.id, network.name)
        Network.get_or_create_router(oFC, network, subnet)
      rescue => e
        Logging.fatal(1, "Network properly configured is required.", e)
      end

      Logging.info('Configuring keypair \'%s\'' % [oConfig.get('keypair_name')])
      SecurityGroup.hpc_import_key(oConfig)


      Logging.info('Configuring Security Group \'%s\'' % [oConfig.get('security_group')])
      security_group = SecurityGroup.get_or_create_security_group(oFC, oConfig.get('security_group'))
      ports = oConfig.get(:ports)

      ports.each do |port|
        port = port.to_s if port.class != String
        if not (/^\d+(-\d+)?$/ =~ port)
           Logging.error("Port '%s' is not valid. Must be <Port> or <PortMin>-<PortMax>" % [port])
        else
           mPortFound = /^(\d+)(-(\d+))?$/.match(port)
           portmin = mPortFound[1]
           portmax = (mPortFound[3]) ? (mPortFound[3]) : (portmin)
           Network.get_or_create_rule(oFC, security_group.id, 'tcp', portmin, portmax)
        end
      end



      oBuildEnv = BuildEnv.new(oConfig)
      ENV['FORJ_CLI_ENV'] = oBuildEnv.sBuildEnvFile
      oBuildEnv.set('FORJ_HPC',             oFC.sAccountName)
      oBuildEnv.set('FORJ_HPC_NET',         network.name)
      oBuildEnv.set('FORJ_SECURITY_GROUP',  oConfig.get('security_group'))
      oBuildEnv.set('FORJ_KEYPAIR',         oConfig.get('keypair_name'))
      oBuildEnv.set('FORJ_HPC_NOVA_KEYPUB', oConfig.get('keypair_path') + '.pub')
      oBuildEnv.set('FORJ_BASE_IMG',        oConfig.get('image_name'))
      oBuildEnv.set('FORJ_FLAVOR',          oConfig.get('flavor_name'))
      oBuildEnv.set('FORJ_BP_FLAVOR',       oConfig.get('bp_flavor'))
      oBuildEnv.set('FORJ_TENANT_NAME',     oConfig.get(:tenant_name))
      oBuildEnv.set('FORJ_HPC_COMPUTE',     rhGet(oConfig.oConfig.ExtraGet(:hpc_accounts,  oFC.sAccountName, :regions), :compute))


      oBuildEnv.set('FORJ_DOMAIN', yDNS[:domain_name])

      if yDNS[:tenant_id]
         oBuildEnv.set('FORJ_DNS_TENANTID', yDNS[:tenant_id])
         oBuildEnv.set('FORJ_DNS_ZONE',     yDNS[:service])
      end
      oBuildEnv.save()

      # run build.sh to boot maestro
      puts

      build = 'bin/build.sh' unless build

      build_config = oConfig.get('build_config')
      box_name =     oConfig.get('box_name')

      arg = '--meta blueprint=%s ' % [blueprint]
      arg += "--test-box '%s' " % oConfig.get(:test_box) if oConfig.exist?(:test_box)

      command = '%s --build_ID %s --box-name %s --build-conf-dir %s --build-config %s --gitBranch %s --debug-box %s' % [build, name, box_name, infra_dir, build_config, branch, arg]

      maestro_build_path = File.join(maestro_repo, 'build')

      current_dir = Dir.pwd
      Dir.chdir(maestro_build_path)

      Logging.info("Calling '%s' from '%s'" %  [build, Dir.pwd])
      Logging.debug("%s=%s %s" % ['FORJ_CLI_ENV', ENV['FORJ_CLI_ENV'], command])
      Kernel.system(ENV, command)
      Dir.chdir(current_dir)

#      if test
#        Logging.debug 'test flag is on, deleting objects'
#        Network.delete_router_interface(subnet.id, router)
#        Network.delete_subnet(subnet.id)
#        Network.delete_network(network.name)
#      end

    rescue Interrupt
      Logging.message("\n'%s' boot from '%s' interrupted by user" % [name, blueprint])
    rescue => e
      Logging.error("%s\n%s" % [e.message, e.backtrace.join("\n")])
    end
  end

end


class BuildEnv

   attr_reader :sBuildEnvFile

   def initialize(oConfig)

      oConfig.oConfig.fatal_if_inexistent(:infra_repo)
      oConfig.oConfig.fatal_if_inexistent(:account_name)

      sBuildDir = File.expand_path(File.join(oConfig.get(:infra_repo),'build'))
      @sBuildEnvFile = File.join(sBuildDir, oConfig.get(:account_name)+'.build.env')
      AppInit.ensure_dir_exists(sBuildDir)
      @yBuildEnvVar = {}
      @oConfig = oConfig
   end

   def set(key, value)
      # This file creates the in-memory data required to configure specific project build_env.
      # Name : ~/.forj/account/<Account>.build.env
      if value
         Logging.debug("Setting '%s' = '%s'" % [key, value])
         @yBuildEnvVar[key] = value
      else
         Logging.debug("'%s' is not set" % [key])
      end
   end

   def save()
      begin
         File.open(@sBuildEnvFile, 'w') do |out|
            @yBuildEnvVar.each do | key, value |
               desc = @oConfig.oConfig.getAppDefault(:description, key)
               out.write("# %s - %s\n" % [key, desc]) if desc
               value = "" if not value
               out.write("%s='%s'\n\n" % [key, value])
            end
         end
         Logging.debug("'%s' written." % [@sBuildEnvFile])
      rescue => e
         Logging.error("%s\n%s" % [e.message, e.backtrace.join("\n")])
      end
   end
end