RTBHouse: Extract tagid from Imp fields (#4306)

Author: pjaworski-rtbh

src/main/java/org/prebid/server/bidder/rtbhouse/RtbhouseBidder.java

@@ -94,6 +94,110 @@ public Result<List<HttpRequest<BidRequest>>> makeHttpRequests(BidRequest bidRequ
         return Result.withValue(BidderUtil.defaultRequest(outgoingRequest, endpointUrl, mapper));
     }
 
+    private ExtImpRtbhouse parseImpExt(Imp imp) {
+        try {
+            return mapper.mapper().convertValue(imp.getExt(), RTBHOUSE_EXT_TYPE_REFERENCE).getBidder();
+        } catch (IllegalArgumentException e) {
+            throw new PreBidException(e.getMessage());
+        }
+    }
+
+    private Price resolveBidFloor(Imp imp, ExtImpRtbhouse impExt, BidRequest bidRequest) {
+        final List<String> brCur = bidRequest.getCur();
+        final Price initialBidFloorPrice = Price.of(imp.getBidfloorcur(), imp.getBidfloor());
+
+        final BigDecimal impExtBidFloor = impExt.getBidFloor();
+        final String impExtCurrency = impExtBidFloor != null && brCur != null && !brCur.isEmpty()
+                ? brCur.getFirst() : null;
+        final Price impExtBidFloorPrice = Price.of(impExtCurrency, impExtBidFloor);
+        final Price resolvedPrice = initialBidFloorPrice.getValue() == null
+                ? impExtBidFloorPrice : initialBidFloorPrice;
+
+        return BidderUtil.isValidPrice(resolvedPrice)
+                && !StringUtils.equalsIgnoreCase(resolvedPrice.getCurrency(), BIDDER_CURRENCY)
+                ? convertBidFloor(resolvedPrice, imp.getId(), bidRequest)
+                : resolvedPrice;
+    }
+
+    private Price convertBidFloor(Price bidFloorPrice, String impId, BidRequest bidRequest) {
+        final String bidFloorCur = bidFloorPrice.getCurrency();
+        try {
+            final BigDecimal convertedPrice = currencyConversionService
+                    .convertCurrency(bidFloorPrice.getValue(), bidRequest, bidFloorCur, BIDDER_CURRENCY);
+
+            return Price.of(BIDDER_CURRENCY, convertedPrice);
+        } catch (PreBidException e) {
+            throw new PreBidException(String.format(
+                    "Unable to convert provided bid floor currency from %s to %s for imp `%s`",
+                    bidFloorCur, BIDDER_CURRENCY, impId));
+        }
+    }
+
+    private static Imp modifyImp(Imp imp, Price bidFloorPrice) {
+        return imp.toBuilder()
+                .tagid(extractTagId(imp))
+                .bidfloorcur(ObjectUtil.getIfNotNull(bidFloorPrice, Price::getCurrency))
+                .bidfloor(ObjectUtil.getIfNotNull(bidFloorPrice, Price::getValue))
+                .pmp(null)
+                .build();
+    }
+
+    private static String extractTagId(Imp imp) {
+        return Optional.ofNullable(imp.getTagid())
+                .filter(StringUtils::isNotBlank)
+                .or(() -> extractGpid(imp))
+                .or(() -> extractAdslot(imp))
+                .or(() -> extractPbadslot(imp))
+                .or(() -> Optional.ofNullable(imp.getId())
+                        .filter(StringUtils::isNotBlank))
+                .orElse(null);
+    }
+
+    private static Optional<String> extractGpid(Imp imp) {
+        return Optional.ofNullable(imp.getExt())
+                .map(ext -> ext.get("gpid"))
+                .map(JsonNode::textValue)
+                .filter(StringUtils::isNotBlank);
+    }
+
+    private static Optional<String> extractAdslot(Imp imp) {
+        return Optional.ofNullable(imp.getExt())
+                .map(ext -> ext.get("data"))
+                .map(data -> data.get("adserver"))
+                .map(adserver -> adserver.get("adslot"))
+                .map(JsonNode::textValue)
+                .filter(StringUtils::isNotBlank);
+    }
+
+    private static Optional<String> extractPbadslot(Imp imp) {
+        return Optional.ofNullable(imp.getExt())
+                .map(ext -> ext.get("data"))
+                .map(data -> data.get("pbadslot"))
+                .map(JsonNode::textValue)
+                .filter(StringUtils::isNotBlank);
+    }
+
+    private Site modifySite(Site site, String publisherId) {
+        final ObjectNode prebidNode = mapper.mapper().createObjectNode();
+        prebidNode.put("publisherId", publisherId);
+
+        final ExtPublisher extPublisher = ExtPublisher.empty();
+        extPublisher.addProperty("prebid", prebidNode);
+
+        final Publisher publisher = Optional.ofNullable(site)
+                .map(Site::getPublisher)
+                .map(Publisher::toBuilder)
+                .orElseGet(Publisher::builder)
+                .ext(extPublisher)
+                .build();
+
+        return Optional.ofNullable(site)
+                .map(Site::toBuilder)
+                .orElseGet(Site::builder)
+                .publisher(publisher)
+                .build();
+    }
+
     @Override
     public Result<List<BidderBid>> makeBids(BidderCall<BidRequest> httpCall, BidRequest bidRequest) {
         try {
@@ -144,23 +248,6 @@ private BidderBid resolveBidderBid(Bid bid,
                 .build();
     }
 
-    private String resolveNativeAdm(String adm, List<BidderError> bidderErrors) {
-        final JsonNode admNode;
-        try {
-            admNode = mapper.mapper().readTree(adm);
-        } catch (JsonProcessingException e) {
-            bidderErrors.add(BidderError.badServerResponse("Unable to parse native adm: %s".formatted(adm)));
-            return null;
-        }
-
-        final JsonNode nativeNode = admNode.get("native");
-        if (nativeNode != null) {
-            return nativeNode.toString();
-        }
-
-        return adm;
-    }
-
     private static BidType getBidType(String impId, List<Imp> imps) {
         for (Imp imp : imps) {
             if (imp.getId().equals(impId)) {
@@ -176,51 +263,21 @@ private static BidType getBidType(String impId, List<Imp> imps) {
         return BidType.banner;
     }
 
-    private ExtImpRtbhouse parseImpExt(Imp imp) {
+    private String resolveNativeAdm(String adm, List<BidderError> bidderErrors) {
+        final JsonNode admNode;
         try {
-            return mapper.mapper().convertValue(imp.getExt(), RTBHOUSE_EXT_TYPE_REFERENCE).getBidder();
-        } catch (IllegalArgumentException e) {
-            throw new PreBidException(e.getMessage());
+            admNode = mapper.mapper().readTree(adm);
+        } catch (JsonProcessingException e) {
+            bidderErrors.add(BidderError.badServerResponse("Unable to parse native adm: %s".formatted(adm)));
+            return null;
         }
-    }
 
-    private static Imp modifyImp(Imp imp, Price bidFloorPrice) {
-        return imp.toBuilder()
-                .bidfloorcur(ObjectUtil.getIfNotNull(bidFloorPrice, Price::getCurrency))
-                .bidfloor(ObjectUtil.getIfNotNull(bidFloorPrice, Price::getValue))
-                .pmp(null)
-                .build();
-    }
-
-    private Price resolveBidFloor(Imp imp, ExtImpRtbhouse impExt, BidRequest bidRequest) {
-        final List<String> brCur = bidRequest.getCur();
-        final Price initialBidFloorPrice = Price.of(imp.getBidfloorcur(), imp.getBidfloor());
-
-        final BigDecimal impExtBidFloor = impExt.getBidFloor();
-        final String impExtCurrency = impExtBidFloor != null && brCur != null && !brCur.isEmpty()
-                ? brCur.getFirst() : null;
-        final Price impExtBidFloorPrice = Price.of(impExtCurrency, impExtBidFloor);
-        final Price resolvedPrice = initialBidFloorPrice.getValue() == null
-                ? impExtBidFloorPrice : initialBidFloorPrice;
-
-        return BidderUtil.isValidPrice(resolvedPrice)
-                && !StringUtils.equalsIgnoreCase(resolvedPrice.getCurrency(), BIDDER_CURRENCY)
-                ? convertBidFloor(resolvedPrice, imp.getId(), bidRequest)
-                : resolvedPrice;
-    }
-
-    private Price convertBidFloor(Price bidFloorPrice, String impId, BidRequest bidRequest) {
-        final String bidFloorCur = bidFloorPrice.getCurrency();
-        try {
-            final BigDecimal convertedPrice = currencyConversionService
-                    .convertCurrency(bidFloorPrice.getValue(), bidRequest, bidFloorCur, BIDDER_CURRENCY);
-
-            return Price.of(BIDDER_CURRENCY, convertedPrice);
-        } catch (PreBidException e) {
-            throw new PreBidException(String.format(
-                    "Unable to convert provided bid floor currency from %s to %s for imp `%s`",
-                    bidFloorCur, BIDDER_CURRENCY, impId));
+        final JsonNode nativeNode = admNode.get("native");
+        if (nativeNode != null) {
+            return nativeNode.toString();
         }
+
+        return adm;
     }
 
     private static Bid resolveMacros(Bid bid) {
@@ -232,25 +289,4 @@ private static Bid resolveMacros(Bid bid) {
                 .adm(StringUtils.replace(bid.getAdm(), PRICE_MACRO, priceAsString))
                 .build();
     }
-
-    private Site modifySite(Site site, String publisherId) {
-        final ObjectNode prebidNode = mapper.mapper().createObjectNode();
-        prebidNode.put("publisherId", publisherId);
-
-        final ExtPublisher extPublisher = ExtPublisher.empty();
-        extPublisher.addProperty("prebid", prebidNode);
-
-        final Publisher publisher = Optional.ofNullable(site)
-                .map(Site::getPublisher)
-                .map(Publisher::toBuilder)
-                .orElseGet(Publisher::builder)
-                .ext(extPublisher)
-                .build();
-
-        return Optional.ofNullable(site)
-                .map(Site::toBuilder)
-                .orElseGet(Site::builder)
-                .publisher(publisher)
-                .build();
-    }
 }