mcdevs-wiki-extractor 0.2.0 → 0.3.0

package/lib/entities_extractor.js

@@ -1,68 +1,74 @@
-var WikiTextParser = require('parse-wikitext');
-var fs = require('fs');
+const WikiTextParser = require('parse-wikitext')
+const fs = require('fs')
 
-var wikiTextParser = new WikiTextParser('wiki.vg');
+const wikiTextParser = new WikiTextParser('wiki.vg')
 
-module.exports={writeAllEntities:writeAllEntities,getEntities:getEntities};
+module.exports = { writeAllEntities: writeAllEntities, getEntities: getEntities }
 
-var table_parser=require("./common/table_parser");
+const tableParser = require('./common/table_parser')
 
-function getEntities(date,cb)
-{
-  wikiTextParser.getFixedArticle("Entities",date,function(err,pageData,title){
-    var sectionObject=wikiTextParser.pageToSectionObject(pageData);
-    var mobs=sectionObject["Mobs"].content;
-    var objects=sectionObject["Objects"].content;
+function getEntities (date, cb) {
+  wikiTextParser.getFixedArticle('Entity_metadata', date, function (err, pageData, title) {
+    if (err) {
+      cb(err)
+      return
+    }
+    const sectionObject = wikiTextParser.pageToSectionObject(pageData)
+    const mobs = sectionObject.Mobs.content
+    const objects = sectionObject.Objects.content
 
-    var parsed_mobs=table_parser.parseWikiTable(table_parser.getFirstTable(mobs));
-    var parsed_objects=table_parser.parseWikiTable(table_parser.getFirstTable(objects));
+    let parsedMobs = tableParser.parseWikiTable(tableParser.getFirstTable(mobs))
+    let parsedObjects = tableParser.parseWikiTable(tableParser.getFirstTable(objects))
 
-    function transformSize(size) {
-      if(size == 'N/A')
-        return null;
-      var r;
-      if(r=size.match(/^(.+?) \* size/))
-      {
-        return parseFloat(r[1]);
+    function transformSize (size) {
+      if (size === 'N/A') { return null }
+      let r = size.match(/^(.+?) \* size/)
+      if (r) {
+        return parseFloat(r[1])
       }
-      if(r=size.match(/^(.+?) \* (.+?)/))
-      {
-        return parseFloat(r[1])*parseFloat(r[2]);
+      r = size.match(/^(.+?) \* (.+?)/)
+      if (r) {
+        return parseFloat(r[1]) * parseFloat(r[2])
       }
-      return parseFloat(size);
+      return parseFloat(size)
     }
 
-    parsed_mobs=parsed_mobs.map(function(val){
+    parsedMobs = parsedMobs.map(function (val) {
       return {
-        id:parseInt(val["Type"]),
-        displayName:val["Name"],
-        width:transformSize(val["x, z"]),
-        height:transformSize(val["y"]),
-        type:"mob",
-        stringId:val["ID"] ? val["ID"].replace(/<code>(.+)<\/code>/,"$1") : undefined
+        id: parseInt(val.Type),
+        internalId: parseInt(val.Type),
+        name: val.ID ? val.ID.replace(/<code>(.+)<\/code>/, '$1').replace(/minecraft:/g, '') : undefined,
+        displayName: val.Name,
+        width: transformSize(val['bounding box x and z']),
+        height: transformSize(val['bounding box y']),
+        type: 'mob'
       }
-    });
+    })
 
-    parsed_objects=parsed_objects.map(function(val){
+    parsedObjects = parsedObjects.map(function (val) {
       return {
-        id:parseInt(val["ID"]),
-        displayName:val["Name"],
-        width:transformSize(val["x, z"]),
-        height:transformSize(val["y"]),
-        type:"object"
+        id: parseInt(val.ID),
+        internalId: parseInt(val.ID),
+        name: val.Name.replace(/\(.+?\)/, '').trim().replace(/ /, '_').toLowerCase(),
+        displayName: val.Name,
+        width: transformSize(val['bounding box x and z']),
+        height: transformSize(val['bounding box y']),
+        type: 'object'
       }
-    });
-
-    var entities=parsed_mobs.concat(parsed_objects);
+    })
 
+    const entities = parsedMobs.concat(parsedObjects)
 
-    cb(null,entities);
-  });
+    cb(null, entities)
+  })
 }
 
-function writeAllEntities(file,date, cb)
-{
-  getEntities(date,function(err,entities){
-    fs.writeFile(file, JSON.stringify(entities,null,2), cb);
-  });
+function writeAllEntities (file, date, cb) {
+  getEntities(date, function (err, entities) {
+    if (err) {
+      cb(err)
+      return
+    }
+    fs.writeFile(file, JSON.stringify(entities, null, 2), cb)
+  })
 }