diff --git a/pomodorodiscord/src/pomodoro_frame.py b/pomodorodiscord/src/pomodoro_frame.py index a717dc3..ea9a963 100644 --- a/pomodorodiscord/src/pomodoro_frame.py +++ b/pomodorodiscord/src/pomodoro_frame.py @@ -79,7 +79,7 @@ def initialize_rpc(self): def update_rpc(self): while True: if self.break_running: - self.rpc.break_state(self.start_time_timestamp, self.end_time_timestamp) + self.rpc.break_state(self.session_counter, self.start_time_timestamp, self.end_time_timestamp) elif self.running: self.rpc.running_state(self.session_counter + 1, self.start_time_timestamp, self.end_time_timestamp) else: diff --git a/pomodorodiscord/src/richpresence.py b/pomodorodiscord/src/richpresence.py index caed52b..aac0c4f 100644 --- a/pomodorodiscord/src/richpresence.py +++ b/pomodorodiscord/src/richpresence.py @@ -14,10 +14,13 @@ def __init__(self): print(f"Failed to connect to Discord: {e}") def default_state(self): - self.update(state="Idling", details=None, large_image="graytomato") + self.update(state="Idling", details=None, large_image="graytomato", + large_text="github.com/freeram/pomodoro-discord") def running_state(self, session, start_time, end_time): - self.update(state=f"Session {session}", details="Studying", start=start_time, end=end_time, large_image="tomato") + self.update(state=f"Session {session}", details="Studying", start=start_time, + end=end_time, large_image="tomato", large_text="github.com/freeram/pomodoro-discord") - def break_state(self, start_time, end_time): - self.update(state="On break", details=None, start=start_time, end=end_time, large_image="greentomato") + def break_state(self, complete, start_time, end_time): + self.update(state=f"On break. Sessions completed: {complete}", details=None, start=start_time, + end=end_time, large_image="greentomato", large_text="github.com/freeram/pomodoro-discord")