package morph import ( "encoding/json" "fmt" "os" "path" "path/filepath" "github.com/nspcc-dev/neo-go/cli/input" "github.com/nspcc-dev/neo-go/pkg/crypto/keys" "github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest" "github.com/nspcc-dev/neo-go/pkg/util" "github.com/nspcc-dev/neo-go/pkg/wallet" "github.com/spf13/cobra" "github.com/spf13/viper" ) const contractWalletName = "contract.json" func initializeContractWallet(walletDir string) (*wallet.Wallet, error) { var ( password string err error ) if key := "credentials.contract"; viper.IsSet(key) { password = viper.GetString(key) } else { prompt := "Password for contract wallet > " password, err = input.ReadPassword(prompt) if err != nil { return nil, err } } w, err := wallet.NewWallet(path.Join(walletDir, contractWalletName)) if err != nil { return nil, err } acc, err := wallet.NewAccount() if err != nil { return nil, err } err = acc.Encrypt(password, keys.NEP2ScryptParams()) if err != nil { return nil, err } w.AddAccount(acc) if err := w.Save(); err != nil { return nil, err } return w, nil } func openContractWallet(cmd *cobra.Command, walletDir string) (*wallet.Wallet, error) { p := path.Join(walletDir, contractWalletName) w, err := wallet.NewWalletFromFile(p) if err != nil { if !os.IsNotExist(err) { return nil, fmt.Errorf("can't open wallet: %w", err) } cmd.Printf("Contract group wallet is missing, initialize at %s\n", filepath.Join(walletDir, contractWalletName)) w, err = initializeContractWallet(walletDir) if err != nil { return nil, err } } var password string if key := "credentials.contract"; viper.IsSet(key) { password = viper.GetString(key) } else { prompt := "Password for contract wallet > " password, err = input.ReadPassword(prompt) if err != nil { return nil, fmt.Errorf("can't fetch password: %w", err) } } for i := range w.Accounts { if err := w.Accounts[i].Decrypt(password, keys.NEP2ScryptParams()); err != nil { return nil, fmt.Errorf("can't unlock wallet: %w", err) } } return w, nil } func (c *initializeContext) addManifestGroup(h util.Uint160, cs *contractState) error { priv := c.ContractWallet.Accounts[0].PrivateKey() pub := priv.PublicKey() sig := priv.Sign(h.BytesBE()) found := false for i := range cs.Manifest.Groups { if cs.Manifest.Groups[i].PublicKey.Equal(pub) { cs.Manifest.Groups[i].Signature = sig found = true break } } if !found { cs.Manifest.Groups = append(cs.Manifest.Groups, manifest.Group{ PublicKey: pub, Signature: sig, }) } data, err := json.Marshal(cs.Manifest) if err != nil { return err } cs.RawManifest = data return nil }