feat: add processIssuing (#152)

[bytom/vapor.git] / vendor / github.com / bytom / protocol / bc / tx.go

diff --git a/vendor/github.com/bytom/protocol/bc/tx.go b/vendor/github.com/bytom/protocol/bc/tx.go
new file mode 100644 (file)
index 0000000..c73ea3a
--- /dev/null
+++ b/vendor/github.com/bytom/protocol/bc/tx.go
@@ -0,0 +1,73 @@
+package bc
+
+import (
+       "github.com/bytom/crypto/sha3pool"
+       "github.com/bytom/errors"
+)
+
+// Tx is a wrapper for the entries-based representation of a transaction.
+type Tx struct {
+       *TxHeader
+       ID       Hash
+       Entries  map[Hash]Entry
+       InputIDs []Hash // 1:1 correspondence with TxData.Inputs
+
+       SpentOutputIDs []Hash
+       GasInputIDs    []Hash
+}
+
+// SigHash ...
+func (tx *Tx) SigHash(n uint32) (hash Hash) {
+       hasher := sha3pool.Get256()
+       defer sha3pool.Put256(hasher)
+
+       tx.InputIDs[n].WriteTo(hasher)
+       tx.ID.WriteTo(hasher)
+       hash.ReadFrom(hasher)
+       return hash
+}
+
+// Convenience routines for accessing entries of specific types by ID.
+var (
+       ErrEntryType    = errors.New("invalid entry type")
+       ErrMissingEntry = errors.New("missing entry")
+)
+
+// Output try to get the output entry by given hash
+func (tx *Tx) Output(id Hash) (*Output, error) {
+       e, ok := tx.Entries[id]
+       if !ok || e == nil {
+               return nil, errors.Wrapf(ErrMissingEntry, "id %x", id.Bytes())
+       }
+       o, ok := e.(*Output)
+       if !ok {
+               return nil, errors.Wrapf(ErrEntryType, "entry %x has unexpected type %T", id.Bytes(), e)
+       }
+       return o, nil
+}
+
+// Spend try to get the spend entry by given hash
+func (tx *Tx) Spend(id Hash) (*Spend, error) {
+       e, ok := tx.Entries[id]
+       if !ok || e == nil {
+               return nil, errors.Wrapf(ErrMissingEntry, "id %x", id.Bytes())
+       }
+       sp, ok := e.(*Spend)
+       if !ok {
+               return nil, errors.Wrapf(ErrEntryType, "entry %x has unexpected type %T", id.Bytes(), e)
+       }
+       return sp, nil
+}
+
+// Issuance try to get the issuance entry by given hash
+func (tx *Tx) Issuance(id Hash) (*Issuance, error) {
+       e, ok := tx.Entries[id]
+       if !ok || e == nil {
+               return nil, errors.Wrapf(ErrMissingEntry, "id %x", id.Bytes())
+       }
+       iss, ok := e.(*Issuance)
+       if !ok {
+               return nil, errors.Wrapf(ErrEntryType, "entry %x has unexpected type %T", id.Bytes(), e)
+       }
+       return iss, nil
+}