# encoding: utf-8
#--
#   Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies)
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU Affero General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   This program is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#   GNU Affero General Public License for more details.
#
#   You should have received a copy of the GNU Affero General Public License
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.
#++
class SshKeyProcessor < ApplicationProcessor
  subscribes_to :ssh_key_generation

  def on_message(message)
    verify_connections!
    json = ActiveSupport::JSON.decode(message)
    logprefix = "#{self.class.name} ssh-key(#{json['target_id']}): "
    logger.info(logprefix + "Consuming message. Command: #{json['command']}. Arguments: #{json['arguments']}. Target_id: #{json['target_id']}")
    logger.debug(logprefix + "Processing message #{json}")
    unless %w(add_to_authorized_keys delete_from_authorized_keys).include?(json['command'])
      raise "Unknown command"
    end
    logger.debug(logprefix + "Processor sending message: #{json['command']} #{json['arguments']}")
    SshKey.send(json['command'], *json['arguments'])        
    if target_id = json['target_id'] #Existing target id means we are creating a new key
      #Trying max 5 times if find_by_id would return a correct value
      5.times do
        obj = SshKey.find_by_id(target_id.to_i)
        if obj && obj.respond_to?(:ready)
          obj.ready = true
          obj.save!
          logger.info(logprefix + "Success")
          return
        else
          logger.error(logprefix + "Problem: SshKey.find_by_id(target_id.to_i).to_s == #{SshKey.find_by_id(target_id.to_i)}")
          sleep(3)
        end
      end
      logger.fatal(logprefix + "FATAL: SshKey was not marked ready.")
    end
  end
end