Premiumize_buttons.user.js

// ==UserScript==
// @name         Premiumize.me Next File Button
// @namespace    http://tampermonkey.net/
// @version      0.87
// @description  Adds a next and previous button to the premiumize.me file preview page
// @author       xerg0n
// @match        https://www.premiumize.me/*
// @grant        none
// ==/UserScript==

/*
Access to the stored Data
*/
class Store {
    constructor() {
        this.key_main = "folders"
        this.key_last = "lastFile"

        // used to filter files that are supposed to be opend
        var whitelist_items = [
            ".mkv$",
            ".avi$",
            ".mov$",
            ".mp4$",
            ".wmv$",
            ".m4v$",
            ".mpeg$",
            //audio
            ".mp3$",
            ".wav$",
            ".acc$",
            ".m4a$",
            ".wma$",
        ];
        this.whitelist = new RegExp(whitelist_items.join("|"), "i");
    }
    setLastFile(file){
        localStorage.setItem(this.key_last, file.id);
    }

    getLastFile(){
        var id = localStorage.getItem(this.key_last)
        var file = this.getFile(id);
        return file;
    }
    hasFolder(folder_id){
        return (folder_id in this.getFolders())
    }
    addFolder(folder){
        var all_folders = this.getFolders()
        if (!this.hasFolder(folder.id)){
            all_folders[folder.id] = folder
            this.setFiles(all_folders)
        }else{
            console.log('folder already saved')
        }
    }
    getFolder(folder_id) {
        return this.getFolders()[folder_id];
    }

    getFile(id, folder_id=null){
        if (folder_id){
            return this.getFolders()[folder_id].files.filter(file => file.id == id)[0];
        }else{
            return this.getFileFromAll(id);
        }
    }

    getFileFromAll(fileid) {
        var folders = this.getFolders();
        for (var key in folders) {
            if (folders.hasOwnProperty(key)) {
                var id = folders[key].files.findIndex((db_file) => db_file.id == fileid);
                if (id != -1){
                    return folders[key].files[id];
                }
            }
        }
    }

    getNextFile(id, filter=this.whitelist){
        // get all files
       var sorted_files = this.getFiles()
            .sort((a, b) => (a.name.toLowerCase() > b.name.toLowerCase()) ? 1 : -1)
            .filter(file => file.name.match(this.whitelist));
       var idx = sorted_files.findIndex( (element, index, array) => element.id == id);
       return sorted_files[idx+1];
    }

    getNextFolder(id){
        // get all files
        var folders = Object.values(this.getFolders())
                         .filter(folder => folder.files
                                 .filter( file => file.name.match(this.whitelist)).length > 0);
        var sorted = folders.sort((a, b) => (a.name.toLowerCase() > b.name.toLowerCase()) ? 1 : -1)
        console.log(sorted)
        var idx = sorted.findIndex( (element, index, array) => element.id == id);
        return sorted[idx+1];
    }


    getNext(folder_id, id, filter=this.whitelist) {
        var folder_files = this.getFolder(folder_id, filter=filter).files
        if (filter){
            folder_files = folder_files.filter(file => file.name.match(this.whitelist));
        }
        var index = folder_files.findIndex((file) => file.id == id);
        if (index < folder_files.length-1){
            return folder_files[index+1]
        }else{
            return null
        }
    }

    getPrev(folder_id, id, filter=this.whitelist) {
        var folder_files = this.getFolder(folder_id, filter=filter).files
        if (filter){
            folder_files = folder_files.filter(file => file.name.match(this.whitelist));
        }
        var index = folder_files.findIndex((file) => file.id == id);
        if (index != 0){
            return folder_files[index-1]
        }else{
            return null
        }
    }

    getFolders(){
        var files = JSON.parse(localStorage.getItem(this.key_main));
        if (files == null){
            files = {};
        }
        return files
    }
    getFiles(){
        var folders = this.getFolders();
        return Object.keys(folders).map(x=> folders[x].files).flat();
        }

