// ==UserScript==
// @name         Torn War Stuff Enhanced
// @namespace    namespace
// @version      1.7.0
// @description  Show travel status and hospital time and sort by hospital time on war page. Fork of https://greasyfork.org/en/scripts/448681-torn-war-stuff
// @author       xentac
// @license      MIT
// @match        https://www.torn.com/factions.php*
// @grant        GM_addStyle
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_deleteValue
// @grant        GM_listValues
// @grant        GM_registerMenuCommand
// @grant        GM_xmlhttpRequest
// @connect      api.torn.com
// @downloadURL none
// ==/UserScript==

(async function () {
  ("use strict");

  const ffScouterV2DisableWarMonitor = document.createElement("div");
  ffScouterV2DisableWarMonitor.id = "FFScouterV2DisableWarMonitor";
  ffScouterV2DisableWarMonitor.style.display = "none";
  document.documentElement.appendChild(ffScouterV2DisableWarMonitor);

  let apiKey =
    localStorage.getItem("xentac-torn_war_stuff_enhanced-apikey") ??
    "###PDA-APIKEY###";
  const sort_enemies = true;
  let ever_sorted = false;
  const CONTENT = "data-twse-content";
  const TRAVELING = "data-twse-traveling";
  const HIGHLIGHT = "data-twse-highlight";

  try {
    GM_registerMenuCommand("Set Api Key", function () {
      checkApiKey(false);
    });
  } catch (error) {
    // This is fine, but we need to handle torn pda too
  }

  function checkApiKey(checkExisting = true) {
    if (
      !checkExisting ||
      apiKey === null ||
      apiKey.indexOf("PDA-APIKEY") > -1 ||
      apiKey.length != 16
    ) {
      let userInput = prompt(
        "Please enter a PUBLIC Api Key, it will be used to get basic faction information:",
        apiKey ?? "",
      );
      if (userInput !== null && userInput.length == 16) {
        apiKey = userInput;
        localStorage.setItem(
          "xentac-torn_war_stuff_enhanced-apikey",
          userInput,
        );
      } else {
        console.error(
          "[TornWarStuffEnhanced] User cancelled the Api Key input.",
        );
      }
    }
  }

  GM_addStyle(`
.members-list li:has(div.status[data-twse-highlight="true"]) {
  background-color: #afa5 !important;
}
`);

  GM_addStyle(`
.members-list div.status[data-twse-traveling="true"]::after {
  color: #F287FF !important;
}
`);

  GM_addStyle(`
.members-list div.status {
  position: relative !important;
  color: transparent !important;
}
.members-list div.status::after {
  content: attr(data-twse-content);
  position: absolute;
  top: 0;
  left: 0;
  width: calc(100% - 10px);
  height: 100%;
  background: inherit;
  display: flex;
  right: 10px;
  justify-content: flex-end;
  align-items: center;
}
.members-list .ok.status::after {
    color: var(--user-status-green-color);
}


.members-list .not-ok.status::after {
    color: var(--user-status-red-color);
}

.members-list .abroad.status::after, .members-list .traveling.status::after {
    color: var(--user-status-blue-color);
}
`);

  let running = true;
  let found_war = false;

  function get_faction_ids() {
    const nodes = get_member_lists();
    faction_ids = [];
    nodes.forEach((elem) => {
      const id = elem
        .querySelector(`A[href^='/factions.php']`)
        .href.split("ID=")[1];
      if (id) {
        faction_ids.push(id);
      }
    });
    return faction_ids;
  }

  function get_member_lists() {
    return document.querySelectorAll("ul.members-list");
  }

  function get_sorted_column(member_list) {
    const member_div = member_list.parentNode.querySelector("div.member div");
    const level_div = member_list.parentNode.querySelector("div.level div");
    const points_div = member_list.parentNode.querySelector("div.points div");
    const status_div = member_list.parentNode.querySelector("div.status div");

    let column = null;
    let order = null;

    let classname = "";

    if (member_div.className.match(/activeIcon__/)) {
      column = "member";
      classname = member_div.className;
    } else if (level_div.className.match(/activeIcon__/)) {
      column = "level";
      classname = level_div.className;
    } else if (points_div.className.match(/activeIcon__/)) {
      column = "points";
      classname = points_div.className;
    } else if (status_div.className.match(/activeIcon__/)) {
      column = "status";
      classname = status_div.className;
    }

    if (classname.match(/asc__/)) {
      order = "asc";
    } else {
      order = "desc";
    }

    if (column != "score" && order != "desc") {
      ever_sorted = true;
    }

    return { column: column, order: order };
  }

  // Start the dom watcher
  setTimeout(() => {
    requestAnimationFrame(watch);
    requestAnimationFrame(update_statuses);
  }, 250);

  const observer = new MutationObserver((mutations) => {
    for (const mutation of mutations) {
      for (const node of mutation.addedNodes) {
        if (node.classList && node.classList.contains("faction-war")) {
          console.log(
            "[TornWarStuffEnhanced] Observed mutation of .faction-war node",
          );
          found_war = true;
          extract_all_member_lis();
        }
      }
    }
  });

  setTimeout(() => {
    if (document.querySelector(".faction-war")) {
      console.log("[TornWarStuffEnhanced] Found .faction-war");
      found_war = true;
      extract_all_member_lis();
    }
  }, 500);

  function pad_with_zeros(n) {
    if (n < 10) {
      return "0" + n;
    }
    return n;
  }

  const wrapper = document.body; //.querySelector('#mainContainer')
  observer.observe(wrapper, { subtree: true, childList: true });

  const member_lis = new Map();

  let last_request = null;
  const MIN_TIME_SINCE_LAST_REQUEST = 9000;

  async function update_statuses() {
    if (!running) {
      return;
    }
    if (
      last_request &&
      new Date() - last_request < MIN_TIME_SINCE_LAST_REQUEST
    ) {
      requestAnimationFrame(update_statuses);
      return;
    }
    last_request = new Date();
    cache.clear_stale();
    const faction_ids = get_faction_ids();
    for (let i = 0; i < faction_ids.length; i++) {
      if (!update_status(faction_ids[i])) {
        requestAnimationFrame(update_statuses);
        return;
      }
    }
    requestAnimationFrame(update_statuses);
  }

  async function update_status(faction_id) {
    let error = false;
    const status = await fetch(
      `https://api.torn.com/faction/${faction_id}?selections=basic&key=${apiKey}&comment=TornWarStuffEnhanced`,
    )
      .then((r) => r.json())
      .catch((m) => {
        console.error("[TornWarStuffEnhanced] ", m);
        error = true;
      });
    if (error) {
      return true;
    }
    if (status.error) {
      console.log(
        "[TornWarStuffEnhanced] Received error from torn API ",
        status.error,
      );
      if (
        [0, 1, 2, 3, 4, 6, 7, 10, 12, 13, 14, 16, 18, 21].includes(status.error)
      ) {
        console.log(
          "[TornWarStuffEnhanced] Received a non-recoverable error. Giving up.",
        );
        running = false;
        return false;
      }
      if ([5, 8, 9].includes(status.error.code)) {
        // 5: Too many requests error code
        // 8: IP block
        // 9: API disabled
        // Try again in 30 + MIN_TIME_SINCE_LAST_REQUEST seconds
        console.log("[TornWarStuffEnhanced] Retrying in 40 seconds.");
        last_request = new Date() + 30000;
      }
      return false;
    }
    if (!status.members) {
      return false;
    }
    for (const [k, v] of Object.entries(status.members)) {
      v.status.description = v.status.description
        .replace("South Africa", "SA")
        .replace("Cayman Islands", "CI")
        .replace("United Kingdom", "UK")
        .replace("Argentina", "Arg")
        .replace("Switzerland", "Switz");
      cache.update_member_status(k, v);
    }
  }

  class CacheManager {
    constructor(prefix, staleness) {
      this.prefix = prefix;
      this.staleness = staleness;
    }

    make_id(id) {
      return `${this.prefix}${id}`;
    }

    update_member_status(id, data) {
      GM_setValue(
        this.make_id(id),
        JSON.stringify({ data: data, timestamp: new Date().getTime() }),
      );
    }

    extract_value(value) {
      if (value == null) {
        return null;
      }

      let data = {};
      let timestamp = 0;
      try {
        const object = JSON.parse(value);
        data = object.data;
        timestamp = object.timestamp;
      } catch (e) {
        console.log(
          `[TornWarStuffEnhanced] Error parsing cached value from ${this.make_id(id)}`,
        );
        this.delete_member_status(id);
        return null;
      }

      return { data: data, timestamp: timestamp };
    }

    get_member_status(id) {
      const value = GM_getValue(this.make_id(id), null);

      const extracted = this.extract_value(value);

      if (extracted == null) {
        return null;
      }

      const { data, timestamp } = extracted;
      if (new Date().getTime() - timestamp > this.staleness) {
        this.delete_member_status(id);
        return null;
      }

      return data;
    }

    delete_member_status(id) {
      GM_deleteValue(id);
    }

    clear_stale() {
      for (const key of GM_listValues()) {
        if (!key.startsWith(this.prefix)) {
          continue;
        }

        const extracted = this.extract_value(GM_getValue(key, null));

        if (extracted == null) {
          GM_deleteValue(key);
          continue;
        }
        const { timestamp } = extracted;
        if (new Date().getTime() - timestamp > this.staleness) {
          GM_deleteValue(key);
        }
      }
    }
  }
  const cache = new CacheManager("twse-", 30000);

  function extract_all_member_lis() {
    get_member_lists().forEach((ul) => {
      extract_member_lis(ul);
    });
  }

  function extract_member_lis(ul) {
    const lis = ul.querySelectorAll("LI.enemy, li.your");
    lis.forEach((li) => {
      const atag = li.querySelector(`A[href^='/profiles.php']`);
      if (!atag) {
        return;
      }
      const id = atag.href.split("ID=")[1];
      member_lis.set(id, li);
    });
  }

  let last_frame = 0;
  const TIME_BETWEEN_FRAMES = 500;

  function watch() {
    if (!found_war) {
      requestAnimationFrame(watch);
      return;
    }
    // Update no more frequently than every 500ms
    if (new Date() - last_frame < TIME_BETWEEN_FRAMES) {
      requestAnimationFrame(watch);
      return;
    }
    last_frame = new Date();
    member_lis.forEach((li, id) => {
      const state = cache.get_member_status(id);
      const status_DIV = li.querySelector("DIV.status");
      if (!status_DIV) {
        return;
      }
      if (!state || !running) {
        // Make sure the user sees something before we've downloaded state
        status_DIV.setAttribute(CONTENT, status_DIV.innerText);
        return;
      }
      const status = state.status;

      li.setAttribute("data-until", status.until);
      switch (status.state) {
        case "Abroad":
        case "Traveling":
          if (
            !(
              status_DIV.classList.contains("traveling") ||
              status_DIV.classList.contains("abroad")
            )
          ) {
            status_DIV.setAttribute(CONTENT, status_DIV.innerText);
            break;
          }
          if (status.description.includes("Traveling to ")) {
            li.setAttribute("data-sortA", "4");
            const content = "► " + status.description.split("Traveling to ")[1];
            li.setAttribute("data-until", content);
            status_DIV.setAttribute(CONTENT, content);
          } else if (status.description.includes("In ")) {
            li.setAttribute("data-sortA", "3");
            const content = status.description.split("In ")[1];
            li.setAttribute("data-until", content);
            status_DIV.setAttribute(CONTENT, content);
          } else if (status.description.includes("Returning")) {
            li.setAttribute("data-sortA", "2");
            const content =
              "◄ " + status.description.split("Returning to Torn from ")[1];
            li.setAttribute("data-until", content);
            status_DIV.setAttribute(CONTENT, content);
          } else if (status.description.includes("Traveling")) {
            li.setAttribute("data-sortA", "5");
            const content = "Traveling";
            li.setAttribute("data-until", content);
            status_DIV.setAttribute(CONTENT, content);
          }
          break;
        case "Hospital":
        case "Jail":
          if (
            !(
              status_DIV.classList.contains("hospital") ||
              status_DIV.classList.contains("jail")
            )
          ) {
            status_DIV.setAttribute(CONTENT, status_DIV.innerText);
            status_DIV.setAttribute(TRAVELING, "false");
            status_DIV.setAttribute(HIGHLIGHT, "false");
            break;
          }
          li.setAttribute("data-sortA", "1");
          if (status.description.includes("In a")) {
            status_DIV.setAttribute(TRAVELING, "true");
          } else {
            status_DIV.setAttribute(TRAVELING, "false");
          }

          let now = new Date().getTime() / 1000;
          if (window.getCurrentTimestamp) {
            now = window.getCurrentTimestamp() / 1000;
          }
          const hosp_time_remaining = Math.round(status.until - now);
          if (hosp_time_remaining <= 0) {
            status_DIV.setAttribute(HIGHLIGHT, "false");
            return;
          }
          const s = Math.floor(hosp_time_remaining % 60);
          const m = Math.floor((hosp_time_remaining / 60) % 60);
          const h = Math.floor(hosp_time_remaining / 60 / 60);
          const time_string = `${pad_with_zeros(h)}:${pad_with_zeros(m)}:${pad_with_zeros(s)}`;

          if (status_DIV.getAttribute(CONTENT) != time_string) {
            status_DIV.setAttribute(CONTENT, time_string);
          }

          if (hosp_time_remaining < 300) {
            status_DIV.setAttribute(HIGHLIGHT, "true");
          } else {
            status_DIV.setAttribute(HIGHLIGHT, "false");
          }
          break;

        default:
          status_DIV.setAttribute(CONTENT, status_DIV.innerText);
          li.setAttribute("data-sortA", "0");
          status_DIV.setAttribute(TRAVELING, "false");
          status_DIV.setAttribute(HIGHLIGHT, "false");
          break;
      }
    });
    if (sort_enemies) {
      // Only sort if Status is the field to be sorted
      const nodes = get_member_lists();
      for (let i = 0; i < nodes.length; i++) {
        let sorted_column = get_sorted_column(nodes[i]);
        if (!ever_sorted) {
          sorted_column = { column: "status", order: "asc" };
        }
        if (sorted_column["column"] != "status") {
          continue;
        }
        let lis = nodes[i].querySelectorAll("LI.enemy, li.your");
        let sorted_lis = Array.from(lis).sort((a, b) => {
          let left = a;
          let right = b;
          if (sorted_column["order"] == "desc") {
            left = b;
            right = a;
          }
          const sorta =
            left.getAttribute("data-sortA") - right.getAttribute("data-sortA");
          if (sorta != 0) {
            return sorta;
          }
          if (
            left.getAttribute("data-until") < right.getAttribute("data-until")
          ) {
            return -1;
          } else if (
            left.getAttribute("data-until") == right.getAttribute("data-until")
          ) {
            return 0;
          } else {
            return 1;
          }
        });
        let sorted = true;
        for (let j = 0; j < sorted_lis.length; j++) {
          if (nodes[i].children[j] !== sorted_lis[j]) {
            sorted = false;
            break;
          }
        }
        if (!sorted) {
          sorted_lis.forEach((li) => {
            nodes[i].appendChild(li);
          });
        }
      }
    }
    requestAnimationFrame(watch);
  }
  console.log("[TornWarStuffEnhanced] Initialized");

  window.dispatchEvent(new Event("FFScouterV2DisableWarMonitor"));
})();