@ECHO OFF

REM Version:    V07
REM Date:        20190703

REM ============== USAGE ==============
REM flashall_AFT.cmd [Format option] [SN number]
REM     [Format option]:    1=enable, 0=disable. default=0
REM                             Format userdata/cache
REM     [SN number]:        device SN number
REM                             Support multiple download image

REM ============== setting ==============

SETLOCAL

del fastboot.log

ECHO. >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO ###########################  OS INFO  ############################ >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO. >>fastboot.log
ECHO OS_VERSION = %PROCESSOR_ARCHITECTURE% >>fastboot.log

IF "%PROCESSOR_ARCHITECTURE%" == x86 (
    ECHO We don't support 32bit os environment, exit!
    GOTO :EOF
)

IF "%selfWrapped%"=="" (
  REM this is necessary so that we can use "exit" to terminate the batch file,
  REM and all subroutines, but not the original cmd.exe
  SET selfWrapped=true
  %ComSpec% /s /c ""%~0" %*"
  GOTO :EOF
)

REM ============== main() ==============

CLS
SET "dataerase=0"
SET "SSN="
SET "para="
SET "tempSysName=.system_tmp.img"

IF [%1] EQU [1] ( SET dataerase=1 )
IF NOT [%2] EQU [] ( SET para=-s )
IF NOT [%2] EQU [] (
    SET SSN=%2
    SET tempSysName=.system_tmp_%2.img
)

call:flash

EXIT 0

REM ============== flash() ==============
:flash

SET "RAW="
FOR /F "delims=" %%i IN ('dir "%~dp0" /B ^| findstr /I "raw"') DO ( SET RAW=%%i )
IF NOT EXIST "%RAW%" ECHO "FAILED, RAW NOT EXIST!" && EXIT 1

ECHO.
ECHO ##################################################################
ECHO #######################  FLASH RAW FILE   ########################
ECHO ##################################################################
ECHO.
ECHO ##  RAW: %RAW%
ECHO ##  ERASE DATA: %dataerase%
ECHO ##  SSN: %SSN%
ECHO.
ECHO ==================================================================

ECHO. >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO #######################  FLASH RAW FILE   ######################## >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO. >>fastboot.log
ECHO ##  RAW: %RAW% >>fastboot.log
ECHO ##  ERASE DATA: %dataerase% >>fastboot.log
ECHO ##  SSN: %SSN% >>fastboot.log
ECHO. >>fastboot.log
ECHO ================================================================== >>fastboot.log
fastboot version >>fastboot.log
ECHO ================================================================== >>fastboot.log
call:fastboot_tool flash all "%RAW%" 2>>fastboot.log
REM call:fastboot_tool reboot-bootloader 2>>fastboot.log

ping 127.0.0.1 -n 10 > null

IF %dataerase% EQU 1 (
    ECHO. >>fastboot.log
    ECHO ################################################################## >>fastboot.log
    ECHO ##########################  WIPE DATA  ########################### >>fastboot.log
    ECHO ################################################################## >>fastboot.log
    ECHO. >>fastboot.log
    call:fastboot_tool -w 1>>fastboot.log 2>&1
    call:fastboot_tool format asdf 1>>fastboot.log 2>&1
    call:fastboot_tool format ADF 1>>fastboot.log 2>&1
    call:fastboot_tool erase asuskey5 1>>fastboot.log 2>&1
)

set "ERRORLEVEL="
ECHO. >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO #########################  DEVICE INFO  ########################## >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO. >>fastboot.log

fastboot oem device-info 2>>fastboot.log
fastboot oem device-info 2>lock.txt && findstr /C:"(bootloader) Device unlocked: true" lock.txt
echo ERRORLEVEL=%ERRORLEVEL% >>fastboot.log
REM errorlevel == 0 => Device unlocked: true
IF %ERRORLEVEL% == 0 ( goto lock )
REM errorlevel == 1 => Device unlocked: false
IF %ERRORLEVEL% == 1 ( goto reboot )

REM ============== Lock ==================
:lock

IF %dataerase% EQU 1 (
    ECHO. >>fastboot.log
    ECHO ################################################################## >>fastboot.log
    ECHO #########################  DEVICE LOCK  ########################## >>fastboot.log
    ECHO ################################################################## >>fastboot.log
    ECHO. >>fastboot.log
    echo fastboot.exe oem asus-csc_lk >>fastboot.log
    call:fastboot_tool oem asus-csc_lk 2>>fastboot.log
)

REM ============== Reboot ================
:reboot

set "ERRORLEVEL="
findstr /C:"(bootloader) Device authorized: true" lock.txt
echo ERRORLEVEL=%ERRORLEVEL% >>fastboot.log
REM errorlevel == 0 => Device authorized: true
IF %ERRORLEVEL% == 0 ( goto oemlock )
REM errorlevel == 1 => Device authorized: false
IF %ERRORLEVEL% == 1 ( goto adb_enable )

REM ============== oem lock ================
emlock

IF %dataerase% EQU 1 (
    ECHO. >>fastboot.log
    ECHO ################################################################## >>fastboot.log
    ECHO #######################  DEVICE OEM LOCK  ######################## >>fastboot.log
    ECHO ################################################################## >>fastboot.log
    ECHO. >>fastboot.log
    echo fastboot.exe oem reset-dev_info >>fastboot.log
    call:fastboot_tool oem reset-dev_info 2>>fastboot.log
)

REM ============= adb enable ===============
:adb_enable

ECHO. >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO #########################  DEVICE INFO  ########################## >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO. >>fastboot.log
fastboot reboot-bootloader 2>>fastboot.log
ping 127.0.0.1 -n 3 > null
fastboot oem device-info 2>>fastboot.log
ECHO ================================================================== >>fastboot.log
fastboot getvar partition-type:userdata 2>>fastboot.log
fastboot getvar current-slot 2>>fastboot.log
fastboot getvar slot-retry-count:b 2>>fastboot.log
fastboot getvar slot-unbootable:b 2>>fastboot.log
fastboot getvar slot-successful:b 2>>fastboot.log
fastboot getvar slot-retry-count:a 2>>fastboot.log
fastboot getvar slot-unbootable:a 2>>fastboot.log
fastboot getvar slot-successful:a 2>>fastboot.log
fastboot getvar cid 2>>fastboot.log

ECHO. >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO #########################  ADB ENABLE  ########################### >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO. >>fastboot.log
echo fastboot.exe oem adb_enable >>fastboot.log
call:fastboot_tool oem adb_enable 2>>fastboot.log

IF EXIST %tempSysName% DEL %tempSysName%

ECHO.
ECHO ##################################################################
ECHO ##################  FLASH COMPLETE. RESTART!   ###################
ECHO ##################################################################
ECHO.

ECHO. >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO ##################  FLASH COMPLETE. RESTART!   ################### >>fastboot.log
ECHO ################################################################## >>fastboot.log
ECHO. >>fastboot.log

GOTO:EOF

REM ============== fastboot_tool() ==============
:fastboot_tool

ECHO [command] : fastboot.exe %~1 %~2 %~3 %~4 %para% %SSN%

fastboot.exe %~1 %~2 %~3 %~4 %para% %SSN%
IF NOT %ERRORLEVEL% == 0 (
    ECHO "FAILED, fastboot.exe %~1 %~2 %~3 %~4 %para% %SSN% failure, EXIT!"
    IF EXIST %tempSysName% DEL %tempSysName%
    EXIT 1
)

ECHO ==================================================================
GOTO:EOF

:END