    setFiles(files){
        var ser_files = JSON.stringify(files);
        localStorage.setItem(this.key_main, ser_files)
    }

    updateFile(file){
        var all_folders = this.getFolders()
        for (var key in all_folders) {
            if (all_folders.hasOwnProperty(key)) {
                var id = all_folders[key].files.findIndex((db_file) => db_file.id == file.id);
                if (id > -1){
                    all_folders[key].files[id] = file;
                    this.setFiles(all_folders);
                    return
                }
            }
        }
    }
}

/*
* Player wrapper
*/
class Player {
    constructor(){
        this.parser = new Parser();
        var file_id = this.parser.getCurrentFileId();
        var folder_id = this.parser.getFolderId();

        this.store = new Store();
        this.current_file = this.store.getFile(file_id, folder_id);
        this.player = this.parser.getPlayer();
        this.playerElement = this.parser.getPlayerElement();
        this.restoreTime();
        this.registerListeners();
    }
    registerListeners(){
        this.playerElement.addEventListener("timeupdate",
                                            this.timechange.bind(this),
                                            true);
        this.playerElement.addEventListener("ended",
                                            this.ended.bind(this),
                                            true);
    }
    timechange(){
        console.log('[Player] timeupdate')
        var time = this.player.currentTime()
        this.current_file.time = time
        if (this.player.remainingTime() < 120){
            this.current_file.finished = true
        }
        this.store.updateFile(this.current_file);
    }
    restoreTime(){
        if (this.current_file.time){
            this.playerElement.currentTime = this.current_file.time;
        }
    }
    setNext(file){
        this.next_file = file
    }
    ended(){
        // set time to 0
        this.current_file.time = 0
        this.current_file.finished = true
        this.store.updateFile(this.current_file);

        if (this.next_file){
            location.href = "/file?id="+this.next_file.id
        }
    }
}

/*
*  Parsing the elemnts and returning data
*/
class Parser {
    getCurrentFileId(){
        return document.URL.match(/\/file\?id=(\S*)/)[1];
    }
    getBreadcrumb(){
        return document.getElementsByClassName('breadcrumb')[0].childNodes
    }
    getFolderId(){
        var breadcrumb = this.getBreadcrumb()
        var id = breadcrumb[breadcrumb.length-4].firstChild.getAttribute("href")
        .match(/folder_id=(\S*)/)[1];

        return id
    }

    getFolderIdFromUri(){
        return document.URL.match(/folder_id=(\S*)/)[1];
    }
    getPlayerElement(){
        return document.getElementById('player_html5_api');
    }
    getPlayer(){
        return window.player;
    }
    * elementparser(){
        var icon_sel = ".fa-file-alt"
        var folder_id = this.getFolderIdFromUri();
        var icon_elements = document.querySelectorAll(".fa-file-alt");
        for (var icon_el of icon_elements){
            var element = icon_el.parentNode.parentNode
            var text = element.querySelector('div span:nth-child(3)');
            var id = element.getAttribute('href').replace("/file?id=","");
            yield {
                main: element,
                text: text,
                id: element.getAttribute('href').replace("/file?id=","")
            };
        }
    }
    markFilesFinished(finished_ids){
        for (var parsed of this.elementparser()){
            if(finished_ids.has(parsed.id)){
                parsed.text.style.textDecoration = 'line-through'
            }
        }
    }

    parseFileLinks(folder_id){
        var file_links = [];
        for (var parsed of this.elementparser()){
            var file = {
                name: parsed.text.innerText,
                id: parsed.id,
                time: 0,
                folder_id: folder_id,
            };
            file_links.push(file);
        }
        return file_links;
    }
    parseFolder(){
        var id = this.getFolderIdFromUri();
        var bread = this.getBreadcrumb()
        var folder = {
            files: this.parseFileLinks(id),
            name: bread[bread.length-1].innerText,
            url: bread[bread.length-1].baseURI,
            id: id,
        }
        return folder
    }
    currentPage(){
        if (/folder_id=/.test(document.URL)){
            return 'folder'
        }
        if (/transfers/.test(document.URL)){
            return 'downloader'
        }
        if (/file\?id=/.test(document.URL)){
            return 'playback'
        }
        return 'unknown'
    }
}
class Logger{
    context(new_context){
        this.context = new_context
    }
    log(message){
       console.log("["+this.context+"] ", message)
    }
}
/*
* Utils
*/
class Utils{
    static async waitFor(sel){
        while(!document.querySelector(sel)) {
            return new Promise(r => setTimeout(r, 500));
        }
    }

