-- TomPoints.lua
-- Parses chat channels for X,Y coordinate locations to create TomTom waypoints.
-- Heavily patterned off the ClickLinks addon.

local addonName, addon = ...

local PATTERNS = {
    "((%d+(%.?%d*)),(%d+(%.?%d*)))",
}

local zoneMapIDs = {
    ["Azj%-Kahet"]         = 2255,
    ["Siren Isle"]        = 2369,
    ["Isle of Dorn"]      = 2248,
    ["Ringing Deeps"] = 2214,
    ["The Ringing Deeps"] = 2214,
    ["Hallowfall"]        = 2215,
    ["Undermine"]         = 2346,
}

local function getCustomMapID(msg)
    for zoneName, mapID in pairs(zoneMapIDs) do
        if string.find(msg, zoneName) then
            return mapID
        end
    end
    return nil
end

local LINK_PATTERN = "[^|]|c.-|h|r";

local lastAddedWaypoint = nil;

-- formatXYLink
-- Formats the TomTom coordinates x and y into the link text format WoW needs to turn it into a clickable link.
local function formatXYLink(x, y, mapID)
    local curMap = C_Map.GetBestMapForUnit("player");
    local iconString = "";
    -- If the link is for a different zone, show an icon to distinguish it from same-zone links
    if (mapID == curMap) then
        -- Only show an icon if the user is choosing the Blizzard-style links
        if addonConfig["WhichStyle"] == "blizzard" then
            iconString = CreateAtlasMarkup("Waypoint-MapPin-ChatIcon", 13, 13);
        end
        return "|cff" .. addonConfig["LinkColor"] .. "|Hgarrmission:TomPoints:tpx:"..format("%.02f", x).."tpy:"..format("%.02f", y).."tpw:"..format("%d", mapID).." |h[" .. iconString .. x .. ", " .. y .. "]|h|r";
    else
        iconString = CreateAtlasMarkup("warlockportalalliance", 15, 15);
        return "|cff" .. addonConfig["LinkColor"] .. "|Hgarrmission:TomPoints:tpx:"..format("%.02f", x).."tpy:"..format("%.02f", y).."tpw:"..format("%d", mapID).." |h[" .. iconString .. x .. ", " .. y .. "]|h|r";
    end
end

-- formatPinLink
-- Formats the coordinates x and y into the link text format WoW needs to turn it into a clickable link for MapPins.
-- Don't use this link format because you can't easily share it for now. I have to assume they'll allow it in the future.
local function formatPinLink(x, y)
    map = C_Map.GetBestMapForUnit("player");
    --print("Map : " .. map);
    return "|cffffff00|Hworldmap:" .. map .. ":"..format("%d", tonumber(x)*100)..":"..format("%d", tonumber(y)*100).."|h[|A:Waypoint-MapPin-ChatIcon:13:13:0:0|a " .. x .. ", " .. y .. "]|h|r";
end

-- formatBlizzPinLink
-- Formats the coordinates x and y into the Blizzard map pin link format with "Map Pin Location" text.
local function formatBlizzPinLink(x, y, mapID)
    return "|cffffff00|Hworldmap:" .. mapID .. ":"..format("%d", tonumber(x)*100)..":"..format("%d", tonumber(y)*100).."|h[|A:Waypoint-MapPin-ChatIcon:13:13:0:0|a Map Pin Location]|h|r";
end

-- addTomTomWaypoint
-- Helper function to add the waypoint in TomTom
-- coordXFromLink - The X coordinate from the link, this is not in units that TomTom expects.
-- coordYFromLink - The Y coordinate from the link, this is not in units that TomTom expects.
-- TODO: a custom icon would be nice to distinguish from normal waypoints, since these may be temporal in nature, like rares in Nazjatar, Mechagon, etc.
local function addTomTomWaypoint(coordXFromLink, coordYFromLink, mapIDFromLink, text)
    if (TomTom) then
        -- Remove the previous waypoint if it exists
        if lastAddedWaypoint then
            TomTom:RemoveWaypoint(lastAddedWaypoint)
            lastAddedWaypoint = nil
        end
        -- Add the new waypoint and store it for future updates
        lastAddedWaypoint = TomTom:AddWaypoint(tonumber(mapIDFromLink), tonumber(coordXFromLink)/100, tonumber(coordYFromLink)/100, {
            title = text,
            persistent = nil,
            minimap = true,
            world = true
        })
    else
        --print("TomTom not loaded!")
    end
