Bot.py

import requests
import time
import os
from datetime import datetime

class CodeforcesFriendsMonitor:
    def __init__(self, your_handle):
        self.your_handle = your_handle
        self.base_url = "https://codeforces.com/api/"
        self.last_submission_ids = set()
        self.friends = []

    def get_user_submissions(self, handle, count=10):
        """Retrieve the latest submissions of a user."""
        try:
            url = f"{self.base_url}user.status?handle={handle}&from=1&count={count}"
            response = requests.get(url)
            data = response.json()

            if data['status'] == 'OK':
                return data['result']
            return []
        except Exception as e:
            print(f"Error fetching submissions for {handle}: {e}")
            return []

    def play_sound(self, sound_type="default"):
        """Play notification sound on macOS."""
        try:
            # macOS system sounds:
            sounds = {
                "default": "Ping",
                "glass": "Glass",
                "hero": "Hero",
                "sosumi": "Sosumi",
                "submarine": "Submarine",
                "blow": "Blow",
                "bottle": "Bottle",
                "frog": "Frog",
                "funk": "Funk",
                "pop": "Pop",
                "purr": "Purr",
                "tink": "Tink"
            }

            sound = sounds.get(sound_type, "Ping")
            os.system(f'afplay /System/Library/Sounds/{sound}.aiff')

        except Exception as e:
            print(f"Error playing sound: {e}")

    def send_notification(self, title, message):
        """Send macOS notification."""
        try:
            # Using osascript for native macOS notification
            script = f'''
            display notification "{message}" with title "{title}" sound name "Ping"
            '''
            os.system(f"osascript -e '{script}'")
        except Exception as e:
            print(f"Error sending notification: {e}")

    def monitor_submissions(self, check_interval=60, sound_type="default", show_notifications=True):
        """Monitor friends' submissions."""
        print(f"\n[INFO] Starting submission monitoring (every {check_interval} seconds)\n")
        print("Press Ctrl+C to stop.\n")

        # Fetch initial submissions to avoid alerting on old ones
        for friend in self.friends:
            submissions = self.get_user_submissions(friend, count=5)
            for sub in submissions:
                self.last_submission_ids.add(sub['id'])

        print(f"[OK] {len(self.last_submission_ids)} initial submissions recorded.\n")

        try:
            while True:
                for friend in self.friends:
                    submissions = self.get_user_submissions(friend, count=5)

                    for sub in submissions:
                        if sub['id'] not in self.last_submission_ids:
                            # New submission found!
                            self.last_submission_ids.add(sub['id'])

                            problem_name = sub['problem'].get('name', 'Unknown')
                            problem_index = sub['problem'].get('index', '?')
                            contest_id = sub.get('contestId', sub.get('problem', {}).get('contestId', '?'))
                            verdict = sub.get('verdict', 'TESTING')

                            timestamp = datetime.fromtimestamp(sub['creationTimeSeconds'])

                            # Display in terminal
                            print("-" * 60)
                            print("[ALERT] New Submission!")
                            print(f"User: {friend}")
                            print(f"Problem: {contest_id}{problem_index} - {problem_name}")
                            print(f"Verdict: {verdict}")
                            print(f"Time: {timestamp.strftime('%Y-%m-%d %H:%M:%S')}")
                            print("-" * 60)
                            print()

                            # Play sound
                            self.play_sound(sound_type)

                            # Send macOS notification
                            if show_notifications:
                                notification_msg = f"{friend} -> {contest_id}{problem_index} ({verdict})"
                                self.send_notification("Codeforces Submission", notification_msg)

                time.sleep(check_interval)

        except KeyboardInterrupt:
            print("\n\nMonitoring stopped.")


def main():
    print("=" * 60)
    print("CF Monitor Bot - macOS Edition")
    print("=" * 60)

    # Enter your handle
    your_handle = input("\nEnter your Codeforces handle: ").strip()

    if not your_handle:
        print("Error: Handle cannot be empty!")
        return

    monitor = CodeforcesFriendsMonitor(your_handle)

    print("\nEnter your friends' handles:")
    friends_input = input("Separate handles with comma: ").strip()

    if friends_input:
        monitor.friends = [f.strip() for f in friends_input.split(',')]
        print(f"[OK] {len(monitor.friends)} friend(s) added.")
    else:
        print("Error: No friends entered!")
        return

    check_interval = input("\nEnter check interval in seconds [default: 60]: ").strip()
    check_interval = int(check_interval) if check_interval.isdigit() else 60

    print("\nSelect notification sound:")
    print("1. Ping (Default)")
    print("2. Glass")
    print("3. Hero")
    print("4. Submarine")
    print("5. Funk")

    sound_choice = input("Choice (1-5): ").strip()
    sound_map = {
        "1": "default",
        "2": "glass",
        "3": "hero",
        "4": "submarine",
        "5": "funk"
    }
    sound_type = sound_map.get(sound_choice, "default")

    monitor.monitor_submissions(check_interval, sound_type)


if __name__ == "__main__":
    main()