Track colony versions in the database

amplify/backend/api/colonycdapp/schema.graphql

@@ -45,6 +45,7 @@ input CreateUniqueColonyInput {
   type: ColonyType
   status: ColonyStatusInput
   meta: MetadataInput
+  version: Int!
 }
 
 input ProfileInput {
@@ -82,7 +83,7 @@ input UpdateExtensionByColonyAndHashInput {
 }
 
 input SetCurrentVersionInput {
-  item: String!
+  key: String!
   version: Int!
 }
 
@@ -224,6 +225,7 @@ type Colony @model {
   balances: ColonyBalances @function(name: "fetchColonyBalances")
   meta: Metadata
   extensions: [ColonyExtension!] @hasMany(indexName: "byColony", fields: ["id"])
+  version: Int!
 }
 
 type User @model {
@@ -331,6 +333,6 @@ type ColonyExtension @model {
 
 type CurrentVersion @model {
   id: ID!
-  item: String! @index(name: "byItem", queryField: "getCurrentVersionByItem")
+  key: String! @index(name: "byKey", queryField: "getCurrentVersionByKey")
   version: Int!
 }

amplify/backend/function/createUniqueColony/src/index.js

@@ -22,6 +22,7 @@ exports.handler = async (event) => {
     profile,
     colonyNativeTokenId,
     type = 'COLONY',
+    version,
   } = event.arguments?.input || {};
 
   /*
@@ -140,6 +141,7 @@ exports.handler = async (event) => {
           chainId: 2656691,
           network: 'GANACHE',
         },
+        version,
       },
     },
     GRAPHQL_URI,

amplify/backend/function/setCurrentVersion/src/graphql.js

@@ -1,7 +1,7 @@
 module.exports = {
   getCurrentVersion: /* GraphQL */ `
-    query GetCurrentVersion($item: CurrentVersionItem!) {
-      getCurrentVersionByItem(item: $item) {
+    query GetCurrentVersion($key: String!) {
+      getCurrentVersionByKey(key: $key) {
         items {
           id
         }

amplify/backend/function/setCurrentVersion/src/index.js

@@ -17,19 +17,19 @@ const GRAPHQL_URI = 'http://localhost:20002/graphql';
  * @type {import('@types/aws-lambda').APIGatewayProxyHandler}
  */
 exports.handler = async (event) => {
-  const { item, version } = event.arguments?.input || {};
+  const { key, version } = event.arguments?.input || {};
 
   const getCurrentVersionData = await graphqlRequest(
     getCurrentVersion,
     {
-      item,
+      key,
     },
     GRAPHQL_URI,
     API_KEY,
   );
 
   const existingEntryId =
-    getCurrentVersionData?.data?.getCurrentVersionByItem?.items?.[0]?.id;
+    getCurrentVersionData?.data?.getCurrentVersionByKey?.items?.[0]?.id;
 
   if (existingEntryId) {
     const updateCurrentVersionData = await graphqlRequest(
@@ -51,7 +51,7 @@ exports.handler = async (event) => {
     createCurrentVersion,
     {
       input: {
-        item,
+        key,
         version,
       },
     },

package.json

@@ -33,7 +33,7 @@
   },
   "homepage": "https://colony.io",
   "devDependencies": {
-    "@aws-amplify/cli": "^10.2.1",
+    "@aws-amplify/cli": "^10.6.1",
     "@babel/core": "^7.13.10",
     "@babel/node": "^7.13.10",
     "@babel/preset-env": "^7.13.10",

scripts/temp-create-data.js

@@ -1,9 +1,9 @@
-const { utils, Wallet, Contract, ContractFactory, providers } = require('ethers');
+const { utils, Wallet, Contract, ContractFactory, providers, BigNumber } = require('ethers');
 
 const colonyJSExtras = require('@colony/colony-js/extras')
 const colonyJSIColony = require('../node_modules/@colony/colony-js/dist/cjs/contracts/IColony/9/factories/IColony__factory.js')
 const { addAugmentsB } = require('../node_modules/@colony/colony-js/dist/cjs/clients/Core/augments/AddDomain.js');
-const { getExtensionHash } =require('@colony/colony-js');
+const { getExtensionHash } = require('@colony/colony-js');
 
 /*
  * @NOTE To preserve time, I just re-used a script I wrote for one of the lambda functions
@@ -57,7 +57,7 @@ const createUniqueDomain = /* GraphQL */ `
 const setCurrentVersion = /* GraphQL */`
   mutation SetCurrentVersion($extensionHash: String!, $version: Int!) {
     setCurrentVersion(input: {
-      item: $extensionHash,
+      key: $extensionHash,
       version: $version
     })
   }
@@ -230,6 +230,7 @@ const createMetacolony = async (singerOrWallet) => {
         name: 'meta',
         profile: { displayName: 'Metacolony' },
         type: 'METACOLONY',
+        version: BigNumber.from(metacolonyVersion).toNumber()
       }
     },
     GRAPHQL_URI,
@@ -300,6 +301,7 @@ const createColony = async (colonyName, tokenAddress, singerOrWallet) => {
         colonyNativeTokenId: tokenAddress,
         name: colonyName,
         profile: { displayName: `Colony ${colonyName.toUpperCase()}` },
+        version: BigNumber.from(currentNetworkVersion).toNumber()
       }
     },
     GRAPHQL_URI,

src/components/common/Extensions/ExtensionActionButton/ExtensionActionButton.tsx

@@ -39,6 +39,8 @@ const ExtensionActionButton = ({ extensionData }: Props) => {
     );
   };
 
+  const isSupportedColonyVersion = colony.version >= 5;
+
   if (!isInstalledExtensionData(extensionData)) {
     return (
       <ActionButton
@@ -51,6 +53,7 @@ const ExtensionActionButton = ({ extensionData }: Props) => {
           extensionData,
         }}
         text={MSG.install}
+        disabled={!isSupportedColonyVersion}
       />
     );
   }
@@ -65,7 +68,7 @@ const ExtensionActionButton = ({ extensionData }: Props) => {
         appearance={{ theme: 'primary', size: 'medium' }}
         onClick={handleEnableButtonClick}
         text={MSG.enable}
-        // disabled={!isSupportedColonyVersion}
+        disabled={!isSupportedColonyVersion}
       />
     );
   }

src/components/common/Extensions/ExtensionUpgradeButton/ExtensionUpgradeButton.tsx

@@ -28,9 +28,7 @@ const ExtensionUpgradeButton = ({ extensionData }: Props) => {
     [],
   );
 
-  // const isSupportedColonyVersion =
-  //   parseInt(colonyVersion || '1', 10) >= ColonyVersion.LightweightSpaceship;
-  const isSupportedColonyVersion = true;
+  const isSupportedColonyVersion = (colony?.version || 1) >= 5;
 
   const extensionCompatible = isExtensionCompatible(
     Extension[extensionData.extensionId],

src/graphql/fragments/colony.graphql

@@ -47,6 +47,7 @@ fragment Colony on Colony {
       ...Watcher
     }
   }
+  version
 }
 
 fragment Watcher on WatchedColonies {