Update/db schema

apps/web/src/composables/users.ts

@@ -11,71 +11,6 @@ const { usersBaseURL, ethereumURL } = useEnvironment()
 // 0xd557a5745d4560B24D36A68b52351ffF9c86A212
 const session = ref<boolean>(false)
 const user = ref<UserWithAccounts>()
-// const user = ref(
-//     {
-//         address: '0xd557a5745d4560B24D36A68b52351ffF9c86A212'.toLowerCase(),
-//         accounts: [
-//             {
-//                 address: '0xd557a5745d4560B24D36A68b52351ffF9c86A212'.toLowerCase(),
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'MetaMask'
-//             },
-//             {
-//                 address: '0x1dc336d94890b10e1a47b6e34cdee1009ee7b942'.toLowerCase(),
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'CoinbaseWallet'
-//             },
-//             {
-//                 address: '0x1dc336d94890b10e1a47b6e34cdee1009ee7b942'.toLowerCase(),
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'CoinbaseWallet'
-//             },
-//             {
-//                 address: '0x1dc336d94890b10e1a47b6e34cdee1009ee7b942'.toLowerCase(),
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'CoinbaseWallet'
-//             },
-//             {
-//                 address: '0x8222Ef172A2117D1C4739E35234E097630D94376'.toLowerCase(),
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'Ledger'
-//             },
-//             {
-//                 address: '0x8222Ef172A2117D1C4739E35234E097630D94377'.toLowerCase(), // Fake address
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'Trezor'
-//             },
-//             {
-//                 address: '0x8222Ef172A2117D1C4739E35234E097630D94378'.toLowerCase(), // Fake address
-//                 currency: 'ETH',
-//                 balance: '1000000000000000000',
-//                 balanceSnapshots: [{ date: '2023-02-06', balance: '1000000000000000000' }, { date: '2023-02-05', balance: '100000000000000000' }],
-//                 roi: 0.25,
-//                 walletProvider: 'WalletConnect'
-//             },
-//         ],
-//         nonce: '1234567890',
-//         pools: []
-//     }
-// )
 const { casimirManager, getPools } = useSSV()
 
 export default function useUsers () {
@@ -190,15 +125,16 @@ export default function useUsers () {
         })
     }
 
