TurtleMail/Postal.lua

Postal = AceLibrary('AceAddon-2.0'):new('AceEvent-2.0', 'AceHook-2.0')
Postal.open, Postal.send = {}, {}

function Postal:OnInitialize()

	-- Allows the mail frame to be pushed
	if UIPanelWindows['MailFrame'] then
		UIPanelWindows['MailFrame'].pushable = 1
	else
		UIPanelWindows['MailFrame'] = { area = 'left', pushable = 1 }
	end

	-- Close FriendsFrame will close if you try to open a mail with mailframe+friendsframe open
	if UIPanelWindows['FriendsFrame'] then
		UIPanelWindows['FriendsFrame'].pushable = 2
	else
		UIPanelWindows['FriendsFrame'] = { area = 'left', pushable = 2 }
	end

	MailItem1:SetPoint('TOPLEFT', 'InboxFrame', 'TOPLEFT', 48, -80)
	for i=1,7 do
		getglobal('MailItem' .. i .. 'ExpireTime'):SetPoint('TOPRIGHT', 'MailItem' .. i, 'TOPRIGHT', 10, -4)
		getglobal('MailItem' .. i):SetWidth(280)
	end

	ATTACHMENTS_MAX = 21
	ATTACHMENTS_PER_ROW_SEND = 7
	ATTACHMENTS_MAX_ROWS_SEND = 3

    Postal_MiniMapMailFrame_Show_Orig = MiniMapMailFrame.Show

	self.Inbox_selectedItems = {}
    Postal.Send_Ready = true

    CreateFrame('GameTooltip', 'PostalTooltip', nil, 'GameTooltipTemplate')
    PostalTooltip:SetOwner(WorldFrame, 'ANCHOR_NONE')
end

