From abd737589bc2693c7e3d05175df9c90e8dc28d8a Mon Sep 17 00:00:00 2001
From: Rossen Georgiev <rossen@rgp.io>
Date: Sat, 18 Apr 2020 16:21:59 +0100
Subject: [PATCH] client: add block/unblock methods for steam users

Close #243
Close #225
---
 steam/client/builtins/friends.py | 40 ++++++++++++++++++++++++++++++++
 steam/client/user.py             |  8 +++++++
 2 files changed, 48 insertions(+)

diff --git a/steam/client/builtins/friends.py b/steam/client/builtins/friends.py
index 144cdaf..e483731 100644
--- a/steam/client/builtins/friends.py
+++ b/steam/client/builtins/friends.py
@@ -176,3 +176,43 @@ class SteamFriendlist(EventEmitter):
 
         self._steam.send(MsgProto(EMsg.ClientRemoveFriend), {'friendid': steamid})
 
+    def block(self, steamid):
+        """
+        Block Steam user
+
+        :param steamid: their steamid
+        :type  steamid: :class:`int`, :class:`.SteamID`, :class:`.SteamUser`
+        """
+        if isinstance(steamid, SteamUser):
+            steamid = steamid.steam_id
+
+        resp = self._steam.send_um_and_wait("Player.IgnoreFriend#1",
+                                            {"steamid": steamid},
+                                            timeout=10)
+
+        if not resp:
+            return EResult.Timeout
+        else:
+            if steamid in self:
+                self[steamid].relationship = EFriendRelationship(resp.body.friend_relationship)
+            return resp.header.eresult
+
+    def unblock(self, steamid):
+        """
+        Unblock Steam user
+
+        :param steamid: their steamid
+        :type  steamid: :class:`int`, :class:`.SteamID`, :class:`.SteamUser`
+        """
+        if isinstance(steamid, SteamUser):
+            steamid = steamid.steam_id
+
+        resp = self._steam.send_um_and_wait("Player.IgnoreFriend#1",
+                                            {"steamid": steamid, "unignore": True},
+                                            timeout=10)
+        if not resp:
+            return EResult.Timeout
+        else:
+            if steamid in self:
+                self[steamid].relationship = EFriendRelationship(resp.body.friend_relationship)
+            return resp.header.eresult
diff --git a/steam/client/user.py b/steam/client/user.py
index 1847d02..cc576f4 100644
--- a/steam/client/user.py
+++ b/steam/client/user.py
@@ -126,3 +126,11 @@ class SteamUser(object):
                 'chat_entry_type': EChatEntryType.ChatMsg,
                 'message': message.encode('utf8'),
                 })
+
+    def block(self):
+        """Block user"""
+        self._steam.friends.block(self)
+
+    def unblock(self):
+        """Unblock user"""
+        self._steam.friends.unblock(self)