Added build call handling to get account module data for smart accounts

schema.graphql

@@ -11,9 +11,11 @@ type SmartAccount @entity {
 
   owner: User!
 
-  creator: User!
+  creator: User
 
   origin: String!
+
+  accountModule: AccountModule
 }
 
 type AccountModule @entity {
@@ -27,6 +29,8 @@ type AccountModule @entity {
   check: String
 
   instaIndex: InstaIndex!
+
+  accountsCloned: [SmartAccount!]! @derivedFrom(field: "accountModule")
 }
 
 type InstaIndex @entity {

src/index.ts

@@ -4,5 +4,6 @@ export {
   handleLogNewCheck,
   handleLogNewMaster,
   handleLogUpdateMaster,
-  handleSetBasics
+  handleSetBasics,
+  handleBuild
 } from "./mappings/instaIndex";

src/mappings/instaIndex.ts

@@ -6,6 +6,7 @@ import {
   LogNewMaster,
   LogUpdateMaster,
   SetBasicsCall,
+  BuildCall,
   InstaIndex
 } from "../../generated/InstaIndex/InstaIndex";
 import {
@@ -13,7 +14,7 @@ import {
   getOrCreateUser,
   getOrCreateSmartAccount,
   getOrCreateInstaIndex
-} from '../utils/helpers'
+} from "../utils/helpers";
 
 // - event: LogAccountCreated(address,indexed address,indexed address,indexed address)
 //   handler: handleLogAccountCreated
@@ -21,9 +22,11 @@ import {
 //  event LogAccountCreated(address sender, address indexed owner, address indexed account, address indexed origin);
 
 export function handleLogAccountCreated(event: LogAccountCreated): void {
-  let smartAccount = getOrCreateSmartAccount(event.params.account.toHexString());
-  let owner = getOrCreateUser(event.params.owner.toHexString())
-  let sender = getOrCreateUser(event.params.sender.toHexString())
+  let smartAccount = getOrCreateSmartAccount(
+    event.params.account.toHexString()
+  );
+  let owner = getOrCreateUser(event.params.owner.toHexString());
+  let sender = getOrCreateUser(event.params.sender.toHexString());
 
   smartAccount.owner = owner.id;
   smartAccount.creator = sender.id;
@@ -40,7 +43,7 @@ export function handleLogAccountCreated(event: LogAccountCreated): void {
 export function handleLogNewAccount(event: LogNewAccount): void {
   // current account version has to be retrieved from the contract
   let accountVersion = InstaIndex.bind(event.address).versionCount();
-  let accountModule = getOrCreateAccountModule(accountVersion.toString())
+  let accountModule = getOrCreateAccountModule(accountVersion.toString());
 
   accountModule.address = event.params._newAccount.toHexString();
   accountModule.connectors = event.params._connectors.toHexString();
@@ -53,7 +56,9 @@ export function handleLogNewAccount(event: LogNewAccount): void {
 //   handler: handleLogNewCheck
 
 export function handleLogNewCheck(event: LogNewCheck): void {
-  let accountModule = getOrCreateAccountModule(event.params.accountVersion.toString());
+  let accountModule = getOrCreateAccountModule(
+    event.params.accountVersion.toString()
+  );
 
   accountModule.check = event.params.check.toHexString();
 
@@ -82,15 +87,25 @@ export function handleLogUpdateMaster(event: LogUpdateMaster): void {
   index.save();
 }
 
-// - event: LogUpdateMaster(indexed address)
-//   handler: handleLogUpdateMaster
-
 export function handleSetBasics(call: SetBasicsCall): void {
   let accountVersion = InstaIndex.bind(call.to).versionCount();
-  let accountModule = getOrCreateAccountModule(accountVersion.toString())
+  let accountModule = getOrCreateAccountModule(accountVersion.toString());
 
   accountModule.address = call.inputs._account.toHexString();
   accountModule.connectors = call.inputs._connectors.toHexString();
 
   accountModule.save();
 }
+
+export function handleBuild(call: BuildCall): void {
+  let smartAccount = getOrCreateSmartAccount(
+    call.outputs._account.toHexString()
+  );
+  let owner = getOrCreateUser(call.inputs._owner.toHexString());
+
+  smartAccount.owner = owner.id;
+  smartAccount.origin = call.inputs._origin.toHexString();
+  smartAccount.accountModule = call.inputs.accountVersion.toString();
+
+  smartAccount.save();
+}

subgraph.yaml

@@ -36,3 +36,5 @@ dataSources:
       callHandlers:
         - function: setBasics(address,address,address,address)
           handler: handleSetBasics
+        - function: build(address,uint256,address)
+          handler: handleBuild