// ==UserScript==
// @name           Display shortcut for relationships on MusicBrainz
// @description    Display icon shortcut for relationships of release-group, release, recording and work: e.g. Amazon, Discogs, Wikipedia, ... links. This allows to access some relationships without opening the entity page.
// @version        2020.3.31.1
// @author         Aurelien Mino <aurelien.mino@gmail.com>
// @licence        GPL (http://www.gnu.org/copyleft/gpl.html)
// @downloadURL    https://raw.github.com/murdos/musicbrainz-userscripts/master/mb_relationship_shortcuts.user.js
// @updateURL      https://raw.github.com/murdos/musicbrainz-userscripts/master/mb_relationship_shortcuts.user.js
// @include        http*://*musicbrainz.org/artist/*
// @include        http*://*musicbrainz.org/release-group/*
// @include        http*://*musicbrainz.org/label/*
// @require        https://code.jquery.com/jquery-3.2.1.min.js
// ==/UserScript==

// Definitions: relations-type and corresponding icons we are going to treat
var relationsIconsURLs = {
    url: {
        'amazon asin': 'https://musicbrainz.org/static/images/favicons/amazon-32.png',
        discogs: 'https://musicbrainz.org/static/images/favicons/discogs-32.png',
        wikidata: 'https://musicbrainz.org/static/images/favicons/wikidata-32.png',
        imdb: 'https://musicbrainz.org/static/images/favicons/imdb-32.png',
        'creative commons licensed download': 'http://creativecommons.org/favicon.ico',
        'cover art link': 'http://www.cdcovers.cc/favicon.ico',
        secondhandsongs: 'https://musicbrainz.org/static/images/favicons/secondhandsongs-32.png',
        lyrics: 'http://www.nomy.nu/img/lyrics-icon.gif',
        allmusic: 'https://musicbrainz.org/static/images/favicons/allmusic-16.png',
    },
    'release-group': {
        'single from': 'http://www.amaesingtools.com/images/left_arrow_icon.gif',
    },
    release: {
        'part of set': 'http://web.archive.org/web/20060709091901/http://wiki.musicbrainz.org/-/musicbrainz/img/moin-inter.png',
        remaster: 'http://web.archive.org/web/20060708200714/http://wiki.musicbrainz.org/-/musicbrainz/img/moin-www.png',
    },
};

var otherDatabasesIconURLs = {
    'd-nb.info': 'https://musicbrainz.org/static/images/favicons/dnb-16.png',
    'www.musik-sammler.de': 'https://musicbrainz.org/static/images/favicons/musiksammler-32.png',
    'www.worldcat.org': 'https://musicbrainz.org/static/images/favicons/worldcat-32.png',
    'rateyourmusic.com': 'https://musicbrainz.org/static/images/favicons/rateyourmusic-32.png',
};

var incOptions = {
    'release-group': ['release-group-rels', 'url-rels'],
    release: ['release-rels', 'url-rels', 'discids'],
    recording: ['work-rels'],
    work: ['url-rels'],
};

// prevent JQuery conflicts, see http://wiki.greasespot.net/@grant
this.$ = this.jQuery = jQuery.noConflict(true);

if (!unsafeWindow) unsafeWindow = window;

