package lookout import ( "github.com/btcsuite/btcd/wire" "github.com/lightningnetwork/lnd/lnwallet" ) // PunisherConfig houses the resources required by the Punisher. type PunisherConfig struct { // PublishTx provides the ability to send a signed transaction to the // network. PublishTx func(*wire.MsgTx) error // TODO(conner) add DB tracking and spend ntfn registration to see if // ours confirmed or not } // BreachPunisher handles the responsibility of constructing and broadcasting // justice transactions. Justice transactions are constructed from previously // accepted state updates uploaded by the watchtower's clients. type BreachPunisher struct { cfg *PunisherConfig } // NewBreachPunisher constructs a new BreachPunisher given a PunisherConfig. func NewBreachPunisher(cfg *PunisherConfig) *BreachPunisher { return &BreachPunisher{ cfg: cfg, } } // Punish constructs a justice transaction given a JusticeDescriptor and // publishes is it to the network. func (p *BreachPunisher) Punish(desc *JusticeDescriptor, quit <-chan struct{}) error { justiceTxn, err := desc.CreateJusticeTxn() if err != nil { log.Errorf("Unable to create justice txn for "+ "client=%s with breach-txid=%x: %v", desc.SessionInfo.ID, desc.BreachedCommitTx.TxHash(), err) return err } log.Infof("Publishing justice transaction txid=%x for client=%s", justiceTxn.TxHash(), desc.SessionInfo.ID) err = p.cfg.PublishTx(justiceTxn) if err != nil && err != lnwallet.ErrDoubleSpend { log.Errorf("Unable to publish justice txn for client=%s"+ "with breach-txid=%x: %v", desc.SessionInfo.ID, desc.BreachedCommitTx.TxHash(), err) return err } // TODO(conner): register for spend and remove from db after // confirmation return nil }