Merge pull request #44 from Muzikie/43-add-support-for-wallet

Add support for wallet

config/middlewares.ts

@@ -9,10 +9,4 @@ export default [
   'strapi::session',
   'strapi::favicon',
   'strapi::public',
-  {
-    resolve: './src/middlewares/custom/signup',
-    config: {
-      enabled: true,
-    },
-  },
 ];

package.json

@@ -13,11 +13,13 @@
     "lint": "eslint"
   },
   "dependencies": {
+    "@solana/web3.js": "^1.95.3",
     "@strapi/plugin-cloud": "4.25.1",
     "@strapi/plugin-i18n": "4.25.1",
     "@strapi/plugin-users-permissions": "^4.25.6",
     "@strapi/strapi": "4.25.1",
     "better-sqlite3": "8.6.0",
+    "crypto": "^1.0.1",
     "pg": "^8.12.0",
     "react": "^18.0.0",
     "react-dom": "^18.0.0",

src/api/profile/content-types/profile/schema.json

@@ -28,12 +28,6 @@
       "relation": "oneToOne",
       "target": "plugin::users-permissions.user"
     },
-    "wallet": {
-      "type": "relation",
-      "relation": "oneToOne",
-      "target": "api::wallet.wallet",
-      "mappedBy": "profile"
-    },
     "points": {
       "type": "integer",
       "required": true,

src/api/profile/services/profile.ts

@@ -3,5 +3,50 @@
  */
 
 import { factories } from '@strapi/strapi';
+import {Keypair} from '@solana/web3.js';
+import crypto from 'crypto';
 
-export default factories.createCoreService('api::profile.profile');
+import {encryptPrivateKey} from '../../../utils/crypto';
+import {SupportedBlockchains} from '../../../utils/types';
+
+export default factories.createCoreService('api::profile.profile', {
+  addProfileAndWallet: async (userId: string) => {
+    try {
+      const now = new Date().getTime();
+
+      // Create a profile for the new user
+      await strapi.entityService.create('api::profile.profile', {
+        data: {
+          first_name: '',
+          las_name: '',
+          points: 0,
+          users_permissions_user: userId,
+          createdAt: now,
+          publishedAt: now,
+        },
+      });
+
+      // Create a wallet for the new user
+      const wallet = Keypair.generate();
+      const iv = crypto.randomBytes(16);
+      const privateKey = wallet.secretKey.toString();
+      await strapi.entityService.create('api::wallet.wallet', {
+        data: {
+          public_key: wallet.publicKey.toString(),
+          encrypted_private_key: encryptPrivateKey(privateKey, iv),
+          address: wallet.publicKey.toBase58(),
+          blockchain: SupportedBlockchains.Solana,
+          encryption_metadata: {},
+          users_permissions_user: userId,
+          createdAt: now,
+          publishedAt: now,
+        },
+      });
+
+      strapi.log.info(`Profile and wallet created for user ${userId}`);
+    } catch (error) {
+      strapi.log.error('Error during profile or wallet creation:', error);
+      // @todo we should delete the user and inform the client about the error
+    }
+  },
+});

src/api/wallet/content-types/wallet/schema.json

@@ -12,23 +12,35 @@
   },
   "pluginOptions": {},
   "attributes": {
+    "public_key": {
+      "type": "string",
+      "required": true,
+      "unique": true
+    },
+    "encrypted_private_key": {
+      "type": "json",
+      "required": true
+    },
     "address": {
       "type": "string",
       "required": true
     },
-    "balance": {
-      "type": "biginteger",
-      "default": "0",
+    "encryption_metadata": {
+      "type": "json",
       "required": true
     },
-    "seed": {
-      "type": "string",
+    "blockchain": {
+      "type": "enumeration",
+      "enum": [
+        "Solana",
+        "Klayr"
+      ],
       "required": true
     },
-    "profile": {
+    "users_permissions_user": {
       "type": "relation",
       "relation": "oneToOne",
-      "target": "api::profile.profile",
+      "target": "plugin::users-permissions.user",
       "inversedBy": "wallet"
     }
   }

src/extensions/users-permissions/content-types/user/schema.json

@@ -95,6 +95,12 @@
       "relation": "oneToMany",
       "target": "api::contribution.contribution",
       "mappedBy": "users_permissions_user"
+    },
+    "wallet": {
+      "type": "relation",
+      "relation": "oneToOne",
+      "target": "api::wallet.wallet",
+      "mappedBy": "users_permissions_user"
     }
   }
 }