$(document).ready(function () {
    // Get pageType (label or artist)
    let parent = {};
    let child = {};
    if ((m = window.location.href.match('/artist/(.{36})[^/]*$'))) {
        parent.type = 'artist';
        parent.mbid = m[1];
        child.type = 'release-group';
    } else if ((m = window.location.href.match('/(release-group|label)/(.{36})[^/]*$'))) {
        parent.type = m[1];
        parent.mbid = m[2];
        child.type = 'release';
    } else if ((m = window.location.href.match('/artist/(.{36})/(releases|recordings|works)'))) {
        parent.type = 'artist';
        parent.mbid = m[1];
        child.type = m[2].replace(/s$/, '');
    } else {
        // Not supported
        return;
    }

    let mbidRE = /(release|release-group|work)\/([a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12})/;

    // Determine target column
    let columnindex = 0;
    $("table.tbl tbody tr[class!='subh']").each(function () {
        $(this)
            .children('td')
            .each(function () {
                if ($(this).find('a').attr('href') !== undefined && $(this).find('a').attr('href').match(mbidRE)) {
                    return false;
                }
                columnindex++;
            });
        return false;
    });

    // Set MBID to row in tables to get easiest fastest access
    $("table.tbl tr[class!='subh']").each(function () {
        let $tr = $(this);

        $tr.children(`th:eq(${columnindex})`).after("<th style='width: 150px;'>Relationships</th>");
        $tr.children(`td:eq(${columnindex})`).after("<td class='relationships'></td>");

        $(this)
            .find('a')
            .each(function () {
                let href = $(this).attr('href');
                if ((m = href.match(mbidRE))) {
                    $tr.attr('id', m[2]);
                    return false;
                }
            });
    });

    // Adapt width of subheader rows by incrementing the colspan of a cell
    $("table.tbl tr.subh").each(function() {
        $(this).children("th[colspan]").attr('colspan', function(index, oldValue) {
            if (index === 0) {
                return Number(oldValue) + 1;
            } else {
                return oldValue;
            }
        });
    });

    // Calculate offset for multi-page lists
    let page = 1;
    if ((m = window.location.href.match('[?&]page=([0-9]*)'))) {
        page = m[1];
    }
    let offset = (page - 1) * 100;

    // Call the MB webservice
    let url = `/ws/2/${child.type}?${parent.type}=${parent.mbid}&inc=${incOptions[child.type].join('+')}&limit=100&offset=${offset}`;
    //console.log("MB WS url: " + url);

    $.get(url, function (data, textStatus, jqXHR) {
        // Parse each child
        $(data)
            .find(child.type)
            .each(function () {
                let mbid = $(this).attr('id');

                // URL relationships
                $(this)
                    .find("relation-list[target-type='url'] relation")
                    .each(function () {
                        let reltype = $(this).attr('type');
                        let target = $(this).children('target').text();
                        if (Object.prototype.hasOwnProperty.call(relationsIconsURLs.url, reltype)) {
                            $(`#${mbid} td.relationships`).append(
                                `<a href='${target.replace(/'/g, '&apos;')}'>` +
                                    `<img style='max-height: 16px;' src='${relationsIconsURLs.url[reltype]}' />&nbsp;` +
                                    `</a>`
                            );
                        } else
                            for (let rel in otherDatabasesIconURLs) {
                                if (target.indexOf(rel) != -1) {
                                    $(`#${mbid} td.relationships`).append(
                                        `<a href='${target.replace(/'/g, '&apos;')}'>` +
                                            `<img style='max-height: 16px;' src='${otherDatabasesIconURLs[rel]}' />&nbsp;` +
                                            `</a>`
                                    );
                                }
                            }
                    });

                // Other relationships
                $(this)
                    .find("relation-list[target-type!='url']")
                    .each(function () {
                        let targettype = $(this).attr('target-type').replace('release_group', 'release-group');
                        let relations = {};

                        if (relationsIconsURLs[targettype] === undefined) {
                            return;
                        }

                        $(this)
                            .children('relation')
                            .each(function () {
                                let reltype = $(this).attr('type');
                                let target = $(this).children('target').text();
                                let url = targettype == 'url' ? target : `/${targettype}/${target}`;

                                if (Object.prototype.hasOwnProperty.call(relationsIconsURLs[targettype], reltype)) {
                                    if (!Object.prototype.hasOwnProperty.call(relations, reltype)) relations[reltype] = [url];
                                    else relations[reltype].push(url);
                                }
                            });

                        $.each(relations, function (reltype, urls) {
                            let html = '';
                            if (urls.length < -1) {
                                html += `<img src='${relationsIconsURLs[targettype][reltype]}' />(${urls.length})&nbsp;`;
                            } else {
                                $.each(urls, function (index, url) {
                                    html += `<a href='${url}'><img src='${relationsIconsURLs[targettype][reltype]}' /></a>&nbsp;`;
                                });
                            }
                            $(`#${mbid} td.relationships`).append(html);
                        });
                    });
            });
    });
});