cli-plugin-network/cmd/network_project_update.go at main · ignite/cli-plugin-network · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package cmd

import (
	"fmt"
	"strings"

	sdk "github.com/cosmos/cosmos-sdk/types"
	"github.com/ignite/cli/ignite/pkg/cliui"
	"github.com/ignite/cli/ignite/pkg/yaml"
	"github.com/spf13/cobra"

	"github.com/ignite/cli-plugin-network/network"
)

const (
	flagProjectName        = "name"
	flagProjectMetadata    = "metadata"
	flagProjectTotalSupply = "total-supply"
)

func NewNetworkProjectUpdate() *cobra.Command {
	c := &cobra.Command{
		Use:   "update [project-id]",
		Short: "Update details fo the project of the project",
		Args:  cobra.ExactArgs(1),
		RunE:  networkProjectUpdateHandler,
	}
	c.Flags().String(flagProjectName, "", "update the project name")
	c.Flags().String(flagProjectMetadata, "", "update the project metadata")
	c.Flags().String(flagProjectTotalSupply, "", "update the total of the mainnet of a project")
	c.Flags().AddFlagSet(flagNetworkFrom())
	c.Flags().AddFlagSet(flagSetKeyringBackend())
	c.Flags().AddFlagSet(flagSetKeyringDir())
	return c
}

func networkProjectUpdateHandler(cmd *cobra.Command, args []string) error {
	session := cliui.New(cliui.StartSpinner())
	defer session.End()

	var (
		projectName, _        = cmd.Flags().GetString(flagProjectName)
		metadata, _           = cmd.Flags().GetString(flagProjectMetadata)
		projectTotalSupply, _ = cmd.Flags().GetString(flagProjectTotalSupply)
	)

	totalSupply, err := sdk.ParseCoinsNormalized(projectTotalSupply)
	if err != nil {
		return err
	}

	nb, err := newNetworkBuilder(cmd, CollectEvents(session.EventBus()))
	if err != nil {
		return err
	}

	// parse project ID
	projectID, err := network.ParseID(args[0])
	if err != nil {
		return err
	}

	if projectName == "" && metadata == "" && totalSupply.Empty() {
		return fmt.Errorf("at least one of the flags %s must be provided",
			strings.Join([]string{
				flagProjectName,
				flagProjectMetadata,
				flagProjectTotalSupply,
			}, ", "),
		)
	}

	n, err := nb.Network()
	if err != nil {
		return err
	}

	var proposals []network.Prop

	if projectName != "" {
		proposals = append(proposals, network.WithProjectName(projectName))
	}
	if metadata != "" {
		proposals = append(proposals, network.WithProjectMetadata(metadata))
	}
	if !totalSupply.Empty() {
		proposals = append(proposals, network.WithProjectTotalSupply(totalSupply))
	}

	if err = n.UpdateProject(cmd.Context(), projectID, proposals...); err != nil {
		return err
	}

	project, err := n.Project(cmd.Context(), projectID)
	if err != nil {
		return err
	}
	session.Println()

	info, err := yaml.Marshal(cmd.Context(), project)
	if err != nil {
		return err
	}

	return session.Print(info)
}