#!/usr/bin/env python3
import requests
import time
import re
import subprocess
from threading import Thread
from pydbus import SessionBus
from pydbus.generic import signal
from gi.repository import GLib
from bs4 import BeautifulSoup

# --- CONFIGURATION ---
BASE_URL = "http://localhost:5666"
AUTH = ('winamp', 'llama')
APP_ID = "org.mpris.MediaPlayer2.winamp"

class WinampMPRIS:
    """
    MPRIS2 specification implementation for Winamp Web Interface.
    Optimized for Plasma 6 property change detection.
    """
    dbus = f"""
    <node>
      <interface name="org.mpris.MediaPlayer2">
        <property name="CanQuit" type="b" access="read" />
        <property name="CanRaise" type="b" access="read" />
        <property name="HasTrackList" type="b" access="read" />
        <property name="Identity" type="s" access="read" />
        <property name="DesktopEntry" type="s" access="read" />
        <property name="SupportedUriSchemes" type="as" access="read" />
        <property name="SupportedMimeTypes" type="as" access="read" />
      </interface>
      <interface name="org.mpris.MediaPlayer2.Player">
        <method name="Next" />
        <method name="Previous" />
        <method name="Pause" />
        <method name="PlayPause" />
        <method name="Stop" />
        <method name="Play" />
        <method name="Seek">
            <arg type="x" name="Offset" direction="in" />
        </method>
        <method name="SetPosition">
            <arg type="o" name="TrackId" direction="in" />
            <arg type="x" name="Position" direction="in" />
        </method>
        <method name="OpenUri">
            <arg type="s" name="Uri" direction="in" />
        </method>
        <property name="PlaybackStatus" type="s" access="read">
            <annotation name="org.freedesktop.DBus.Property.EmitsChangedSignal" value="true"/>
        </property>
        <property name="LoopStatus" type="s" access="read" />
        <property name="Rate" type="d" access="read" />
        <property name="Shuffle" type="b" access="read" />
        <property name="Metadata" type="a{{sv}}" access="read">
            <annotation name="org.freedesktop.DBus.Property.EmitsChangedSignal" value="true"/>
        </property>
        <property name="CanGoNext" type="b" access="read" />
        <property name="CanGoPrevious" type="b" access="read" />
        <property name="CanPlay" type="b" access="read" />
        <property name="CanPause" type="b" access="read" />
        <property name="CanControl" type="b" access="read" />
        <property name="CanSeek" type="b" access="read" />
        <property name="Position" type="x" access="read">
            <annotation name="org.freedesktop.DBus.Property.EmitsChangedSignal" value="false"/>
        </property>
        <property name="Volume" type="d" access="read" />
      </interface>
    </node>
    """

    # Signal for org.freedesktop.DBus.Properties.PropertiesChanged
    PropertiesChanged = signal()

    def __init__(self):
        self._title = "Winamp"
        self._artist = "Unknown"
        self._album = ""
        self._status = "Stopped"
        self._last_position_us = 0
        self._total_length_us = 0
        self._last_update_ts = time.time()

    def _request(self, endpoint):
        try:
            requests.get(f"{BASE_URL}/{endpoint}", auth=AUTH, timeout=2)
        except Exception as e:
            print(f"Request failed: {e}")

    # MPRIS Methods
    def Next(self): self._request("next")
    def Previous(self): self._request("prev")
    def Pause(self): self._request("pause")
    def PlayPause(self): self._request("pause")
    def Play(self): self._request("play")
    def Stop(self): self._request("stop")
    def Seek(self, offset): pass
    def SetPosition(self, track_id, position): pass
    def OpenUri(self, uri): pass

    # --- Root Properties ---
    @property
    def CanQuit(self): return False
    @property
    def CanRaise(self): return False
    @property
    def HasTrackList(self): return False
    @property
    def Identity(self): return "Winamp Web Bridge"
    @property
    def DesktopEntry(self): return ""
    @property
    def SupportedUriSchemes(self): return []
    @property
    def SupportedMimeTypes(self): return []

    # --- Player Properties ---
    @property
    def PlaybackStatus(self): return self._status
    @property
    def LoopStatus(self): return "None"
    @property
    def Rate(self): return 1.0
    @property
    def Shuffle(self): return False
    @property
    def CanGoNext(self): return True
    @property
    def CanGoPrevious(self): return True
    @property
    def CanPlay(self): return True
    @property
    def CanPause(self): return True
    @property
    def CanControl(self): return True
    @property
    def CanSeek(self): return False
    
    @property
    def Position(self):
        if self._status == "Playing":
            elapsed_us = int((time.time() - self._last_update_ts) * 1e6)
            current_pos = self._last_position_us + elapsed_us
            if self._total_length_us > 0:
                return min(current_pos, self._total_length_us)
            return current_pos
        return self._last_position_us

    @property
    def Volume(self): return 1.0

    @property
    def Metadata(self):
        # Metadata is a{sv}, so values MUST be wrapped in GLib.Variant
        return {
            'mpris:trackid': GLib.Variant('o', '/org/mpris/MediaPlayer2/winamp/track/0'),
            'mpris:length': GLib.Variant('x', self._total_length_us),
            'xesam:title': GLib.Variant('s', self._title),
            'xesam:artist': GLib.Variant('as', [self._artist]),
            'xesam:album': GLib.Variant('s', self._album),
            'mpris:artUrl': GLib.Variant('s', 'https://webamp.org/favicon.ico')
        }