    static makeButton(text, file, style=null){
        var btn = document.createElement('a');
        btn.className = "btn btn-primary";
        btn.innerText = text
        btn.style = 'font-size:10px;'+style
        btn.setAttribute("href","/file?id="+file.id);
        btn.title = file.name;
        return btn
    }
}

async function main(){
    var logger = new Logger();
    var store = new Store();
    var parser = new Parser();
    var page = parser.currentPage();

    switch (page){
        case 'folder':
            await Utils.waitFor(".glyphicon-file");
            logger.context = 'folder'
            var folder_id = parser.getFolderIdFromUri();
            console.log('[folder] id:'+folder_id)
            if (store.hasFolder(folder_id)){
                console.log('[folder] marking files')
                var ids = new Set()
                store.getFolder(folder_id).files
                    .forEach((f) => f.finished && ids.add(f.id))
                parser.markFilesFinished(ids)
            }else{
                var folder = parser.parseFolder();
                store.addFolder(folder)
            }
            break
        case 'downloader':
            await Utils.waitFor(".btn-primary");
            logger.context = 'downloader'
            var lastFile = store.getLastFile();
            var btn_cont = Utils.makeButton("Reopen last", lastFile);

            btn_cont.innerHTML = '<span class="glyphicon glyphicon glyphicon-chevron-right"></span>'
                +'<span>Reopen last</span>';
            btn_cont.style = "margin-left: 6px";

            var container = document.querySelectorAll('[data-reactid=".0.1.0"]')[0];
            container.appendChild(btn_cont);
            break
        case 'playback':
            await Utils.waitFor(".vjs-control-bar");
            logger.context = 'playback'
            var player_container = new Player(store);

            var current_file = store.getFile(parser.getCurrentFileId(),
                                             parser.getFolderId())

            store.setLastFile(current_file);


            // button container
            container = document.createElement('div');
            container.style = "margin-bottom: 5px";

            var next_file = store.getNext(parser.getFolderId(),
                                          parser.getCurrentFileId())
            var prev_file = store.getPrev(parser.getFolderId(),
                                          parser.getCurrentFileId())
            var btn_next;
            if (prev_file){
                var btn_prev = Utils.makeButton("🡄", prev_file,
                                                style='border-radius: 20px;')
                container.appendChild(btn_prev);
            }
            if (next_file){
                player_container.setNext(next_file);

                btn_next = Utils.makeButton("🡆", next_file,
                                            style='border-radius: 20px;')
                btn_next.style.float = "right";
                container.appendChild(btn_next);
            }else{
                // insert buttons for next episode
                var maybe_next = store.getNextFile(current_file.id);
                var maybe_next_folder = store.getNextFolder(parser.getFolderId());

                var btn_style = 'background-color:rgb(63, 67, 70); float:right; margin-right: 2px; '

                if (maybe_next){
                    //player_container.setNext(maybe_next);
                    var btn_file = Utils.makeButton("Next File?",
                                                maybe_next,
                                                style=btn_style)
                    container.appendChild(btn_file);
                    //container.appendChild(btn_next);
                }
                if (maybe_next_folder){
                    var btn_folder = Utils.makeButton("Next Folder?",
                                                maybe_next_folder.files[0],
                                                style=btn_style)
                    container.appendChild(btn_folder);
                    //container.appendChild(btn_next);
                }
            }

            var main_container = document.getElementsByClassName('container')[1];
            main_container.insertBefore(container, main_container.childNodes[5]);
            break
        default:
            console.log('unknown page')
            break
    }

};
main();