Fix #496 (#497)

* Fix #496 * SchemRegistrar test * bump version * Bump version to 39 * version fix
daostack · Mar 15, 2020 · dbcbd3c · dbcbd3c
1 parent e275cfc
commit dbcbd3c
Show file tree

Hide file tree

Showing 9 changed files with 111 additions and 45 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@daostack/subgraph",
-  "version": "0.0.38-3",
+  "version": "0.0.39-0",
   "author": "DAOstack (https://www.daostack.io)",
   "license": "GPL-3.0",
   "description": "A caching layer for daostack using The Graph",

diff --git a/src/mappings/Controller/mapping.ts b/src/mappings/Controller/mapping.ts
@@ -69,6 +69,7 @@ function insertScheme(
     controllerScheme.numberOfPreBoostedProposals = BigInt.fromI32(0);
     controllerScheme.numberOfBoostedProposals = BigInt.fromI32(0);
     controllerScheme.numberOfExpiredInQueueProposals = BigInt.fromI32(0);
+    controllerScheme.isRegistered = true;
   }
   controllerScheme.dao = avatarAddress.toHex();
   controllerScheme.paramsHash = paramsHash;
@@ -91,11 +92,12 @@ function insertScheme(
   controllerScheme.save();
 }
 
-function deleteScheme(avatarAddress: Address, scheme: Address): void {
-  store.remove(
-    'ControllerScheme',
-    crypto.keccak256(concat(avatarAddress, scheme)).toHex(),
-  );
+function unregisterScheme(avatarAddress: Address, scheme: Address): void {
+  let controllerScheme = ControllerScheme.load(crypto.keccak256(concat(avatarAddress, scheme)).toHex());
+  if (controllerScheme != null) {
+    controllerScheme.isRegistered = false;
+    controllerScheme.save();
+  }
 }
 
 function insertOrganization(
@@ -215,7 +217,7 @@ export function handleRegisterScheme(event: RegisterScheme): void {
 export function handleUnregisterScheme(event: UnregisterScheme): void {
   let controller = Controller.bind(event.address);
   let avatar = controller.avatar();
-  deleteScheme(avatar, event.params._scheme);
+  unregisterScheme(avatar, event.params._scheme);
 
   let ent = new ControllerUnregisterScheme(eventId(event));
   ent.txHash = event.transaction.hash;

diff --git a/src/mappings/Controller/schema.graphql b/src/mappings/Controller/schema.graphql
@@ -28,6 +28,7 @@ type ControllerScheme @entity {
 	numberOfPreBoostedProposals: BigInt!
 	numberOfBoostedProposals: BigInt!
 	numberOfExpiredInQueueProposals: BigInt!
+	isRegistered: Boolean!
 }
 
 type ControllerGlobalConstraint @entity {