module ForemanSalt
  module Orchestration
    module Salt
      extend ActiveSupport::Concern
      include ::Orchestration

      included do
        after_validation :queue_salt_autosign
        before_destroy   :queue_salt_destroy
      end

      def salt?
        name.present? && salt_proxy.present?
      end

      def initialize_salt
        @salt_api ||= ProxyAPI::Salt.new :url => salt_proxy.url
      end

      def queue_salt_autosign
        return unless salt? && errors.empty?
        new_record? ? queue_salt_autosign_create : queue_salt_autosign_update
      end

      def queue_salt_autosign_create
        # do nothing - we'll set autosign at the last second: when a host requests a provision URL
      end

      def queue_salt_autosign_update
        # Host has been built --> remove auto sign
        if old.build? and !build?
          queue.create(:name => _("Remove autosign entry for %s") % self, :priority => 50, :action => [self, :salt_autosign_remove])
        end
      end

      def queue_salt_destroy
        queue.create(:name => _("Remove autosign entry for %s") % self, :priority => 50, :action => [self, :salt_autosign_remove])
        queue.create(:name => _("Delete existing salt key for %s") % self, :priority => 50, :action => [self, :salt_key_delete])
      end

      def queue_salt_autosign_remove
        return unless salt? && errors.empty?
        queue.create(:name => _("Remove autosign entry for %s") % self, :priority => 50, :action => [self, :salt_autosign_remove])
      end

      def salt_autosign_create
        logger.info "Create autosign entry for #{name}"
        initialize_salt
        salt_key_delete # if there's already an existing key
        @salt_api.autosign_create name
      rescue => e
        failure _("Failed to create %{name}'s Salt autosign entry: %{e}") % { :name => name, :e => proxy_error(e) }
      end

      def salt_autosign_remove
        logger.info "Remove autosign entry for #{name}"
        initialize_salt
        @salt_api.autosign_remove name
      rescue => e
        failure _("Failed to remove %{name}'s Salt autosign entry: %{e}") % { :name => name, :e => proxy_error(e) }
      end

      def salt_key_delete
        logger.info "Delete salt key for #{name}"
        initialize_salt
        @salt_api.key_delete name
      rescue => e
        failure _("Failed to delete %{name}'s Salt key: %{e}") % { :name => name, :e => proxy_error(e) }
      end
    end
  end
end