// ==UserScript==
// @name         GenBooklist
// @namespace    http://tampermonkey.net/
// @version      0.9
// @description  Generate a copy of book list from various sites.
// @author       You
// @match        https://read.readmoo.com/*
// @match        https://www.kobo.com/tw/zh/library*
// @match        https://viewer-ebook.books.com.tw/viewer/*
// @match        https://ebook.hyread.com.tw/Template/store/member/my_bookcase_column_list.jsp*
// @match        https://ebook.hyread.com.tw/Template/store/member/my_bookcase_cover_list.jsp*
// @match        https://www.pubu.com.tw/bookshelf*
// @match        https://play.google.com/books*
// @match        https://mybook.taiwanmobile.com/bookcase/list*
// @match        https://www.amazon.com/hz/mycd/digital-console/contentlist/*
// @match        https://webreader.hamibook.com.tw/HamiBookcase*
// @match        https://www.amazon.cn/hz/mycd/digital-console/contentlist/*
// @match        https://www.bookwalker.com.tw/bookcase/available_book_list*
// @icon         data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==
// @homepageURL  https://github.com/plasmabal/gen_booklist
// @updateURL    https://raw.githubusercontent.com/plasmabal/gen_booklist/master/genbooklist.js
// @grant        GM_registerMenuCommand
// @grant        GM_setClipboard
// @grant        GM_getValue
// @grant        GM_setValue
// @grant        GM_unregisterMenuCommand
// ==/UserScript==

