module Zabbix
  class ZabbixApi

    def get_screen_id(screen_name)
      message = {
          'method' => 'screen.get',
          'params' => {
              'filter' => {
                  'name' => screen_name
              }
          }
      }
      response = send_request(message)
      response.empty? ? nil : response[0]['screenid']
    end

    def get_screen_parameter(screen_name, param_name)
      message = {
          'method' => 'screen.get',
          'params' => {
              'extendoutput' => '1',
              'filter' => {
                  'name' => screen_name
              }
          }
      }
      response = send_request(message)
      response.empty? ? nil : response[0][param_name]
    end

    def get_screen_graph_ids(screen_id)
      message = {
          'method' => 'screen.get',
          'params' => {
              'extendoutput' => '1',
              'select_screenitems' => '1',
              'screenids' => [screen_id]
          }
      }
      response = send_request(message)
      if response.empty?
        result = nil
      else
        result = []
        screenitems = response[0]['screenitems']
        screenitems.each() do |item|
          result << item['resourceid'] if item['resourcetype'].to_i == 0
        end
      end
      return result
    end

    def set_screen_parameter(screen_id, param_name, param_value)
      message = {
          'method' => 'screen.update',
          'params' => {
              param_name => param_value,
              'screenid' => screen_id
          }
      }
      response = send_request(message)
      response.empty? ? nil : response[0]['screenids'].to_i
    end

    def del_all_graphs_from_screen(screen_id)
      message = {
          'method' => 'screen.deleteItems',
          'params' => {
              'screenids' => [screen_id],
          }
      }
      response = send_request(message)
      response ? response[0]['screenids'].to_i : nil
    end

    def add_graph_to_screen(screen_id, graph_id, x, y)
      message = {
          'method' => 'screen.addItems',
          'params' => {
              'screenids' => [screen_id],
              'screenitems' => [
                  {
                      'resourcetype' => 'graph',
                      'resourceid' => graph_id,
                      'width' => '800',
                      'height' => '200',
                      'x' => x,
                      'y' => y,
                      'valign' => 'Middle',
                      'halign' => 'Centre',
                      'colspan' => '0',
                      'rowspan' => '0',
                      'elements' => '0',
                      'dynamic' => '0',
                      'url' => '0',
                      'style' => '0'
                  }
              ]
          }
      }
      response = send_request(message)
      return response
    end

    def add_screen(screen_name, hsize, vsize)
      message = {
          'method' => 'screen.create',
          'params' => {
              'name' => screen_name,
              'hsize' => hsize,
              'vsize' => vsize
          }
      }
      response = send_request(message)
      response.empty? ? nil : response['screenids'][0]
    end

  end
end