Scripts for GNS working for phase1

davekay100 · davekay100 · commit 9a17b4012e40 · 2020-08-07T11:47:18.000-04:00
diff --git a/scripts/contracts/connectedContracts.ts b/scripts/contracts/connectedContracts.ts
@@ -9,7 +9,7 @@ import { StakingFactory } from '../../build/typechain/contracts/StakingContract'
 import { GraphTokenFactory } from '../../build/typechain/contracts/GraphTokenContract'
 import { IEthereumDidRegistryFactory } from '../../build/typechain/contracts/IEthereumDidRegistryContract'
 
-import { getNetworkAddresses, IPFS } from './helpers'
+import { getNetworkAddresses, IPFS, basicOverrides } from './helpers'
 import { connectContracts } from './connectedNetwork'
 import {
   SubgraphMetadata,
@@ -88,36 +88,41 @@ class ConnectedENS extends ConnectedContract {
   setTestRecord = async (name: string): Promise<ContractTransaction> => {
     const contracts = await connectContracts(this.configuredWallet, this.network)
     const normalizedName = name.toLowerCase()
-    const labelNameFull = `${normalizedName}.${'test'}`
+    const labelNameFull = `${normalizedName}.${'eth'}`
     const labelHashFull = utils.namehash(labelNameFull)
     console.log(`Namehash for ${labelNameFull}: ${labelHashFull}`)
     const signerAddress = await contracts.testRegistrar.signer.getAddress()
     const label = utils.keccak256(utils.toUtf8Bytes(normalizedName))
     return contracts.testRegistrar.register(label, signerAddress)
   }
 
-  setText = async (name: string): Promise<ContractTransaction> => {
-    const contracts = await connectContracts(this.configuredWallet, this.network)
-    const normalizedName = name.toLowerCase()
-    const labelNameFull = `${normalizedName}.${'test'}`
-    const labelHashFull = utils.namehash(labelNameFull)
-    console.log(`Setting text name: ${labelNameFull} with node: ${labelHashFull}`)
-    const key = 'GRAPH NAME SERVICE'
-    const signerAddress = await contracts.publicResolver.signer.getAddress()
-    return contracts.publicResolver.setText(labelHashFull, key, signerAddress)
-  }
+  // TODO - remove, not in use anymore
+  // setText = async (name: string): Promise<ContractTransaction> => {
+  //   const contracts = await connectContracts(this.configuredWallet, this.network)
+  //   const normalizedName = name.toLowerCase()
+  //   const labelNameFull = `${normalizedName}.${'eth'}`
+  //   const labelHashFull = utils.namehash(labelNameFull)
+  //   console.log(`Setting text name: ${labelNameFull} with node: ${labelHashFull}`)
+  //   const key = 'GRAPH NAME SERVICE'
+  //   const signerAddress = await contracts.publicResolver.signer.getAddress()
+  //   return contracts.publicResolver.setText(labelHashFull, key, signerAddress)
+  // }
 
   checkOwner = async (name: string): Promise<void> => {
     const contracts = await connectContracts(this.configuredWallet, this.network)
     try {
-      const node = utils.namehash(`${name}.test`)
+      const node = utils.namehash(`${name}.eth`)
       console.log(`Node: ${node}`)
       const res = await contracts.ens.owner(node)
-      console.log(`Owner of ${name}.test is: ${res}`)
+      console.log(`Owner of ${name}.eth is: ${res}`)
     } catch (e) {
       console.log(`  ..failed on checkOwner: ${e.message}`)
     }
   }
+
+  getNode = (name: string): string => {
+    return utils.namehash(`${name}.eth`)
+  }
 }
 
 class ConnectedGraphToken extends ConnectedContract {
@@ -208,7 +213,7 @@ class ConnectedGNS extends ConnectedContract {
     ipfs: string,
     graphAccount: string,
     subgraphDeploymentID: string,
-    versionMetadata: string | SubgraphMetadata,
+    versionMetadata: string | VersionMetadata,
     subgraphMetadata: string | SubgraphMetadata,
   ): Promise<ContractTransaction> => {
     const subgraphDataHash = await this.handleSubgraphMetadata(ipfs, subgraphMetadata)
@@ -226,7 +231,7 @@ class ConnectedGNS extends ConnectedContract {
     ipfs: string,
     graphAccount: string,
     subgraphDeploymentID: string,
-    versionMetadata: string | SubgraphMetadata,
+    versionMetadata: string | VersionMetadata,
     subgraphNumber: string,
   ): Promise<ContractTransaction> => {
     const metaHashBytes = await this.handleVersionMetadata(ipfs, versionMetadata)
@@ -236,6 +241,7 @@ class ConnectedGNS extends ConnectedContract {
       subgraphNumber,
       subgraphDeploymentIDBytes,
       metaHashBytes,
+      basicOverrides(),
     )
   }
 
diff --git a/scripts/contracts/ens.ts b/scripts/contracts/ens.ts
@@ -47,7 +47,7 @@ const main = async () => {
     if (func == 'registerName') {
       console.log(`Setting owner for ${name} and the text record...`)
       await executeTransaction(ens.setTestRecord(name), network)
-      await executeTransaction(ens.setText(name), network)
+      // await executeTransaction(ens.setText(name), network)
     } else if (func == 'checkOwner') {
       console.log(`Checking owner of ${name} ...`)
       await ens.checkOwner(name)
diff --git a/scripts/contracts/ethereumDIDRegistry.ts b/scripts/contracts/ethereumDIDRegistry.ts
@@ -53,7 +53,10 @@ const main = async () => {
   try {
     if (func == 'setAttribute') {
       console.log(`Setting attribute on ethereum DID registry ...`)
-      await executeTransaction(ethereumDIDRegistry.pinIPFSAndSetAttribute(ipfs, metadataPath), network)
+      await executeTransaction(
+        ethereumDIDRegistry.pinIPFSAndSetAttribute(ipfs, metadataPath),
+        network,
+      )
     } else {
       console.log(`Wrong func name provided`)
       process.exit(1)
diff --git a/scripts/contracts/gns.ts b/scripts/contracts/gns.ts
@@ -9,30 +9,37 @@ import {
   configureGanacheWallet,
   configureWallet,
   buildNetworkEndpoint,
+  basicOverrides,
 } from './helpers'
-import { ConnectedGNS } from './connectedContracts'
+import { ConnectedGNS, ConnectedENS, ConnectedGraphToken } from './connectedContracts'
+import { connectContracts } from './connectedNetwork'
+import { Wallet, utils } from 'ethers'
 
 const {
   network,
   func,
   ipfs,
   graphAccount,
   subgraphDeploymentID,
-  nameIdentifier,
   name,
-  metadataPath,
+  versionPath,
+  subgraphPath,
   subgraphNumber,
+  tokens,
+  nSignal,
 } = minimist.default(process.argv.slice(2), {
   string: [
     'network',
     'func',
     'ipfs',
     'graphAccount',
     'subgraphDeploymentID',
-    'nameIdentifier',
     'name',
-    'metadataPath',
+    'versionPath',
+    'subgraphPath',
     'subgraphNumber',
+    'tokens',
+    'nSignal',
   ],
 })
 
@@ -41,35 +48,61 @@ if (!network || !func || !graphAccount) {
     `
 Usage: ${path.basename(process.argv[1])}
   --network  <string> - options: ganache, kovan, rinkeby
-  --func <text> - options: publishNewSubgraph, publishVersion, deprecate
+  --func <text> - options: setDefaultName, publishNewSubgraph, publishVersion, deprecate,
+                           updateSubgraphMetadata, mintNSignal, burnNSignal, withdrawGRT
 
   Function arguments:
+  setDefaultName
+    --graphAccount <address>        - graph account address
+    --nameSystem <number>           - 0 = ENS
+    --name <string>                 - name that has been registered and is getting set as default
+
   publishNewSubgraph
     --ipfs <url>                    - ex. https://api.thegraph.com/ipfs/
-    --graphAccount <address>        - erc1056 identity, often just the transacting account
+    --graphAccount <address>        - graph account address
     --subgraphDeploymentID <base58> - subgraphID in base58
-    --nameIdentifier <string>       - ex. the node value in ENS
-    --name <string>                 - name of the subgraph
-    --metadataPath <path>           - filepath to metadata. JSON format:
+    --versionPath <path>            - filepath to metadata. JSON format:
+                                        {
+                                          "description": "",
+                                          "label": ""
+                                        }
+    --subgraphPath <path>           - filepath to metadata. JSON format:
                                         {
                                           "description": "",
+                                          "displayName": "",
                                           "image": "",
                                           "codeRepository": "",
                                           "website": "",
-                                          "description": "",
-                                          "label": ""
                                         }
+
   publishVersion
     --ipfs <url>                    - ex. https://api.thegraph.com/ipfs/
-    --graphAccount <address>        - erc1056 identity, often just the transacting account
+    --graphAccount <address>        - graph account address
     --subgraphDeploymentID <base58> - subgraphID in base58
-    --nameIdentifier <string>       - ex. the node value in ENS
-    --name <string>                 - name of the subgraph
-    --metadataPath <path>           - filepath to metadata. Same format as above
+    --subgraphPath <path>           - filepath to version metadata. Same format as above
     --subgraphNumber <number>       - numbered subgraph for the graph account
     
   deprecate
-    --graphAccount <address>        - erc1056 identity, often just the transacting account
+    --graphAccount <address>        - graph account address
+    --subgraphNumber <number>       - numbered subgraph for the graph account
+
+  updateSubgraphMetadata
+    --graphAccount <address>        - graph account address
+    --subgraphNumber <number>       - numbered subgraph for the graph account
+    --subgraphPath <path>           - filepath to subgraph metadata. Same format as above
+
+  mintNSignal
+    --graphAccount <address>        - graph account address
+    --subgraphNumber <number>       - numbered subgraph for the graph account
+    --tokens <number>               - tokens being deposited. Script adds 10^18
+  
+  burnNSignal
+    --graphAccount <address>        - graph account address
+    --subgraphNumber <number>       - numbered subgraph for the graph account
+    --nSignal <number>              - tokens being burnt. Script adds 10^18
+
+  withdrawGRT
+    --graphAccount <address>        - graph account address
     --subgraphNumber <number>       - numbered subgraph for the graph account
   `,
   )
@@ -78,64 +111,105 @@ Usage: ${path.basename(process.argv[1])}
 
 const main = async () => {
   let gns: ConnectedGNS
+  let connectedGT: ConnectedGraphToken
   let provider
+  let mnemonicWallet: Wallet
+  const networkContracts = await connectContracts(mnemonicWallet, network)
+
   if (network == 'ganache') {
     provider = buildNetworkEndpoint(network)
+    mnemonicWallet = configureWallet(process.env.MNEMONIC, provider)
     gns = new ConnectedGNS(network, configureGanacheWallet())
+    connectedGT = new ConnectedGraphToken(network, configureGanacheWallet())
   } else {
     provider = buildNetworkEndpoint(network, 'infura')
-    gns = new ConnectedGNS(network, configureWallet(process.env.MNEMONIC, provider))
+    mnemonicWallet = configureWallet(process.env.MNEMONIC, provider)
+    gns = new ConnectedGNS(network, mnemonicWallet)
+    connectedGT = new ConnectedGraphToken(network, configureWallet(process.env.MNEMONIC, provider))
   }
 
   try {
+    if (func == 'setDefaultName') {
+      checkFuncInputs([name], ['name'], func)
+      console.log(`Setting default name for ${name}`)
+      const ens = new ConnectedENS(network, mnemonicWallet)
+      await executeTransaction(
+        networkContracts.gns.setDefaultName(graphAccount, 0, ens.getNode(name), name),
+        network,
+      )
+    }
     if (func == 'publishNewSubgraph') {
       checkFuncInputs(
-        [ipfs, subgraphDeploymentID, nameIdentifier, name, metadataPath],
-        ['ipfs', 'subgraphDeploymentID', 'nameIdentifier', 'name', 'metadataPath'],
-        'publishNewSubgraph',
+        [ipfs, subgraphDeploymentID, versionPath, subgraphPath],
+        ['ipfs', 'subgraphDeploymentID', 'versionPath', 'subgraphPath'],
+        func,
       )
       console.log(`Publishing 1st version of subgraph ${name} ...`)
       await executeTransaction(
         gns.pinIPFSAndNewSubgraph(
           ipfs,
           graphAccount,
           subgraphDeploymentID,
-          nameIdentifier,
-          name,
-          metadataPath,
+          versionPath,
+          subgraphPath,
         ),
         network,
       )
     } else if (func == 'publishNewVersion') {
       checkFuncInputs(
-        [ipfs, subgraphDeploymentID, nameIdentifier, name, metadataPath, subgraphNumber],
-        [
-          'ipfs',
-          'subgraphDeploymentID',
-          'nameIdentifier',
-          'name',
-          'metadataPath',
-          'subgraphNumber',
-        ],
-        'publishNewVersion',
+        [ipfs, subgraphDeploymentID, versionPath, subgraphNumber],
+        ['ipfs', 'subgraphDeploymentID', 'versionPath', 'subgraphNumber'],
+        func,
       )
       console.log(`Publishing a new version for subgraph ${name} ...`)
       await executeTransaction(
         gns.pinIPFSAndNewVersion(
           ipfs,
           graphAccount,
           subgraphDeploymentID,
-          nameIdentifier,
-          name,
-          metadataPath,
+          versionPath,
           subgraphNumber,
         ),
         network,
       )
     } else if (func == 'deprecate') {
-      checkFuncInputs([subgraphNumber], ['subgraphNumber'], 'deprecate')
+      checkFuncInputs([subgraphNumber], ['subgraphNumber'], func)
       console.log(`Deprecating subgraph ${graphAccount}-${subgraphNumber}`)
-      await executeTransaction(gns.gns.deprecate(graphAccount, subgraphNumber), network)
+      await executeTransaction(gns.gns.deprecateSubgraph(graphAccount, subgraphNumber), network)
+    } else if (func == 'updateSubgraphMetadata') {
+      checkFuncInputs([subgraphNumber, subgraphPath], ['subgraphNumber', 'subgraphPath'], func)
+      console.log(`Updating subgraph metadata for ${graphAccount}-${subgraphNumber}`)
+      await executeTransaction(
+        gns.gns.updateSubgraphMetadata(graphAccount, subgraphNumber, subgraphPath),
+        network,
+      )
+    } else if (func == 'mintNSignal') {
+      checkFuncInputs([subgraphNumber, tokens], ['subgraphNumber', 'tokens'], func)
+      console.log(
+        '  First calling approve() to ensure curation contract can call transferFrom()...',
+      )
+      const tokensWithDecimal = utils.parseUnits(tokens as string, 18).toString()
+      await executeTransaction(
+        connectedGT.approveWithDecimals(networkContracts.gns.address, tokensWithDecimal),
+        network,
+      )
+      console.log(`Minting nSignal with ${tokens} on ${graphAccount}-${subgraphNumber}`)
+      await executeTransaction(
+        gns.gns.mintNSignal(graphAccount, subgraphNumber, tokensWithDecimal, basicOverrides()),
+        network,
+      )
+    } else if (func == 'burnNSignal') {
+      checkFuncInputs([subgraphNumber, nSignal], ['subgraphNumber', nSignal], func)
+      console.log(`Burning ${nSignal} nSignal on ${graphAccount}-${subgraphNumber}`)
+      const nSignalWithDecimal = utils.parseUnits(nSignal as string, 18).toString()
+      await executeTransaction(
+        gns.gns.burnNSignal(graphAccount, subgraphNumber, nSignalWithDecimal),
+        network,
+      )
+    } else if (func == 'withdrawGRT') {
+      checkFuncInputs([subgraphNumber], ['subgraphNumber'], func)
+      console.log(`Withdrawing GRT from deprecated subgraph ${graphAccount}-${subgraphNumber}`)
+      await executeTransaction(gns.gns.withdraw(graphAccount, subgraphNumber), network)
     } else {
       console.log(`Wrong func name provided`)
       process.exit(1)
