1 files changed, 41 insertions, 0 deletions
diff --git a/lib/filterSource.js b/lib/filterSource.js
new file mode 100644
index 0000000..bf5d113
--- /dev/null
+++ b/lib/filterSource.js
@@ -0,0 +1,41 @@
+// whitelist of building unit types to include
+// other building unit types like carspace we skip
+const buildingUnitTypeWhitelist = [
+  'APT', // apartment
+  'BLDG', // building
+  'CHAL', // chalet
+  'CTGE', // cottage
+  'FLAT', // flat
+  'HSE', // house
+  'OFFC', // office
+  'SAPT', // studio apartment
+  'SE', // suite
+  'SHOP', // shop
+  'STR', // strata unit
+  'TNHS', // townhouse
+  'UNIT', // unit
+  'VLLA' // villa
+]
+
+/**
+ * Filters features based on the source schema
+ * 
+ * @param {Object} feature 
+ * @returns {boolean}
+ */
+module.exports = (feature) => {
+
+  // if the address has a building unit type, only allow a few whitelisted types
+  if ('BLGUNTTYP' in feature.properties) {
+    if (buildingUnitTypeWhitelist.includes(feature.properties.BLGUNTTYP)) {
+      // building unit type in the whitelist, include feature
+      return true
+    } else {
+      // building unit type is set and not in the whitelist, drop feature
+      return false
+    }
+  } else {
+    // building unit type not set, include feature
+    return true
+  }
+}