background.js

if(localStorage.allowGreybar === "true")
  chrome.downloads.setShelfEnabled(true);
else
  chrome.downloads.setShelfEnabled(false);

var maxTimeLeftInMs=0;

function formatTimeLeft(ms) {
  if (ms < 1000) {
    return '0s';
  }
  var days = parseInt(ms / (24 * 60 * 60 * 1000));
  var hours = parseInt(ms / (60 * 60 * 1000)) % 24;
  if (days) {
    return days+'d';
  }
  var minutes = parseInt(ms / (60 * 1000)) % 60;
  if (hours) {
    return hours+'h';
  }
  var seconds = parseInt(ms / 1000) % 60;
  if (minutes) {
    return minutes+'m';
  }
  return seconds+'s';
}

function setTimeLeft(ms){
	if(ms>maxTimeLeftInMs){
		maxTimeLeftInMs=ms;
	}
}

var lightColors = {
  progressColor: '#0d0',
  arrow: '#333',
  danger: 'red',
  complete: '#8FED24',
  paused: 'grey',
  background: 'white',
  progressBar: '#ddd',
  timeLeft: '#fff',	
}

var darkColors = {
  progressColor: '#0d0',
  arrow: '#333',
  danger: 'red',
  complete: '#8FED24',
  paused: 'grey',
  background: 'white',
  progressBar: '#ddd',
  timeLeft: '#444',
}

var colors;
setColors();
function setColors(){
	if(localStorage.theme == "light"){
		colors = lightColors;
	} else {
		colors = darkColors;
	}
}
function drawLine(ctx, x1, y1, x2, y2) {
  ctx.beginPath();
  ctx.moveTo(x1, y1);
  ctx.lineTo(x2, y2);
  ctx.stroke();
}


function drawProgressSpinner(ctx, stage,color) {
  var center = ctx.canvas.width/2;
  var radius = center*0.9;
  const segments = 16;
  var segArc = 2 * Math.PI / segments;
  ctx.lineWidth = Math.round(ctx.canvas.width*0.1);

	var total_width=ctx.canvas.width;
	var y = ctx.canvas.height-7;

	ctx.fillStyle=color;
	var fill_width=(stage*total_width)/16;
	ctx.fillRect(0,y,fill_width,6);

	ctx.fillStyle=colors.progressBar;
	ctx.fillRect(fill_width,y,total_width-fill_width,6);
}

function drawTimeLeft(ctx){
	ctx.fillStyle=colors.timeLeft;
	ctx.font = "9px Arial";
	ctx.fillText(formatTimeLeft(maxTimeLeftInMs),0,8);
}

function drawArrow(ctx,color) {
  ctx.beginPath();
  ctx.lineWidth = Math.round(ctx.canvas.width*0.15);
  ctx.lineJoin = 'round';
  ctx.strokeStyle = ctx.fillStyle = color;
  var center = ctx.canvas.width/2;
  var minw2 = center*0.20;
  var maxw2 = center*0.60;
  var height2 = maxw2+1;
  ctx.moveTo(center-minw2, center-height2);
  ctx.lineTo(center+minw2, center-height2);
  ctx.lineTo(center+minw2, center);
  ctx.lineTo(center+maxw2, center);
  ctx.lineTo(center, center+height2);
  ctx.lineTo(center-maxw2, center);
  ctx.lineTo(center-minw2, center);
  ctx.lineTo(center-minw2, center-height2);
  ctx.lineTo(center+minw2, center-height2);
  ctx.stroke();
  ctx.fill();
}

function drawDangerBadge(ctx) {
  //setInterval(function() {
        doDrawDangerBadge(ctx,colors.danger);
  //      doDrawDangerBadge(ctx,'#CC0000');
  //}, 500 );  
}

function doDrawDangerBadge(ctx,color) {
  var s = ctx.canvas.width/100;
  ctx.fillStyle = color;
  ctx.strokeStyle = colors.background;
  ctx.lineWidth = Math.round(s*5);
  var edge = ctx.canvas.width-ctx.lineWidth;
  ctx.beginPath();
  ctx.moveTo(s*75, s*55);
  ctx.lineTo(edge, edge);
  ctx.lineTo(s*55, edge);
  ctx.lineTo(s*75, s*55);
  ctx.lineTo(edge, edge);
  ctx.fill();
  ctx.stroke();	
}

