Feat/ra 58 add withdraws to transactions (#118)

* feat: Initial Implementation for adding withdrawals to Transactions.

* fix: merged main, generated new testdata due to yaci changes

* fix: added tests, renamed class due to typo

Author: Kammerlo

api/src/main/java/org/cardanofoundation/rosetta/api/block/mapper/BlockTxToEntity.java

@@ -28,8 +28,8 @@ public BlockTx fromEntity(TxnEntity model) {
           mapper.map(TxnEntity::getTxHash, BlockTx::setHash);
           mapper.map(tx -> tx.getBlock().getHash(), BlockTx::setBlockHash);
           mapper.map(tx -> tx.getBlock().getNumber(), BlockTx::setBlockNo);
-          mapper.map(tx -> 0L, BlockTx::setSize); // TODO saa: why is this 0L?
-          mapper.map(tx -> 0L, BlockTx::setScriptSize); // TODO why is this 0L?
+          mapper.map(tx -> 0L, BlockTx::setSize); // will be calcualted, within the population method
+          mapper.map(tx -> 0L, BlockTx::setScriptSize); // TODO Needs to be calulated if needed
 
           mapper.map(TxnEntity::getInvalid, BlockTx::setValidContract);
 
@@ -43,7 +43,6 @@ public BlockTx fromEntity(TxnEntity model) {
               .ofNullable(source.getFee())
               .map(BigInteger::toString)
               .orElse(null));
-
           return dest;
         })
         .map(model);

api/src/main/java/org/cardanofoundation/rosetta/api/block/mapper/BlockTxToRosettaTransaction.java

