From c2f37f784e7effcd83c0edd795d50ec7969b02f6 Mon Sep 17 00:00:00 2001
From: 10001shh <sghsri@gmail.com>
Date: Thu, 10 Jan 2019 17:18:58 -0600
Subject: [PATCH] badge number

---
 js/background.js | 12 ++++++++++--
 js/content.js    |  2 +-
 js/popup.js      |  2 ++
 3 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/js/background.js b/js/background.js
index 5cbdd454..0f552aea 100644
--- a/js/background.js
+++ b/js/background.js
@@ -1,6 +1,13 @@
+chrome.storage.sync.get('savedCourses', function (data) {
+	if (data.savedCourses && data.savedCourses.length) {
+		chrome.browserAction.setBadgeBackgroundColor({
+			color: '#bf5700'
+		});
+		chrome.browserAction.setBadgeText({ text: "" + data.savedCourses.length });
+	}
+});
 /* Handle messages and their commands from content and popup scripts*/
 chrome.runtime.onMessage.addListener(function (request, sender, response) {
-
 	switch (request.command) {
 		case "courseStorage":
 			if (request.action == "add") {
@@ -150,13 +157,13 @@ function add(request, sender, sendResponse) {
 				savedCourses: courses
 			});
 		}
+		chrome.browserAction.setBadgeText({ text: "" + courses.length });
 		sendResponse({
 			done: "Added: (" + request.course.unique + ") " + request.course.coursename,
 			label: "Remove Course -"
 		});
 	});
 }
-
 /* Find and Remove the requested course from the storage*/
 function remove(request, sender, sendResponse) {
 	chrome.storage.sync.get('savedCourses', function (data) {
@@ -170,6 +177,7 @@ function remove(request, sender, sendResponse) {
 		chrome.storage.sync.set({
 			savedCourses: courses
 		});
+		chrome.browserAction.setBadgeText({ text: "" + courses.length });
 		sendResponse({
 			done: "Removed: (" + request.course.unique + ") " + request.course.coursename,
 			label: "Add Course +"
diff --git a/js/content.js b/js/content.js
index d3561da3..2750a077 100644
--- a/js/content.js
+++ b/js/content.js
@@ -191,7 +191,7 @@ function saveCourse() {
 		$("#snackbar").text(response.done);
 		setTimeout(function () {
 			$("#snackbar").attr("class", "show");
-		}, 150);
+		}, 200);
 		setTimeout(function () {
 			$("#snackbar").attr("class", "");
 		}, 3000);
diff --git a/js/popup.js b/js/popup.js
index b3811d0d..505905a8 100644
--- a/js/popup.js
+++ b/js/popup.js
@@ -261,6 +261,7 @@ $("#importOrig").change(function (e) {
 				chrome.storage.sync.set({
 					savedCourses: impCourses
 				});
+				chrome.browserAction.setBadgeText({ text: "" + impCourses.length });
 				chrome.tabs.query({}, function (tabs) {
 					for (var i = 0; i < tabs.length; i++) {
 						chrome.tabs.sendMessage(tabs[i].id, {
@@ -369,6 +370,7 @@ function clear() {
 			});
 		}
 	});
+	chrome.browserAction.setBadgeText({ text: "" });
 	$("#courseList").empty()
 	console.log("cleared");
 	showEmpty();