#APK Updater v1.1
#(C) 2023 by Manuel Kamper (kmpr.at)
#Do not remove copyright when distributing!
import feedparser, mysql.connector, os, glob, time, shutil
from datetime import datetime
from selenium import webdriver
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
from selenium.webdriver.common.by import By

metadata_path = "/mnt/data/metadata"
repo_path = "/mnt/data/repo"
temp_path = "/tmp/apkdown"

mydb = mysql.connector.connect(
  host="localhost",
  user="appstore",
  password="<your-pwd>",
  database="appstore"
)

### NO EDITS BELOW NECESSARY ###

url = "https://apkcombo.com/latest-updates/feed"

def get_last_filename_and_rename(new_filename):
    global temp_path, repo_path
    files = glob.glob(temp_path + '/*')
    max_file = max(files, key=os.path.getctime)
    filename_stripped = max_file.replace("_apkcombo.com","")
    filename = filename_stripped.split("/")[-1].split("_")[-2]
    new_filename2 = max_file.replace(filename, new_filename)
    new_filename2 = new_filename2.replace("_apkcombo.com","")
    print(new_filename2)
    new_path = new_filename2.replace(temp_path, repo_path)
    shutil.move(max_file, new_path)
    return new_path

mycursor = mydb.cursor()

print("APK Updater started.")
print("Clearing APK Database from previous updates")
sql = "TRUNCATE appstore"
mycursor.execute(sql)
mydb.commit()

#fetch updated apps from rss feed
print("Fetching updated APKs from RSS...")
feed = feedparser.parse(url)
updates = 0
for entry in feed.entries:
    updates += 1
    appname = str(entry.link).split("/")[-2]
    sql = "INSERT INTO appstore (url, app_name, updated) VALUES (%s, %s, %s)"
    date_str = datetime.strptime(entry.published, "%a, %d %b %Y %H:%M:%S %z")
    val = (entry.link+"download/apk", appname, date_str)
    mycursor.execute(sql, val)
mydb.commit()
print("Imported " + str(updates) + " updated APK infos from RSS into Database.")

#find local available apps from metadata directory (filename without .yml)
print("Searching for Apps in local Appstore...")
list_apps = ['.'.join(x.split('.')[:-1]) for x in os.listdir(metadata_path) if os.path.isfile(os.path.join(metadata_path, x))]
print("Found " + str(len(list_apps)) + " Apps in local Appstore.")

#iterate throug local apps and query db to find matches. if match found, mark app for update
real_updates = 0
for app in list_apps:
    sql = "SELECT * FROM appstore WHERE app_name='" + app + "'"
    mycursor.execute(sql)
    row = mycursor.fetchone()
    if row is not None:
        print("Local App " + app + " marked for update!")
        sql2 = "UPDATE appstore SET marked_for_update=1 WHERE id=" + str(row[0])
        mycursor.execute(sql2)
        real_updates +=1
mydb.commit()
print("Found " + str(real_updates) + " new updates for Apps in local Appstore")

if real_updates > 0:
    #select all marked_for_update entries from db and download apk from apk_url into repo_path
    #with filename: app_name_version.apk/.xapk
    #make dir temp_path if it does not exist
    if not os.path.exists(temp_path):
        os.makedirs(temp_path)

    #load firefox selenium and download apk
    options=Options()
    options.set_preference("browser.download.folderList", 2)
    options.set_preference("browser.download.manager.showWhenStarting", False)
    options.set_preference("browser.download.dir", temp_path)
    options.set_preference("browser.helperApps.neverAsk.saveToDisk", "application/octet-stream")
    options.add_argument("--headless")
    driver = webdriver.Firefox(options=options)
    driver.set_window_size(1200, 700)

    #get download-url for every file marked for update
    sql = "SELECT * FROM appstore WHERE marked_for_update=1"
    mycursor.execute(sql)
    rows = mycursor.fetchall()
    for row in rows:
        #make sure temp_path is empty
        for file in os.scandir(temp_path):
            if file.name.endswith(".apk") or file.name.endswith(".xapk"):
                os.unlink(file.path)
        driver.get(row[1])
        time.sleep(5)
        cookiebutton = driver.find_element(By.CSS_SELECTOR, 'svg.icon.ic-download-right')
        cookiebutton.click()
        print("APK " + row[2] + " downloaded.")
        time.sleep(30)
        #rename and move file to repo
        print("Moved APK to: " + get_last_filename_and_rename(row[2]))
    driver.close()
    driver.quit()

    #delete temp_path directory and all files in it
    shutil.rmtree(temp_path)

    #kill all running firefox processes to free RAM
    os.system("pkill -f firefox")

    #todo run updating f-droid appstore command
    os.system("cd /mnt/data && fdroid update -c && cp /mnt/data/icons/icon.png /mnt/data/repo/icons/icon.png")
print("Finished APK update.")