feat: convenience methods to get/set operator PI split (#406)

Author: gpabst

chainio/clients/elcontracts/reader.go

@@ -354,3 +354,20 @@ func (r *ChainReader) GetOperatorAVSSplit(
 
 	return split, nil
 }
+
+func (r *ChainReader) GetOperatorPISplit(
+	ctx context.Context,
+	operator gethcommon.Address,
+) (uint16, error) {
+	if r.rewardsCoordinator == nil {
+		return 0, errors.New("RewardsCoordinator contract not provided")
+	}
+
+	split, err := r.rewardsCoordinator.GetOperatorPISplit(&bind.CallOpts{Context: ctx}, operator)
+
+	if err != nil {
+		return 0, err
+	}
+
+	return split, nil
+}

chainio/clients/elcontracts/writer.go

@@ -385,6 +385,33 @@ func (w *ChainWriter) SetOperatorAVSSplit(
 	return receipt, nil
 }
 
+func (w *ChainWriter) SetOperatorPISplit(
+	ctx context.Context,
+	operator gethcommon.Address,
+	split uint16,
+	waitForReceipt bool,
+) (*gethtypes.Receipt, error) {
+	if w.rewardsCoordinator == nil {
+		return nil, errors.New("RewardsCoordinator contract not provided")
+	}
+
+	noSendTxOpts, err := w.txMgr.GetNoSendTxOpts()
+	if err != nil {
+		return nil, utils.WrapError("failed to get no send tx opts", err)
+	}
+
+	tx, err := w.rewardsCoordinator.SetOperatorPISplit(noSendTxOpts, operator, split)
+	if err != nil {
+		return nil, utils.WrapError("failed to create SetOperatorAVSSplit tx", err)
+	}
+	receipt, err := w.txMgr.Send(ctx, tx, waitForReceipt)
+	if err != nil {
+		return nil, utils.WrapError("failed to send tx", err)
+	}
+
+	return receipt, nil
+}
+
 func (w *ChainWriter) ProcessClaims(
 	ctx context.Context,
 	claims []rewardscoordinator.IRewardsCoordinatorRewardsMerkleClaim,