feat: distribution airdrop details

atomone-hub · Sep 3, 2024 · 0043f98 · 0043f98
1 parent 3468d08
commit 0043f98
Show file tree

Hide file tree

Showing 2 changed files with 92 additions and 17 deletions.
diff --git a/distribution.go b/distribution.go
@@ -33,7 +33,8 @@ type airdrop struct {
 	// params hold the distribution parameters that resulted in this airdrop
 	params distriParams
 	// addresses contains the airdrop amount per address.
-	addresses map[string]sdk.Int
+	addresses       map[string]sdk.Int
+	addressesDetail map[string]airdropDetail
 	// nonVotersMultiplier ensures that non-voters don't hold more than 1/3 of
 	// the supply
 	nonVotersMultiplier sdk.Dec
@@ -49,6 +50,24 @@ type airdrop struct {
 	reservedAddr sdk.Dec
 }
 
+type airdropDetail struct {
+	YesDetail    amtDetail `json:"yesDetail"`
+	NoDetail     amtDetail `json:"noDetail"`
+	NWVDetail    amtDetail `json:"nwvDetail"`
+	AbsDetail    amtDetail `json:"absDetail"`
+	DnvDetail    amtDetail `json:"dnvDetail"`
+	LiquidDetail amtDetail `json:"liquidDetail"`
+	Total        sdk.Dec   `json:"total"`
+}
+
+type amtDetail struct {
+	AtomAmt    sdk.Dec `json:"atomAmt"`
+	Multiplier sdk.Dec `json:"multiplier"`
+	BonusMalus sdk.Dec `json:"bonusMalus"`
+	Factor     sdk.Dec `json:"factor"`
+	AtoneAmt   sdk.Dec `json:"atoneAmt"`
+}
+
 type distrib struct {
 	// total supply of the distrib
 	supply sdk.Dec
@@ -93,9 +112,10 @@ func (d distrib) votePercentages() map[govtypes.VoteOption]sdk.Dec {
 
 func distribution(accounts []Account, params distriParams, prefix string) (airdrop, error) {
 	airdrop := airdrop{
-		params:    params,
-		addresses: make(map[string]sdk.Int),
-		icfSlash:  sdk.ZeroDec(),
+		params:          params,
+		addresses:       make(map[string]sdk.Int),
+		addressesDetail: make(map[string]airdropDetail),
+		icfSlash:        sdk.ZeroDec(),
 		atom: distrib{
 			supply:   sdk.ZeroDec(),
 			votes:    newVoteMap(),
@@ -188,17 +208,62 @@ func distribution(accounts []Account, params distriParams, prefix string) (airdr
 		airdrop.atone.unstaked = airdrop.atone.unstaked.Add(liquidAirdropAmt)
 		// add address and amount (skipping 0 balance)
 		if amtInt := airdropAmt.RoundInt(); !amtInt.IsZero() {
-			if prefix == "" {
-				// Fill with "cosmos" prefixed address
-				airdrop.addresses[acc.Address] = amtInt
-				continue
+			addr := acc.Address
+			if prefix != "" {
+				// Derive address from "cosmos" to prefix parameter
+				var err error
+				addr, err = convertBech32(acc.Address, "cosmos", prefix)
+				if err != nil {
+					return airdrop, err
+				}
 			}
-			// Derive address from "cosmos" to prefix parameter
-			atomOneAddr, err := convertBech32(acc.Address, "cosmos", prefix)
-			if err != nil {
-				return airdrop, err
+			// Fill with "cosmos" prefixed address
+			airdrop.addresses[addr] = amtInt
+			airdrop.addressesDetail[addr] = airdropDetail{
+				YesDetail: amtDetail{
+					AtomAmt:    yesAtomAmt,
+					Multiplier: params.yesVotesMultiplier,
+					BonusMalus: sdk.OneDec(),
+					Factor:     params.supplyFactor,
+					AtoneAmt:   yesAirdropAmt,
+				},
+				NoDetail: amtDetail{
+					AtomAmt:    noAtomAmt,
+					Multiplier: params.noVotesMultiplier,
+					BonusMalus: sdk.OneDec(),
+					Factor:     params.supplyFactor,
+					AtoneAmt:   noAirdropAmt,
+				},
+				NWVDetail: amtDetail{
+					AtomAmt:    noWithVetoAtomAmt,
+					Multiplier: params.noVotesMultiplier,
+					BonusMalus: params.bonus,
+					Factor:     params.supplyFactor,
+					AtoneAmt:   noWithVetoAtomAmt,
+				},
+				AbsDetail: amtDetail{
+					AtomAmt:    abstainAtomAmt,
+					Multiplier: airdrop.nonVotersMultiplier,
+					BonusMalus: sdk.OneDec(),
+					Factor:     params.supplyFactor,
+					AtoneAmt:   abstainAtomAmt,
+				},
+				DnvDetail: amtDetail{
+					AtomAmt:    noVoteAtomAmt,
+					Multiplier: airdrop.nonVotersMultiplier,
+					BonusMalus: params.malus,
+					Factor:     params.supplyFactor,
+					AtoneAmt:   noVoteAirdropAmt,
+				},
+				LiquidDetail: amtDetail{
+					AtomAmt:    acc.LiquidAmount,
+					Multiplier: airdrop.nonVotersMultiplier,
+					BonusMalus: params.malus,
+					Factor:     params.supplyFactor,
+					AtoneAmt:   liquidAirdropAmt,
+				},
+				Total: airdropAmt,
 			}
-			airdrop.addresses[atomOneAddr] = amtInt
 		}
 	}
 	// Compute minted part

diff --git a/main.go b/main.go
@@ -190,10 +190,11 @@ func distributionCmd() *ffcli.Command {
 				}
 			}
 			var (
-				datapath     = fs.Arg(0)
-				accountsFile = filepath.Join(datapath, "accounts.json")
-				airdropFile  = filepath.Join(datapath, "airdrop.json")
-				airdrops     []airdrop
+				datapath          = fs.Arg(0)
+				accountsFile      = filepath.Join(datapath, "accounts.json")
+				airdropFile       = filepath.Join(datapath, "airdrop.json")
+				airdropDetailFile = filepath.Join(datapath, "airdrop_detail.json")
+				airdrops          []airdrop
 			)
 			accounts, err := parseAccounts(accountsFile)
 			if err != nil {
@@ -219,6 +220,15 @@ func distributionCmd() *ffcli.Command {
 					return err
 				}
 				fmt.Printf("⚠ '%s' has been created/updated, don't forget to update S3 ⚠\n", airdropFile)
+
+				bz, err = json.MarshalIndent(airdrops[0].addressesDetail, "", "  ")
+				if err != nil {
+					return err
+				}
+				if err := os.WriteFile(airdropDetailFile, bz, 0o666); err != nil {
+					return err
+				}
+				fmt.Printf("⚠ '%s' has been created/updated, don't forget to update S3 ⚠\n", airdropDetailFile)
 			}
 			return nil
 		},