HairOnScreen/background.js at master · 0x00019913/HairOnScreen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
// generate the context menus brought up by clicking the browser action
chrome.storage.sync.get("numHairImages", function(items) {
  var hairCount = items.hasOwnProperty("numHairImages") ? items.numHairImages : 1;

  chrome.contextMenus.removeAll();
  chrome.contextMenus.create({
    title: hairCountText(hairCount),
    id: "hair_count",
    contexts: ["action"],
    enabled: false
  });
  chrome.contextMenus.create({
    title: "More hair!",
    id: "more_hair",
    contexts: ["action"]
  });
  chrome.contextMenus.create({
    title: "Less hair!",
    id: "less_hair",
    contexts: ["action"]
  });
  chrome.contextMenus.create({
    title: "Reset hair!",
    id: "reset_hair",
    contexts: ["action"]
  });
  chrome.contextMenus.create({
    title: "Randomize hair!",
    id: "randomize_hair",
    contexts: ["action"]
  });
  chrome.contextMenus.create({
    title: "(reload after updating)",
    id: "reload_prompt",
    contexts: ["action"],
    enabled: false
  });
});

// handle context menu clicks
chrome.contextMenus.onClicked.addListener(function (info) {
  if (info.menuItemId == "more_hair") incrementNumImages();
  else if (info.menuItemId == "less_hair") decrementNumImages();
  else if (info.menuItemId == "randomize_hair") randomizeHair();
  else if (info.menuItemId == "reset_hair") resetNumImages();
});

// displays count
function hairCountText(num) {
  return "Hair count: " + num;
}

// general function for updating the hair count
function updateNumImages(updateFunction) {
  chrome.storage.sync.get("numHairImages", function(items) {
    var numHairImages;
    if (items.hasOwnProperty("numHairImages")) {
      numHairImages = updateFunction(items.numHairImages);
    }
    else {
      numHairImages = updateFunction(1);
    }

    chrome.storage.sync.set( {"numHairImages": numHairImages } );

    chrome.contextMenus.update("hair_count", { title: hairCountText(numHairImages) });
  });
}

// increment count
function incrementNumImages() {
  updateNumImages(function(num) {
    return num + 1;
  });
}

// decrement count
function decrementNumImages() {
  updateNumImages(function(num) {
    return Math.max(0, num-1);
  });
}

// reset count to 1
function resetNumImages() {
  updateNumImages( function() { return 1; } );
}

// randomize positions
function randomizeHair() {
  chrome.storage.sync.set( { "hairImages": [] } );
}