Hulk did something

[bytom/vapor.git] / asset / annotate.go

diff --git a/asset/annotate.go b/asset/annotate.go
new file mode 100644 (file)
index 0000000..514166e
--- /dev/null
+++ b/asset/annotate.go
@@ -0,0 +1,47 @@
+package asset
+
+import (
+       "encoding/json"
+
+       "github.com/vapor/blockchain/query"
+       chainjson "github.com/vapor/encoding/json"
+       "github.com/vapor/protocol/vm/vmutil"
+)
+
+func isValidJSON(b []byte) bool {
+       var v interface{}
+       err := json.Unmarshal(b, &v)
+       return err == nil
+}
+
+//Annotated annotate the asset
+func Annotated(a *Asset) (*query.AnnotatedAsset, error) {
+       jsonDefinition := json.RawMessage(`{}`)
+
+       // a.RawDefinitionByte is the asset definition as it appears on the
+       // blockchain, so it's untrusted and may not be valid json.
+       if isValidJSON(a.RawDefinitionByte) {
+               jsonDefinition = json.RawMessage(a.RawDefinitionByte)
+       }
+
+       annotatedAsset := &query.AnnotatedAsset{
+               ID:                a.AssetID,
+               Alias:             *a.Alias,
+               VMVersion:         a.VMVersion,
+               RawDefinitionByte: a.RawDefinitionByte,
+               Definition:        &jsonDefinition,
+               IssuanceProgram:   chainjson.HexBytes(a.IssuanceProgram),
+       }
+
+       annotatedAsset.LimitHeight = vmutil.GetIssuanceProgramRestrictHeight(a.IssuanceProgram)
+       if a.Signer != nil {
+               annotatedAsset.AnnotatedSigner = query.AnnotatedSigner{
+                       Type:       a.Signer.Type,
+                       XPubs:      a.Signer.XPubs,
+                       Quorum:     a.Signer.Quorum,
+                       KeyIndex:   a.Signer.KeyIndex,
+                       DeriveRule: a.Signer.DeriveRule,
+               }
+       }
+       return annotatedAsset, nil
+}