def parse_time_to_us(time_str):
    """Parses MM:SS or H:MM:SS to microseconds."""
    try:
        parts = list(map(int, time_str.split(':')))
        if len(parts) == 2: # MM:SS
            return (parts[0] * 60 + parts[1]) * 1_000_000
        elif len(parts) == 3: # H:MM:SS
            return (parts[0] * 3600 + parts[1] * 60 + parts[2]) * 1_000_000
    except Exception:
        pass
    return 0

def get_winamp_window_title():
    """Extracts the Winamp window title using wmctrl."""
    try:
        output = subprocess.check_output(["wmctrl", "-l"], stderr=subprocess.STDOUT).decode('utf-8')
        for line in output.splitlines():
            # Match the typical Winamp window title pattern
            if " - Winamp" in line:
                # wmctrl output: 0x03000001  0 Compy-686 ARTMS–<Dall>–Unf/Air | 0:06/2:45 - Winamp
                parts = line.split(None, 3)
                if len(parts) >= 4:
                    return parts[3]
    except Exception:
        pass
    return None

def update_loop(player):
    last_known_title = ""
    last_known_status = ""
    offline_logged = False

    while True:
        try:
            # 1. Try Window Title (Primary)
            window_title = get_winamp_window_title()
            title_parsed = False
            
            if window_title:
                # Pattern: Artist–Album–Title | 0:06/2:45 - Winamp
                match = re.search(r"(.*?) \| (\d+:?\d*:\d*)/(\d+:?\d*:\d*) - Winamp", window_title)
                if match:
                    metadata_raw = match.group(1).strip()
                    new_pos_us = parse_time_to_us(match.group(2))
                    new_len_us = parse_time_to_us(match.group(3))
                    
                    # Split metadata (Artist–Album–Title)
                    meta_parts = [p.strip() for p in re.split(r'[–—-]', metadata_raw)]
                    
                    if len(meta_parts) >= 3:
                        player._artist = meta_parts[0]
                        player._album = meta_parts[1]
                        player._title = " – ".join(meta_parts[2:])
                    elif len(meta_parts) == 2:
                        player._artist = meta_parts[0]
                        player._title = meta_parts[1]
                        player._album = ""
                    else:
                        player._title = metadata_raw
                        player._artist = "Unknown"
                        player._album = ""
                        
                    player._last_position_us = new_pos_us
                    player._total_length_us = new_len_us
                    player._last_update_ts = time.time()
                    title_parsed = True

            # 2. Poll Web UI (Fallback and Controls/Status)
            try:
                r = requests.get(f"{BASE_URL}/main", auth=AUTH, timeout=1)
                if r.status_code == 200:
                    offline_logged = False
                    soup = BeautifulSoup(r.text, 'html.parser')
                    p_tags = soup.find_all('p')
                    status_raw = p_tags[0].text if p_tags else ""

                    new_status = "Stopped"
                    if "Playing" in status_raw: new_status = "Playing"
                    elif "Paused" in status_raw: new_status = "Paused"
                    
                    player._status = new_status

                    if not title_parsed:
                        # Web UI Fallback for metadata
                        match_meta = re.search(r"track \d+ - (.*?) - (.*?) -", status_raw)
                        if match_meta:
                            player._artist = match_meta.group(1).strip()
                            player._title = match_meta.group(2).strip()
                        elif "track" in status_raw:
                            player._title = status_raw.split('-')[0].replace("Playing track ", "").strip()

                        match_time = re.search(r"\((\d+:?\d*:\d+) / (\d+:?\d*:\d+)\)", status_raw)
                        if match_time:
                            player._last_position_us = parse_time_to_us(match_time.group(1))
                            player._total_length_us = parse_time_to_us(match_time.group(2))
                            player._last_update_ts = time.time()
            except requests.exceptions.RequestException:
                if not window_title:
                    if not offline_logged:
                        print("Winamp Web Interface offline and no window found.")
                        offline_logged = True
                    player._status = "Stopped"

            if player._status != last_known_status or player._title != last_known_title:
                last_known_status = player._status
                last_known_title = player._title

                print(f"UPDATE: [{player._status}] {player._artist} - {player._title}")

                player.PropertiesChanged(
                    "org.mpris.MediaPlayer2.Player",
                    {
                        "PlaybackStatus": player.PlaybackStatus,
                        "Metadata": player.Metadata
                    },
                    []
                )

        except Exception as e:
            print(f"Update error: {e}")
            
        time.sleep(1)

if __name__ == "__main__":
    bus = SessionBus()
    player_logic = WinampMPRIS()
    bus.publish(APP_ID, ("/org/mpris/MediaPlayer2", player_logic))

    GLib.timeout_add(1500, lambda: player_logic.PropertiesChanged(
        "org.mpris.MediaPlayer2.Player",
        {
            "PlaybackStatus": player_logic.PlaybackStatus,
            "Metadata": player_logic.Metadata
        },
        []
    ))

    thread = Thread(target=update_loop, args=(player_logic,), daemon=True)
    thread.start()

    print(f"--- Winamp Bridge Started (Window Title + Web UI) ---")
    loop = GLib.MainLoop()
    try:
        loop.run()
    except KeyboardInterrupt:
        pass