function Postal:SendMailFrame_Update()

	MoneyFrame_Update('SendMailCostMoneyFrame', GetSendMailPrice() * max(1, self:GetNumMails()))

	local itemCount = 0
	local itemTitle
	local gap
	-- local last = 0
	local last = self:GetNumMails()
	for i=1,ATTACHMENTS_MAX do
		local btn = getglobal('PostalAttachment' .. i)

		local texture, count
		if btn.bag and btn.slot then
			texture, count = GetContainerItemInfo(btn.bag, btn.slot)
		end
		if not texture then
			btn:SetNormalTexture(nil)
			getglobal(btn:GetName()..'Count'):Hide()
			btn.slot = nil
			btn.bag = nil
		else
			btn:SetNormalTexture(texture)
			if count > 1 then
				getglobal(btn:GetName()..'Count'):Show()
				getglobal(btn:GetName()..'Count'):SetText(count)
			else
				getglobal(btn:GetName()..'Count'):Hide()
			end
		end
	end

	-- Determine how many rows of attachments to show
	local itemRowCount = 1
	local temp = last
	while temp > ATTACHMENTS_PER_ROW_SEND and itemRowCount < ATTACHMENTS_MAX_ROWS_SEND do
		itemRowCount = itemRowCount + 1
		temp = temp - ATTACHMENTS_PER_ROW_SEND
	end

	if not gap and temp == ATTACHMENTS_PER_ROW_SEND and itemRowCount < ATTACHMENTS_MAX_ROWS_SEND then
		itemRowCount = itemRowCount + 1
	end
	if SendMailFrame.maxRowsShown and last > 0 and itemRowCount < SendMailFrame.maxRowsShown then
		itemRowCount = SendMailFrame.maxRowsShown
	else
		SendMailFrame.maxRowsShown = itemRowCount
	end

	-- Compute sizes
	local cursorx = 0
	local cursory = itemRowCount - 1
	local marginxl = 8 + 6
	local marginxr = 40 + 6
	local areax = SendMailFrame:GetWidth() - marginxl - marginxr
	local iconx = PostalAttachment1:GetWidth() + 2
	local icony = PostalAttachment1:GetHeight() + 2
	local gapx1 = floor((areax - (iconx * ATTACHMENTS_PER_ROW_SEND)) / (ATTACHMENTS_PER_ROW_SEND - 1))
	local gapx2 = floor((areax - (iconx * ATTACHMENTS_PER_ROW_SEND) - (gapx1 * (ATTACHMENTS_PER_ROW_SEND - 1))) / 2)
	local gapy1 = 5
	local gapy2 = 6
	local areay = (gapy2 * 2) + (gapy1 * (itemRowCount - 1)) + (icony * itemRowCount)
	local indentx = marginxl + gapx2 + 17
	local indenty = 170 + gapy2 + icony - 13
	local tabx = (iconx + gapx1) - 3 --this magic number changes the attachment spacing
	local taby = (icony + gapy1)
	local scrollHeight = 249 - areay


	PostalHorizontalBarLeft:SetPoint('TOPLEFT', SendMailFrame, 'BOTTOMLEFT', 2 + 15, 184 + areay - 14)

	SendMailScrollFrame:SetHeight(scrollHeight)
	SendMailScrollChildFrame:SetHeight(scrollHeight)

	local SendMailScrollFrameTop = ({SendMailScrollFrame:GetRegions()})[3]
	SendMailScrollFrameTop:SetHeight(scrollHeight)
	SendMailScrollFrameTop:SetTexCoord(0, 0.484375, 0, scrollHeight / 256)

	StationeryBackgroundLeft:SetHeight(scrollHeight)
	StationeryBackgroundLeft:SetTexCoord(0, 1.0, 0, scrollHeight / 256)


	StationeryBackgroundRight:SetHeight(scrollHeight)
	StationeryBackgroundRight:SetTexCoord(0, 1.0, 0, scrollHeight / 256)


		-- Set Items
	for i=1,ATTACHMENTS_MAX do
		if cursory >= 0 then
			getglobal('PostalAttachment'..i):Enable()
			getglobal('PostalAttachment'..i):Show()
			getglobal('PostalAttachment'..i):SetPoint('TOPLEFT', 'SendMailFrame', 'BOTTOMLEFT', indentx + (tabx * cursorx), indenty + (taby * cursory))
			
			cursorx = cursorx + 1
			if cursorx >= ATTACHMENTS_PER_ROW_SEND then
				cursory = cursory - 1
				cursorx = 0
			end
		else
			getglobal('PostalAttachment'..i):Hide()
		end
	end
	for i=ATTACHMENTS_MAX+1,ATTACHMENTS_MAX do
		getglobal('PostalAttachment'..i):Hide()
	end

	if self:GetNumMails() > 0 then
		SendMailCODButton:Enable()
		SendMailCODButtonText:SetTextColor(NORMAL_FONT_COLOR.r, NORMAL_FONT_COLOR.g, NORMAL_FONT_COLOR.b)
	else
		SendMailRadioButton_OnClick(1)
		SendMailCODButton:Disable()
		SendMailCODButtonText:SetTextColor(GRAY_FONT_COLOR.r, GRAY_FONT_COLOR.g, GRAY_FONT_COLOR.b)
	end

	Postal:SendMailFrame_CanSend()
end