function drawPausedBadge(ctx) {
  var s = ctx.canvas.width/100;
  ctx.beginPath();
  ctx.strokeStyle = colors.background;
  ctx.lineWidth = Math.round(s*5);
  ctx.rect(s*30, s*5, s*15, s*45);
  ctx.fillStyle = colors.paused;
  ctx.fill();
  ctx.stroke();
  ctx.rect(s*50, s*5, s*15, s*45);
  ctx.fill();
  ctx.stroke();
}

function drawCompleteBadge(ctx) {
  var s = ctx.canvas.width/100;
  ctx.beginPath();
  ctx.arc(s*75, s*75, s*15, 0, 2*Math.PI, false);
  ctx.fillStyle = colors.complete;
  ctx.fill();
  ctx.strokeStyle = colors.background;
  ctx.lineWidth = Math.round(s*5);
  ctx.stroke();
}

function drawIcon(side, stage, badge) {
	
  var canvas = document.createElement('canvas');
  canvas.width = canvas.height = side;
  document.body.appendChild(canvas);
  var ctx = canvas.getContext('2d');
  
  if(stage=='n'){
	drawArrow(ctx,colors.arrow);
  }
  if (stage != 'n' && badge!='p' && badge!='d') {
    drawProgressSpinner(ctx, stage,colors.complete);
	  drawTimeLeft(ctx);
	
  }
  //drawArrow(ctx);
  if (badge == 'd') {
	  drawArrow(ctx,colors.danger);
    drawDangerBadge(ctx);
  } else if (badge == 'p') {
	drawProgressSpinner(ctx, stage,'yellow');
    drawPausedBadge(ctx);
  } else if (badge == 'c') {
	drawArrow(ctx,colors.complete);
    //drawCompleteBadge(ctx);
  } 
  return canvas;
}

function maybeOpen(id) {
  var openWhenComplete = [];
  try {
    openWhenComplete = JSON.parse(localStorage.openWhenComplete);
  } catch (e) {
    localStorage.openWhenComplete = JSON.stringify(openWhenComplete);
  }
  var openNowIndex = openWhenComplete.indexOf(id);
  if (openNowIndex >= 0) {
    chrome.downloads.open(id);
    openWhenComplete.splice(openNowIndex, 1);
    localStorage.openWhenComplete = JSON.stringify(openWhenComplete);
  }
}

function setBrowserActionIcon(stage, badge) {
	
  var canvas1 = drawIcon(19, stage, badge);
  var canvas2 = drawIcon(38, stage, badge);
  var imageData = {};
  imageData['' + canvas1.width] = canvas1.getContext('2d').getImageData(
        0, 0, canvas1.width, canvas1.height);
  imageData['' + canvas2.width] = canvas2.getContext('2d').getImageData(
        0, 0, canvas2.width, canvas2.height);
  chrome.browserAction.setIcon({imageData:imageData});
  canvas1.parentNode.removeChild(canvas1);
  canvas2.parentNode.removeChild(canvas2);
}

function pollProgress() {
	
  pollProgress.tid = -1;
  chrome.downloads.search({}, function(items) {
    var popupLastOpened = parseInt(localStorage.popupLastOpened);
    var totalTotalBytes = 0;
    var totalBytesReceived = 0;
    var anyMissingTotalBytes = false;
    var anyDangerous = false;
    var anyPaused = false;
    var anyRecentlyCompleted = false;
    var anyInProgress = false;
	maxTimeLeftInMs=0;
    items.forEach(function(item) {
      if (item.state == 'in_progress') {
        anyInProgress = true;
		
		setTimeLeft(new Date(item.estimatedEndTime).getTime() - new Date().getTime());
        if (item.totalBytes) {
          totalTotalBytes += item.totalBytes;
          totalBytesReceived += item.bytesReceived;
		  
        } else {
          anyMissingTotalBytes = true;
        }
        var dangerous = ((item.danger != 'safe') &&
                         (item.danger != 'accepted'));
        anyDangerous = anyDangerous || dangerous;
        anyPaused = anyPaused || item.paused;
      } else if ((item.state == 'complete') && item.endTime && !item.error) {
	
        var ended = (new Date(item.endTime)).getTime();
        var recentlyCompleted = (ended >= popupLastOpened);
        var showNoti = true;
        if(localStorage.notiications === "false")
            showNoti=false;
		if(recentlyCompleted && showNoti) showNotification(item);
        anyRecentlyCompleted = anyRecentlyCompleted || recentlyCompleted;
        maybeOpen(item.id);
      }
    });
    var stage = !anyInProgress ? 'n' : (anyMissingTotalBytes ? 'm' :
      parseInt((totalBytesReceived / totalTotalBytes) * 15)+1);
    var badge = anyDangerous ? 'd' : (anyPaused ? 'p' :
      (anyRecentlyCompleted ? 'c' : ''));
	
    //var targetIcon = stage + ' ' + badge;
    //if (sessionStorage.currentIcon != targetIcon) {
      setBrowserActionIcon(stage, badge);
      //sessionStorage.currentIcon = targetIcon;
    //}

    if (anyInProgress &&
        (pollProgress.tid < 0)) {
      pollProgress.start();
    }
  });
}
pollProgress.tid = -1;
pollProgress.MS = 200;

