feat: raw and process snapshot commands

.github/workflows/test.yml

@@ -34,19 +34,6 @@ jobs:
         run: |
           go test -v -coverprofile=coverage.txt -covermode=atomic -coverpkg=./... $(go list ./...)
 
-      - name: filter non-testable files
-        run: |
-          excludelist="$(find ./ -type f -name '*.go' | xargs grep -l 'DONTCOVER')"
-          excludelist+=" $(find ./ -type f -name '*.pb.go')"
-          excludelist+=" $(find ./ -type f -name '*.pb.gw.go')"
-          excludelist+=" $(find ./cmd -type d)"
-          for filename in ${excludelist}; do
-            filename=${filename#".//"}
-            echo "Excluding ${filename} from coverage report..."
-            filename=$(echo "$filename" | sed 's/\//\\\//g')
-            sed -i.bak "/""$filename""/d" coverage.txt
-          done
-
       - name: Upload coverage to Codecov
         uses: codecov/[email protected]
         with:

.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,152 @@
 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 NewAirdropGenerate() *cobra.Command {
+	return &cobra.Command{
+		Use:   "generate [airdrop-config] [input-genesis]",
+		Short: "Utility tool to create snapshots for an airdrop",
+		Args:  cobra.ExactArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			var (
+				airdropConfig = args[0]
+				inputGenesis  = args[1]
+			)
+
+			c, err := config.ParseConfig(airdropConfig)
+			if err != nil {
+				return err
+			}
+
+			genState, err := genesis.GetGenStateFromPath(inputGenesis)
+			if err != nil {
+				return err
+			}
+
+			s, err := snapshot.Generate(genState)
+			if err != nil {
+				return err
+			}
+
+			filters := make(snapshot.Filters, 0)
+			for _, snap := range c.Snapshots {
+				filter := s.Filter(snapshot.FilterType(snap.Type), snap.Denom, snap.Formula, snap.Excluded)
+				filters = append(filters, filter)
+			}
+			filter := filters.Sum()
+
+			// export snapshot json
+			filterJSON, err := json.MarshalIndent(filter, "", "    ")
+			if err != nil {
+				return fmt.Errorf("failed to marshal snapshot: %w", err)
+			}
+
+			cmd.Println(string(filterJSON))
+			// TODO generate the genesis
+			return nil
+		},
+	}
+}
+
+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 {
+			genState, err := genesis.GetGenStateFromPath(args[0])
+			if err != nil {
+				return err
+			}
+
+			s, err := snapshot.Generate(genState)
+			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 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
+			}
+
+			filters := make(snapshot.Filters, 0)
+			for _, snap := range c.Snapshots {
+				filter := s.Filter(snapshot.FilterType(snap.Type), snap.Denom, snap.Formula, snap.Excluded)
+				filters = append(filters, filter)
+			}
+			filter := filters.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] [input-genesis]",
+		Short: "Generate a genesis based on processed files and airdrop config",
+		Args:  cobra.ExactArgs(3),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			fmt.Println("Hello from airdrop command")
 			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)
+	}
+}