end

-- addUiPinWaypoint
-- Helper function to add the waypoint using the new Pin system
-- coordXFromLink - The X coordinate from the link, this is not in units that WoW expects.
-- coordYFromLink - The Y coordinate from the link, this is not in units that WoW expects.
-- quiet          - should this function not announce it's creating a pin? since these may be temporal in nature, like rares in Nazjatar, Mechagon, etc.
local function addUiPinWaypoint(coordXFromLink, coordYFromLink, mapIDFromLink, quiet)
    map = tonumber(mapIDFromLink);
    if C_Map.CanSetUserWaypointOnMap(map) then
        local waypoint = UiMapPoint.CreateFromCoordinates(map, tonumber(coordXFromLink)/100, tonumber(coordYFromLink)/100)
        C_Map.SetUserWaypoint(waypoint);
        C_SuperTrack.SetSuperTrackedUserWaypoint(true); -- Ensure the map pin is automatically selected
        if not(quiet) then
            if addonConfig["ActivateMapPin"] then
                C_SuperTrack.SetSuperTrackedUserWaypoint(true);
                print("TomPoints created and activated Map Pin: " .. formatXYLink(tonumber(coordXFromLink), tonumber(coordYFromLink), map) .. " in " .. C_Map.GetMapInfo(map).name);
            else
                print("TomPoints created Map Pin: " .. formatXYLink(tonumber(coordXFromLink), tonumber(coordYFromLink), map) .. " in " .. C_Map.GetMapInfo(map).name);
            end
        end
        -- Automatically click the map pin
        C_Timer.After(0.1, function()
            if C_Map.SetUserWaypointHyperlink then
                C_Map.SetUserWaypointHyperlink(C_Map.GetUserWaypointHyperlink());
            else
                --print("C_Map.SetUserWaypointHyperlink is not available in this version of WoW.");
            end
            C_SuperTrack.SetSuperTrackedUserWaypoint(true); -- Ensure the map pin is automatically selected
        end)
    else
        print("TomPoints can't create Map Pin in this zone");
    end
end

-- TomPoints_SetItemRef
-- Handles when the user clicks on the link
local function TomPoints_SetItemRef(link, text, button)
    -- print("link: " .. link .. " text: " .. text .. " button: " .. button);
    local linkType, addon, param1 = strsplit(":", link)
    --print("Type: " .. linkType);
    --if (linkType == "worldmap" and IsModifiedClick("CHATLINK") and ChatEdit_InsertLink(C_Map.GetUserWaypointHyperlink())) then
    --    print("Pasted in link");
    --else
    if (linkType == "garrmission" and addon == "TomPoints") then
        local x, y, m = strmatch(link, "tpx:(%d+%.%d+)tpy:(%d+%.%d+)tpw:(%d+)")
        if (x and y and m) then
            map = tonumber(m);
            -- If the person is trying to link the waypoint like it's an achievement, attempt to add to chat
            -- Otherwise, add the waypoint like normal if the chat edit area isn't open (maybe the shift was down by accident)
            linkAdded = false; 
            if IsModifiedClick("CHATLINK") then
                if (addonConfig["ShareAsPin"]) then -- share map pins instead of raw text
                    if C_Map.CanSetUserWaypointOnMap(map) then
                        curPin = C_Map.GetUserWaypoint();
                        -- Have to create the pin to be able to share it
                        addUiPinWaypoint(x, y, map, true);
                        linkAdded = ChatEdit_InsertLink(C_Map.GetUserWaypointHyperlink());
                        if curPin then
                            C_Map.SetUserWaypoint(curPin);
                        end
                    else
                        print("TomPoints can't create Map Pin in this zone; can't share as map pin");
                    end
                end
                if not(linkAdded) then
                    linkAdded = ChatEdit_InsertLink(" " .. x .. ", " .. y .. " ");
                end
            end
            if not(linkAdded) then
                if (TomTom) then
                    addTomTomWaypoint(x, y, m, text);
                    if (addonConfig["AlwaysCreateMapPin"]) then
                        addUiPinWaypoint(x, y, m);
                    end
                else
                    addUiPinWaypoint(x, y, m);
                end
            end
        else
          print("X or Y coordinate is invalid");
        end
    end
