forked from TrueCloudLab/frostfs-node
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package morph
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core/native/nativenames"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/native/noderoles"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
)
|
|
|
|
func (c *initializeContext) setNotaryAndAlphabetNodes() error {
|
|
if ok, err := c.setRolesFinished(); ok || err != nil {
|
|
if err == nil {
|
|
c.Command.Println("Stage 2: already performed.")
|
|
}
|
|
return err
|
|
}
|
|
|
|
designateHash, err := c.Client.GetNativeContractHash(nativenames.Designation)
|
|
if err != nil {
|
|
return fmt.Errorf("can't fetch %s hash: %w", nativenames.Designation, err)
|
|
}
|
|
|
|
var pubs []interface{}
|
|
for _, w := range c.Wallets {
|
|
acc, err := getWalletAccount(w, singleAccountName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pubs = append(pubs, acc.PrivateKey().PublicKey().Bytes())
|
|
}
|
|
|
|
w := io.NewBufBinWriter()
|
|
emit.AppCall(w.BinWriter, designateHash, "designateAsRole",
|
|
callflag.States|callflag.AllowNotify, int64(noderoles.P2PNotary), pubs)
|
|
emit.AppCall(w.BinWriter, designateHash, "designateAsRole",
|
|
callflag.States|callflag.AllowNotify, int64(noderoles.NeoFSAlphabet), pubs)
|
|
|
|
if err := c.sendCommitteeTx(w.Bytes(), -1); err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.awaitTx()
|
|
}
|
|
|
|
func (c *initializeContext) setRolesFinished() (bool, error) {
|
|
height, err := c.Client.GetBlockCount()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
pubs, err := c.Client.GetDesignatedByRole(noderoles.NeoFSAlphabet, height)
|
|
return len(pubs) == len(c.Wallets), err
|
|
}
|