feat: add vesting account handling

database/auth.go

@@ -1,6 +1,7 @@
 package database
 
 import (
+	"encoding/json"
 	"fmt"
 
 	dbutils "github.com/forbole/bdjuno/v2/database/utils"
@@ -52,6 +53,41 @@ func (db *Db) saveAccounts(paramsNumber int, accounts []types.Account) error {
 	return nil
 }
 
+func (db *Db) SaveVestingAccounts(paramsNumber int, vestingAccount []types.VestingAccount) error {
+	if len(vestingAccount) == 0 {
+		return nil
+	}
+	stmt := `INSERT INTO vesting_account (address, original_vesting, end_time, start_time, vesting_periods) VALUES `
+	var params []interface{}
+
+	for i, account := range vestingAccount {
+		ai := i * paramsNumber
+		stmt += fmt.Sprintf("($%d,$%d,$%d,$%d,$%d),", ai+1, ai+2, ai+3, ai+4, ai+5)
+
+		originalVestingBz, err := json.Marshal(account.OriginalVesting)
+		if err != nil {
+			return err
+		}
+
+		VestingPeriodsBz, err := json.Marshal(account.VestingPeriods)
+		if err != nil {
+			return err
+		}
+
+		params = append(params, account.Address, string(originalVestingBz), account.EndTime, account.StartTime, string(VestingPeriodsBz))
+	}
+
+	stmt = stmt[:len(stmt)-1]
+	stmt += `ON CONFLICT (address) DO NOTHING`
+
+	_, err := db.Sql.Exec(stmt, params...)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // GetAccounts returns all the accounts that are currently stored inside the database.
 func (db *Db) GetAccounts() ([]string, error) {
 	var rows []string

database/schema/01-auth.sql

@@ -1,4 +1,14 @@
 CREATE TABLE account
 (
     address TEXT NOT NULL PRIMARY KEY
+);
+
+
+CREATE TABLE vesting_account
+(
+    address             TEXT        NOT NULL REFERENCES account (address) PRIMARY KEY,
+    original_vesting    JSONB       NOT NULL DEFAULT '{}',
+    end_time            TIMESTAMP   NOT NULL,
+    start_time          TIMESTAMP   NOT NULL,
+    vesting_periods     JSONB       NOT NULL DEFAULT '{}'
 );

modules/auth/auth_accounts.go

@@ -2,6 +2,7 @@ package auth
 
 import (
 	"encoding/json"
+	"strings"
 
 	"github.com/cosmos/cosmos-sdk/codec"
 	authttypes "github.com/cosmos/cosmos-sdk/x/auth/types"
@@ -32,6 +33,42 @@ func GetGenesisAccounts(appState map[string]json.RawMessage, cdc codec.Marshaler
 	return accounts, nil
 }
 
+// GetGenesisVestingAccounts parses the given appState and returns the desired genesis vesting account details
+func GetGenesisVestingAccounts(appState map[string]json.RawMessage) ([]types.VestingAccount, error) {
+	log.Debug().Str("module", "auth (vesting)").Msg("parsing genesis")
+
+	var authState Accounts
+	if err := json.Unmarshal(appState[authttypes.ModuleName], &authState); err != nil {
+		return nil, err
+	}
+
+	// Store the accounts
+	accounts := []types.VestingAccount{}
+	for _, account := range authState.Accounts {
+		accountType := strings.Split(account.AccountType, ".")[3]
+
+		if accountType == "PeriodicVestingAccount" {
+			// Prepare vesting periods array for creating new Account instance
+			vestingPeriods := make([]types.VestingPeriod, len(account.VestingPeriods))
+			for index, period := range account.VestingPeriods {
+				vestingPeriods[index] = types.NewVestingPeriod(period.Length, period.Amount)
+			}
+
+			// Create new Account instance
+			vestingAccount := types.NewVestingAccount(
+				account.BaseVestingAccount.BaseAccount.Address,
+				account.BaseVestingAccount.OriginalVesting,
+				account.BaseVestingAccount.EndTime,
+				account.StartTime,
+				vestingPeriods,
+			)
+			accounts = append(accounts, vestingAccount)
+		}
+	}
+
+	return accounts, nil
+}
+
 // --------------------------------------------------------------------------------------------------------------------
 
 // GetAccounts returns the account data for the given addresses

modules/auth/handle_genesis.go

@@ -13,6 +13,7 @@ import (
 func (m *Module) HandleGenesis(_ *tmtypes.GenesisDoc, appState map[string]json.RawMessage) error {
 	log.Debug().Str("module", "auth").Msg("parsing genesis")
 
+	// Handle Genesis account addresses
 	accounts, err := GetGenesisAccounts(appState, m.cdc)
 	if err != nil {
 		return fmt.Errorf("error while getting genesis accounts: %s", err)
@@ -23,5 +24,17 @@ func (m *Module) HandleGenesis(_ *tmtypes.GenesisDoc, appState map[string]json.R
 		return fmt.Errorf("error while storing genesis accounts: %s", err)
 	}
 
+	// Handle Genesis vesting account details
+	vestingAccounts, err := GetGenesisVestingAccounts(appState)
+	if err != nil {
+		return fmt.Errorf("error while getting genesis vesting accounts: %s", err)
+	}
+
+	paramsNumber := 5
+	err = m.db.SaveVestingAccounts(paramsNumber, vestingAccounts)
+	if err != nil {
+		return fmt.Errorf("error while storing genesis vesting accounts: %s", err)
+	}
+
 	return nil
 }

modules/auth/types.go

@@ -0,0 +1,31 @@
+package auth
+
+import sdk "github.com/cosmos/cosmos-sdk/types"
+
+// --- Types are defined according to genesis file structure ---
+
+type Accounts struct {
+	Accounts []AccountDetails `json:"accounts"`
+}
+
+type AccountDetails struct {
+	AccountType        string             `json:"@type"`
+	BaseVestingAccount BaseVestingAccount `json:"base_vesting_account"`
+	StartTime          string             `json:"start_time"`
+	VestingPeriods     []Period           `json:"vesting_periods"`
+}
+
+type BaseVestingAccount struct {
+	BaseAccount     BaseAccount `json:"base_account"`
+	OriginalVesting []sdk.Coin  `json:"original_vesting"`
+	EndTime         string      `json:"end_time"`
+}
+
+type BaseAccount struct {
+	Address string `json:"address"`
+}
+
+type Period struct {
+	Length string     `json:"length"`
+	Amount []sdk.Coin `json:"amount"`
+}

types/auth.go

@@ -1,5 +1,7 @@
 package types
 
+import sdk "github.com/cosmos/cosmos-sdk/types"
+
 // Account represents a chain account
 type Account struct {
 	Address string
@@ -11,3 +13,50 @@ func NewAccount(address string) Account {
 		Address: address,
 	}
 }
+
+// --------------- For Vesting Account ---------------
+
+// Account represents a chain account
+type VestingAccount struct {
+	Address         string
+	OriginalVesting sdk.Coins
+	EndTime         string
+	StartTime       string
+	VestingPeriods  []VestingPeriod
+}
+
+// NewAccount builds a new VestingAccount instance
+func NewVestingAccount(address string, originalVesting sdk.Coins, endTime string, startTime string, vestingPeriods []VestingPeriod) VestingAccount {
+	return VestingAccount{
+		Address:         address,
+		OriginalVesting: originalVesting,
+		EndTime:         endTime,
+		StartTime:       startTime,
+		VestingPeriods:  vestingPeriods,
+	}
+}
+
+func (u VestingAccount) Equal(v VestingAccount) bool {
+	for index, periodU := range u.VestingPeriods {
+		periodV := v.VestingPeriods[index]
+		if periodU.Amounts.String() != periodV.Amounts.String() {
+			return false
+		}
+	}
+	return u.Address == v.Address &&
+		u.OriginalVesting.String() == v.OriginalVesting.String() &&
+		u.EndTime == v.EndTime &&
+		u.StartTime == v.StartTime
+}
+
+type VestingPeriod struct {
+	Length  string
+	Amounts sdk.Coins
+}
+
+func NewVestingPeriod(length string, amounts sdk.Coins) VestingPeriod {
+	return VestingPeriod{
+		Length:  length,
+		Amounts: amounts,
+	}
+}