feat: raw and process snapshot commands

.gitignore

@@ -1,3 +1,5 @@
 cli-plugin-airdrop
 .idea
-.DS_STORE
+.DS_STORE
+coverage.txt
+coverage.out

Makefile

@@ -17,15 +17,12 @@ FIND_ARGS := -name '*.go' -type f -not -name '*.pb.go' -not -name '*.pb.gw.go'
 ## format: Run gofmt and goimports.
 format:
 	@echo Formatting...
-	@go install mvdan.cc/gofumpt
-	@go install golang.org/x/tools/cmd/goimports
 	@find . $(FIND_ARGS) | xargs gofumpt -w .
 	@find . $(FIND_ARGS) | xargs goimports -w -local github.com/ignite/cli-plugin-airdrop
 
 ## lint: Run Golang CI Lint.
 lint:
 	@echo Running gocilint...
-	@go install github.com/golangci/golangci-lint/cmd/golangci-lint
 	@golangci-lint run --out-format=tab --issues-exit-code=0
 
 help: Makefile

cmd/cmd.go

@@ -1,17 +1,213 @@
 package cmd
 
 import (
+	"encoding/json"
 	"fmt"
 
 	"github.com/spf13/cobra"
+
+	"github.com/ignite/cli-plugin-airdrop/pkg/config"
+	"github.com/ignite/cli-plugin-airdrop/pkg/genesis"
+	"github.com/ignite/cli-plugin-airdrop/pkg/snapshot"
 )
 
 func NewAirdrop() *cobra.Command {
-	return &cobra.Command{
+	c := &cobra.Command{
 		Use:   "airdrop",
 		Short: "Utility tool to create snapshots for an airdrop",
+	}
+
+	c.AddCommand(
+		NewAirdropGenerate(),
+		NewAirdropRaw(),
+		NewAirdropProcess(),
+		NewAirdropGenesis(),
+	)
+
+	return c
+}
+
+func NewAirdropRaw() *cobra.Command {
+	return &cobra.Command{
+		Use:   "raw [input-genesis]",
+		Short: "Generate raw airdrop data based on the input genesis",
+		Args:  cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			// parse the genesis input to get the current stake state
+			genState, err := genesis.GetGenStateFromPath(args[0])
+			if err != nil {
+				return err
+			}
+
+			// get only the essential info from input genesis
+			// to generate the snapshot the raw snapshot object
+			s, err := snapshot.Generate(genState.AppState)
+			if err != nil {
+				return err
+			}
+
+			// export snapshot json
+			snapshotJSON, err := json.MarshalIndent(s, "", "    ")
+			if err != nil {
+				return fmt.Errorf("failed to marshal snapshot: %w", err)
+			}
+
+			cmd.Println(string(snapshotJSON))
+			return nil
+		},
+	}
+}
+
+func NewAirdropProcess() *cobra.Command {
+	return &cobra.Command{
+		Use:   "process [airdrop-config] [raw-snapshot]",
+		Short: "Process the airdrop raw data based on the config file",
+		Args:  cobra.ExactArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var (
+				airdropConfig = args[0]
+				rawSnapshot   = args[1]
+			)
+
+			c, err := config.ParseConfig(airdropConfig)
+			if err != nil {
+				return err
+			}
+
+			s, err := snapshot.ParseSnapshot(rawSnapshot)
+			if err != nil {
+				return err
+			}
+
+			records := make(snapshot.Records, 0)
+			for _, snap := range c.Snapshots {
+				record := s.ApplyConfig(snapshot.ConfigType(snap.Type), snap.Denom, snap.Formula, snap.Excluded)
+				records = append(records, record)
+			}
+			filter := records.Sum()
+
+			// export filter json
+			filterJSON, err := json.MarshalIndent(filter, "", "    ")
+			if err != nil {
+				return fmt.Errorf("failed to marshal snapshot: %w", err)
+			}
+
+			cmd.Println(string(filterJSON))
+			return nil
+		},
+	}
+}
+
+func NewAirdropGenesis() *cobra.Command {
+	return &cobra.Command{
+		Use:   "genesis [airdrop-config] [raw-snapshot] [output-genesis]",
+		Short: "Generate and add quadratic airdrop claim record to the output genesis based on the raw data and airdrop config",
+		Args:  cobra.ExactArgs(3),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var (
+				airdropConfig = args[0]
+				rawSnapshot   = args[1]
+				outputGenesis = args[2]
+			)
+
+			c, err := config.ParseConfig(airdropConfig)
+			if err != nil {
+				return err
+			}
+
+			// parse the genesis output to get the current stake state
+			genState, err := genesis.GetGenStateFromPath(outputGenesis)
+			if err != nil {
+				return err
+			}
+
+			s, err := snapshot.ParseSnapshot(rawSnapshot)
+			if err != nil {
+				return err
+			}
+
+			// apply the config file to the raw data to generate the claim records
+			records := make(snapshot.Records, 0)
+			for _, snap := range c.Snapshots {
+				record := s.ApplyConfig(snapshot.ConfigType(snap.Type), snap.Denom, snap.Formula, snap.Excluded)
+				records = append(records, record)
+			}
+			record := records.Sum()
+
+			// add claim records to the output genesis
+			if err := genState.AddFromClaimRecord(c.AirdropToken, record.ClaimRecords()); err != nil {
+				return err
+			}
+
+			// export snapshot json
+			genesisJSON, err := json.MarshalIndent(genState, "", "    ")
+			if err != nil {
+				return fmt.Errorf("failed to marshal snapshot: %w", err)
+			}
+
+			cmd.Println(string(genesisJSON))
+			return nil
+		},
+	}
+}
+
+func NewAirdropGenerate() *cobra.Command {
+	return &cobra.Command{
+		Use:   "generate [airdrop-config] [input-genesis] [output-genesis]",
+		Short: "Generate and add quadratic airdrop claim records to the output genesis based on the input genesis stake values and airdrop config",
+		Args:  cobra.ExactArgs(3),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			fmt.Println("Hello from airdrop command")
+			var (
+				airdropConfig = args[0]
+				inputGenesis  = args[1]
+				outputGenesis = args[2]
+			)
+
+			// parse the airdrop config yaml
+			c, err := config.ParseConfig(airdropConfig)
+			if err != nil {
+				return err
+			}
+
+			// parse the genesis input to get the current stake state
+			inGenState, err := genesis.GetGenStateFromPath(inputGenesis)
+			if err != nil {
+				return err
+			}
+
+			// parse the genesis output state to be modified
+			outGenState, err := genesis.GetGenStateFromPath(outputGenesis)
+			if err != nil {
+				return err
+			}
+
+			// get only the essential info from input genesis
+			// to generate the snapshot the raw snapshot object
+			s, err := snapshot.Generate(inGenState.AppState)
+			if err != nil {
+				return err
+			}
+
+			// apply the config file to the raw data to generate the claim records
+			records := make(snapshot.Records, 0)
+			for _, snap := range c.Snapshots {
+				record := s.ApplyConfig(snapshot.ConfigType(snap.Type), snap.Denom, snap.Formula, snap.Excluded)
+				records = append(records, record)
+			}
+			filter := records.Sum()
+
+			// add claim records to the output genesis
+			if err := outGenState.AddFromClaimRecord(c.AirdropToken, filter.ClaimRecords()); err != nil {
+				return err
+			}
+
+			// export snapshot json
+			genesisJSON, err := json.MarshalIndent(outGenState, "", "    ")
+			if err != nil {
+				return fmt.Errorf("failed to marshal snapshot: %w", err)
+			}
+
+			cmd.Println(string(genesisJSON))
 			return nil
 		},
 	}

cmd/debug/debug.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+
+	"github.com/ignite/cli-plugin-airdrop/cmd"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "cli-plugin-airdrop",
+	Short: "debug command for CLI airdrop plugin",
+}
+
+func main() {
+	rootCmd.AddCommand(cmd.NewAirdrop())
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}