(function() {
    'use strict';

    var shouldTrimSubtitle = GM_getValue('trimSubtitle', false);

    // ==== Site Configuration ====
    //
    // Note:
    //
    // Each site configuration contains the following fields
    //
    // * isCorrectHost: Function receives a host string and returns bool. Indicate if the host is the expected one.
    // * hostErrMsg: Function returns a string.  The message is displayed if the site configuration is used at the wrong host.
    // * getBookList: Function returns HTMLCollection.
    // * noListMsg: Functions returns a string.  The message is displayed if the HTMLCollection returned from `getBookList` contains no element.
    // * transformBookItem: Convert each HTMLElement returned `getBookList` to an object contains the following fields:
    //    * name: Book name
    //    * author: Book author
    //   Or returns `undefined` if the eleemnt should be skipped.

    // readmoo.com
    let readmooSite = {
        isCorrectHost: function(host) {
            return (host == "read.readmoo.com");
        },
        hostErrMsg: function() {
            return "請在讀墨書櫃裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('library-item-info');
        },
        noListMsg: function() {
            return '找不到書籍列表. 請切換到書櫃並選擇為列表顯示.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByClassName('title')[0].innerText;
            let author = item.getElementsByClassName('author')[0].innerText;
            return {name: name, author: author};
        }
    }

    // kobo.com
    let koboSite = {
        isCorrectHost: function(host) {
            return (host == "www.kobo.com");
        },
        hostErrMsg: function() {
            return "請在 Kobo 裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('book-list')[0].getElementsByClassName('element-flipper');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            if (item.getElementsByClassName('buy-now').length > 0) {
                return undefined;
            }
            let itemInfo = item.getElementsByClassName('item-info')[0];
            let name = itemInfo.getElementsByClassName('title')[0].innerText;
            let author = itemInfo.getElementsByClassName('authors')[0].innerText;
            return {name: name, author: author};
        }
    }

    // books.com.tw
    let booksSite = {
        isCorrectHost: function(host) {
            return (host == "viewer-ebook.books.com.tw");
        },
        hostErrMsg: function() {
            return "請在博客來電子書櫃裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('bookshelf__book');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let bookmarks = item.getElementsByClassName('book__bookmark');
            if (bookmarks.length > 0 && bookmarks[0].innerText == '試閱') {
                return undefined;
            }
            let name = item.getElementsByClassName('book__description__title')[0].innerText;
            let author = item.getElementsByClassName('book__description__author')[0].innerText.replace('作者：', '');
            return {name: name, author: author};
        }
    }

        // ebook.hyread.com.tw
    let hyreadSite = {
        isCorrectHost: function(host) {
            return (host == "ebook.hyread.com.tw");
        },
        hostErrMsg: function() {
            return "請在 HyRead 書櫃裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('bookcase_list_item');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByClassName('bookname')[0].innerText;
            let author = item.getElementsByClassName('bookname')[1].innerText;
            return {name: name, author: author};
        }
    }

    // www.pubu.com.tw
    let pubuSite = {
        isCorrectHost: function(host) {
            return (host == "www.pubu.com.tw");
        },
        hostErrMsg: function() {
            return "請在 Pubu 我的內容裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('allBook');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByTagName('h3')[0].innerText;
            return {name: name, author: ''};
        }
    }

    // https://play.google.com/books
    let googleBooksSite = {
        isCorrectHost: function(host) {
            return (host == "play.google.com");
        },
        hostErrMsg: function() {
            return "請在 Google Play Books 我的書籍裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('ebook');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let coverBadges = item.getElementsByClassName('cover-badge');
            if (coverBadges.length > 0 && coverBadges[0].innerText == '試閱內容') {
                return undefined;
            }
            let name = item.getElementsByClassName('title')[0].innerText;
            let author = item.getElementsByClassName('author')[0].innerText;
            return {name: name, author: author};
        }
    }

    // https://mybook.taiwanmobile.com/bookcase/list
    let myBookSite = {
        isCorrectHost: function(host) {
            return (host == "mybook.taiwanmobile.com");
        },
        hostErrMsg: function() {
            return "請在 myBook 我的書籍裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('bookshelf-list-grid')[0].getElementsByClassName('item');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByTagName('h3')[0].innerText;
            let author = item.getElementsByClassName('author')[0].innerText.replace('作者：', '');
            return {name: name, author: author};
        }
    }

    // https://www.amazon.com/hz/mycd/digital-console/contentlist/
    let amazonUSSite = {
        isCorrectHost: function(host) {
            return (host == "www.amazon.com");
        },
        hostErrMsg: function() {
            return "請在 Amazon My Digital Content 裡使用";
        },
        getBookList: function() {
            return document.getElementById('CONTENT_LIST').getElementsByTagName('tbody')[0].getElementsByTagName('tr');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByClassName('digital_entity_title')[0].innerText;
            let author = item.getElementsByClassName('information_row')[0].innerText;
            return {name: name, author: author};
        }
    }

    // https://webreader.hamibook.com.tw/HamiBookcase#/
    let hamiBookSite = {
        isCorrectHost: function(host) {
            return (host == "webreader.hamibook.com.tw");
        },
        hostErrMsg: function() {
            return "請在 HamiBook 的個人書櫃裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('one_classification_book_content')[0].getElementsByTagName('li');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByClassName('index_book_title')[0].innerText;
            return {name: name, author: ''};
        }
    }

    // https://www.amazon.cn/hz/mycd/digital-console/contentlist/
    let amazonCNSite = {
        isCorrectHost: function(host) {
            return (host == "www.amazon.cn");
        },
        hostErrMsg: function() {
            return "請在 Amazon CN 的 管理我的内容和设备 的 设备 裡使用";
        },
        getBookList: function() {
            return document.getElementById("CONTENT_LIST").getElementsByClassName("digital_entity_details");
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByClassName('digital_entity_title')[0].innerText;
            let author = item.getElementsByClassName('information_row')[0].innerText;
            return {name: name, author: author};
        }
    }

    // https://www.bookwalker.com.tw/bookcase/available_book_list
    let bookwalkerTWSite = {
        isCorrectHost: function(host) {
            return (host == "www.bookwalker.com.tw");
        },
        hostErrMsg: function() {
            return "請在 BOOK☆WALKER TAIWAN 的 線上PC閱讀書櫃 裡使用";
        },
        getBookList: function() {
            return document.getElementsByClassName('readerBookListType');
        },
        noListMsg: function() {
            return '找不到書籍列表.';
        },
        transformBookItem: function(item) {
            let name = item.getElementsByClassName('readerBookName')[0].innerText;
            let author = item.getElementsByClassName('readerBookAuthor')[0].innerText;
            return {name: name, author: author};
        }
    }

    // ==== Booklist Generator ====

    let makeGenerator = function (site) {
        return function() {
            if (!site.isCorrectHost(window.location.host)) {
                alert(site.hostErrMsg());
                return;
            }
            let bookList = site.getBookList();
            if (bookList.length == 0) {
                alert(site.noListMsg());
                return;
            }

            const books = []
            for (let i = 0; i < bookList.length; i++) {
                let item = site.transformBookItem(bookList[i]);
                if (item === undefined) { continue; }
                books.push(item);
            }

            function trimSubtitle(title) {
                if (shouldTrimSubtitle) {
                    return title.replace(/：.+$/, '');
                } else {
                    return title;
                }
            }

            let msg = books
                .map(b => trimSubtitle(b.name) + '\t' + b.author)
                .join('\n') + "\n";

            GM_setClipboard(msg);

            if (books.length == 1) {
                alert(books.length + " book is collected.");
            } else {
                alert(books.length + " books are collected.");
            }
        }
    }

    let sites = {
        'Amazon CN': amazonCNSite,
        'Amazon US': amazonUSSite,
        'BOOK☆WALKER TAIWAN': bookwalkerTWSite,
        'Books': booksSite,
        'Google Play Books': googleBooksSite,
        'HamiBook': hamiBookSite,
        'HyRead': hyreadSite,
        'Kobo': koboSite,
        'MyBook': myBookSite,
        'Pubu': pubuSite,
        'Readmoo': readmooSite,
    }
    for (const [name, site] of Object.entries(sites)) {
        if (site.isCorrectHost(window.location.host)) {
            GM_registerMenuCommand(name, makeGenerator(site));
        }
    }{}

    var trimSubtitleMenuId;
    function trimSubtitleMenuTitle() {
        return "Setting: Trim Subtitle " + (shouldTrimSubtitle ? "on" : "off");
    }
    function toggleTrimSubtitle() {
        shouldTrimSubtitle = !shouldTrimSubtitle;
        GM_unregisterMenuCommand(trimSubtitleMenuId);
        trimSubtitleMenuId = GM_registerMenuCommand(trimSubtitleMenuTitle(), toggleTrimSubtitle);
    }
    trimSubtitleMenuId = GM_registerMenuCommand(trimSubtitleMenuTitle(), toggleTrimSubtitle);
})();