first trade preview

RaidLedger.toc

@@ -42,4 +42,5 @@ db.lua
 options.lua
 gui.lua
 
+trade.lua
 cli.lua

db.lua

@@ -162,6 +162,24 @@ local function GetFilteritemsSet(s)
     return set
 end
 
+function db:AddOrUpdateLoot(item, count, beneficiary, cost)
+    local itemName, itemLink, itemRarity, _, _, _, _, itemStackCount = GetItemInfo(item)
+
+    local ledger = self:GetCurrentLedger()
+    for _, entry in pairs(ledger["items"]) do
+        if entry.detail then
+            if entry.detail.item == itemLink and entry.cost == 0 and entry.detail.count == count then
+                entry.beneficiary = beneficiary
+                entry.cost = cost
+                self:OnLedgerItemsChange()
+                return
+            end
+        end
+    end
+
+    self:AddLoot(item, count, beneficiary, cost, true)
+end
+
 function db:AddLoot(item, count, beneficiary, cost, force)
     local itemName, itemLink, itemRarity, _, _, _, _, itemStackCount = GetItemInfo(item)
     itemStackCount = itemStackCount or 0

trade.lua

@@ -0,0 +1,113 @@
+local _, ADDONSELF = ...
+
+local L = ADDONSELF.L
+local GUI = ADDONSELF.gui
+local Database = ADDONSELF.db
+local Print = ADDONSELF.print
+local RegEvent = ADDONSELF.regevent
+
+local f = CreateFrame("Frame", nil, TradeFrame)
+f:SetWidth(172)
+f:SetHeight(40)
+f:SetBackdrop({
+    -- bgFile = "Interface\\FrameGeneral\\UI-Background-Marble",
+    bgFile = "Interface\\DialogFrame\\UI-DialogBox-Background",
+
+    edgeFile = "Interface\\DialogFrame\\UI-DialogBox-Border",
+    tile = true,
+    tileEdge = true,
+    tileSize = 16,
+    edgeSize = 24,
+    insets = { left = 4, right = 4, top = 4, bottom = 4 },   
+})
+
+f:SetPoint("TOPRIGHT", TradeFrame, "BOTTOMRIGHT", 0, 0)
+f:SetToplevel(true)
+f:EnableMouse(true)
+f:SetFrameLevel(TradeFrame:GetFrameLevel() + 1)
+
+local addtoCheckbtn = nil
+do
+    local b = CreateFrame("CheckButton", nil, f, "UICheckButtonTemplate")
+    b.text = b:CreateFontString(nil, "OVERLAY", "GameFontNormal")
+    b.text:SetPoint("LEFT", b, "RIGHT", 0, 1)
+    b:SetPoint("LEFT", f, 5, 0)
+    b.text:SetText(L["Add to"] .. " " .. L["Raid Ledger"])
+
+    f:SetWidth(b.text:GetStringWidth() + 50)
+    addtoCheckbtn = b
+end
+
+
+local currTrade = nil
+
+local function UpdateCheckButton()
+    -- print(1)
+end
+
+RegEvent("TRADE_SHOW", function()
+    addtoCheckbtn:SetChecked(UnitInRaid("player"))
+end)
+
+RegEvent("TRADE_UPDATE", UpdateCheckButton)
+RegEvent("TRADE_PLAYER_ITEM_CHANGED", UpdateCheckButton)
+RegEvent("TRADE_MONEY_CHANGED", UpdateCheckButton)
+
+RegEvent("TRADE_ACCEPT_UPDATE", function(p, t)
+    currTrade = {
+        target = GetUnitName("NPC", true),
+        targetmoney = GetTargetTradeMoney(),
+        playermoney = GetPlayerTradeMoney(),
+
+        targetitems = {},
+        playeritems = {},
+    }
+
+    for i = 1, 6 do
+        local t = GetTradeTargetItemLink(i)
+
+        if t then
+            table.insert(currTrade.targetitems, {
+                item = t,
+                count = select(3, GetTradeTargetItemInfo(i)),
+            })
+        end
+
+        local p = GetTradePlayerItemLink(i)
+        if p then
+            table.insert(currTrade.playeritems, {
+                item = p,
+                count = select(3, GetTradePlayerItemInfo(i)),
+            })
+        end
+    end
+end)
+
+local function AddLootFromTrade(beneficiary, cost, items) 
+    if cost == 0 then
+        return
+    end
+
+    for i, item in ipairs(items) do
+        if i > 1 then
+            cost = 0
+        end
+
+        Database:AddOrUpdateLoot(item.item, item.count, beneficiary, cost / 10000)
+        Print(L["Item added"] .. " " .. item.item .. " " .. L["Beneficiary"] .. " " .. beneficiary .. " " .. GetMoneyString(cost))
+    end
+end
+
+RegEvent("UI_INFO_MESSAGE", function(_, text)
+    if text == ERR_TRADE_COMPLETE then
+
+        if not addtoCheckbtn:GetChecked() then
+            return
+        end
+
+        AddLootFromTrade(currTrade.target, currTrade.targetmoney, currTrade.playeritems)
+        AddLootFromTrade(UnitName("player"), currTrade.playermoney, currTrade.targetitems)
+    end
+
+    currTrade = nil
+end)