function Postal:OnEnable()
	self:RegisterEvent('UI_ERROR_MESSAGE')
	self:RegisterEvent('MAIL_SEND_SUCCESS')
	self:RegisterEvent('MAIL_CLOSED')
    self:RegisterEvent('CURSOR_UPDATE')

	self:Hook('ContainerFrameItemButton_OnClick')
	self:Hook('PickupContainerItem')
	self:Hook('UseContainerItem')
	self:Hook('ClickSendMailItemButton')
	self:Hook('SendMailFrame_Update')
	self:Hook('SendMailFrame_CanSend')
	self:Hook('SetItemButtonDesaturated')
	self:Hook('InboxFrame_OnClick')
	self:Hook('InboxFrameItem_OnEnter')
	self:Hook('InboxFrame_Update')

	SendMailFrame:CreateTexture('PostalHorizontalBarLeft', 'BACKGROUND')
	PostalHorizontalBarLeft:SetTexture([[Interface\ClassTrainerFrame\UI-ClassTrainer-HorizontalBar]])
	PostalHorizontalBarLeft:SetWidth(256)
	PostalHorizontalBarLeft:SetHeight(16)
	PostalHorizontalBarLeft:SetTexCoord(0, 1, 0, 0.25)
	SendMailFrame:CreateTexture('PostalHorizontalBarRight', 'BACKGROUND')
	PostalHorizontalBarRight:SetTexture([[Interface\ClassTrainerFrame\UI-ClassTrainer-HorizontalBar]])
	PostalHorizontalBarRight:SetWidth(75)
	PostalHorizontalBarRight:SetHeight(16)
	PostalHorizontalBarRight:SetTexCoord(0, 0.29296875, 0.25, 0.5)
	PostalHorizontalBarRight:SetPoint('LEFT', PostalHorizontalBarLeft, 'RIGHT')

	do
		local background = ({SendMailPackageButton:GetRegions()})[1]
		background:Hide()
		local count = ({SendMailPackageButton:GetRegions()})[3]
		count:Hide()
		SendMailPackageButton:Disable()
		SendMailPackageButton:SetScript('OnReceiveDrag', nil)
		SendMailPackageButton:SetScript('OnDragStart', nil)
	end

	do
		SendMailMoneyText:SetPoint('TOPLEFT', 0, -2)
		SendMailMoney:ClearAllPoints()
		SendMailMoney:SetPoint('TOPLEFT', SendMailMoneyText, 'BOTTOMLEFT', 5, -3)
		SendMailSendMoneyButton:SetPoint('TOPLEFT', SendMailMoney, 'TOPRIGHT', 0, 12)
	end

	PostalMailButton:SetScript('OnClick', function()
		local attachments = self:AttachmentList()
		self.Send_State = {
			first = true,
			to = SendMailNameEditBox:GetText(),
			subject = PostalSubjectEditBox:GetText(),
			body = SendMailBodyEditBox:GetText(),
			money = MoneyInputFrame_GetCopper(SendMailMoney),
			cod = SendMailCODButton:GetChecked(),
			attachments = attachments,
			total = getn(attachments),
		}

		self:SendMail_Clear()
		self:SendMailFrame_Update()
	end)

	-- hack to avoid automatic subject setting/button enabling
	SendMailMailButton:Hide()
	SendMailSubjectEditBox:Hide()
	SendMailSubjectEditBox.SetText = function(self, text) PostalSubjectEditBox:SetText(text) end
	SendMailNameEditBox:SetScript('OnTabPressed', function()
		PostalSubjectEditBox:SetFocus()
	end)
	SendMailNameEditBox:SetScript('OnEnterPressed', function()
		PostalSubjectEditBox:SetFocus()
	end)
	SendMailBodyEditBox:SetScript('OnTabPressed', function()
		if IsShiftKeyDown() then
			PostalSubjectEditBox:SetFocus()
		else
			SendMailMoneyGold:SetFocus()
		end
	end)

	SendMailFrame_Update()
end

function Postal:MAIL_CLOSED()
	self.Send_State = nil
	self:Inbox_Abort()
	self:SendMail_Clear()

	-- Hides the minimap unread mail button if there are no unread mail on closing the mailbox.
	-- Does not scan past the first 50 items since only the first 50 are viewable.
	for i=1,GetInboxNumItems() do
		local wasRead = ({ GetInboxHeaderInfo(i) })[9]
		if not wasRead then
			return
		end
	end
	MiniMapMailFrame:Hide()
	-- There may be an UPDATE PENDING MAIL event after closing which would make the frame reappear, the following prevents that
	local t = GetTime()
	MiniMapMailFrame.Show = function()
		if GetTime() - t > 2 then
			MiniMapMailFrame.Show = Postal_MiniMapMailFrame_Show_Orig
			MiniMapMailFrame:Show()
		end
	end
end

function Postal:MAIL_SEND_SUCCESS() 
	Postal.Send_Ready = true
end

function Postal:ContainerFrameItemButton_OnClick(btn, ignore)
	local item = {this:GetParent():GetID(), this:GetID()}
	if self:Send_Attached(item) then
		return
	else
	    return self.hooks['ContainerFrameItemButton_OnClick'].orig(btn, ignore)
    end
end

function Postal:Send_AttachItem(item)
	for i = 1,ATTACHMENTS_MAX do
		if not getglobal('PostalAttachment'..i).item then
			if not self:ItemIsMailable(item) then
                return self:Print('Postal: Cannot attach item.', 1, 0.5, 0)
			end
			getglobal('PostalAttachment'..i).item = item
            return SendMailFrame_Update()
		end
	end
end