@@ -30,6 +30,7 @@ public class BlockTxToRosettaTransaction {
   final DelegationToOperation delegationToOperation;
   final PoolRetirementToOperation poolRetirementToOperation;
   final PoolRegistrationToOperation poolRegistrationToOperation;
+  final WithdrawalToOperation withdrawalToOperation;
 
   private static final OperationStatus status = OperationStatus.builder()
       .status(SUCCESS_OPERATION_STATUS.getStatus()) // TODO saa: need to check the right status
@@ -61,6 +62,7 @@ public Transaction toDto(BlockTx model, @Deprecated String poolDeposit) {
           MutableInt ix = new MutableInt(0);
           @NotNull @Valid List<Operation> destOp = ctx.getDestination().getOperations();
           destOp.addAll(inputToOperation.convert(model.getInputs(), status, ix));
+          destOp.addAll(withdrawalToOperation.convert(model.getWithdrawals(), status, ix));
           destOp.addAll(stakeToOperation.convert(model.getStakeRegistrations(), status, ix));
           destOp.addAll(delegationToOperation.convert(model.getDelegations(), status, ix));
           destOp.addAll(poolRegistrationToOperation.convert(model.getPoolRegistrations(), status, ix));

api/src/main/java/org/cardanofoundation/rosetta/api/block/mapper/StakeRegistrationToOperation.java

@@ -5,6 +5,7 @@
 import lombok.AllArgsConstructor;
 
 import com.bloxbean.cardano.yaci.core.model.certs.CertificateType;
+import org.cardanofoundation.rosetta.common.services.ProtocolParamService;
 import org.modelmapper.ModelMapper;
 import org.modelmapper.spi.MappingContext;
 import org.openapitools.client.model.Amount;
@@ -20,7 +21,6 @@
 public class StakeRegistrationToOperation extends AbstractToOperation<StakeRegistration> {
 
   final ModelMapper modelMapper;
-
   @Override
   public Operation toDto(StakeRegistration model, OperationStatus status, int index) {
     return Optional

api/src/main/java/org/cardanofoundation/rosetta/api/block/mapper/WithdrawalEntityToWithdrawal.java

@@ -0,0 +1,25 @@
+package org.cardanofoundation.rosetta.api.block.mapper;
+
+import java.util.Optional;
+import lombok.AllArgsConstructor;
+import org.cardanofoundation.rosetta.api.block.model.domain.Withdrawal;
+import org.cardanofoundation.rosetta.api.block.model.entity.WithdrawalEntity;
+import org.cardanofoundation.rosetta.common.annotation.OpenApiMapper;
+import org.modelmapper.ModelMapper;
+
+@OpenApiMapper
+@AllArgsConstructor
+public class WithdrawalEntityToWithdrawal {
+
+  private final ModelMapper modelMapper;
+
+  public Withdrawal fromEntity(WithdrawalEntity model) {
+    return Optional
+        .ofNullable(modelMapper.getTypeMap(WithdrawalEntity.class, Withdrawal.class))
+        .orElseGet(() -> modelMapper.createTypeMap(WithdrawalEntity.class, Withdrawal.class))
+        .addMappings(mp -> {
+          mp.map(WithdrawalEntity::getAddress, Withdrawal::setStakeAddress);
+        })
+        .map(model);
+  }
+}

api/src/main/java/org/cardanofoundation/rosetta/api/block/mapper/WithdrawalToOperation.java

@@ -0,0 +1,35 @@
+package org.cardanofoundation.rosetta.api.block.mapper;
+
+import java.util.Optional;
+import lombok.AllArgsConstructor;
+import org.cardanofoundation.rosetta.api.block.model.domain.Withdrawal;
+import org.cardanofoundation.rosetta.common.annotation.OpenApiMapper;
+import org.cardanofoundation.rosetta.common.enumeration.OperationType;
+import org.modelmapper.ModelMapper;
+import org.modelmapper.spi.MappingContext;
+import org.openapitools.client.model.Amount;
+import org.openapitools.client.model.Operation;
+import org.openapitools.client.model.OperationStatus;
+
+@OpenApiMapper
+@AllArgsConstructor
+public class WithdrawalToOperation extends AbstractToOperation<Withdrawal>{
+
+  private final ModelMapper modelMapper;
+
+  @Override
+  public Operation toDto(Withdrawal model, OperationStatus status, int index) {
+    return Optional
+        .ofNullable(modelMapper.getTypeMap(Withdrawal.class, Operation.class))
+        .orElseGet(() -> modelMapper.createTypeMap(Withdrawal.class, Operation.class))
+        .addMappings(mp -> {
+          mp.map(f -> status.getStatus(), Operation::setStatus);
+          mp.map(f -> OperationType.WITHDRAWAL.getValue(), Operation::setType);
+          mp.<String>map(Withdrawal::getStakeAddress, (d, v) -> d.getAccount().setAddress(v));
+          mp.<Amount>map(f -> getDepositAmount("-" + f.getAmount()), (d, v) -> d.getMetadata().setWithdrawalAmount(v));
+          mp.<Long>map(f -> index, (d, v) -> d.getOperationIdentifier().setIndex(v));
+        })
+        .setPostConverter(MappingContext::getDestination)
+        .map(model);
+  }
+}

api/src/main/java/org/cardanofoundation/rosetta/api/block/model/domain/BlockTx.java

@@ -31,5 +31,6 @@ public class BlockTx {
   protected List<Delegation> delegations;
   protected List<PoolRegistration> poolRegistrations;
   protected List<PoolRetirement> poolRetirements;
+  protected List<Withdrawal> withdrawals;
 
 }

api/src/main/java/org/cardanofoundation/rosetta/api/block/model/domain/Withdrawal.java

@@ -0,0 +1,17 @@
+package org.cardanofoundation.rosetta.api.block.model.domain;
+
+import java.math.BigInteger;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import org.cardanofoundation.rosetta.api.block.model.entity.WithdrawalEntity;
+
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class Withdrawal {
+  private String stakeAddress;
+  private BigInteger amount;
+}

api/src/main/java/org/cardanofoundation/rosetta/api/block/model/entity/WithdrawalEntity.java

@@ -0,0 +1,39 @@
+package org.cardanofoundation.rosetta.api.block.model.entity;
+
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.IdClass;
+import jakarta.persistence.Table;
+import java.math.BigInteger;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+
+@Data
+@NoArgsConstructor
+@AllArgsConstructor
+@SuperBuilder
+@Entity
+@Table(name = "withdrawal")
+@IdClass(WithdrawalId.class)
+public class WithdrawalEntity extends BlockAwareEntity {
+  @Id
+  @Column(name = "address")
+  private String address;
+
+  @Id
+  @Column(name = "tx_hash")
+  private String txHash;
+
+  @Column(name = "amount")
+  private BigInteger amount;
+
+  @Column(name = "epoch")
+  private Integer epoch;
+
+  @Column(name = "slot")
+  private Long slot;
+
+}

api/src/main/java/org/cardanofoundation/rosetta/api/block/model/entity/WithdrawalId.java

@@ -0,0 +1,14 @@
+package org.cardanofoundation.rosetta.api.block.model.entity;
+
+import jakarta.persistence.Column;
+import java.io.Serializable;
+import lombok.EqualsAndHashCode;
+
+@EqualsAndHashCode
+public class WithdrawalId implements Serializable {
+    @Column(name = "address")
+    private String address;
+
+    @Column(name = "tx_hash")
+    private String txHash;
+}

api/src/main/java/org/cardanofoundation/rosetta/api/block/model/repository/WithdrawalRepository.java

@@ -0,0 +1,11 @@
+package org.cardanofoundation.rosetta.api.block.model.repository;
+
+import java.util.List;
+import org.cardanofoundation.rosetta.api.block.model.entity.WithdrawalEntity;
+import org.cardanofoundation.rosetta.api.block.model.entity.WithdrawalId;
+import org.springframework.data.jpa.repository.JpaRepository;
+
+public interface WithdrawalRepository extends JpaRepository<WithdrawalEntity, WithdrawalId> {
+
+  List<WithdrawalEntity> findByTxHash(String txHash);
+}