// ==UserScript==
// @version       1.9.0
// @include       *.fanfiction.net/s/*
// @namespace     ffnet
// @name          Fanfiction.net story export script.
// @author        Alssn
// @description   Writes all chapters of the story on one page.
// @require		  http://ajax.googleapis.com/ajax/libs/jquery/1.10.2/jquery.min.js
// @grant         GM.setClipboard
// @grant         GM_xmlhttpRequest
// @connect       fanfiction.net
// @downloadURL none
// ==/UserScript==

//Chapters are downloaded with a delay to prevent ff.net from limiting connections
var delay = 500;

var chapters=[];

var style = $("<style type='text/css'> .ffne_action{padding-right: 8px; cursor:pointer;} .ffne_action:hover{} #ffne_export{ } #ffne{float:right;margin-left: 0.9em;} #ffne_button{font-size:1.3em;cursor:pointer;line-height: 1em;padding-right: 7px;} .ffne_hidden{display:none;} #ffneInputDelay{width: 50px}</style>");
$('body').append(style);

function addButtons(){
    // Adding buttons
    let res = document.getElementById('f_size');
    // creating links
    var node = $('.lc').first();
    var exportMenu = $('<span id="ffne"><span id="ffne_button" class="xcontrast_txt">fE</span></span>');
    var exportContainer = $('<span id="ffne_export"></span>');
    var addIndexButton = $('<span href="javascript:" class="ffne_action" title="Create table of contents">Index</span>');
    var expAllButton = $('<span href="javascript:" class="ffne_action" id="exportAllButton" title="Show the whole story on one page">Story</span>');
    var expRestButton = $('<span href="javascript:" class="ffne_action" id="exportRestButton" title="Export chapters from the current to the last one">Rest</span>');
    var expButton = $('<span href="javascript:" class="ffne_action" title="Leave only raw text">Text</span>');
    var headersSwitch = $('<span class="ffne_action" title="Add headers when exporting">Add headers <input type="checkbox" id="checkbox_headers" name="checkbox_headers" checked></span>');
    var delayInput = $('<span class="ffne_action" title="Request delay in milliseconds">Delay <input type="text" id="ffneInputDelay" name="ffne_input_delay" value="'+delay+'"></span>');
    var copyButton = $('<span href="javascript:" class="ffne_action" id="ffneCopyButton" title="Copy compiled text to the clipboard">Copy</span>');
    exportMenu.append(exportContainer);
    exportContainer.append(expAllButton,expRestButton, headersSwitch, delayInput,'|&nbsp;',addIndexButton,'|&nbsp;',expButton, copyButton);
    node.append(exportMenu);
    expAllButton.click(exportChapters);
    expRestButton.click(exportRest);
    expButton.click(exportCh);
    addIndexButton.click(addIndex);
    copyButton.click(copyText);
    $('#ffne_button').click(function(){
        var cont = $('#ffne_export');
        if (cont.hasClass('ffne_hidden')){cont.removeClass('ffne_hidden');}else{cont.addClass('ffne_hidden')}
    });
}

//Adding buttons to page;
addButtons();

//Adding table of contents
function addIndex(){
    var chapters = $('div[name="ffnee_chapter"]');
    var index = $('<div id="ffnee_index"><h2>Table of contents</h2></div>');
    var toC = $('<ol></ol>');
    index.append(toC);
    for (var i=0;i<chapters.length;i++){
        var item = $(chapters[i]); //chapter we are currently processing
        toC.append($('<li><a href="#'+item.attr('id')+'">'+item.attr('title')+'</a></li>'));
    }
    $('#storytext').prepend(index);
}
//adding headers, as entered by author
function addHeaders(){
    var chapters = document.getElementsByName('ffnee_chapter');
    for (var i=0;i<chapters.length;i++){
        var item = chapters.item(i); //chapter to which we are adding a header
        var header = document.createElement('p');
        header.innerHTML = '<h2>Chapter '+(i+1)+': '+item.getAttribute('title')+'</h2>';
        item.insertBefore(header,item.firstChild);
    }
}
function addTitle(){
    var titleText = $('b.xcontrast_txt','#profile_top').first().html();
    var title = $('<h1>'+titleText+'</h1>');
    var authorText = $('a.xcontrast_txt[href^="/u/"]','#profile_top').first().html();
    var author = $('<h2>'+authorText+'</h2>');
    var storytext = $('#storytext');
    storytext.prepend(title, author);
}
function exportCh(){
    document.body.innerHTML='<div style=\'padding-left:2em;padding-right:2em;padding-top:1em;\'>'+document.getElementById('storytextp').innerHTML+'</div>';
}
function copyText(){
    GM.setClipboard(document.getElementById('storytextp').innerText);
}

