Implement a new setImage method for the UIButton class (#9)

* Implement a new `setImage` method for the `UIButton` class

* Update `CHANGELOG.md`

Author: ns-vasilev

CHANGELOG.md

@@ -4,6 +4,8 @@ All notable changes to this project will be documented in this file.
 ## [Unreleased]
 
 #### Added
+- Implement a new `setImage` method for the `UIButton` class.
+  - Added in Pull Request [#9](https://github.com/space-code/flex-ui/pull/9).
 - Implement setting an optional value as the UILabel's text.
   - Added in Pull Request [#8](https://github.com/space-code/flex-ui/pull/8).
 

Sources/FlexUI/Classes/Extensions/FlexUI+UIButton.swift

@@ -205,4 +205,17 @@ public extension FlexUI where Component: UIButton {
         component.isEnabled = isEnable
         return self
     }
+
+    /// Sets the image for a specific button state.
+    ///
+    /// - Parameters:
+    ///   - image: The image to set.
+    ///   - controlState: The state for which to set the background image (default is `.normal`).
+    /// - Returns: The current instance of `FlexUI` for further configuration.
+    @discardableResult
+    @MainActor
+    func setImage(_ image: UIImage, for controlState: UIControl.State = .normal) -> Self {
+        component.setImage(image, for: controlState)
+        return self
+    }
 }