end
-- hooksecurefunc("ChatFrame_OnHyperlinkShow", TomPoints_SetItemRef); -- Doesn't work because WoW's code doesn't know how to handle the custom link type
 -- Doesn't work because WoW's code doesn't know how to handle the custom link type



-- findLinks
-- Find the location of links in the message text, or nil if none found. Returns an array of begin and end index pairs, so
-- the length will always be number of links * 2
-- This makes the potentially ignorant assumption that every link begins with |c and ends with |h|r
-- This is also required, since some achievements, quests, items, etc. can contain text that looks like a waypoint, and will trick the parser
-- into turning it into a waypoint link. The Achievement, "10,000 World Quests Completed" is one such link.
-- msgText - the full message text, including being modified from this script itself, so the new links are included in the
-- list of links, so they don't get double-replaced.
-- returns the list indices indicating where the links exist in the string or nil if no links are found.
local function findLinks(msgText)
  local result = {}
  local lastIdx = 1;
  -- Append the space at the beginning so people can't unknowingly trick the code into thinking there's a link
  -- this only matters at the beginning of the string, Lua's pattern parsing stuff isn't that robust.
  local beginI, endI = string.find(" " .. msgText, LINK_PATTERN, lastIdx);
  local tableIdx = 0;
  while (beginI) do
    -- Subtract 1 for the space we appended above.
    result[tableIdx] = beginI - 1;
    tableIdx = tableIdx + 1;
    -- Subtract 1 for the space we appended above.
    result[tableIdx] = endI - 1;
    tableIdx = tableIdx + 1;
    --print ("Found link starting at " .. (beginI - 1) .. " ending at: " .. (endI - 1));
    -- Subtract 1 for the space we appended above.
    lastIdx = endI - 1;
    -- Append the space at the beginning so people can't unknowingly trick the code into thinking there's a link
    -- this only matters at the beginning of the string, Lua's pattern parsing stuff isn't that robust.
    beginI, endI = string.find(" " ..msgText, LINK_PATTERN, lastIdx)
  end
  if ((table.getn(result) or 0) > 0) then
    return result;
  else
    return nil;
  end
end

-- isInsideLink
-- Determines if the given index is within a link found in the message.
-- linkTable - the array of links created from findLinks (this can be nil)
-- startIndex - the starting index of the text to determine if it exists in a link.
local function isInsideLink(linkTable, startIndex)
  if (linkTable) then
    for i = 0, #(linkTable), 2 do
      --print ("Link i " .. i .. " = begin: " .. linkTable[i] .. " end: " .. linkTable[i + 1]);
      if (startIndex >= linkTable[i] and startIndex <= linkTable[i+1]) then
        --print ("Found that match starting at " .. startIndex .. " was inside a link, ignoring, hopefully!")
        -- return the end index of the match
        return linkTable[i+1] + 1;
      end
    end
  else
    return nil;
  end
end

-- doReplaceLink
-- Replaces the matching waypoint text with the clickable waypoint link.
-- This is favorable over gsub, since gsub only replaces patterns, without an idea of where to start replacing from.
-- msgText - the full message text
-- replaceStartIdx - the index where to start replacing text
-- replaceEndIdx - the index where to stop replacing text
-- returns the message text now with the waypoint link instead of the text that matched.
local function doReplaceLink(msgText, replaceStartIdx, replaceEndIdx, replaceWithText)
  local preText, postText;
  preText = strsub(msgText, 0, replaceStartIdx);
  postText = strsub(msgText, replaceEndIdx);
  return preText .. replaceWithText .. postText;
end