function exportRest(e){
    var chap_select = document.getElementById('chap_select');
    console.log('exporting rest');
    exportChapters(e,chap_select.value-1);
}
function exportChapters(e,start,end){
    // Main actions
    // Progress indicator
    delay = document.getElementById('ffneInputDelay').value;
    var expDiv = document.getElementById('exportAllButton');
    var expText = expDiv.childNodes[0];
    var hr=location.href;
    var chapterNumIndex=hr.search(/\/\d{1,3}\//);
    //Getting number of chapters
    var storyLength=getLength();
    if (storyLength == 1){
        expText.nodeValue = 'Oneshot';
        return;
    }
    if (start==undefined){
        start=0;
    }
    if (end==undefined){
        end=storyLength;
    }
    storyLength = end-start;
    let i = start;
    var totalStoryLength = storyLength;//reference
    console.log('retrieving '+totalStoryLength+' chapters');
    console.log('start index is: '+start+', end is: '+end);
    setTimeout(function tick(){
        console.log(`Starting to load chapter ${i}`);
        loadChapter(i+1,function(response,num){
            console.log('Loaded chapter '+num);
            chapters[num]=parseChapter(response, num+1);
            expText.nodeValue = 'Export: Chapter '+String(totalStoryLength-storyLength+1)+' out of '+totalStoryLength;
            storyLength--;
            if (storyLength==0){
                console.log(chapters);
                parseStory(chapters);
                expText.nodeValue='Story (again)';
            }
        });
        i++;
        if (i<end){
            setTimeout(tick, delay);
        }
    }, delay);

}
// Converting chapters' array into a whole;
function parseStory(chapters){
    var numCh= chapters.length;
    //document.body.innerHTML=chapters[0];
    var appendNode=document.getElementById('storytext');
    appendNode.innerHTML= '';
    var firstChapter=true;
    for (var i=0;i<numCh;i++){
        if (chapters[i]!=undefined){
            //findHeader(chapters[i]);  //smart header search
            var st=chapters[i];
            st.setAttribute('name','ffnee_chapter');
            st.setAttribute('id','ffnee_ch'+i);
            if (firstChapter){
                firstChapter=false;
            }else {
                st.style.marginTop='10em';
            }
            appendNode.appendChild(st);
        }
    }
    let headersEnabled = document.getElementById('checkbox_headers').checked;
    if (headersEnabled){
        addHeaders();
    }
    addTitle();
}
function parseChapter(chapterHtml, chapterNumber){

    var t=document.createElement('div');
    t.innerHTML=chapterHtml;
    //extracting text only
    var ev='.//div[@id=\'storytext\']';
    var xpathResult = document.evaluate(ev,t,null,XPathResult.ORDERED_NODE_SNAPSHOT_TYPE, null);
    var chapterContent=document.createElement('div');
    const chapName = getChapterName(t);
    chapterContent.setAttribute('title',chapName);
    chapterContent.innerHTML = xpathResult.snapshotItem(0).innerHTML;
    console.log(`Parsing chapter ${chapterNumber}. Title: ${chapName}`);
    return chapterContent;
}
function getChapterName(obj){
    let select = obj.querySelector('#chap_select');
    return select.options[select.selectedIndex].innerHTML.split(/[. ]{2}/)[1];
}
//  Getting number of chapters;
function getLength(){
    var chNum = document.getElementById('chap_select');
    if (chNum==null){
        numChapters = 1;
    }else {
        var numChapters = chNum.getElementsByTagName('option').length;
    }
    return (numChapters);
}
function testReaponseHandler(){
    console.log(this.responseText);
}
// This function loads chapters and extracts chapter's number and title
function loadChapter(num,callback){
    var replStr='\/'+String(num)+'\/';
    var hr=location.href;
    var currentURL=hr.replace(/\/\d{1,3}\//,replStr);
    try{
        var req = new XMLHttpRequest();
        req.open('get',currentURL,true);
        req.onload= function(){
            callback(req.responseText,num-1);
        };
        req.send();
    }catch (e) {
        console.log(e);
    }
}