do
	local last_picked_up

    function Postal:CURSOR_UPDATE()
        Aux.log(tostring(last_picked_up))
        last_picked_up = nil
    end

	function Postal:CursorItem()
		if CursorHasItem() then
			return last_picked_up
		end
	end

	function Postal:PickupContainerItem(bag, slot, force)
		local item = {bag, slot}
		if not force and self:Send_Attached(item) then
			return
		end
		if not CursorHasItem() then
            self:Wait(function()
                last_picked_up = item
            end)
		end
		self.hooks['PickupContainerItem'].orig(unpack(item))
	end
end

function Postal:UseContainerItem(bag, slot)
    local item = {bag, slot}
    if self:Send_Attached(item) then
        return
    end

    if IsShiftKeyDown() or IsControlKeyDown() or IsAltKeyDown() then
        return self.hooks['UseContainerItem'].orig(unpack(item))
    end

    if SendMailFrame:IsVisible() and not CursorHasItem() then
        self:Send_AttachItem(item)
        return
    elseif TradeFrame:IsVisible() and not CursorHasItem() then
        for i = 1,6 do
            if not GetTradePlayerItemLink(i) then
                self.hooks['PickupContainerItem'].orig(unpack(item))
                ClickTradeButton(i)
                return
            end
        end
    end

    self.hooks['UseContainerItem'].orig(unpack(item))
end

-- Handle the dragging of items
function Postal:AttachmentButton_OnClick(button)
	button = button or this
	local buttonItem = button.item
	local cursorItem = self:CursorItem()
	if cursorItem then
		if not self:ItemIsMailable(unpack(cursorItem)) then
			self:Print('Postal: Cannot attach item.', 1, 0.5, 0)
			ClearCursor()
			return
		end

		button.item = cursorItem
		ClearCursor() -- triggers lock changed event to update the bag frame
	else
		button.item = nil
	end

	if buttonItem then
		PickupContainerItem(unpack(buttonItem))
	end
	SendMailFrame_Update()
end

function Postal:ItemIsMailable(item)
	-- Make sure tooltip is cleared
--	PostalTooltip:ClearLines() TODO
	PostalTooltip:SetBagItem(unpack(item))
	for i=1,PostalTooltip:NumLines() do
		local text = getglobal('PostalTooltipTextLeft' .. i):GetText()
		if text == ITEM_SOULBOUND or text == ITEM_BIND_QUEST or text == ITEM_CONJURED or text == ITEM_BIND_ON_PICKUP then
			return false
		end
	end
	return true
end

function Postal:Send_Attached(item)
	for i=1,ATTACHMENTS_MAX do
		local btn = getglobal('PostalAttachment' .. i)
		if btn.item and btn.item[1] == item[1] and btn.item[2] == item[2] then
			return true
		end
    end

    if not self.Send_State then
        return
    end

    for _, attachment in self.Send_State.attachments do
        if attachment.item[1] == item[1] and attachment.item[2] == item[2] then
            return true
        end
    end
end

function Postal:GetNumMails()
	local num = 0
	for i=1,ATTACHMENTS_MAX do
		local btn = getglobal('PostalAttachment'..i)
		if btn.item then
			num = num + 1
		end
	end
	return num
end

function Postal:SendMail_Clear()
	local num = 0
	for i=1,ATTACHMENTS_MAX do
		local btn = getglobal('PostalAttachment'..i)
		btn.item = nil
	end
	PostalMailButton:Disable()
	SendMailNameEditBox:SetText('')
	SendMailNameEditBox:SetFocus()
	PostalSubjectEditBox:SetText('')
	SendMailBodyEditBox:SetText('')
	MoneyInputFrame_ResetMoney(SendMailMoney)
	SendMailRadioButton_OnClick(1)
end

function Postal:SendMailFrame_CanSend()
	if strlen(SendMailNameEditBox:GetText()) > 0 and (SendMailSendMoneyButton:GetChecked() and MoneyInputFrame_GetCopper(SendMailMoney) or 0) + GetSendMailPrice() * self:GetNumMails() <= GetMoney() then
		PostalMailButton:Enable()
	else
		PostalMailButton:Disable()
	end
end

