229 lines
9.0 KiB
Lua
229 lines
9.0 KiB
Lua
local _G = ShaguTweaks.GetGlobalEnv()
|
|
local T = ShaguTweaks.T
|
|
local L = ShaguTweaks.L
|
|
local gfind = string.gmatch or string.gfind
|
|
local GetUnitData = ShaguTweaks.GetUnitData
|
|
local hooksecurefunc = ShaguTweaks.hooksecurefunc
|
|
local GetExpansion = ShaguTweaks.GetExpansion
|
|
local cmatch = ShaguTweaks.cmatch
|
|
local rgbhex = ShaguTweaks.rgbhex
|
|
local strsplit = ShaguTweaks.strsplit
|
|
local friendinfo = gsub(gsub(FRIENDS_LEVEL_TEMPLATE,"%%s","%%s %%s"),"%%d","%%s")
|
|
|
|
local module = ShaguTweaks:register({
|
|
title = T["Social Colors"],
|
|
description = T["Show class colors in Who, Guild, Friends and Chat."],
|
|
expansions = { ["vanilla"] = true, ["tbc"] = true },
|
|
category = T["Social & Chat"],
|
|
enabled = true,
|
|
})
|
|
|
|
module.enable = function(self)
|
|
do -- add class colors to chat
|
|
for i=1,NUM_CHAT_WINDOWS do
|
|
if _G["ChatFrame"..i] and not _G["ChatFrame"..i].HookAddMessageColor and not Prat then
|
|
_G["ChatFrame"..i].HookAddMessageColor = _G["ChatFrame"..i].AddMessage
|
|
_G["ChatFrame"..i].AddMessage = function(frame, text, a1, a2, a3, a4, a5)
|
|
if text then
|
|
for name in gfind(text, "|Hplayer:(.-)|h") do
|
|
local real = strsplit(":", name)
|
|
local class = GetUnitData(real)
|
|
if class and class ~= UNKNOWN then
|
|
local color = rgbhex(RAID_CLASS_COLORS[class])
|
|
text = string.gsub(text,
|
|
"|Hplayer:"..name.."|h%["..real.."%]|h(.-:-)",
|
|
"|r["..color.."|Hplayer:"..name.."|h"..color..real.."|h|r]|r%1")
|
|
end
|
|
end
|
|
end
|
|
_G["ChatFrame"..i].HookAddMessageColor(frame, text, a1, a2, a3, a4, a5)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
ShaguTweaks_cache = ShaguTweaks_cache or {}
|
|
ShaguTweaks_cache["players"] = ShaguTweaks_cache["players"] or {}
|
|
|
|
local playerdb = ShaguTweaks_cache["players"]
|
|
local socialmod = CreateFrame("Frame", "ShaguTweaksSocialMod", UIParent)
|
|
socialmod:RegisterEvent("CHAT_MSG_SYSTEM")
|
|
socialmod:SetScript("OnEvent", function()
|
|
local name = cmatch(arg1, _G.ERR_FRIEND_ONLINE_SS)
|
|
name = cmatch(arg1, _G.ERR_FRIEND_OFFLINE_S)
|
|
if name and playerdb[name] and playerdb[name].cname then
|
|
playerdb[name].lastseen = date("%a %d-%b-%Y")
|
|
end
|
|
end)
|
|
|
|
do -- add colors to guild list
|
|
hooksecurefunc("GuildStatus_Update", function()
|
|
local playerzone = GetRealZoneText()
|
|
local off = FauxScrollFrame_GetOffset(GuildListScrollFrame)
|
|
local _, _, playerrankindex = GetGuildInfo("player")
|
|
|
|
for i=1, GUILDMEMBERS_TO_DISPLAY, 1 do
|
|
local name, _, rankindex, level, class, zone, _, _, online = GetGuildRosterInfo(off + i)
|
|
|
|
class = L["class"][class]
|
|
|
|
if name then
|
|
if class then
|
|
local color = RAID_CLASS_COLORS[class]
|
|
if online then
|
|
_G["GuildFrameButton"..i.."Name"]:SetTextColor(color.r,color.g,color.b,1)
|
|
_G["GuildFrameButton"..i.."Class"]:SetTextColor(color.r,color.g,color.b,1)
|
|
_G["GuildFrameGuildStatusButton"..i.."Name"]:SetTextColor(color.r,color.g,color.b,1)
|
|
else
|
|
_G["GuildFrameButton"..i.."Name"]:SetTextColor(color.r,color.g,color.b,.5)
|
|
_G["GuildFrameButton"..i.."Class"]:SetTextColor(color.r,color.g,color.b,.5)
|
|
_G["GuildFrameGuildStatusButton"..i.."Name"]:SetTextColor(color.r,color.g,color.b,.5)
|
|
end
|
|
end
|
|
|
|
if level then
|
|
local color = GetDifficultyColor(level)
|
|
if online then
|
|
_G["GuildFrameButton"..i.."Level"]:SetTextColor(color.r + .2, color.g + .2, color.b + .2, 1)
|
|
else
|
|
_G["GuildFrameButton"..i.."Level"]:SetTextColor(color.r + .2, color.g + .2, color.b + .2, .5)
|
|
end
|
|
end
|
|
|
|
if zone and zone == playerzone then
|
|
if online then
|
|
_G["GuildFrameButton"..i.."Zone"]:SetTextColor(.5, 1, 1, 1)
|
|
else
|
|
_G["GuildFrameButton"..i.."Zone"]:SetTextColor(.5, 1, 1, .5)
|
|
end
|
|
end
|
|
|
|
if rankindex and rankindex == playerrankindex then
|
|
if online then
|
|
_G["GuildFrameGuildStatusButton"..i.."Rank"]:SetTextColor(.5, 1, 1, 1)
|
|
else
|
|
_G["GuildFrameGuildStatusButton"..i.."Rank"]:SetTextColor(.5, 1, 1, .5)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end)
|
|
end
|
|
|
|
do -- add colors to friend list
|
|
local FRIENDS_NAME_LOCATION = GetExpansion() == "vanilla" and "ButtonTextNameLocation" or "ButtonTextLocation"
|
|
hooksecurefunc("FriendsList_Update", function()
|
|
if GetNumFriends() == 0 then return end
|
|
|
|
local playerzone = GetRealZoneText()
|
|
local off = FauxScrollFrame_GetOffset(FriendsFrameFriendsScrollFrame)
|
|
|
|
for i=1, FRIENDS_TO_DISPLAY do
|
|
local name, level, class, zone, connected, status = GetFriendInfo(off + i)
|
|
if not name or name == _G.UNKNOWN then break end
|
|
local friendName = _G["FriendsFrameFriendButton"..i.."ButtonTextName"]
|
|
local friendLoc = _G["FriendsFrameFriendButton"..i..FRIENDS_NAME_LOCATION]
|
|
local friendInfo = _G["FriendsFrameFriendButton"..i.."ButtonTextInfo"]
|
|
local caption = friendName or friendLoc
|
|
|
|
if connected then
|
|
if not class or class == _G.UNKNOWN then break end
|
|
local ccolor = RAID_CLASS_COLORS[L["class"][class]] or { 1, 1, 1 }
|
|
local lcolor = GetDifficultyColor(tonumber(level)) or { 1, 1, 1 }
|
|
|
|
zone = ( zone == playerzone and "|cffffffff" or "|cffcccccc" ) .. zone .. "|r"
|
|
local cname = rgbhex(ccolor) .. name .. "|r"
|
|
local clevel = rgbhex(lcolor) .. level .. "|r"
|
|
|
|
if playerdb[name] then
|
|
playerdb[name].lastseen = date("%a %d-%b-%Y")
|
|
playerdb[name].cname = cname
|
|
playerdb[name].clevel = clevel
|
|
playerdb[name].cclass = ccolor
|
|
end
|
|
|
|
if friendName then
|
|
friendName:SetText(cname)
|
|
friendLoc:SetText(format(TEXT(FRIENDS_LIST_TEMPLATE), zone, status))
|
|
else
|
|
friendLoc:SetText(format(TEXT(FRIENDS_LIST_TEMPLATE), cname, zone, status))
|
|
end
|
|
|
|
friendInfo:SetText(format(TEXT(friendinfo), clevel, class, ""))
|
|
caption:SetVertexColor(1,1,1,.9)
|
|
friendInfo:SetVertexColor(1,1,1,.9)
|
|
else
|
|
if playerdb[name] and playerdb[name].cname and playerdb[name].clevel and playerdb[name].lastseen then
|
|
caption:SetText(format(TEXT(FRIENDS_LIST_OFFLINE_TEMPLATE), playerdb[name].cname))
|
|
friendInfo:SetText(format(TEXT(friendinfo), playerdb[name].clevel, playerdb[name].lastseen, ""))
|
|
else
|
|
caption:SetText(format(TEXT(FRIENDS_LIST_OFFLINE_TEMPLATE), name.."|r"))
|
|
friendInfo:SetText(TEXT(UNKNOWN))
|
|
end
|
|
|
|
caption:SetVertexColor(1,1,1,.4)
|
|
friendInfo:SetVertexColor(1,1,1,.4)
|
|
end
|
|
end
|
|
end)
|
|
end
|
|
|
|
do -- add colors to who list
|
|
hooksecurefunc("WhoList_Update", function()
|
|
local num, max = GetNumWhoResults()
|
|
local off = FauxScrollFrame_GetOffset(WhoListScrollFrame)
|
|
|
|
local playerzone = GetRealZoneText()
|
|
local playerrace = UnitRace("player")
|
|
local playerguild = GetGuildInfo("player")
|
|
|
|
for i=1, WHOS_TO_DISPLAY do
|
|
local name, guild, level, race, class, zone = GetWhoInfo(off + i)
|
|
local displayedText = ""
|
|
|
|
if num + 1 >= MAX_WHOS_FROM_SERVER then
|
|
displayedText = format(WHO_FRAME_SHOWN_TEMPLATE, MAX_WHOS_FROM_SERVER)
|
|
WhoFrameTotals:SetText("|cffffffff" .. format(GetText("WHO_FRAME_TOTAL_TEMPLATE", nil, num), max).." |cffaaaaaa"..displayedText)
|
|
else
|
|
displayedText = format(WHO_FRAME_SHOWN_TEMPLATE, num)
|
|
WhoFrameTotals:SetText("|cffffffff" .. format(GetText("WHO_FRAME_TOTAL_TEMPLATE", nil, num), num).." |cffaaaaaa"..displayedText)
|
|
end
|
|
|
|
class = L["class"][class]
|
|
|
|
_G["WhoFrameButton"..i.."Name"]:SetTextColor(NORMAL_FONT_COLOR.r, NORMAL_FONT_COLOR.g, NORMAL_FONT_COLOR.b)
|
|
|
|
if (UIDropDownMenu_GetSelectedID(WhoFrameDropDown) == 1) then
|
|
if (zone == playerzone) then
|
|
_G["WhoFrameButton"..i.."Variable"]:SetTextColor(.5, 1, 1)
|
|
else
|
|
_G["WhoFrameButton"..i.."Variable"]:SetTextColor(1, 1, 1)
|
|
end
|
|
|
|
elseif (UIDropDownMenu_GetSelectedID(WhoFrameDropDown) == 2) then
|
|
if (guild == playerguild) then
|
|
_G["WhoFrameButton"..i.."Variable"]:SetTextColor(.5, 1, 1)
|
|
else
|
|
_G["WhoFrameButton"..i.."Variable"]:SetTextColor(1, 1, 1)
|
|
end
|
|
|
|
elseif (UIDropDownMenu_GetSelectedID(WhoFrameDropDown) == 3) then
|
|
if (race == playerrace) then
|
|
_G["WhoFrameButton"..i.."Variable"]:SetTextColor(.5, 1, 1)
|
|
else
|
|
_G["WhoFrameButton"..i.."Variable"]:SetTextColor(1, 1, 1)
|
|
end
|
|
end
|
|
|
|
if class then
|
|
local color = RAID_CLASS_COLORS[class]
|
|
_G["WhoFrameButton"..i.."Class"]:SetTextColor(color.r,color.g,color.b,1)
|
|
end
|
|
|
|
local color = GetDifficultyColor(level)
|
|
_G["WhoFrameButton"..i.."Level"]:SetTextColor(color.r, color.g, color.b)
|
|
end
|
|
end)
|
|
end
|
|
end
|