-- replaceBlizzPinLink
-- Replaces Blizzard's Map Pin link with a TomPoints link, so we can share it like other waypoint links
local function replaceBlizzPinLink(msgText)
    -- Use the [^%]]+ to match pin links in (hopefully) any language
    local matchPattern = "|cffffff00|Hworldmap:(%d+):(%d+):(%d+)|h%[|A:Waypoint%-MapPin%-ChatIcon:13:13:0:0|a [^%]]+%]|h|r";
    local stopCount = 0;
    local searchFromIdx = 1;
    local replaceBegin, replaceEnd = strfind(msgText, matchPattern, searchFromIdx);
    while replaceBegin and stopCount < 50 do
        --print("Begin: " .. tostring(replaceBegin) .. " end: " .. tostring(replaceEnd));
        stopCount = stopCount + 1;
        local linkText = msgText:sub(replaceBegin, replaceEnd);
        local mapID, xCoord, yCoord = strmatch(linkText, matchPattern, 1);
        if mapID ~= nil and xCoord ~= nil and yCoord ~= nil then
            xCoordFloat = tonumber(xCoord) / 100.;
            yCoordFloat = tonumber(yCoord) / 100.;
            --print("Found link: with xCoord: " .. xCoordFloat .. " yCoord: " .. yCoordFloatStr);
            local replacementString = formatXYLink(xCoordFloat, yCoordFloat, tonumber(mapID));
            msgText = doReplaceLink(msgText, replaceBegin - 1, replaceEnd + 1, replacementString);
            searchFromIdx = searchFromIdx + replacementString:len();
            -- Automatically click the map pin
            C_Timer.After(0.1, function()
                if C_Map.SetUserWaypointHyperlink then
                    C_Map.SetUserWaypointHyperlink(C_Map.GetUserWaypointHyperlink());
                else
                    print("C_Map.SetUserWaypointHyperlink is not available in this version of WoW.");
                end
                C_SuperTrack.SetSuperTrackedUserWaypoint(true); -- Automatically set to the detected pin
            end)
            --print("Replaced, msgText now: " .. msgText);
        else
            --print("Coords were nil");
            -- nudge it forward
            searchFromIdx = searchFromIdx + 1;
        end
        replaceBegin, replaceEnd = strfind(msgText, matchPattern, searchFromIdx);
    end
    return msgText;
end