-    async function updatePrimaryAddress(primaryAddress: string, updatedProvider: ProviderString, updatedAddress: string) {
+    async function updatePrimaryAddress(updatedAddress: string) {
+        const userId = user?.value?.id
         const requestOptions = {
             method: 'PUT',
             headers: { 
                 'Content-Type': 'application/json'
             },
-            body: JSON.stringify({ primaryAddress, updatedProvider, updatedAddress })
+            body: JSON.stringify({ userId, updatedAddress })
         }
-        return await fetch(`${usersBaseURL}/users/update-primary-account`, requestOptions)
+        return await fetch(`${usersBaseURL}/user/update-primary-account`, requestOptions)
     }
 
     return {

apps/web/src/composables/wallet.ts

@@ -107,6 +107,7 @@ export default function useWallet() {
           // If account doesn't exist, add account using users api
           console.log('adding sub account')
           const account = {
+            userId: user?.value?.id,
             address: connectedAddress.toLowerCase() as string,
             currency: connectedCurrency,
             ownerAddress: user?.value?.address.toLowerCase() as string,
@@ -206,13 +207,12 @@ export default function useWallet() {
     }
   }
 
-  // TODO: What is this used for? 
   // Do we need balance of active address only? 
   // Or do we need balance of all addresses in accounts associated with user? 
   // Is this calculated on front end or back end or both?
   async function getUserBalance() {
     if (ethersProviderList.includes(selectedProvider.value)){
-      const walletBalance = await getEthersBalance(selectedProvider.value, selectedAddress.value)
+      const walletBalance = await getEthersBalance(selectedAddress.value)
       console.log('walletBalance in wei in wallet.ts :>> ', walletBalance)
     return walletBalance
     } else {
@@ -312,14 +312,14 @@ export default function useWallet() {
   async function setPrimaryWalletAccount() {
     if (!user?.value?.address) {
       alert('Please login first')
-    }
-    return alert('Not yet implemented for this wallet provider')
-    if (ethersProviderList.includes(selectedProvider.value)) {
-      const result = await updatePrimaryAddress(primaryAddress.value, selectedProvider.value, selectedAddress.value)
+    } else if (ethersProviderList.includes(selectedProvider.value)) {
+      const result = await updatePrimaryAddress(selectedAddress.value)
       const { data } = await result.json()
       if (data) {
         primaryAddress.value = data.address
       }
+    } else {
+      return alert('Not yet implemented for this wallet provider')
     }
   }
 

common/data/scripts/postgres.ts

@@ -1,7 +1,7 @@
 import minimist from 'minimist'
 import fs from 'fs'
 import { run } from '@casimir/helpers'
-import { JsonSchema, Schema, accountSchema, nonceSchema, userSchema } from '@casimir/data'
+import { JsonSchema, Schema, accountSchema, nonceSchema, userSchema, userAccountSchema } from '@casimir/data'
 
 /** Resource path from package caller */
 const resourcePath = './scripts'
@@ -10,7 +10,8 @@ const resourcePath = './scripts'
 const tableSchemas = {
     accounts: accountSchema,
     nonces: nonceSchema,
-    users: userSchema
+    users: userSchema,
+    user_accounts: userAccountSchema
 }
 
 /**
@@ -25,7 +26,7 @@ void async function () {
     const argv = minimist(process.argv.slice(2))
 
     /** Default to all tables */
-    const tables = argv.tables ? argv.tables.split(',') : ['accounts', 'nonces', 'users']
+    const tables = argv.tables ? argv.tables.split(',') : ['accounts', 'nonces', 'users', 'user_accounts']
     let sqlSchema = '-- Generated by @casimir/data/scripts/postgres.ts\n\n'
     for (const table of tables) {
         const tableSchema = tableSchemas[table] as JsonSchema

common/data/src/index.ts

@@ -3,6 +3,7 @@ import nonceSchema from './schemas/nonce.schema.json'
 import userSchema from './schemas/user.schema.json'
 import eventSchema from './schemas/event.schema.json'
 import aggSchema from './schemas/agg.schema.json'
+import userAccountSchema from './schemas/user_account.schema.json'
 import operatorStore from './mock/operator.store.json'
 import validatorStore from './mock/validator.store.json'
 import accountStore from './mock/account.store.json'
@@ -15,6 +16,7 @@ export {
   accountSchema,
   nonceSchema,
   userSchema,
+  userAccountSchema,
   eventSchema,
   aggSchema,
   accountStore,

common/data/src/interfaces/JsonSchema.ts

@@ -9,4 +9,6 @@ export interface JsonSchema {
             description: string
         }
     },
+    uniqueFields?: string[],
+    composite_key?: string
 }

common/data/src/providers/schema.ts

@@ -1,5 +1,6 @@
 import * as glue from '@aws-cdk/aws-glue-alpha'
 import { JsonSchema } from '../interfaces/JsonSchema'
+import { snakeCase } from '@casimir/helpers'
 
 export type JsonType = 'string' | 'number' | 'integer' | 'boolean' | 'object' | 'array' | 'null'
 export type GlueType = glue.Type
@@ -61,6 +62,8 @@ export class Schema {
      * ```
      */
     getPostgresTable(): string {
+        const compositeKey = this.jsonSchema.composite_key
+        const uniqueFields = this.jsonSchema.uniqueFields || []
         const columns = Object.keys(this.jsonSchema.properties).map((name: string) => {
             const property = this.jsonSchema.properties[name]
             let type = {
@@ -70,7 +73,8 @@ export class Schema {
                 boolean: 'BOOLEAN',
                 object: 'JSON',
                 array: 'JSON',
-                null: 'VARCHAR'
+                null: 'VARCHAR',
+                serial: 'SERIAL'
             }[property.type as JsonType] as PostgresType
 
             if (name.endsWith('_at')) type = 'TIMESTAMP'
@@ -80,20 +84,24 @@ export class Schema {
 
             const comment = property.description
             if (comment.includes('PK')) column += ' PRIMARY KEY'
-            
+
             return column
         })
 
+        /** Check for composite key property and add the primary key if so */
+        if (compositeKey) columns.push(`PRIMARY KEY (${compositeKey})`)
+
         /** Make table name plural of schema objects (todo: check edge-cases) */
-        const tableName = this.getTitle().toLowerCase() + 's'
+        const tableName = this.getTitle()
 
-        return `CREATE TABLE ${tableName} (\n\t${columns.join(',\n\t')}\n);`
+        const queryString = uniqueFields.length > 0 ? `CREATE TABLE ${tableName} (\n\t${columns.join(',\n\t')}, \n\tUNIQUE (${uniqueFields.join(', ')}));` : `CREATE TABLE ${tableName} (\n\t${columns.join(',\n\t')}\n);`
+        return queryString
     }
 
     /**
      * Get the title of the JSON schema object.
      */
     getTitle(): string {
-        return this.jsonSchema.title
+        return snakeCase(this.jsonSchema.title) + 's'
     }
 }

common/data/src/schemas/account.schema.json

@@ -6,7 +6,7 @@
     "properties": {
         "address": {
             "type": "string",
-            "description": "The account address (PK)"
+            "description": "The account address"
         },
         "balance": {
             "type": "number",
@@ -20,9 +20,9 @@
             "type": "string",
             "description": "The account creation date in ISO 8601 format"
         },
-        "owner_address": {
-            "type": "string",
-            "description": "The account owner address"
+        "id": {
+            "type": "serial",
+            "description": "The account ID (PK)"
         },
         "snapshots": {
             "type": "array",
@@ -49,15 +49,27 @@
             "type": "string",
             "description": "The account last update date in ISO 8601 format"
         },
+        "user_id": {
+            "type": "integer",
+            "description": "The account user ID (FK)"
+        },
         "wallet_provider": {
             "type": "string",
             "description": "The account wallet provider"
         }
     },
+    "uniqueFields": [
+        "address",
+        "user_id",
+        "wallet_provider"
+    ],
     "required": [
+        "id",
         "address",
+        "created_at",
         "currency",
-        "owner_address",
+        "updated_at",
+        "user_id",
         "wallet_provider"
     ]
 }

common/data/src/schemas/user.schema.json

@@ -6,18 +6,25 @@
     "properties": {
         "address": {
             "type": "string",
-            "description": "The user address (PK)"
+            "description": "The user address used to sign in"
         },
         "created_at": {
             "type": "string",
             "description": "The account creation date in ISO 8601 format"
         },
+        "id": {
+            "type": "serial",
+            "description": "The user ID (PK)"
+        },
         "updated_at": {
             "type": "string",
             "description": "The account last update date in ISO 8601 format"
         }
     },
     "required": [
-        "address"
+        "address",
+        "created_at",
+        "id",
+        "updated_at"
     ]
 }

common/data/src/schemas/user_account.schema.json

@@ -0,0 +1,27 @@
+{
+    "$id": "https://casimir.co/user_account.schema.json",
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "title": "User Account",
+    "type": "object",
+    "properties": {
+        "account_id": {
+            "type": "integer",
+            "description": "The unique identifier for an account (pk part 2) and the FK to the account table"
+        },
+        "created_at": {
+            "type": "string",
+            "description": "The date and time the user_account was created"
+        },
+        "user_id": {
+            "type": "integer",
+            "description": "The unique identifier for a user (pk part 1) and the FK to the user table"
+        }
+    },
+    "composite_key": "user_id, account_id",
+    "required": [
+        "account_id",
+        "created_at",
+        "user_id"
+    ]
+  }
+

common/types/src/index.ts

@@ -1,4 +1,5 @@
 import { Account } from './interfaces/Account'
+import { AccountWithStakingInfo } from './interfaces/AccountWithStakingInfo'
 import { AddAccountOptions } from './interfaces/AddAccountOptions'
 import { BalanceSnapshot } from './interfaces/BalanceSnapshot'
 import { ContractArgs } from './interfaces/ContractArgs'
@@ -18,6 +19,7 @@ import { Validator } from './interfaces/Validator'
 
 export type {
     Account,
+    AccountWithStakingInfo,
     AddAccountOptions,
     BalanceSnapshot,
     ContractArgs,