// ==UserScript==
// @name			  AO3 author+tags quick-search
// @version		  1.2
// @include		  https://archiveofourown.org/works*
// @include		  https://archiveofourown.org/chapters*
// @description Generates quick links from AO3 fics to more by the same author in the same fandom (or character/pairing/any other tag). 
// @namespace   rallamajoop
// @downloadURL none
// ==/UserScript==

/* Adds extra links to the tags section at the top of an AO3 work page, redirecting user to any use of same tags by the same author.
* ie. a quick way to find out if the writer of the great fic you just read has written anything else for the same fandom/pairing/trope/etc.

* If printCounts is set to 'true', code will also calculate and display the number of uses of each tag in the fandom/relationship/character sections.
* This will also make the links take longer to load, as code must pre-load each link individually. To limit load, I have not implemented this for freeform 
* or other tags.

* Author-based search links are a bit of a hack, but most cases should now select correctly in the tags menu on the right. I've added a subheading identifing the tag to cover all bases.
*/

//Setting variables! Change these if you like
var printCounts = true; 		// setting this to false will prevent script for printing tag counts, which may decrease loading times and server load
var quickSearchText = "*"; 	// text for new links - change this if you want something other than an *. You can put <sup>TEXT</sup> to make it superscript
var autoSortResultsBy = ""; // valid options are "Kudos", "Word Count", "Date" or "" (none, usually defaults to date)
var importantTags = ["fandom","relationship","character"]; // these are the tags which the script prints counts for

//Proper code starts here
var loc = location.href;
var href="https://archiveofourown.org/works?utf8=%E2%9C%93&commit=Sort+and+Filter&work_search"; //[relationship_names]=";
var workPage=["https://archiveofourown.org/works/", "https://archiveofourown.org/chapters/"];
var scriptTag = "&greasemonkey"; //tag to flag that we've clicked an author quicktag link

var sortOpts = {"Kudos": "&work_search[sort_column]=kudos_count", "Word Count": "&work_search[sort_column]=word_count", "Date" : "&work_search[sort_column]=revised_at" };

if (isWorkPage(loc)) {
  // Story page, add extra links user-based search to each tag

  var h3=document.body.getElementsByClassName("byline heading");
  var authors=h3.item(0).getElementsByTagName('a');
  authors=parseAuthors(authors);

  var allTags=document.body.getElementsByClassName("work meta group").item(0);
  var dds = allTags.getElementsByTagName('dd');

  var links = dds.item(1).getElementsByTagName('a');

  for (var j=0; j<dds.length; j++) {
    var tags = dds.item(j);
    var tagType = getTagType(tags);
    if (!tagType) break;

    links = tags.getElementsByTagName('a');
    for (var i=links.length-1; i>=0; i--) {

      for (var a=0; a<authors.length; a++) {

        var author=authors[a];
        var tag = getTag(links.item(i).href);
        var newHref = buildSearch(tag, author, tagType);
        
        var isCreatedLink = tag.includes(links.item(i).href);
        var existing = document.querySelectorAll('[href="' + newHref + '"]');
        if (!isCreatedLink && existing.length==0) { //check that we aren't creating a duplicate link - can happen when page partially loads from cache 

          var text = quickSearchText;
          var newlink=document.createElement('span');
          
          if (printCounts && printCountFor(tags) && !ignoreAuthor(author)) {
            //If this is an important tag, count search results
            var count = getNumFics(newHref);
            if (count>1) {
              text = "<b><sup>(" + count.toString() + ")<sup></b>";
            }
            else {
              text = "<sup>(1)<sup></b>";
            }
          }

          var toolTip = "'" + unformatTag(tag) + "' fic by " + author;
          newlink.innerHTML = " <a href=\"" + newHref + "\" title=\"" + toolTip + "\">" + text + "</a>";

          links.item(i).parentNode.appendChild(newlink);
        }
      }
    }
  }
}


else if (loc.includes(scriptTag)) {
  // Quicktags search results page, add title to show which tag is in use

  //var sTag = getBetween("work_search[relationship_names]=", loc, "&user_id=");
  let sTag, tagType;
  [tagType, sTag] = parseLoc(loc);
  var main = document.getElementById("main");
  var heading = main.getElementsByClassName("heading").item(0);

  sTag = unformatTag(sTag);
  
  var subTitle = "<br> Author QuickSearch: " + sTag;
  var span=document.createElement('h3');
  span.innerHTML = subTitle;
  
//  const sortBys = sortLinks(loc);
//  heading.parentNode.insertBefore(sortBys, heading.nextSibling); //not really necessary as of v1.2
  heading.parentNode.insertBefore(span, heading.nextSibling);

  checkTagBox(tagType, sTag);
}

//don't bother counting fics by orphan_account - these won't all be by same author anyway
function ignoreAuthor(author) {
  if (author == "orphan_account") return true;
  return false;
}

