From ba4e75063ac8f6d8b60c0826595e51ccddf24057 Mon Sep 17 00:00:00 2001
From: taizan-hokuto <55448286+taizan-hokuto@users.noreply.github.com>
Date: Mon, 11 Nov 2019 21:04:20 +0900
Subject: [PATCH] Make it possible to pause /resume  fetching chat

---
 pytchat/core_multithread/livechat.py   |  1 -
 pytchat/core_multithread/replaychat.py | 24 ++++++++++++++++++++++--
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/pytchat/core_multithread/livechat.py b/pytchat/core_multithread/livechat.py
index 0e4e076..8357191 100644
--- a/pytchat/core_multithread/livechat.py
+++ b/pytchat/core_multithread/livechat.py
@@ -3,7 +3,6 @@ import datetime
 import json
 import random
 import signal
-import threading
 import time
 import traceback
 import urllib.parse
diff --git a/pytchat/core_multithread/replaychat.py b/pytchat/core_multithread/replaychat.py
index 7489761..4c518c5 100644
--- a/pytchat/core_multithread/replaychat.py
+++ b/pytchat/core_multithread/replaychat.py
@@ -3,11 +3,11 @@ import datetime
 import json
 import random
 import signal
-import threading
 import time
 import traceback
 import urllib.parse
 from concurrent.futures import CancelledError, ThreadPoolExecutor
+from queue import Queue
 from .buffer import Buffer
 from .replayparser import Parser
 from .. import config
@@ -83,8 +83,10 @@ class ReplayChat:
         self._direct_mode = direct_mode
         self._is_alive   = True
         self._parser = Parser()
+        self._pauser = Queue()
+        self._pauser.put_nowait(None)
+       
         self._setup()
-
         if not ReplayChat._setup_finished:
             ReplayChat._setup_finished = True
             if interruptable:
@@ -156,6 +158,12 @@ class ReplayChat:
         try:
             with requests.Session() as session:
                 while(continuation and self._is_alive):
+                    if self._pauser.empty():
+                        #pauseが呼ばれて_pauserが空状態のときは一時停止する
+                        self._pauser.get()
+                        #resumeが呼ばれて_pauserにitemが入ったら再開する
+                        #直後に_pauserにitem(None)を入れてブロックを防ぐ
+                        self._pauser.put_nowait(None)
                     livechat_json = (
                       self._get_livechat_json(continuation, session, headers)
                     )
@@ -244,6 +252,18 @@ class ReplayChat:
         raise IllegalFunctionCall(
             "既にcallbackを登録済みのため、get()は実行できません。")
 
+    def pause(self):
+        '''チャット取得を一時停止する。'''
+        if not self._pauser.empty():
+            self._pauser.get()
+
+
+    def resume(self):
+        '''チャット取得を再開する。'''
+        if self._pauser.empty():
+            self._pauser.put_nowait(None)
+        
+
     def is_alive(self):
         return self._is_alive