NearBindgen refactored to factory decorator

examples/__tests__/test-clean-state.ava.js

@@ -11,9 +11,6 @@ test.beforeEach(async t => {
     // Deploy the clean-state contract.
     const cleanState = await root.devDeploy('./build/clean-state.wasm');
 
-    // Init the contract
-    await cleanState.call(cleanState, 'init', {});
-
     // Save state for test runs, it is unique for each test
     t.context.worker = worker;
     t.context.accounts = {

examples/src/clean-state.js

@@ -1,23 +1,19 @@
-import { NearContract, NearBindgen, call, view, near } from 'near-sdk-js'
+import { NearBindgen, call, view, near } from 'near-sdk-js'
 
-@NearBindgen
-class CleanState extends NearContract {
+@NearBindgen({ requireInit: false })
+class CleanState {
     @call
-    clean({keys}) {
+    clean({ keys }) {
         keys.forEach(key => near.storageRemove(key))
     }
 
     @call
-    put({key, value}) {
+    put({ key, value }) {
         near.storageWrite(key, value)
     }
 
     @view
-    get({key}) {
+    get({ key }) {
         return near.storageRead(key)
     }
-
-    default() {
-        return new CleanState()
-    }
 }

src/build-tools/near-bindgen-exporter.js

@@ -4,7 +4,7 @@ export default function () {
     visitor: {
       ClassDeclaration(path) {
         let classNode = path.node;
-        if (classNode.decorators && classNode.decorators[0].expression.name == 'NearBindgen') {
+        if (classNode.decorators && classNode.decorators[0].expression.callee.name == 'NearBindgen') {
           let classId = classNode.id;
           let contractMethods = {};
           for (let child of classNode.body.body) {

src/index.ts

@@ -1,4 +1,9 @@
-import { call, view, initialize, NearBindgen } from "./near-bindgen";
+import {
+  call,
+  view,
+  initialize,
+  NearBindgen
+} from "./near-bindgen";
 
 import * as near from "./api";
 import {

src/near-bindgen.ts

@@ -9,33 +9,37 @@ export function call(target: Object, key: string | symbol, descriptor: TypedProp
 export function view(target: Object, key: string | symbol, descriptor: TypedPropertyDescriptor<Function>): void {
 }
 
-
-export function NearBindgen<T extends { new(...args: any[]): {} }>(target: T) {
-    return class extends target {
-        static _create() {
-            return new target();
-        }
-
-        static _getState(): Object {
-            const rawState = near.storageRead("STATE");
-            return rawState ? this._deserialize(rawState) : null;
-        }
-
-        static _saveToStorage(obj: Object): void {
-            near.storageWrite("STATE", this._serialize(obj));
-        }
-
-        static _getArgs(): JSON {
-            return JSON.parse(near.input() || "{}");
-        }
-
-        static _serialize(value: Object): string {
-            return JSON.stringify(value);
-        }
-
-        static _deserialize(value: string): Object {
-            return JSON.parse(value);
+export function NearBindgen({ requireInit = false }: { requireInit: boolean }) {
+    return <T extends { new(...args: any[]): {} }>(target: T) => {
+        return class extends target {
+            static _create() {
+                return new target();
+            }
+
+            static _getState(): Object {
+                const rawState = near.storageRead("STATE");
+                return rawState ? this._deserialize(rawState) : null;
+            }
+
+            static _saveToStorage(obj: Object): void {
+                near.storageWrite("STATE", this._serialize(obj));
+            }
+
+            static _getArgs(): JSON {
+                return JSON.parse(near.input() || "{}");
+            }
+
+            static _serialize(value: Object): string {
+                return JSON.stringify(value);
+            }
+
+            static _deserialize(value: string): Object {
+                return JSON.parse(value);
+            }
+
+            static _requireInit(): boolean {
+                return requireInit;
+            }
         }
     }
-}
-
+}