The Syndicate

Code: Select all

// ==UserScript==
// @name         Secret Playerlist Vote Counter
// @namespace    mailto:luminouslag@gmail.com
// @version      1.0
// @description  Automatic Vote Counting for Secret Plyerlist Mafia game on The Syndicate
// @author       Lumi
// @match        https://mafiathesyndicate.com/viewtopic.php?*
// @icon         https://www.google.com/s2/favicons?sz=64&domain=mafiathesyndicate.com
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_setClipboard
// ==/UserScript==

console.log('Script Initialized');

async function getDocument(thread, start) {
  return await window.fetch(`https://mafiathesyndicate.com/viewtopic.php?t=${thread}&start=${start}`).then(r => r.text()).then(html => (new DOMParser()).parseFromString(html, 'text/html'));
}

function getDataFromPost(post) {
  let postNum = post.getElementsByClassName('post-number')[0].textContent.match(/[0-9]+/g)[0];
  let contentChildren = post.getElementsByClassName('content')[0].children;
  let author = post.getElementsByClassName('author')[1].getElementsByTagName('a')[1].textContent;
  let currentVoteTarget = null;
  for (const contentChild of contentChildren) {
    if (contentChild.classList.contains('mention')) {
      let voteTarget = contentChild.textContent.replaceAll('\n','').matchAll(/\[VOTE: ([^\]]*)\]/g).next().value;
      if (voteTarget) currentVoteTarget = voteTarget[1];
    }
  }
  return {
    number: postNum,
    target: currentVoteTarget,
    author: author,
  };
}

function isCorrectThread(targetNum) {
  let topicNum = document.getElementsByClassName("topic-title")[0].children[0].href.matchAll(/\?t=([0-9]+)&/g).next().value[1];
  return topicNum == targetNum;
};

function getNumPosts() {
  return parseInt(document.getElementsByClassName('pagination')[0].textContent.matchAll(/([0-9]*) posts/g).next().value[1]);
};

function getFirstPostNumOfThisPage() {
  return parseInt(document.getElementsByClassName('post')[0].getElementsByClassName('post-number')[0].textContent.match(/[0-9]+/g)[0]);
}

function tallyPage(voteHistory, page) {
  let posts = page.getElementsByClassName('post');
  for (const post of posts) {
    let postData = getDataFromPost(post);
    if (parseInt(postData.number) == voteHistory.length + 1) {
      voteHistory.push(postData);
    }
  }
  return voteHistory;
}

function handleKeyUp(e) {
    let isNum = (e.keyCode >= 48 && e.keyCode <= 57);
    let isC = (e.keyCode == 67);
    let isR = (e.keyCode == 82);
    let isX = (e.keyCode == 88);
    if (isC && e.altKey) {
      let currentDay = 1;
      for(let i=1; i<=10; i++) {
        if (GM_getValue(`EoD${i}`, 0) != 0) currentDay++;
      }
      let voteCount = calcVoteCount(currentDay);
      GM_setClipboard(voteCount);
    } else if (isR && e.altKey) {
      let day = parseInt(prompt('Enter Day Number'));
      let voteCount = calcVoteCount(day);
      GM_setClipboard(voteCount);
    } else if ( isNum && e.altKey) {
      let num = parseInt(e.key);
      if (num == 0) num = 10;
      let postNum = prompt(`Enter Post Number for EoD ${num}`)
      GM_setValue(`EoD${num}`, parseInt(postNum));
    } else if ( isX && e.altKey) {
      window.location.href = 'https://www.youtube.com/watch?v=dQw4w9WgXcQ';
    }
};

function calcVoteCount(day) {
  let SoD = GM_getValue(`EoD${day-1}`, 0);
  let EoD = GM_getValue(`EoD${day}`, 0);
  voteHistory = JSON.parse(GM_getValue(`voteHistory`), '[]');

  let voteTargets = {};
  let postCounts = {};

  for (const vote of voteHistory) {
    if ((vote.number >= SoD || SoD == 0) && (vote.number <= EoD || EoD == 0)) {
      if (!(vote.author in voteTargets)) {
        voteTargets[vote.author] = null;
        postCounts[vote.author] = 1;
      } else {
        postCounts[vote.author] += 1;
      }
      if (vote.target != null) voteTargets[vote.author] = [vote.target, vote.number];
    }
  }

  let voteCount = {};

  for (const [key, value] of Object.entries(voteTargets) ) {
    if (value != null) {
      let target = value[0].toLowerCase().trim();
      if (target in voteCount) {
        voteCount[target].push([key, value[1], postCounts[key]]);
      } else {
        voteCount[target] = [[key, value[1], postCounts[key]]];
      }
    } else {
      if ('Not Voting' in voteCount) {
        voteCount['Not Voting'].push([key, null, postCounts[key]]);
      } else {
        voteCount['Not Voting'] = [[key, null, postCounts[key]]];
      }
    }
  }

  return voteCountToString(voteCount);
}

function voteToString(vote) {
  if (vote[1] == null) return `${vote[0]} (${vote[2]})`
  return `[url=https://mafiathesyndicate.com/viewtopic.php?t=2451&start=${vote[1]-1}]${vote[0]}[/url] (${vote[2]})`
}

function voteCountToString(voteCount) {
  let output = '';
  for (const [key, value] of Object.entries(voteCount)) {
    let valueStrings = [];
    for (const vote of value) {
      valueStrings.push(voteToString(vote));
    }
    output += `${key} (${value.length}): ${valueStrings.join(', ')}\n`
  }
  return output;
}

if (isCorrectThread(2451)) {
  let voteHistory = JSON.parse(GM_getValue(`voteHistory`, '[]'));
  let numPosts = getNumPosts();
  let curVoteHistoryLength = voteHistory.length;
  while (voteHistory.length < numPosts) {
    let page = await getDocument(2451, voteHistory.length);
    voteHistory = tallyPage(voteHistory, page);
    if (voteHistory.length == curVoteHistoryLength) break;
  }
  GM_setValue(`voteHistory`, JSON.stringify(voteHistory));

  document.addEventListener('keyup', handleKeyUp, false);
};