Level20/VersionCheck.lua at main · semyon422/Level20 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
local addonName, addon = ...
local L = addon.L

local COMM_PREFIX = "L20VER"

local versionState = {
	initialized = false,
	currentVersion = 0,
	latestVersion = nil,
	latestSource = nil,
}

local function GetMetadataVersion()
	local version

	if C_AddOns and C_AddOns.GetAddOnMetadata then
		version = C_AddOns.GetAddOnMetadata(addonName, "Version")
	elseif GetAddOnMetadata then
		version = GetAddOnMetadata(addonName, "Version")
	end

	return tonumber(version)
end

local function RefreshInfoPanel()
	if addon.RefreshInfoPanel then
		addon.RefreshInfoPanel()
	end
end

function addon.GetCurrentVersion()
	return versionState.currentVersion
end

function addon.HasVersionUpdate()
	return versionState.latestVersion ~= nil
end

function addon.GetVersionStatusText()
	if versionState.latestVersion then
		if versionState.latestSource then
			return string.format(L.VERSION_STATUS_UPDATE_SOURCE, versionState.latestVersion, versionState.latestSource)
		end

		return string.format(L.VERSION_STATUS_UPDATE, versionState.latestVersion)
	end

	return string.format(L.VERSION_STATUS_CURRENT, versionState.currentVersion)
end

function addon.GetVersionStatusColor()
	if versionState.latestVersion then
		return 1, 0.82, 0.2
	end

	return 0.35, 0.82, 1
end

function addon.ProcessRemoteVersion(version, sender)
	version = tonumber(version)
	if not version then
		return
	end

	if version <= versionState.currentVersion then
		return
	end

	if versionState.latestVersion and version <= versionState.latestVersion then
		return
	end

	versionState.latestVersion = version
	versionState.latestSource = sender
	Level20DB.lastSeenRemoteVersion = version
	Level20DB.lastSeenRemoteVersionSource = sender
	RefreshInfoPanel()
end

function addon.OnVersionCheckMessage(prefix, message, channel, sender)
	if prefix ~= COMM_PREFIX then
		return
	end

	local playerName = UnitName("player")
	if not playerName or not sender then
		return
	end

	if Ambiguate(sender, "short") == Ambiguate(playerName, "short") then
		return
	end

	addon.ProcessRemoteVersion(message, sender)
end

local function SendVersion(distribution)
	if not distribution then
		return
	end

	C_ChatInfo.SendAddonMessage(COMM_PREFIX, tostring(versionState.currentVersion), distribution)
end

function addon.BroadcastVersionCheck(sendToGuild)
	if not versionState.initialized then
		return
	end

	if IsInRaid(LE_PARTY_CATEGORY_INSTANCE) then
		SendVersion("INSTANCE_CHAT")
	elseif IsInRaid() then
		SendVersion("RAID")
	elseif IsInGroup(LE_PARTY_CATEGORY_INSTANCE) then
		SendVersion("INSTANCE_CHAT")
	elseif IsInGroup() then
		SendVersion("PARTY")
	end

	if sendToGuild and IsInGuild() then
		SendVersion("GUILD")
	end
end

function addon.InitializeVersionCheck()
	if versionState.initialized then
		return
	end

	if not C_ChatInfo or not C_ChatInfo.RegisterAddonMessagePrefix or not C_ChatInfo.SendAddonMessage then
		return
	end

	versionState.currentVersion = GetMetadataVersion() or versionState.currentVersion
	versionState.latestVersion = tonumber(Level20DB.lastSeenRemoteVersion)
	versionState.latestSource = Level20DB.lastSeenRemoteVersionSource

	if versionState.latestVersion and versionState.latestVersion <= versionState.currentVersion then
		versionState.latestVersion = nil
		versionState.latestSource = nil
		Level20DB.lastSeenRemoteVersion = nil
		Level20DB.lastSeenRemoteVersionSource = nil
	end

	C_ChatInfo.RegisterAddonMessagePrefix(COMM_PREFIX)
	versionState.initialized = true
	RefreshInfoPanel()
end