Exemplo n.º 1
0
// Register will handle the registration of a new name in the Zooko server.
func (c *Client) Register(name string, reg interface{}, acc *account.Account) error {
	if _, found, _ := c.lookupString(name); found {
		return errors.New("zooko: cannot register name that already exists")
	}

	jsonRegistration, err := json.Marshal(reg)
	if err != nil {
		return err
	}

	if err := c.checkAccountBalance(acc, true); err != nil {
		return err
	}

	nameNew, rand, err := acc.CreateNameNew(name)
	if err != nil {
		return err
	}

	// Temporarily commit to the new transaction so that we can
	// successfully create the name_firstupdate transaction.
	oldUnspent := acc.Unspent
	acc.Commit(nameNew)
	undoCommit := func() {
		acc.Unspent = oldUnspent
	}

	nameUpdate, err := acc.CreateNameFirstUpdate(rand, name, string(jsonRegistration))
	if err != nil {
		undoCommit()
		return err
	}

	nameNewBytes := &bytes.Buffer{}
	if err := nameNew.MsgTx.Serialize(nameNewBytes); err != nil {
		undoCommit()
		return err
	}

	nameFirstUpdateBytes := &bytes.Buffer{}
	if err := nameUpdate.MsgTx.Serialize(nameFirstUpdateBytes); err != nil {
		undoCommit()
		return err
	}

	registrationMsg, err := message.CreateMessage(&message.RegisterName{
		Name:            &name,
		Value:           jsonRegistration,
		NameNew:         nameNewBytes.Bytes(),
		NameFirstupdate: nameFirstUpdateBytes.Bytes(),
	}, c.Origin, config.ServerAddress())
	if err != nil {
		undoCommit()
		return err
	}

	if err := c.getRegistrationResponse(registrationMsg); err != nil {
		undoCommit()
		return err
	}

	// Add the name_firstupdate transaction to the list of pending
	// transactions.
	acc.Pending = append(acc.Pending, nameUpdate)

	return nil
}