pollProgress.start = function() {
  if (pollProgress.tid < 0) {
    pollProgress.tid = setTimeout(pollProgress, pollProgress.MS);
  }
};

function isNumber(n) {
  return !isNaN(parseFloat(n)) && isFinite(n);
}

if (!isNumber(localStorage.popupLastOpened)) {
  localStorage.popupLastOpened = '' + (new Date()).getTime();
}

chrome.downloads.onCreated.addListener(function(item) {
  pollProgress();
});

pollProgress();

function openWhenComplete(downloadId) {
  var ids = [];
  try {
    ids = JSON.parse(localStorage.openWhenComplete);
  } catch (e) {
    localStorage.openWhenComplete = JSON.stringify(ids);
  }
  pollProgress.start();
  if (ids.indexOf(downloadId) >= 0) {
    return;
  }
  ids.push(downloadId);
  localStorage.openWhenComplete = JSON.stringify(ids);
}

function showNotification(item){
	item.basename = item.filename.substring(Math.max(
      item.filename.lastIndexOf('\\'),
      item.filename.lastIndexOf('/')) + 1);
	if(item.basename.length>50){
		item.basename = item.basename.substr(0,40)+"..."+item.basename.substr(item.basename.length-7,item.basename.length);
	}
	   
	chrome.downloads.getFileIcon(
      item.id,
      {'size':32},
      function(icon_url) {
	   var icon="icon19.png"
	  if(icon_url!==undefined) icon= icon_url;
		var opt = {
			type: "basic",
			title: item.basename,
			message: "Download Complete!",
			eventTime : Date.now() + 5000,
			iconUrl: "icon128.png"
		}

	chrome.notifications.create(""+item.id, opt, function(id){setTimeout(function(){
chrome.notifications.clear(id, function(wasCleared){} );
},5000);});
	
    });
}

chrome.downloads.onChanged.addListener(function(e) {
	if (e.filename && e.filename.previous == '') {
		//console.log(localStorage.animation);
		if(!localStorage.animation || localStorage.animation==="true"){
			sendAnimMsg('safe');
		}
	}
	if (e.danger && e.danger.current != 'safe') {
		console.log('download not safe ' + e.id)
		sendAnimMsg('danger');
	}
});

function sendAnimMsg(msg) {
	console.log('sending msg');
	chrome.tabs.query({active: true}, function (tabs) {
		tabs.forEach(function (tab) {
			chrome.tabs.sendMessage(tab.id, msg);
			console.log('msg send');
		})
	})
}

chrome.notifications.onClicked.addListener(function(itemId){
	if(localStorage.notificationAction == "open-folder"){
		chrome.downloads.show(parseInt(itemId));			
	} else {
		chrome.downloads.open(parseInt(itemId));		
	}
	chrome.notifications.clear(itemId, function(wasCleared){} );
	localStorage.popupLastOpened = Date.now();
});

chrome.notifications.onClosed.addListener(function(notificationId,byUser){
	//chrome.notifications.clear(notificationId, function(wasCleared){} );
	localStorage.popupLastOpened = Date.now();
});

chrome.runtime.onMessage.addListener(function(request) {
  if(request=='theme'){
	setColors();
	return;
  }
  if (request == 'poll') {
    pollProgress.start();
  }
  if (request == 'icons') {
    [16, 19, 38, 128].forEach(function(s) {
      var canvas = drawIcon(s, 'n', '');
      chrome.downloads.download({
        url: canvas.toDataURL('image/png', 1.0),
        filename: 'icon' + s + '.png',
      });
      canvas.parentNode.removeChild(canvas);
    });
  }
  if (isNumber(request.openWhenComplete)) {
    openWhenComplete(request.openWhenComplete);
  }
});