source: other-projects/hathitrust/wcsa/vol-checker/WebContent/HT-HTRC_Mashup.user.js@ 31350

// ==UserScript==
// @name        HT-HTRC Mashup
// @author      David Bainbridge
// @namespace   org.hathitrust.researchcenter.mashup
// @description Hybrid interface between Hathitrust and HTRC
// @version     1.0
// @grant       GM_xmlhttpRequest
// @require     http://ajax.googleapis.com/ajax/libs/jquery/1.11.3/jquery.min.js
// @match       https://www.hathitrust.org/*
// @match       https://babel.hathitrust.org/cgi/ls*
// @match       https://babel.hathitrust.org/cgi/mb*
// ==/UserScript==


var https_servlet_url="https://bedrock.resnet.cms.waikato.ac.nz/vol-checker/";

function mashupInit() {

    var $fieldset = $('form fieldset').first();
    var $search_button = $fieldset.find('button');
    $search_button.css("width","47px");
    $search_button.css("right","-55px");
    $search_button.css("font-size","95%");
    $search_button.find('span').first().hide();

    $fieldset.append('<button id="htrc-bw" class="button search" style="width:47px; right: -105px; background-image:url('+https_servlet_url + 'bookworm.png);" title="Bookworm"></button>');

    var $bw_button = $('#htrc-bw');
    $bw_button.click(function(event) {
    var input=$('#q1-input').val();
    if (input === "") {
                return;
        }
    event.preventDefault();

    var url_front = "https://bookworm.htrc.illinois.edu/#?%7B%22counttype%22%3A%22WordsPerMillion%22%2C%22search_limits%22%3A%5B";
    var url_end   = "%5D%7D";

    var word_front = "%7B%22word%22%3A%5B%22";
    var word_end   = "%22%5D%2C%22date_year%22%3A%7B%22%24gte%22%3A1750%2C%22%24lte%22%3A1923%7D%7D";

    var input_words = input.split(' ');
    var word_len = input_words.length;

    var full_url = url_front;

    for (var i=0; i<word_len; i++) {
                var word = input_words[i];
                if (i>0) { full_url += "%2C"; } // append comma
                full_url += word_front + word + word_end;
            }

    full_url += url_end;

    // window.location.href = "https://bookworm.htrc.illinois.edu/#?%7B%22search_limits%22%3A%5B%7B%22word%22%3A%5B%22"+input+"%22%5D%2C%22date_year%22%3A%7B%22%24gte%22%3A1750%2C%22%24lte%22%3A1923%7D%7D%5D%7D";

    window.location.href = full_url;

    });

    var url = window.location.href;
    var home_url_test = url.replace(/^https:\/\/www.hathitrust.org\//,"");
    //console.log("*** home urltest = " + home_url_test);
    if (home_url_test !== "") {
    var $login_button = $('#login-button');
    if ($login_button.length) {
                $login_button.css('font-size','95%');
                $login_button.css('right','-45px');
                $login_button.css('width','80px');
            }
    }



    /*
    each(function() {

    $('form button').each(function() {
    console.log("*** this = " + this + "jquery this = " + $(this).val());
    });
    */


}

function mashupAugmentResults()
{
    //console.log("**** mashupAugmentResults() called");
    var $results_a = $('#results_A');
    var results_a_exists = $results_a.length;

    if (!results_a_exists) {
    // Not a main search result page
    // => See if part of the collections page area
    // => If it is, set things up so it will use that instead
        $results_a = $('#form1');
        results_a_exists = $results_a.length;
    }

    if (results_a_exists) {
    //console.log("*** results_a = " + $results_a.html());

    var ids = [];
    //var file_safe_ids = [];

    $results_a.find('.result').each(function() {
                var $result_access_link = $(this).find('.result-access-link');
                //console.log("*** result access link = " + $result_access_link.html());

                var $id_link = $result_access_link.find('ul>li>a').first();
                //console.log("*** id_link = " + $id_link.html());
                var data_clicklog = $id_link.attr("data_clicklog");
                //console.log("*** data clicklog= " + data_clicklog);

                var data_json_str = data_clicklog.replace(/^[a-z]+\|/,"");
                //console.log("*** data json= " + data_json_str);

                var id = null;
                if (data_json_str !== "") {
                var data_json = JSON.parse(data_json_str);
                id = data_json.id;
                //console.log("*** (catalog extracted) id= '" + id + "'");
                    }
                else {
                var $second_id_link = $result_access_link.find('ul>li>a').eq(1);
                console.log("*** 2nd link len = " + $second_id_link);
                var id_href = $second_id_link.attr("href");
                console.log("*** id href = " + id_href);
                id = id_href.replace(/^.*id=/,"");
                    }

                ids.push(id);
                //var file_safe_id = id.replace(/:/g,"+").replace(/\//g,"=");
                //file_safe_ids.push(file_safe_id);

                $result_access_link.attr("id","htrc-mashup-" + id);
                //$result_access_link.attr("id","htrc-mashup-" + file_safe_id);

            });

    var ids_str = ids.join(",");
    //var file_safe_ids_str = file_safe_ids.join(",");

    //console.log("*** ids= " + JSON.stringify(ids));
    //console.log("*** ids_str = " + ids_str);

    console.log("**** Away to request: " + https_servlet_url + "VolumeCheck?ids=" + encodeURIComponent(ids_str));

    var encoded_ids=encodeURIComponent(ids_str);
    var full_url = https_servlet_url + "VolumeCheck?ids=" + encoded_ids;

    $.ajax(full_url)
        .done(function(data,textStatus,response) {
        console.log("Adding in HTRC cross-checks");
        var ids_exist = data;
        for (var k in ids_exist) {
            //console.log("*** k = '" + k + "'");

            //var id_str = '#htrc-mashup-'+k;
            //console.log("*** id str = " + id_str);
            // var $id_div = $(id_str);
            
            //var $id_div = $('#'+'htrc-mashup-'+k);
            var id_div = document.getElementById('htrc-mashup-'+k);
            //console.log("*** id_div = " + id_div);
            var $id_div = $(id_div);

            if (ids_exist[k]) {
            var encoded_id=encodeURIComponent(k);
            var ef_url = https_servlet_url + "VolumeCheck?download-id=" + encoded_id;
            var atag = "<a href=\""+ ef_url +"\"><span class=\"icomoon icomoon-download\"></span>Download Extracted Features</a>";

            $id_div.find("ul").append("<li title=\""+k+"\" style=\"color: #924a0b;\">"+atag+"</li>"); // ✓
            }
            else {
            // ✗, ✘
            $id_div.find("ul").append("<li title=\""+k+"\" style=\"color: red;\">HTRC unfriendly ✘</li>");
            }

            //console.log("*** id div len = " + $id_div.length);

            //$.append("<li>HTRC friendly</li>");
        }
        })
        .fail(function(esponse,textStatus,errorThrown) {
        alert( "error:"  + errorThrown);
        });


    //console.log("*** GM httpRequest made ");
    }

    // Look for collection set download button


    var $col_download_button = $('form button[data-tracking-action="MB Download Metadata');
    if ($col_download_button.length >0) {
        var $download_form=$col_download_button.parent();
        $download_form.append("<button id=\"col-to-workset-download\" class=\"btn btn-mini\" style=\"margin-top: 12px;\"><i class=\"icomoon icomoon-download\"></i> Convert to HTRC Workset</button>");
    var $workset_download_button = $('#col-to-workset-download');
    $workset_download_button.click(function(event) {
            event.preventDefault();

        var action = $download_form.attr("action");
        //var action_url = action;

    var action_url = https_servlet_url + "VolumeCheck";

        // Extract hidden elems from, e.g. 
    //   <input type="hidden" name="c" value="464226859" />
        //   <input type="hidden" name="a" value="download" />
        //   <input type="hidden" name="format" value="text" />
        var $hidden_inputs = $download_form.find("input[type=\"hidden\"]");
        for (var i=0; i<$hidden_inputs.length; i++) {
            var hidden_input = $hidden_inputs[i];

            if (i===0) {
                action_url += "?";
            }
            else {
                action_url += "&";
            }

            var $hidden_input = $(hidden_input);
        var hi_name = $hidden_input.attr("name");
        var hi_val  = $hidden_input.attr("value");
        if (hi_name === "c") {
        hi_name = "convert-col";
        }
            action_url += hi_name + "=" + hi_value;

        }


        console.log("workset download url: " + action_url);
        window.location.href = action_url;
        });

    }


}

mashupInit();

mashupAugmentResults();