federation/util/script.go

   1 package util
   2
   3 import (
   4         "sort"
   5
   6         log "github.com/sirupsen/logrus"
   7
   8         "github.com/vapor/crypto"
   9         "github.com/vapor/crypto/ed25519/chainkd"
  10         "github.com/vapor/federation/config"
  11         "github.com/vapor/protocol/vm/vmutil"
  12 )
  13
  14 func ParseFedProg(warders []config.Warder, quorum int) []byte {
  15         SortWarders(warders)
  16
  17         xpubs := []chainkd.XPub{}
  18         for _, w := range warders {
  19                 xpubs = append(xpubs, w.XPub)
  20         }
  21
  22         scirpt, err := vmutil.P2SPMultiSigProgram(chainkd.XPubKeys(xpubs), quorum)
  23         if err != nil {
  24                 log.Panicf("fail to generate federation scirpt for federation: %v", err)
  25         }
  26
  27         scriptHash := crypto.Sha256(scirpt)
  28         wscript, err := vmutil.P2WSHProgram(scriptHash)
  29         if err != nil {
  30                 log.Panicf("Fail converts scriptHash to witness: %v", err)
  31         }
  32
  33         return wscript
  34 }
  35
  36 type byPosition []config.Warder
  37
  38 func (w byPosition) Len() int           { return len(w) }
  39 func (w byPosition) Swap(i, j int)      { w[i], w[j] = w[j], w[i] }
  40 func (w byPosition) Less(i, j int) bool { return w[i].Position < w[j].Position }
  41
  42 func SortWarders(warders []config.Warder) []config.Warder {
  43         sort.Sort(byPosition(warders))
  44         return warders
  45 }