-- processMessage
-- Helper function that does all the message processing.
-- msgText - the full message text to process.
-- returns the msgText whether it has been processed or not, if it has been processed, and a location is found, the resulting text
-- will include the links that will be display in the chat window.
local function processMessage(msgText)
  -- If you want to see the full, un-decorated text, uncomment:
    -- print("Here's what it really looks like: \"" .. gsub(msgText, "\124", "\124\124") .. "\"");
    if addonConfig["ReplaceBlizzardLinks"] then
        msgText = replaceBlizzPinLink(msgText);
    end
    local customMapID = getCustomMapID(msgText);
    local mapID = customMapID or C_Map.GetBestMapForUnit("player");
    local stopCount = 0;
    for key, val in pairs(PATTERNS) do
      -- Attempt to limit the function so it doesn't potentially infinitely-loop
      stopCount = 0;
      local searchFromIdx = 1;

      local startIndex = string.find(msgText, val, searchFromIdx);
      -- print ("Testing string starting from " .. searchFromIdx .. " : " .. strsub(msgText, searchFromIdx))
      while startIndex and stopCount < 50 do
        --print("message len: " .. tostring(msgText:len()) .. " text: " .. msgText);
        stopCount = stopCount + 1;
        local linkLocs = findLinks(msgText);
        local nextSafeIdx = isInsideLink(linkLocs, startIndex);
        if (not(nextSafeIdx)) then
          -- Move the searchFromIdx up so if we replace a pair in a previous loop, this index doesn't lag so far behind
          -- if one of the bad matches below, that changes searchFromIdx is hit
          -- testing with "50 50 test2 50 50" resulted in one link created only instead of 2 without this change
          searchFromIdx = startIndex;
          -- msgText = string.gsub(msgText, val, formatURL("%1"))
          local match1, match2, match3, match4, match5 = strmatch(msgText, val, searchFromIdx)
          if match2 and match4 then
            local match2Value = tonumber(match2);
            local match4Value = tonumber(match4);
            -- Need to verify that coordinates are valid.
            -- Without this, values greater than 100 would be matched, or equal to 0 would match.
            if (match2Value > 0 and match2Value < 100  and match4Value > 0 and match4Value < 100) then
              local replaceBegin, replaceEnd = strfind(msgText, val, searchFromIdx);
              -- Ignore percentages
              -- The only place the percent sign could be is after the last match
              -- Enforcer KX-T57 announcements can cause a match to fire
              -- Like: Enforcer KX-T57 98% ...
              if (msgText:sub(replaceEnd+1, replaceEnd+1) == "%") then
                searchFromIdx = searchFromIdx + match2:len();
                --print("1 Skipping 1: " .. match1 .. " 2: " .. match2);
              elseif (replaceBegin > 1 and tonumber(msgText:sub(replaceBegin-1, replaceBegin-1))) then
                -- This means the first match would be invalid, the pattern matched on a message like:
                -- 123 45, but found match2 = 23 and match 4 is 45
                searchFromIdx = searchFromIdx + match2:len();
                --print("2 Skipping 1: " .. match1 .. " 2: " .. match2);
              elseif (replaceBegin > 1 and msgText:sub(replaceBegin-1, replaceBegin-1):find("%a")) then
                -- This means the first match would be invalid, the pattern matched on a message like:
                -- Enforcer KX-T57 54 98.3, but found match2 = 57 and match 4 is 54 instead of 54 98.3
                searchFromIdx = searchFromIdx + match2:len();
                --print("3 Skipping 1: " .. match1 .. " 2: " .. match2 .. " limit: " .. tostring(stopCount) .. " searchFromIdx: " .. tostring(searchFromIdx));
              else
                -- Replace the text with a Blizzard map pin link for display with "Map Pin Location"
                msgText = doReplaceLink(msgText, replaceBegin - 1, replaceEnd + 1, formatBlizzPinLink(match2Value, match4Value, mapID));
                -- Automatically update TomTom waypoint with the new coordinates
                local messageWithoutCoordinates = msgText:gsub(val, "")
                addTomTomWaypoint(match2Value, match4Value, mapID, messageWithoutCoordinates);
                -- Always create a map pin if a custom zone is specified
                local zoneToUse = customMapID or mapID
                addUiPinWaypoint(match2Value, match4Value, zoneToUse, true);
              end
            else
              -- Move past the first token of the invalid match
              searchFromIdx = searchFromIdx + match2:len();
            end

          end
          --local preSI = startIndex
          startIndex = string.find(msgText, val, searchFromIdx);
          --print("changing startIndex, pre: " .. tostring(preSI) .. " post: " .. tostring(startIndex));
        else
          -- print ("Not safe index, moving to: " .. nextSafeIdx .. " stopCount: " .. stopCount);
          searchFromIdx = nextSafeIdx;
          startIndex = string.find(msgText, val, searchFromIdx);
        end
      end
    end
    return msgText;
end

-- Override the default print function to process zone names and coordinates
local originalPrint = print
local suppressPrint = false -- Prevent recursive prints

print = function(...)
    if suppressPrint then
        return -- Avoid recursive calls
    end

    local args = {...}
    for i, msg in ipairs(args) do
        if type(msg) == "string" then
            -- Process the message to replace coordinates with clickable links
            msg = processMessage(msg)
        end
        args[i] = msg
    end
    originalPrint(unpack(args))

    -- Automatically click the map pin if a waypoint is detected
    C_Timer.After(0.1, function()
        if C_Map.SetUserWaypointHyperlink then
            C_Map.SetUserWaypointHyperlink(C_Map.GetUserWaypointHyperlink())
        else
            suppressPrint = true -- Suppress recursive prints
            --originalPrint("C_Map.SetUserWaypointHyperlink is not available in this version of WoW.")
            suppressPrint = false
        end
        C_SuperTrack.SetSuperTrackedUserWaypoint(true) -- Automatically set to the detected pin
    end)
end

-- Ensure addonConfig is initialized to prevent nil errors
addonConfig = addonConfig or {
    AlwaysCreateMapPin = true,
    ActivateMapPin = true,
    ShareAsPin = true,
    ReplaceBlizzardLinks = false,
    LinkColor = "149bfd",
    WhichStyle = "blizzard"
}

