import adbutils
import os
import sys
import time
import re
import requests
from lxml import html

def adb_start():
    if "platform-tools" in os.environ['PATH']:
        print("ADB found in PATH")
    else:
        os.environ['PATH'] += ';'+os.getcwd()+'\\platform-tools'


def adb_connect():
    adb = adbutils.AdbClient(host="127.0.0.1", port=5037)
    d = adb.device()
    if d != None:
        print("Device connected.")
    return d


def twrp_download(d):
    found = 0
    files = os.listdir(os.curdir)
    for file in files:
        if "twrp" in file:
            found = 1
    if found != 1:
        cpu = d.shell('cat /proc/cpuinfo | grep Hardware')
        cpu = cpu.replace(" ", "")
        cpu = re.sub(r'(.+:)', '', cpu)
        r = requests.get('https://dl.twrp.me/'+cpu)
        tree = html.fromstring(r.text)
        urls = tree.xpath('//a/@href')
        downloads = []
        for i in urls:
            if "img" in i:
                downloads.append(i)
        url_to_download = "https://dl.twrp.me"+downloads[0]
        url_to_download = url_to_download.replace('.html', '')
        print("Use this link to download twrp for your connected device: "+url_to_download)
        print("Ensure that the downloaded file is moved to the same folder as the script before continuing")
        input("Press Enter to continue...")
        files = os.listdir(os.curdir)
        for file in files:
            if "twrp" in file:
                found = 1
        while found != 1:
            print("File not found. Please confirm it has been moved to the correct directory")
            input("Press Enter to continue...")
            files = os.listdir(os.curdir)
            for file in files:
                if "twrp" in file:
                    found = 1
    else:
        print("twrp already downloaded")


def push_files(d):
    initcheck_magisk = d.shell("cd /sdcard && ls | grep Magisk")
    if initcheck_magisk == None and initcheck_twrp == None:
        d.sync.push("Magisk-v19.3.zip", "/sdcard/Magisk.zip")
        check_magisk = d.shell("cd /sdcard && ls | grep Magisk")
        if check != None:
            print("File copied successfully.")
        else:
            print("Something went wrong. Please try again.")
    else:
        print("Magisk already copied")


def reboot_bootloader():
    adb = "platform-tools\\adb.exe"
    fastboot = "platform-tools\\fastboot.exe"
    os.system(adb+" reboot bootloader")
    input("Press Enter when the device has rebooted")
    os.system(fastboot+" devices")


def root_device():
    device = adb_connect()
    twrp_download(device)
    push_files(device)
    reboot_bootloader()
    adb = "platform-tools\\adb.exe"
    fastboot = "platform-tools\\fastboot.exe"
    files = os.listdir(os.curdir)
    for file in files:
        if "twrp" in file:
            twrp = file
    os.system(fastboot + " boot "+twrp)
    input("Press Enter when TWRP has booted")
    print("Follow the onscreen directions to install Magisk (Located at the bottom of the install window)")
    print("After Magisk installs click [Reboot] then [Do Not Install]")
    input("Press Enter when the device has rebooted")