import subprocess
import re
import time
import sqlite3
import os
import requests

def get_working_directory():
    # If you see this function, that means you are using the pyinstaller. Use this command to create the executable:
    # pyinstaller -s -F localwrapped-pyi.py --runtime-tmpdir /data/local/tmp
    if os.path.exists("/sdcard/localwrapped"):
        return "/sdcard/localwrapped"
    elif os.path.exists("/storage/emulated/0/localwrapped"):
        return "/storage/emulated/0/localwrapped"
    elif os.path.exists("/data/media/0/localwrapped"):
        return "/data/media/0/localwrapped"
    elif os.path.exists("/mnt/user/0/emulated/0/localwrapped"):
        return "/mnt/user/0/emulated/0/localwrapped"
    else:
        print("Path not found.")
        try:
            if os.path.exists("/sdcard"):
                os.mkdir("/sdcard/localwrapped")
                return "/sdcard/localwrapped"
            else:
                raise Exception
        except:
            try:
                if os.path.exists("/storage/emulated/0"):
                    os.mkdir("/storage/emulated/0/localwrapped")
                    return "/storage/emulated/0/localwrapped"
                else:
                    raise Exception
            except:
                try:
                    if os.path.exists("/data/media/0"):
                        os.mkdir("/data/media/0/localwrapped")
                        return "/data/media/0/localwrapped"
                    else:
                        raise Exception
                except:
                    try:
                        if os.path.exists("/mnt/user/0/emulated/0"):
                            os.mkdir("/mnt/user/0/emulated/0/localwrapped")
                            return "/mnt/user/0/emulated/0/localwrapped"
                        else:
                            raise Exception
                    except:
                        print("Path not found and could not be created.")
                        exit()

def get_music():
    dumpsys_output = subprocess.check_output("dumpsys media_session", shell=True).decode("utf-8").split("\n")
    if not [line for line in dumpsys_output if "PLAYING" in line]:
        return None
    else:
        description = [line for line in dumpsys_output if "description=" in line]
        try:
            return description[0].split("description=")[1]
        except:
            return None

def get_db_values(url: str = "https://halhadus.rocks/assets/localwrapped/music.db"):
    if not os.path.exists(get_working_directory() + "/music.db"):
        try:
            download = requests.get(url)
            with open(get_working_directory() + "/music.db", "wb") as f:
                f.write(download.content)
        except:
            print("Could not download the database.")
            exit()
    musicconn = sqlite3.connect(get_working_directory() + "/music.db")
    musicc = musicconn.cursor()
    musicc.execute("SELECT * FROM music")
    musicrows = musicc.fetchall()
    musicconn.close()
    musicvalues = []
    for row in musicrows:
        musicvalues.append({
            "listmusicname": row[0],
            "fileartistname": row[4],
            "filealbumname": row[5],
            "filemusictitle": row[6]
        })
    return musicvalues

if not os.path.exists(get_working_directory() + "/count.db"):
    countconn = sqlite3.connect(get_working_directory() + "/count.db")
    countc = countconn.cursor()
    countc.execute("CREATE TABLE count (musicname TEXT, count INTEGER)")
    countconn.commit()
    countconn.close()

a = 0
last_written = None
while True:
    if get_music() == None:
        continue
    last_music = get_music()
    time.sleep(3)
    if last_music == get_music():
        if last_written == last_music:
            continue
        a += 1
        if a == 20:
            for i in get_db_values():
                if i["filealbumname"] == None:
                    full_name = i["filemusictitle"] + ", " + i["fileartistname"] + ", " + "Music"
                else:
                    full_name = i["filemusictitle"] + ", " + i["fileartistname"] + ", " + i["filealbumname"]
                if full_name == last_music:
                    countconn = sqlite3.connect(get_working_directory() + "/count.db")
                    countc = countconn.cursor()
                    countc.execute("SELECT * FROM count")
                    countdbrows = countc.fetchall()
                    countdbvalues = []
                    for row in countdbrows:
                        countdbvalues.append({
                            "musicname": row[0],
                            "count": row[1]
                        })
                    if not [row for row in countdbvalues if row["musicname"] == i["listmusicname"]]:
                        countc.execute("INSERT INTO count VALUES (?, ?)", (i["listmusicname"], 1))
                        countconn.commit()
                        countconn.close()
                    else:
                        countc.execute("UPDATE count SET count = count + 1 WHERE musicname = ?", (i["listmusicname"],))
                        countconn.commit()
                        countconn.close()
                    print(f"Local Wrapped: {i['listmusicname']} has been played.")
                    os.system(f"echo {i['listmusicname']} has been played. >> /cache/magisk.log")
            last_written = last_music
            a = 0
        continue
    else:
        a = 0
        continue