import time
from datetime import datetime
import requests
from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler
import argparse


class DiscordNotifierHandler(FileSystemEventHandler):
    def __init__(self, webhook_url):
        self.webhook_url = webhook_url

    def notify_discord(self, message):
        data = {"content": message}
        try:
            response = requests.post(self.webhook_url, json=data)
            if response.status_code == 204:
                print("Discord notification sent.")
            else:
                print(f"Failed Discord notification: {response.status_code}")
        except Exception as e:
            print(f"Error sending Discord notification: {e}")

    def on_modified(self, event):
        if not event.is_directory:
            msg = f":warning: File modified: `{event.src_path}` at {datetime.now()}"
            print(msg)
            self.notify_discord(msg)

    def on_created(self, event):
        if not event.is_directory:
            msg = f":new: File created: `{event.src_path}` at {datetime.now()}"
            print(msg)
            self.notify_discord(msg)

    def on_deleted(self, event):
        if not event.is_directory:
            msg = f":x: File deleted: `{event.src_path}` at {datetime.now()}"
            print(msg)
            self.notify_discord(msg)


def watch_directory(path_to_watch, webhook_url):
    event_handler = DiscordNotifierHandler(webhook_url)
    observer = Observer()
    observer.schedule(event_handler, path=path_to_watch, recursive=True)
    observer.start()
    print(f"Started watching directory: {path_to_watch}")

    try:
        while True:
            time.sleep(1)
    except KeyboardInterrupt:
        observer.stop()
        print("Stopped watching directory.")

    observer.join()


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Watch a directory and send Discord notifications on file changes.")
    parser.add_argument("--watch", required=True, help="Path to the directory to watch")
    parser.add_argument("--webhook", required=True, help="Discord webhook URL")

    args = parser.parse_args()

    watch_directory(args.watch, args.webhook)