// ==UserScript==
// @name         arxiv-download-rename-tool
// @namespace    Sean
// @version      0.2
// @description  It can rename the pdf name when you download paper from arxiv.org
// @author       Sean
// @match        *://arxiv.org/abs/*
// @match        *://arxiv.org/search/*
// @match        *://arxiv.org/list/*
// @icon         data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==
// @grant        none
// @license      AGPL License
// @downloadURL none
// ==/UserScript==



(function() {
    'use strict';
    const url = location.pathname,webTitle = document.title
    var downloadName = '',downloadPath = ''
    var papertitle = '',papertime = ''
    if(url.search('/abs/')!=-1){
        papertitle = document.querySelector("#abs > h1").innerText
        downloadPath = window.location.href.replace('abs','pdf') + '.pdf' //document.querySelector("#abs-outer > div.extra-services > div.full-text > ul > li:nth-child(1) > a")+'.pdf'
        papertime = window.location.pathname.slice(5,9) //document.querySelector("#abs > div.metatable > table > tbody > tr:nth-child(3) > td.tablecell.arxivid > span > a").innerText.slice(6,10)
        downloadName = renamePaperFile(papertitle,papertime)
        addDownloadButton(downloadPath,downloadName,document.querySelector("#abs-outer > div.extra-services > div.full-text"))
    }
    if(url.search('/search/')!=-1){
        var paperlist = document.querySelectorAll("#main-container > div.content > ol > li")
        for(let paper in paperlist){
            papertitle = paperlist[paper].children[1].innerText
            console.log(papertitle)
            papertime = paperlist[paper].children[0].innerText.slice(6,10)
            downloadName = renamePaperFile(papertitle,papertime)
           // downloadPath = paperlist[paper].children[0].children[0].children[1].children[0].href+'.pdf'
            downloadPath= "https://arxiv.org/pdf/"+paperlist[paper].children[0].children[0].children[1].getAttribute("paper_id")+".pdf"
            addDownloadButton(downloadPath,downloadName,paperlist[paper].children[0])
        }
    }
    if(url.search('/list/')!=-1){
        let paperlist = document.querySelectorAll(".list-identifier")
        for (let i = 0, len = paperlist.length; i < len; i++){
            try {
                let paper = paperlist[i]
                // console.log(paper)
                papertitle = paper.parentNode.nextElementSibling.querySelector('.list-title').innerText
                downloadPath = paper.querySelector('a[title="Download PDF"]').href + '.pdf'
                papertime = downloadPath.split('/').pop().split('.')[0]
                downloadName = renamePaperFile(papertitle,papertime)
                addDownloadButton(downloadPath,downloadName,paper)
            } catch (error) {
                console.warn('AUTO download rename raise warning at : ' + papertitle)
            }
        }
    }

    function addDownloadButton(downloadPath,downloadName,element){
        var button = document.createElement("a"); 
        button.id = "downloadPaper";
        button.textContent = "##Download paper with a new name##";
        button.setAttribute("href", downloadPath)
        button.setAttribute("download", downloadName)
        element.append(button);
    }
    function renamePaperFile(name,time){
        var downloadName = name.replace(': ','：')
        downloadName = downloadName.replace(':','：')
        downloadName = downloadName.replace('?','？')
        downloadName = downloadName.replace('/',' OR ')
        downloadName = downloadName.replace('"','“')+'.pdf'
        return '['+time+']'+downloadName
    }
})();