-- handle the weird built-in mail body textbox onclick
function Postal:ClickSendMailItemButton()
	ClearCursor()
	self:When(function() return not ({GetContainerItemInfo(unpack(self:CursorItem()))})[3] end, function()
		if self:CursorItem() then
			self:Send_AttachItem(self:CursorItem())
		end
	end)
end

function Postal:Send_SendMail()

	local item = tremove(self.Send_State.attachments, 1)

	if item or self.Send_State.first then
		local subject = self.Send_State.subject
		subject = subject ~= '' and subject or '[No Subject]'
		if self.Send_State.total > 1 then
			subject = subject..format(' (Part %d of %d)', self.Send_State.total - getn(self.Send_State.attachments), self.Send_State.total)
		end

		if item then
			ClearCursor()
			self.hooks['ClickSendMailItemButton'].orig()
			ClearCursor()
			self.hooks['PickupContainerItem'].orig(unpack(item))
			self.hooks['ClickSendMailItemButton'].orig()

			local name, _, count = GetSendMailItem()

			if not name then 
				self:Print('Postal: An error occured in POSTAL. This might be related to lag, trying to send items with an item placed in the normal send mail window, or trying to send items that cannot be sent.', 1, 0, 0)
				return
			end
		end

		if self.Send_State.first then
			self.Send_State.first = false

			if self.Send_State.money then
				if self.Send_State.cod then
					SetSendMailCOD(self.Send_State.money)
				else
					SetSendMailMoney(self.Send_State.money)
				end
			end
		end

		return SendMail(self.Send_State.to, subject, self.Send_State.body)
	end
end

function Postal:AttachmentList()
	local arr = {}
	for i = 1, ATTACHMENTS_MAX do
		local btn = getglobal('PostalAttachment' .. i)
		if btn.item then
			tinsert(arr, btn.item)
		end
	end
	return arr
end

function Postal:When(p, callback)
	self.state = {
		p = p,
		callback = callback,
	}
end

function Postal:Wait(callback)
	self.state = {
		p = function() return true end,
		callback = callback,
	}
end

function Postal:UPDATE()

	if self.state and self.state.p() then
		return self.state.callback()
	end

	if self.Send_State and self.Send_Ready then
		self.Send_Ready = nil
		self:Send_SendMail()
	end
end

function Postal:Inbox_OpenMail(selected)
	if getn(selected) == 0 then
		self.opening = false
		self:Inbox_DisableClicks()
	else
		self:Inbox_OpenItem(selected[1], GetInboxNumItems(), selected)
	end
end

function Postal:Inbox_OpenItem(i, inboxCount, selected)
	self:Wait(function()
		local _, _, _, _, money, COD, _, item = GetInboxHeaderInfo(i)
		local newInboxCount = GetInboxNumItems()

		if newInboxCount < inboxCount or COD > 0 then
			tremove(selected, 1)
			if newInboxCount < inboxCount then
				for j, _ in selected do
					selected[j] = selected[j] - 1
				end
			end
			return self:Inbox_OpenMail(selected)
		elseif item then
			TakeInboxItem(i)
			self:When(function() return not ({GetInboxHeaderInfo(i)})[8] or GetInboxNumItems() < inboxCount end, function()
				return self:Inbox_OpenItem(i, inboxCount, selected)
			end)
		elseif money > 0 then
			TakeInboxMoney(i)
			self:When(function() return ({GetInboxHeaderInfo(i)})[5] == 0 or GetInboxNumItems() < inboxCount end, function()
				return self:Inbox_OpenItem(i, inboxCount, selected)
			end)
		else
			DeleteInboxItem(i)
			self:When(function() return GetInboxNumItems() < inboxCount end, function()
				return self:Inbox_OpenItem(i, inboxCount, selected)
			end)
		end
	end)
end

function Postal:SetItemButtonDesaturated(itemButton, locked)
	local item = { itemButton:GetParent():GetID(), itemButton:GetID() }
	if self:Send_Attached(item) then
		return self.hooks['SetItemButtonDesaturated'].orig(itemButton, true)
	end
	return self.hooks['SetItemButtonDesaturated'].orig(itemButton, locked)
end

