Merge pull request #35 from Tianweiii/wilsonDev

Add back account entity

Author: wilsonhow123

packages/subgraph/lyra-token/build/LyraToken/LyraToken.wasm

@@ -44,3 +44,8 @@ type Transfer @entity(immutable: true) {
   blockTimestamp: BigInt!
   transactionHash: Bytes!
 }
+
+type Account @entity(immutable: false) {
+  id: Bytes!           # address
+  balance: BigInt!     # latest token balance
+}

packages/subgraph/lyra-token/build/schema.graphql

@@ -578,3 +578,58 @@ export class Transfer extends Entity {
     this.set("transactionHash", Value.fromBytes(value));
   }
 }
+
+export class Account extends Entity {
+  constructor(id: Bytes) {
+    super();
+    this.set("id", Value.fromBytes(id));
+  }
+
+  save(): void {
+    let id = this.get("id");
+    assert(id != null, "Cannot save Account entity without an ID");
+    if (id) {
+      assert(
+        id.kind == ValueKind.BYTES,
+        `Entities of type Account must have an ID of type Bytes but the id '${id.displayData()}' is of type ${id.displayKind()}`,
+      );
+      store.set("Account", id.toBytes().toHexString(), this);
+    }
+  }
+
+  static loadInBlock(id: Bytes): Account | null {
+    return changetype<Account | null>(
+      store.get_in_block("Account", id.toHexString()),
+    );
+  }
+
+  static load(id: Bytes): Account | null {
+    return changetype<Account | null>(store.get("Account", id.toHexString()));
+  }
+
+  get id(): Bytes {
+    let value = this.get("id");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBytes();
+    }
+  }
+
+  set id(value: Bytes) {
+    this.set("id", Value.fromBytes(value));
+  }
+
+  get balance(): BigInt {
+    let value = this.get("balance");
+    if (!value || value.kind == ValueKind.NULL) {
+      throw new Error("Cannot return null for a required field.");
+    } else {
+      return value.toBigInt();
+    }
+  }
+
+  set balance(value: BigInt) {
+    this.set("balance", Value.fromBigInt(value));
+  }
+}

packages/subgraph/lyra-token/generated/schema.ts

@@ -44,3 +44,8 @@ type Transfer @entity(immutable: true) {
   blockTimestamp: BigInt!
   transactionHash: Bytes!
 }
+
+type Account @entity(immutable: false) {
+  id: Bytes!           # address
+  balance: BigInt!     # latest token balance
+}

packages/subgraph/lyra-token/schema.graphql

@@ -1,3 +1,4 @@
+import { BigInt, Bytes } from "@graphprotocol/graph-ts"
 import {
   Approval as ApprovalEvent,
   GovernmentUpdated as GovernmentUpdatedEvent,
@@ -10,9 +11,20 @@ import {
   GovernmentUpdated,
   MerchantUpdated,
   OwnershipTransferred,
-  Transfer
+  Transfer,
+  Account
 } from "../generated/schema"
 
+function getOrCreateAccount(address: Bytes): Account {
+  let account = Account.load(address)
+  if (account == null) {
+    account = new Account(address) // now it's Bytes, not string
+    account.balance = BigInt.zero()
+    account.save()
+  }
+  return account
+}
+
 export function handleApproval(event: ApprovalEvent): void {
   let entity = new Approval(
     event.transaction.hash.concatI32(event.logIndex.toI32())
@@ -85,4 +97,19 @@ export function handleTransfer(event: TransferEvent): void {
   entity.transactionHash = event.transaction.hash
 
   entity.save()
+
+  const zeroAddress = Bytes.fromHexString("0x0000000000000000000000000000000000000000");
+
+  // Update balances
+  if (event.params.from != zeroAddress) {
+    let fromAccount = getOrCreateAccount(event.params.from)
+    fromAccount.balance = fromAccount.balance.minus(event.params.value)
+    fromAccount.save()
+  }
+
+  if (event.params.to != zeroAddress) {
+    let toAccount = getOrCreateAccount(event.params.to)
+    toAccount.balance = toAccount.balance.plus(event.params.value)
+    toAccount.save()
+  }
 }