src/index.ts

@@ -14,5 +14,11 @@ export default {
    * This gives you an opportunity to set up your data model,
    * run jobs, or perform some special logic.
    */
-  bootstrap(/*{ strapi }*/) {},
+  bootstrap({ strapi }) {
+    strapi.db.lifecycles.subscribe((event) => {
+      if (event.action === 'afterCreate' && event.model.collectionName === 'up_users') {
+        strapi.service('api::profile.profile').addProfileAndWallet(event.result.id);
+      }
+    });
+  },
 };

src/utils/crypto.ts

@@ -0,0 +1,18 @@
+import crypto from 'crypto';
+
+const algorithm = 'aes-256-cbc';
+const PVK_ENCRYPTION_SALT = process.env.PVK_ENCRYPTION_SALT;
+const pvkEncryptionSalt = Buffer.from(PVK_ENCRYPTION_SALT, 'hex');
+
+export const encryptPrivateKey = (privateKey: string, iv: Buffer) => {
+  if (!PVK_ENCRYPTION_SALT) {
+    throw new Error('Error: The env variable PVK_ENCRYPTION_SALT is missing');
+  }
+
+  const cipher = crypto.createCipheriv(algorithm, pvkEncryptionSalt, iv);
+  const encrypted = Buffer.concat([cipher.update(privateKey), cipher.final()]);
+  return {
+    iv: iv.toString('hex'),
+    encryptedData: encrypted.toString('hex'),
+  };
+};

src/utils/types.ts

@@ -0,0 +1,3 @@
+export enum SupportedBlockchains {
+  Solana = 'Solana',
+}

types/generated/contentTypes.d.ts

@@ -749,6 +749,11 @@ export interface PluginUsersPermissionsUser extends Schema.CollectionType {
       'oneToMany',
       'api::contribution.contribution'
     >;
+    wallet: Attribute.Relation<
+      'plugin::users-permissions.user',
+      'oneToOne',
+      'api::wallet.wallet'
+    >;
     createdAt: Attribute.DateTime;
     updatedAt: Attribute.DateTime;
     createdBy: Attribute.Relation<
@@ -1080,11 +1085,6 @@ export interface ApiProfileProfile extends Schema.CollectionType {
       'oneToOne',
       'plugin::users-permissions.user'
     >;
-    wallet: Attribute.Relation<
-      'api::profile.profile',
-      'oneToOne',
-      'api::wallet.wallet'
-    >;
     points: Attribute.Integer &
       Attribute.Required &
       Attribute.SetMinMax<
@@ -1414,15 +1414,15 @@ export interface ApiWalletWallet extends Schema.CollectionType {
     draftAndPublish: true;
   };
   attributes: {
+    public_key: Attribute.String & Attribute.Required & Attribute.Unique;
+    encrypted_private_key: Attribute.JSON & Attribute.Required;
     address: Attribute.String & Attribute.Required;
-    balance: Attribute.BigInteger &
-      Attribute.Required &
-      Attribute.DefaultTo<'0'>;
-    seed: Attribute.String & Attribute.Required;
-    profile: Attribute.Relation<
+    encryption_metadata: Attribute.JSON & Attribute.Required;
+    blockchain: Attribute.Enumeration<['Solana', 'Klayr']> & Attribute.Required;
+    users_permissions_user: Attribute.Relation<
       'api::wallet.wallet',
       'oneToOne',
-      'api::profile.profile'
+      'plugin::users-permissions.user'
     >;
     createdAt: Attribute.DateTime;
     updatedAt: Attribute.DateTime;