--local lastEvent = nil
-- TODO: throttle parsing. This gets called multiple times per message, I'm sure that can't be good for performance reasons.
local function tomPoints_OnEvent(self, event, msg,...)
  --if (lastEvent ~= event) then
    msg = processMessage(msg);
    local channel = select(3, ...)
    if channel == "PARTY" or channel == "RAID" then
        C_Timer.After(0.1, function()
            if C_Map.SetUserWaypointHyperlink then
                C_Map.SetUserWaypointHyperlink(C_Map.GetUserWaypointHyperlink());
            else
                print("C_Map.SetUserWaypointHyperlink is not available in this version of WoW.");
            end
            C_SuperTrack.SetSuperTrackedUserWaypoint(true);
        end)
    end
  return false, msg, ...
end

local CHANNELS = {
    "WHISPER",
    "PARTY",
    "PARTY_LEADER",
    "RAID",
    "RAID_LEADER"
}
-- print("loaded")
for _, type in pairs(CHANNELS) do
    ChatFrame_AddMessageEventFilter("CHAT_MSG_" .. type, tomPoints_OnEvent)
end
--ChatFrame_AddMessageEventFilter("CHAT_MSG_CHANNEL", tomPoints_OnEvent)
--ChatFrame_AddMessageEventFilter("CHAT_MSG_SAY", tomPoints_OnEvent)

local frame = CreateFrame("Frame")
frame:RegisterEvent("VARIABLES_LOADED")

frame:SetScript("OnEvent", function(self, event, arg1)
    if event == "VARIABLES_LOADED" then
        -- Our saved variables, if they exist, have been loaded at this point.
      if addonConfig == nil then
        -- This is the first time this addon is loaded; set SVs to default values
        addonConfig = {};
      end
      if addonConfig["AlwaysCreateMapPin"] == nil then
        addonConfig["AlwaysCreateMapPin"] = true; -- Set to true for automatic creation of map pins
      end
      if addonConfig["ActivateMapPin"] == nil then
        addonConfig["ActivateMapPin"] = true;
      end
      if addonConfig["ShareAsPin"] == nil then
        addonConfig["ShareAsPin"] = true;
      end
      if addonConfig["ReplaceBlizzardLinks"] == nil then
        addonConfig["ReplaceBlizzardLinks"] = false;
      end
      if addonConfig["LinkColor"] == nil then
        addonConfig["LinkColor"] = "149bfd";
      end
      if addonConfig["WhichStyle"] == nil then
        addonConfig["WhichStyle"] = "blizzard"; -- Change to "blizzard" for Blizzard-style map pins
      end
      frame:UnregisterEvent("VARIABLES_LOADED");
    end
end)

--- Configuration/testing functions
local function printUsage(msg)
  print("/tompoints - Print this message");
  print("/tompoints_test, /tptest - Test a string to see how TomPoints parses it, the result will be printed to the chat window.");
  print("/tompoints_removelast, /tpremove_last, /tprl - Removes from TomTom the last waypoint that was clicked on, this isn't remembered between UI reloads, or between logins.");
end

SLASH_TOMPOINTS1 = "/TOMPOINTS";
SLASH_TOMPOINTSTEST1 = "/TOMPOINTS_TEST";
SLASH_TOMPOINTSTEST2 = "/TPTEST";
SLASH_TOMPOINTSREMOVELAST1 = "/TOMPOINTS_REMOVELAST";
SLASH_TOMPOINTSREMOVELAST2 = "/TPREMOVE_LAST";
SLASH_TOMPOINTSREMOVELAST3 = "/TPRL";

SlashCmdList["TOMPOINTS"] = printUsage
SlashCmdList["TOMPOINTSTEST"] = function(msg)
  print("TomPoints test result: " .. processMessage(msg or ""));
end
SlashCmdList["TOMPOINTSREMOVELAST"] = function(msg)
  if (lastAddedWaypoint) then
    TomTom:RemoveWaypoint(lastAddedWaypoint);
    lastAddedWaypoint = nil;
  end
end