require "providers/provider_factory"
require "commands/stack"
require "db/mongo/models/stack/stack_factory"
require "db/mongo/models/project"
require "db/mongo/models/report"

class StackBootstrapWorker < Worker
  include StackCommands

  def perform(options)
    provider_name = options.fetch('provider_name')
    stack_attrs = options.fetch('stack_attributes')

    call(provider_name) do |provider, out, file|
      without_bootstrap = stack_attrs.delete('without_bootstrap')
      save_report(file, stack_attrs)

      stack = Devops::Model::StackFactory.create(provider_name, stack_attrs, out)
      mongo.stack_insert(stack)

      operation_result = sync_stack_proc.call(out, stack, mongo)
      if operation_result == 0
        out << "\nStack '#{stack.name}' has been created\n"
        out.flush
        servers = persist_stack_servers!(stack, provider)
        bootstrap_servers(servers, out) unless without_bootstrap
      end
      operation_result
    end
  end

  private

  def bootstrap_servers(servers, out)
    out << "\n"
    servers.map do |server|
      Devops::Executor::ServerExecutor.new(server, out).two_phase_bootstrap({})
    end
  end

  def save_report(file, stack_attrs)
    report = ::Devops::Model::Report.new(
      "file" => file,
      "_id" => jid,
      "created_by" => stack_attrs['owner'],
      "project" => stack_attrs["project"],
      "deploy_env" => stack_attrs["deploy_env"],
      "type" => ::Devops::Model::Report::STACK_TYPE
    )
    mongo.save_report(report)
  end

  def persist_stack_servers!(stack, provider)
    project = mongo.project(stack.project)
    deploy_env = project.deploy_env(stack.deploy_env)

    provider.stack_servers(stack).map do |extended_info|
      server_attrs = {
        'provider' => provider.name,
        'project' => project.id,
        'deploy_env' => deploy_env.identifier,
        'remote_user' => mongo.image(deploy_env.image).remote_user,
        'key' => extended_info["key_name"] || provider.ssh_key,
        '_id' => extended_info["id"],
        'chef_node_name' => extended_info["name"],
        'private_ip' => extended_info["private_ip"],
        'public_ip' => extended_info["public_ip"],
        'created_by' => stack.owner,
        'run_list' => stack.run_list || [],
        'stack' => stack.name
      }

      server = ::Devops::Model::Server.new(server_attrs)
      mongo.server_insert(server)
      # server.chef_node_name =  provider.create_default_chef_node_name(server)
      server
    end
  end

end