function Postal:InboxFrameItem_OnEnter()
	local didSetTooltip
	GameTooltip:SetOwner(this, 'ANCHOR_RIGHT')
	if this.index then
		if GetInboxItem(this.index) then
			GameTooltip:SetInboxItem(this.index)
			didSetTooltip = 1
		end
	end
	if this.money then
		GameTooltip:AddLine(ENCLOSED_MONEY, '', 1, 1, 1)
		SetTooltipMoney(GameTooltip, this.money)
		SetMoneyFrameColor('GameTooltipMoneyFrame', HIGHLIGHT_FONT_COLOR.r, HIGHLIGHT_FONT_COLOR.g, HIGHLIGHT_FONT_COLOR.b)
	elseif this.cod then
		GameTooltip:AddLine(COD_AMOUNT, '', 1, 1, 1)
		SetTooltipMoney(GameTooltip, this.cod)
		if this.cod > GetMoney() then
			SetMoneyFrameColor('GameTooltipMoneyFrame', RED_FONT_COLOR.r, RED_FONT_COLOR.g, RED_FONT_COLOR.b)
		else
			SetMoneyFrameColor('GameTooltipMoneyFrame', HIGHLIGHT_FONT_COLOR.r, HIGHLIGHT_FONT_COLOR.g, HIGHLIGHT_FONT_COLOR.b)
		end
	end
	if didSetTooltip and (this.money or this.cod) then
		GameTooltip:SetHeight(GameTooltip:GetHeight()+getglobal('GameTooltipTextLeft' .. GameTooltip:NumLines()):GetHeight())
		if GameTooltipMoneyFrame:IsVisible() then
			GameTooltip:SetHeight(GameTooltip:GetHeight()+GameTooltipMoneyFrame:GetHeight())
		end
	end
	GameTooltip:Show()
end

function Postal:UI_ERROR_MESSAGE()
	if event == 'UI_ERROR_MESSAGE' and (arg1 == ERR_INV_FULL or arg1 == ERR_ITEM_MAX_COUNT) then
		if this.num then
			if arg1 == ERR_INV_FULL then
				self:Inbox_Abort()
				self:Print('Postal: Inventory full. Aborting.', 1, 0, 0)
			elseif arg1 == ERR_ITEM_MAX_COUNT then
				self:Print('Postal: You already have the maximum amount of that item. Skipping.', 1, 0, 0)
				if this.lastVal then
					for key, va in this.id do
						if va >= this.lastVal then
							this.id[key] = va + 1
						end
					end
				end
			end
		end
	end
end

function Postal:Print(msg, r, g, b)
	DEFAULT_CHAT_FRAME:AddMessage(msg, r, g, b)
end

function Postal:Inbox_SetSelected()
	local index = this:GetID() + (InboxFrame.pageNum - 1) * 7
	self.Inbox_selectedItems[index] = this:GetChecked()
end

function Postal:Inbox_OpenSelected(all)
	local selected = {}
	if all then
		for i = 1,GetInboxNumItems() do
			tinsert(selected, i)
		end
	else
		for i, _ in self.Inbox_selectedItems do
			tinsert(selected, i)
		end
		sort(selected)
	end
	self.opening = true
	self:Inbox_DisableClicks()
	self:Inbox_OpenMail(selected)
	self.Inbox_selectedItems = {}
end

function Postal:InboxFrame_Update()
	self.hooks['InboxFrame_Update'].orig()
	for i = 1, 7 do
		local index = (i + (InboxFrame.pageNum - 1) * 7)
		if index > GetInboxNumItems() then
			getglobal('PostalBoxItem'..i..'CB'):Hide()
		else
			getglobal('PostalBoxItem'..i..'CB'):Show()
			getglobal('PostalBoxItem'..i..'CB'):SetChecked(self.Inbox_selectedItems[index])
		end
	end
	self:Inbox_DisableClicks()
end

function Postal:Inbox_DisableClicks()
	for i=1,7 do
		getglobal('MailItem'..i..'ButtonIcon'):SetDesaturated(self.opening)
		if self.opening then
			getglobal('MailItem'..i..'Button'):SetChecked(nil)
		end
	end
end

function Postal:InboxFrame_OnClick(index)
	if self.opening then
		this:SetChecked(nil)
		return
	else
		return self.hooks['InboxFrame_OnClick'].orig(index)
	end
end

function Postal:Inbox_Abort()
	self.state = nil
	self.opening = false
	self:Inbox_DisableClicks()
	self.Inbox_selectedItems = {}
end