module MTik

# File lib/mtik.rb, line 265
def self.command(args)
  tk      = MTik::Connection.new(args)
  limit   = args[:limit]  ## Optional reply limit
  cmd     = args[:command]
  replies = Array.new
  if cmd.is_a?(String)
    ## Single command, no arguments
    cmd = [ [ cmd ] ]
  elsif cmd.is_a?(Array) && !cmd[0].is_a?(Array)
    ## Single command, possibly arguments
    cmd = [ cmd ]
  end

  cmd.each_index do |i|
    c = cmd[i]
    replycount = 0
    tk.send_request(false, c[0], c[1,c.length-1]) do |req, sentence|
      replycount += 1
      if c[0] == '/tool/fetch'
        if sentence['status'] == 'finished' && req.state == :sent
          ## Cancel 'finished' fetch commands
          req.cancel
        end
      elsif req.state == :sent && (
        limit.nil? ? (replycount >= MAXREPLIES) : (limit > 0 && replycount >= limit)
      )
        ## Auto-cancel any command after the maximum number of replies:
        req.cancel
      end
      if sentence.key?('!done')
        replies[i] = req.reply
      end
    end
  end

  tk.wait_all ## Event loop -- wait for all commands to finish
  tk.get_reply('/quit')
  tk.close
  return replies
end

# File lib/mtik.rb, line 78
def self.debug
  return @debug
end

# File lib/mtik.rb, line 83
def self.debug=(x)
  return @debug = x
end

# File lib/mtik.rb, line 108
def self.interactive_client(args)
  old_verbose = MTik::verbose
  MTik::verbose = true
  begin
    tk = MTik::Connection.new(args)
  rescue MTik::Error, Errno::ECONNREFUSED => e
    print "=== LOGIN ERROR: #{e.message}\n"
    exit
  end

  while true
    print "\nCommand (/quit to end): "
    cmd = STDIN.gets.sub(/^\s+/, '').sub(/\s*[\r\n]*$/, '')
    maxreply = 0
    m = /^(\d+):/.match(cmd)
    unless m.nil?
      maxreply = m[1].to_i
      cmd.sub!(/^\d+:/, '')
    end
    args  = cmd.split(/\s+/)
    cmd   = args.shift
    next if cmd == ''
    break if cmd == '/quit'
    unless /^(?:\/[a-zA-Z0-9-]+)+$/.match(cmd)
      print "=== INVALID COMMAND: #{cmd}\n" if MTik::debug || MTik::verbose
      break
    end
    print "=== COMMAND: #{cmd}\n" if MTik::debug || MTik::verbose
    trap  = false
    count = 0
    state = 0
    begin
      tk.send_request(false, cmd, args) do |req, sentence|
        if sentence.key?('!trap')
          trap = sentence
          print "=== TRAP: '" + (trap.key?('message') ? trap['message'] : "UNKNOWN") + "'\n\n"
        elsif sentence.key?('!re')
          count += 1
          ## Auto-cancel any '/tool/fetch' commands that have finished,
          ## or commands that have received the specified number of
          ## replies:
          if req.state == :sent && (
            cmd == '/tool/fetch' && sentence['status'] == 'finished'
          ) || (maxreply > 0 && count == maxreply)
            state = 2
            req.cancel do |r, s|
              state = 1
            end
          end
        elsif !sentence.key?('!done') && !sentence.key?('!fatal')
          raise MTik::Error.new("Unknown or unexpected reply sentence type.")
        end
        if state == 0 && req.done?
          state = 1
        end
      end
      while state != 1
        tk.wait_for_reply
      end
    rescue MTik::Error => e
      print "=== ERROR: #{e.message}\n"
    end
    unless tk.connected?
      begin
        tk.login
      rescue MTik::Error => e
        print "=== LOGIN ERROR: #{e.message}\n"
        tk.close
        exit
      end
    end
  end

  reply = tk.get_reply('/quit')
  unless reply[0].key?('!fatal')
    raise MTik::Error.new("Unexpected response to '/quit' command.")
  end

  ## Extract any device-provided message from the '!fatal' response
  ## to the /quit command:
  print "=== SESSION TERMINATED"
  message = ''
  reply[0].each_key do |key|
    next if key == '!fatal'
    message += "'#{key}'"
    unless reply[0][key].nil?
      message += " => '#{reply[0][key]}'"
    end
  end
  if message.length > 0
    print ": " + message
  else
    print " ==="
  end
  print "\n\n"

  unless tk.connected?
    print "=== Disconnected ===\n\n"
  else
    ## In theory, this should never execute:
    tk.close
  end

  MTik::verbose = old_verbose
end

# File lib/mtik.rb, line 68
def self.verbose
  return @verbose
end

# File lib/mtik.rb, line 73
def self.verbose=(x)
  return @verbose = x
end