Merge pull request #52 from near/lookup-set

add lookup set

src/collections/index.js

@@ -1,7 +1,9 @@
 import { LookupMap } from "./lookup-map";
 import { Vector } from "./vector";
+import { LookupSet } from "./lookup-set";
 
 export {
     LookupMap,
-    Vector
+    Vector,
+    LookupSet
 }

src/collections/lookup-set.js

@@ -0,0 +1,34 @@
+import * as near from '../api'
+
+export class LookupSet {
+    constructor(keyPrefix) {
+        this.keyPrefix = keyPrefix
+    }
+
+    contains(key) {
+        let storageKey = this.keyPrefix + key
+        return near.jsvmStorageHasKey(storageKey)
+    }
+
+    remove(key) {
+        let storageKey = this.keyPrefix + key
+        if (near.jsvmStorageRemove(storageKey)) {
+            return near.storageGetEvicted()
+        }
+        return null
+    }
+
+    set(key) {
+        let storageKey = this.keyPrefix + key
+        if (near.jsvmStorageWrite(storageKey, '')) {
+            return near.storageGetEvicted()
+        }
+        return null
+    }
+
+    extend(kvs) {
+        for(let kv of kvs) {
+            this.set(kv[0])
+        }
+    }
+}

src/index.js

@@ -9,13 +9,19 @@ import {
 } from './near-contract'
 
 import * as near from './api'
-import { LookupMap } from './collections/lookup-map'
+import { 
+    LookupMap,
+    Vector,
+    LookupSet
+} from './collections'
 
 export {
     call,
     view,
     NearBindgen,
     NearContract,
     near,
-    LookupMap
+    LookupMap,
+    Vector,
+    LookupSet
 }