update.ahk



#include meta.ahk

IfExist, updater.exe
{
	FileDelete, updater.exe
}
IfExist, ___delete_me___.exe
{
	FileDelete, ___delete_me___.exe
}
IniRead, logLevel, setting.ini, update, log, 0
IniRead, lastUpdate, setting.ini, update, last, 0
IniRead, autoUpdate, setting.ini, update, autoupdate, 1
IniRead, updateMirror, setting.ini, update, mirror, 1
IniWrite, % updateMirror, setting.ini, update, mirror
IniRead, version_str, setting.ini, update, ver, 0
IniRead, inst, setting.ini, update, inst, 11
log_write("Start at " A_YYYY "-" A_MM "-" A_DD, 0)
mirrorList:=["https://github.com"
,"https://ghproxy.com/https://github.com"]
updatemirrorTried:=Array()
today:=A_MM . A_DD
if(betaBuild!=1) {
	if(autoUpdate) {
		if(lastUpdate!=today) {
			log_write("Getting Update",0)
			; MsgBox,,Update,Getting Update`n获取最新版本,2
			get_latest_version()
		} else {
			TrayTip,,% name_en " Start`nv" version "`n" name_zh "启动"
		}
	} else {
		log_write("Update Skiped",0)
		if(lastUpdate!=today) {
			TrayTip, Update,Update Skiped`n跳过升级`n`nCurrent version`n当前版本`nv%version%
			IniWrite, % A_MM A_DD, setting.ini, update, last
		}
	}
} else {
	TrayTip,,% name_en " BETA Start`nv" version "`n" name_zh "内测版本启动"
}

get_latest_version(){
	global
	req := ComObjCreate("MSXML2.ServerXMLHTTP")
	updateMirror:=updateMirror+0
	if(updateMirror > mirrorList.Length() or updateMirror <= 0) {
		updateMirror := 1
	}
	updateSite:=mirrorList[updateMirror]
	updateReqDone:=0
	req.open("GET", updateSite downloadUrl versionFilename, true)
	req.onreadystatechange := Func("updateReady")
	req.send()
	SetTimer, updateTimeout, -10000
	Return
	updateTimeout:
	tryNextUpdate()
	Return
}
tryNextUpdate()
{
	global mirrorList, updateMirror, updatemirrorTried
	updatemirrorTried.Push(updateMirror)
	SetTimer, updateTimeout, Off
	For k, v in mirrorList
	{
		tested:=False
		for _, p in updatemirrorTried
		{
			if(p=k) {
				tested:=True
				break
			}
		}
		if not tested {
			updateMirror:=k
			get_latest_version()
			Return
		}
	}
	TrayTip, , % "Update failed`n`n更新失败",, 0x3
}

; with MSXML2.ServerXMLHTTP method, there would be multiple callback called
updateReady(){
	global req, version, updateReqDone, updateSite, downloadUrl, downloadFilename, binaryFilename
	log_write("update req.readyState=" req.readyState, 1)
    if (req.readyState != 4){  ; Not done yet.
        return
	}
	if(updateReqDone){
		; log_write("state already changed", 1)
		Return
	}
	updateReqDone := 1
	log_write("update req.status=" req.status, 1)
    if (req.status == 200){ ; OK.
		SetTimer, updateTimeout, Off
        ; MsgBox % "Latest version: " req.responseText
		RegExMatch(version, "(\d+)\.(\d+)\.(\d+)", verNow)
		RegExMatch(req.responseText, "^(\d+)\.(\d+)\.(\d+)$", verNew)
		if((verNew1>verNow1)
		|| (verNew1==verNow1 && ((verNew2>verNow2)
			|| (verNew2==verNow2 && verNew3>verNow3)))){
			MsgBox, 0x24, Download, % "Found new version " req.responseText ", download?`n`n发现新版本 " req.responseText " 是否下载?"
			IfMsgBox Yes
			{
				try {
					IfExist, %downloadFilename%
					{
						FileDelete, %downloadFilename%
					}
					UrlDownloadToFile, % updateSite downloadUrl downloadFilename, % "./" downloadFilename
					MsgBox, ,, % "Download finished`n更新下载完成`n`nProgram will restart now`n软件即将重启", 3
					IniWrite, % A_MM A_DD, setting.ini, update, last
					FileInstall, updater.exe, updater.exe, 1
					Run, updater.exe ___ORIGIN_ME___ %downloadFilename% %binaryFilename%, A_ScriptDir, Hide
					ExitApp
				} catch e {
					MsgBox, 16,, % "Upgrade failed`nAn exception was thrown!`nSpecifically: " e
				}
			}
		} else {
			; MsgBox, ,, % "Current version: v" version "`n`nIt is the latest version`n`n软件已是最新版本", 2
			IniWrite, % A_MM A_DD, setting.ini, update, last
		}
	} else {
		tryNextUpdate()
        ; MsgBox, 16,, % "Update failed`n`n更新失败`n`nStatus=" req.status
	}
}