// ==UserScript==
// @name        mmmturkeybacon Seven Days Dashboard and Weekly Total
// @author      mmmturkeybacon
// @description Puts the last seven days, whether they were worked on or not, on the dashboard. Puts the sum of the columns for the week in weekly total fields. Let's you choose which day is the start of the week for calculating the weekly total (the same day is used by mmmturkeybacon Expected Earnings - Projected Earnings This Week). Copy the totals line to generate your own weekly activity report.
// @namespace   http://userscripts.org/users/523367
// @match       https://www.mturk.com/mturk/dashboard
// @require     https://ajax.googleapis.com/ajax/libs/jquery/2.1.3/jquery.min.js
// @version     2.02
// @grant       GM_log
// @downloadURL none
// ==/UserScript==

if (localStorage.getItem('mmmturkeybacon_seven_days_dashboard_start_of_week') == null)
{
    localStorage.setItem('mmmturkeybacon_seven_days_dashboard_start_of_week', '0');
}

$(document).ready(function()
{
    var $src;
    var months_short = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];

    function append_extra_days()
    {
        var $last_status_line_link = $('a[href^="/mturk/statusdetail"]').eq(4);
        var last_status_line_date = $last_status_line_link.attr('href').substring(32);

        var six_days_ago = new Date();
        six_days_ago.setMinutes(six_days_ago.getMinutes() - (480 - six_days_ago.getTimezoneOffset())); // set to mturk timezone
        six_days_ago.setDate(six_days_ago.getDate() - 6);
        //http://stackoverflow.com/questions/3605214/javascript-add-leading-zeroes-to-date
        var six_days_ago_date = ('0' + (six_days_ago.getMonth()+1)).slice(-2) + ('0' + six_days_ago.getDate()).slice(-2) + six_days_ago.getFullYear();

        var $extra_days = $src.find('a[href^="/mturk/statusdetail"]').filter(function(){return ($(this).attr('href').substring(32) >= six_days_ago_date && $(this).attr('href').substring(32) < last_status_line_date);});

        for (var i = $extra_days.length-1; i >= 0; i--)
        {
            $extra_days.eq(i).parent().attr('class', 'metrics-table-first-value');
            $last_status_line_link.parent().parent().after($extra_days.eq(i).parent().parent()[0].outerHTML);
        }
    }

    function insert_missing_days()
    {
        var date = new Date();
        date.setMinutes(date.getMinutes() - (480 - date.getTimezoneOffset())); // set to mturk timezone
        var date_today = date.getDate();

        var $insertion_ref = $('th[id="user_activities.date_column_header.tooltip"][class="metrics-table-first-header"]').parent();
        for (var i = 0; i < 7; i++)
        {
            var date_zero_formatted = ('0' + (date.getMonth()+1)).slice(-2) + ('0' + date.getDate()).slice(-2) + date.getFullYear();
            var date_short_formatted = months_short[date.getMonth()] + ' ' + date.getDate() + ', ' + date.getFullYear();

            if (i == 0)
            {
                date_short_formatted = 'Today';
            }

            var $status_line_link = $('a[href$="'+date_zero_formatted+'"]');
            if ($status_line_link.length <= 0)
            {
                $insertion_ref.after('<tr class="even"><td class="metrics-table-first-value"><a href="/mturk/statusdetail?encodedDate='+date_zero_formatted+'">'+date_short_formatted+'</a></td><td>0</td><td>0</td><td>0</td><td>0</td><td><span class="reward">$0.00</span></td></tr>')
                $insertion_ref = $('a[href$="'+date_zero_formatted+'"]').parent().parent();
            }
            else
            {
                $insertion_ref = $status_line_link.parent().parent();
            }
            date.setDate(date.getDate() - 1);
        }
    }

    function clean_up()
    {
        var $status_line_links = $('a[href^="/mturk/statusdetail"]');
        for (var i = $status_line_links.length-1; i >= 7; i--)
        { // remove all status lines that didn't fall in the last 7 days
            $status_line_links.eq(i).parent().parent().remove();
        }
        
        $('th[id="user_activities.date_column_header.tooltip"][class="metrics-table-first-header"]').parent().nextAll('tr').each(function(index)
        {
            $(this).attr('class', ((index % 2 == 0) ? 'even' : 'odd'));
        });
    }

    function append_weekly_total()
    {
        $('tr[id="mtb_total_line"]').remove();

        var week_start = new Date();
        week_start.setMinutes(week_start.getMinutes() - (480 - week_start.getTimezoneOffset())); // set to mturk timezone
        week_start.setDate(week_start.getDate() - ((7 - Number(localStorage.getItem('mmmturkeybacon_seven_days_dashboard_start_of_week')) + week_start.getDay()) % 7));

        var week_start_date_zero_formatted = ('0' + (week_start.getMonth()+1)).slice(-2) + ('0' + week_start.getDate()).slice(-2) + week_start.getFullYear();
        var week_start_date_short = months_short[week_start.getMonth()] + ' ' + week_start.getDate() + ', ' + week_start.getFullYear();

        var submitted_total = 0;
        var approved_total = 0;
        var rejected_total = 0;
        var pending_total = 0;
        var earnings_total = 0;
        var $this_week = $src.find('a[href^="/mturk/statusdetail"]').filter(function(){return ($(this).attr('href').substring(32) >= week_start_date_zero_formatted);});
        $this_week.each(function()
        {
            submitted_total += parseInt($(this).parent().parent().find('td:eq(1)').text(), 10);
            approved_total  += parseInt($(this).parent().parent().find('td:eq(2)').text(), 10);
            rejected_total  += parseInt($(this).parent().parent().find('td:eq(3)').text(), 10);
            pending_total   += parseInt($(this).parent().parent().find('td:eq(4)').text(), 10);
            earnings_total  += parseInt($(this).parent().parent().find('span[class="reward"]').text().replace(/[^0-9]/g,''), 10);
        });

        $('a[href^="/mturk/statusdetail"]').last().parent().parent().after('<tr id="mtb_total_line" class="grayHead"><td style="font: bold 12px arial, sans-serif; color: #369; width: 60%; text-align: left; padding-left: 5px; vertical-align: middle">Totals For The Week Beginning '+week_start_date_short+' <select id="mtb_day_select"><option value="0">Sunday</option><option value="1">Monday</option><option value="2">Tuesday</option><option value="3">Wednesday</option><option value="4">Thursday</option><option value="5">Friday</option><option value="6">Saturday</option></select></td><td valign="center" style="font: bold 12px arial, sans-serif; color: #369; vertical-align: middle"><img src="#" height="0" width="0" alt="submitted: ">'+submitted_total+'</td><td style="font: bold 12px arial, sans-serif; color: #369; vertical-align: middle"><img src="#" height="0" width="0" alt="approved: ">'+approved_total+'</td><td style="font: bold 12px arial, sans-serif; color: #369; vertical-align: middle"><img src="#" height="0" width="0" alt="rejected: ">'+rejected_total+'</td><td style="font: bold 12px arial, sans-serif; color: #369; vertical-align: middle"><img src="#" height="0" width="0" alt="pending: ">'+pending_total+'</td><td style="font: bold 12px arial, sans-serif; color: #369; vertical-align: middle"><span class="reward"><img src="#" height="0" width="0" alt="earnings: ">'+'$' + (earnings_total/100).toFixed(2)+'</span></td></tr>');

        $('select[id="mtb_day_select"]').prop('selectedIndex', Number(localStorage.getItem('mmmturkeybacon_seven_days_dashboard_start_of_week')));
        $('select[id="mtb_day_select"]')[0].onchange = function(){localStorage.setItem('mmmturkeybacon_seven_days_dashboard_start_of_week', this.selectedIndex); append_weekly_total();};
    }


    $.get('/mturk/status', function(data)
    {
        $src = $(data);
        var maxpagerate = $src.find("td[class='error_title']:contains('You have exceeded the maximum allowed page request rate for this website.')");
        if (maxpagerate.length == 0)
        {
            append_extra_days();
            insert_missing_days();
            clean_up();
            append_weekly_total();
        }
    });
});