ATOM.lua

local addonName, ATOM = ...

ATOM = LibStub('AceAddon-3.0'):NewAddon(ATOM, addonName, 'AceEvent-3.0', 'AceConsole-3.0', 'AceTimer-3.0')

ATOM:SetDefaultModuleLibraries('AceEvent-3.0', 'AceConsole-3.0', 'AceTimer-3.0')

_G['ATOM'] = ATOM;

function ATOM:OnInitialize()
    self.db = LibStub('AceDB-3.0'):New('AtomDB')
end

function ATOM:OnEnable()
    self:RegisterChatCommand('atom', 'SlashCommand')
end

function ATOM:Wait(delay, func)
    ATOM:ScheduleTimer(func or delay, func and delay or 0.5)
end

function ATOM:SlashCommand(msg)
    local cmd, offset = self:GetArgs(msg)
    local args = msg:sub(offset)

    if cmd == 'clear' then
        self:GetModule('Chat'):Clear()
    elseif cmd == 'count' then
        self:CreateItemCountFrame(args)
    elseif cmd == 'destroy' then
        self:GetModule('Bags'):DestroyItems(args == 'true')
    elseif cmd == 'mark' then
        self:GetModule('Macros'):MarkTarget(args ~= '' and args or nil)
    elseif cmd == 'mount' then
        self:GetModule('Mounts'):Mount(args)
    elseif cmd == 'move' then
        SetCVar('autoInteract', GetCVar('autoInteract') ~= '1' and '1' or '0')
    elseif cmd == 'powerleveling' then
        self:GetModule('BattlePets'):GetActivePowerlevelingBattlePetTrainer()
    elseif cmd == 'quest' then
        self:GetModule('Quest'):QuestCompleted(args)
    elseif cmd == 'scoreboard' then
        WorldStateScoreFrame:Show()
    elseif cmd == 'solves' then
        self:TolvirSolves()
    elseif cmd == 'screenshot' then
        Screenshot()
    elseif cmd == 'target' then
        self:GetModule('Macros'):UpdateTargetMacro(args)
    elseif cmd == 'view' then
        self:GetModule('System'):SetView(args ~= '' and tonumber(args) or false)
    elseif cmd == 'volume' then
        self:GetModule('System'):SetVolume(args ~= '' and tonumber(args) or false)
    elseif cmd == 'wago' then
        self:GetModule('Wago'):ShowWindow()
    elseif cmd == 'way' then
        self:SetUserWaypoint(args)
    end
end

function ATOM:TolvirSolves()
    local solves = 0

    for i = 1, GetNumArtifactsByRace(14) do
        solves = solves + select(10, GetArtifactInfoByRace(14, i))
    end

    local change = floor(1000 * (1 - (0.995 ^ solves))) / 10
    ATOM:Print('Tol\'vir: %s solves, (%s%% chance to have found the scepter by now.)', solves, change)
end

function ATOM:SetUserWaypoint(cmd)
    local x, y = self:GetArgs(cmd, 2)

    x = tonumber(x) * 100 / 10000
    y = tonumber(y) * 100 / 10000

    local currentMapAreaID = C_Map.GetBestMapForUnit('player')
    C_Map.SetUserWaypoint(UiMapPoint.CreateFromCoordinates(currentMapAreaID, x, y))
    C_SuperTrack.SetSuperTrackedUserWaypoint(true)
end

function ATOM:ShortenNameplateUnitName(unitId, unitFrame, envTable)
    local function abbriviateName(fullName, maxLength)
        local nameParts = { strsplit(' ', fullName) }

        for i = #nameParts, 1, -1 do
            -- dont abbreviate last names or words less than 5 characters
            if not (i == #nameParts or nameParts[i]:len() < 5) then
                nameParts[i] = nameParts[i]:sub(1, 1) .. '.'
            end

            -- stop abbriviating if the parts fit within the length of the
            -- original truncated unitName that was set for the nameplate
            if table.concat(nameParts, ' '):len() < maxLength then
                break
            end
        end

        return table.concat(nameParts, ' ')
    end

    local unitName = unitFrame.healthBar.unitName:GetText() or ''
    local maxLength = unitName:len()
    local plateFrame = C_NamePlate.GetNamePlateForUnit(unitId)
    local fullName = plateFrame and plateFrame.namePlateUnitName or unitName

    if fullName ~= unitName then
        -- ignore everything after the first comma
        fullName = select(1, strsplit(',', fullName))
        unitFrame.healthBar.unitName:SetText(abbriviateName(fullName, maxLength))
    end
end

--[[
function (self, unitId, unitFrame, envTable)
    local unitName = unitFrame.healthBar.unitName:GetText()
    local plateFrame = C_NamePlate.GetNamePlateForUnit(unitId)

    if plateFrame and plateFrame.namePlateUnitName ~= unitName then
        unitName = plateFrame.namePlateUnitName:gsub('(%S+) ', function (t) return t:sub(1, 1) .. '. ' end)
        unitFrame.healthBar.unitName:SetText(unitName)
    end
end
]]

function ATOM:CreateItemCountFrame(name)
    local frame = CreateFrame('Button', nil, UIParent)
    frame:SetClampedToScreen(true)
    frame:SetPoint('CENTER')
    frame:SetWidth(64)
    frame:SetHeight(64)
    frame:EnableMouse(true)
    frame:SetMovable(true)
    frame:RegisterForDrag('RightButton')
    frame:RegisterEvent('CHAT_MSG_LOOT');

    frame:SetScript('OnDragStop', function(self)
        self:StopMovingOrSizing()
    end)

    frame:SetScript('OnDragStart', function(self)
        self:StartMoving()
    end)

    local icon = frame:CreateTexture('OVERLAY')
    -- icon:SetTexCoord(0.08,0.92,0.08,0.92)
    icon:SetTexture(GetItemIcon(name))
    icon:SetAllPoints()

    local text = frame:CreateFontString('OVERLAY', nil, 'GameFontHighlightLarge')
    text:SetFont('Interface\\AddOns\\ATOM\\Fonts\\Lato-Bold.ttf', 23)
    text:SetPoint('BOTTOM', frame, 'BOTTOM', 0, 4)
    text:SetText(GetItemCount(name))

    frame:SetScript('OnEvent', function(self, event)
        text:SetText(GetItemCount(name))
    end);
end