Fleet numbers are now scraped from the Shipyard page and sent to the …

…OGNext instance, closes #15.
skiwi2 · Jan 8, 2016 · 38b6560 · 38b6560
1 parent 2347387
commit 38b6560
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/functions.js b/functions.js
@@ -242,6 +242,44 @@ function processFleetNodes(nodes) {
     });
 }
 
+function processShipyardNodes(nodes) {
+    var shipyard = [];
+
+    for (var i = 0; i < nodes.length; i++) {
+        var node = nodes[i];
+        var detailButtonElement = node.querySelector(".detail_button");
+        if (detailButtonElement !== null) {
+            var fleetId = detailButtonElement.getAttribute("ref");
+            var cloneLevelNode = detailButtonElement.querySelector(".level").cloneNode(true);
+            var children = [].slice.call(cloneLevelNode.children);
+
+            for (var j = 0; j < children.length; j++) {
+                var child = children[j];
+                if (child.className === "textlabel" || child.className === "undermark") {
+                    cloneLevelNode.removeChild(child);
+                }
+            }
+            var fleetAmount = cloneLevelNode.innerHTML.trim();
+
+            shipyard.push({
+                id: fleetId,
+                amount: fleetAmount
+            });
+        }
+    }
+
+    executeIfNotCached("shipyard", shipyard, function () {
+        var data = {
+            shipyard: shipyard
+        }
+        addPlanetData(data);
+        postData({
+            endpoint: "shipyard",
+            data: data
+        });
+    });
+}
+
 function postData(object) {
     addPlayerData(object.data);
     console.log(JSON.stringify(object.data));

diff --git a/ognext.user.js b/ognext.user.js
@@ -127,6 +127,7 @@ function showOGNextSettings() {
         saveSetting(getFullCacheKey("facility_buildings"), "");
         saveSetting(getFullCacheKey("defences"), "");
         saveSetting(getFullCacheKey("fleet"), "");
+        saveSetting(getFullCacheKey("shipyard"), "");
         showSuccessMessage("Settings have been saved.");
     });
 }
@@ -168,6 +169,10 @@ else if (page === "fleet1") {
     var liButtonDivs = document.querySelectorAll("li[id^='button']");
     processFleetNodes(liButtonDivs);
 }
+else if (page === "shipyard") {
+    var itemBoxDivs = document.querySelectorAll(".item_box");
+    processShipyardNodes(itemBoxDivs);
+}
 
 var planetListDiv = document.getElementById("planetList");
 if (planetListDiv !== null) {