function getTagType(element) {
  var name = element.className;
  if (!name.includes(" tags")) return false;
  name = name.replace(" tags", "");
  return name;
}

//True if work or chapter
function isWorkPage(href) {
  for (var i=0; i<workPage.length; i++) {
    if (href.includes(workPage[i])) return true;
  }
  return false;
}

//Make sure tag is selected on the right menu
function checkTagBox(tagType, tag) {
  if (tagType == "warning") tagType = "archive_warning";
  debugger
/*  let btn = document.getElementById("toggle_include_" + tagType + "_tags");
  btn = btn.getElementsByTagName("button");
  btn = btn[0];
  if (btn.getAttribute("aria-expanded") == "false") {
    btn.click();
  }*/
  const par = document.getElementById("include_" + tagType + "_tags");
  const entries = par.getElementsByTagName("label");
  for (let i=0; i<entries.length; i++) {
    let entry = entries[i];
    if (entry.textContent.includes(tag)) {
      entry.getElementsByTagName("input")[0].checked=true;
      return;
    }
  }
}


//Generate links to sort by kudos etc
function sortLinks(urlStr) {
  var sortBys=document.createElement('div');
  
  let baseUrl = urlStr.replace(scriptTag, "");
  let keys = Object.keys(sortOpts);
  let inc = [true, true, true];
  for (var i =0; i<keys.length; i++) {
    let opt = sortOpts[keys[i]];
    if (baseUrl.includes(opt)) {
      baseUrl = baseUrl.replace(opt, "");
      inc[i] = false;
    }
  }

  for (var i =0; i < keys.length; i++) {
    if(inc[i]) {
      let link=document.createElement('a');
      let key=keys[i];
      link.href=baseUrl + sortOpts[key] + scriptTag;
      link.innerHTML = "Sort by " + key;
      sortBys.append("\u2003");
      sortBys.appendChild(link);
    }
  }
  
  return sortBys;
}

//Author+tag search address
function buildSearch(tag, author, tagType="relationship"){
	var href="https://archiveofourown.org/works?utf8=%E2%9C%93&commit=Sort+and+Filter&work_search[" + tagType + "_names]="
  href+=tag + "&user_id=" + author;
  
  const sortBy = sortOpts[autoSortResultsBy];
  if (sortBy != undefined) href+= sortBy;

  href+=scriptTag;
  return href;
}

//Limit tag counts to fandom, relationship and character tags
function printCountFor(element) {
  var name = getTagType(element);
  for (var i=0; i<importantTags.length; i++) {
    if (importantTags[i] == name) return true;
  }
  return false;

}

//Retrieve page as javascript object
function getSourceAsDOM(url)
{
    var xmlhttp=new XMLHttpRequest();
    xmlhttp.open("GET",url,false);
    xmlhttp.send();
    var parser=new DOMParser();
    return parser.parseFromString(xmlhttp.responseText,"text/html");
}

//Counts results of author+tag search
function getNumFics(url) {
  var results = getSourceAsDOM(url);
  var works = results.getElementsByClassName("work index group")[0];

  if (works.children.length < 20) {
	  return works.children.length;
  }

  var h=results.getElementsByTagName("h2");
  if (h.length>0) {
    h=h[0];
	  var text = h.textContent;
    if (text.includes(" of ")) {
      text = getBetween(" of ", text, " Works");
      return text; //could convert this into an integer, but not much point
    }
  }
	return 20;
}

//May be multiple authors, may have pseudonyms. Returns as array
function parseAuthors(links) {
  var authors = [""];
  for (var i=0; i<links.length; i++) {
    var text = links.item(i).toString();
		authors[i] = getBetween("/users/", text, "/pseuds");
  }
  return authors;
}

//Retrieve tag and tag type from search string
function parseLoc(href) {
  const re=/.+&work_search\[(.+)_names\]=(.+)&user_id=.+/;
  const res = re.exec(href);
  if (res.length == 3) {
    return res.slice(1);
  }
  return ["",""];
}

//Format tag to pass to search
function getTag(href){
  var tag = getBetween("/tags/", href, "/works");
  tag = replaceAll(tag, "*s*","%2F");
  tag = replaceAll(tag, "*d*",".");
  tag = replaceAll(tag, "%20","+");
  return tag;
}

//Format tag from href for display in title
function unformatTag(tag) {
  tag = decodeURIComponent(tag);
  tag = replaceAll(tag, "*s*","/");
  tag = replaceAll(tag, "*a*","&");
  tag = replaceAll(tag, "*d*",".");
  tag = replaceAll(tag, "+", " ");
  return tag;
}


function getBetween(tag1, str, tag2) {
	var ret = str.split(tag1);
  if (ret.length<2) return "";
  ret=ret[1];
	ret = ret.split(tag2);
  if (ret.length<1) return "";
  ret = ret[0];
  return ret;
}

function replaceAll(str, search, replacement) {
	return str.split(search).join(replacement);
}