lib/zabbixapi/classes/users.rb in zabbixapi-4.1.0 vs lib/zabbixapi/classes/users.rb in zabbixapi-4.1.1

@@ -26,40 +26,37 @@
     # @return [String]
     def indentify
       'alias'
     end
 
+    def medias_helper(data, action)
+      result = @client.api_request(
+        method: "user.#{action}",
+        params: {
+          users: data[:userids].map { |t| { userid: t } },
+          medias: data[:media]
+        }
+      )
+      result ? result['mediaids'][0].to_i : nil
+    end
+
     # Add media to users using Zabbix API
     #
     # @param data [Hash] Needs to include userids and media to mass add media to users
     # @raise [ApiError] Error returned when there is a problem with the Zabbix API call.
     # @raise [HttpError] Error raised when HTTP status from Zabbix Server response is not a 200 OK.
     # @return [Integer] Zabbix object id (media)
     def add_medias(data)
-      result = @client.api_request(
-        :method => 'user.addMedia',
-        :params => {
-          :users => data[:userids].map { |t| {:userid => t} },
-          :medias => data[:media],
-        }
-      )
-      result ? result['mediaids'][0].to_i : nil
+      medias_helper(data, 'addMedia')
     end
 
     # Update media for users using Zabbix API
     #
     # @param data [Hash] Needs to include userids and media to mass update media for users
     # @raise [ApiError] Error returned when there is a problem with the Zabbix API call.
     # @raise [HttpError] Error raised when HTTP status from Zabbix Server response is not a 200 OK.
     # @return [Integer] Zabbix object id (user)
     def update_medias(data)
-      result = @client.api_request(
-        :method => 'user.updateMedia',
-        :params => {
-          :users => data[:userids].map { |t| {:userid => t} },
-          :medias => data[:media],
-        }
-      )
-      result ? result['userids'][0].to_i : nil
+      medias_helper(data, 'updateMedia')
     end
   end
 end