rapid7/metasploit-framework

require 'metasploit/framework/data_service/remote/http/response_data_helper'

module RemoteHostDataService
  include ResponseDataHelper

  HOST_API_PATH = '/api/v1/hosts'
  HOST_SEARCH_PATH = HOST_API_PATH + "/search"
  HOST_MDM_CLASS = 'Mdm::Host'
  TAG_MDM_CLASS = 'Mdm::Tag'

  def hosts(opts)
    path = get_path_select(opts, HOST_API_PATH)
    json_to_mdm_object(self.get_data(path, nil, opts), HOST_MDM_CLASS)
  end

  def get_host(opts)
    json_to_mdm_object(self.post_data(HOST_SEARCH_PATH, opts), HOST_MDM_CLASS).first
  end

  def report_host(opts)
    json_to_mdm_object(self.post_data(HOST_API_PATH, opts), HOST_MDM_CLASS).first
  end

  def update_host(opts)
    path = HOST_API_PATH
    if opts && opts[:id]
      id = opts.delete(:id)
      path = "#{HOST_API_PATH}/#{id}"
    end
    json_to_mdm_object(self.put_data(path, opts), HOST_MDM_CLASS)
  end

  def delete_host(opts)
    json_to_mdm_object(self.delete_data(HOST_API_PATH, opts), HOST_MDM_CLASS)
  end

  def get_host_tags(opts)
    path = get_path_select(opts, HOST_API_PATH) + '/tags'
    json_to_mdm_object(self.get_data(path, opts, nil), TAG_MDM_CLASS)
  end

  def add_host_tag(opts)
    path = get_path_select(opts, HOST_API_PATH) + '/tags'
    json_to_mdm_object(self.post_data(path, opts, nil), TAG_MDM_CLASS)
  end

  def delete_host_tag(opts)
    path = get_path_select(opts, HOST_API_PATH) + '/tags'
    json_to_mdm_object(self.delete_data(path, opts, nil), TAG_MDM_CLASS)
  end

end