Add processing of imp.ext.data to mediagrid (#1293)

src/main/java/org/prebid/server/bidder/grid/GridBidder.java

@@ -2,8 +2,12 @@
 
 import com.iab.openrtb.request.Imp;
 import com.iab.openrtb.response.Bid;
+import org.apache.commons.lang3.StringUtils;
 import org.prebid.server.bidder.OpenrtbBidder;
 import org.prebid.server.bidder.grid.model.ExtImpGrid;
+import org.prebid.server.bidder.grid.model.GridExtImp;
+import org.prebid.server.bidder.grid.model.GridExtImpData;
+import org.prebid.server.bidder.grid.model.GridExtImpDataAdServer;
 import org.prebid.server.exception.PreBidException;
 import org.prebid.server.json.JacksonMapper;
 import org.prebid.server.proto.openrtb.ext.response.BidType;
@@ -21,6 +25,26 @@ protected Imp modifyImp(Imp imp, ExtImpGrid impExt) {
         if (impExt.getUid() == null || impExt.getUid() == 0) {
             throw new PreBidException("uid is empty");
         }
+
+        final GridExtImp gridExtImp;
+        try {
+            gridExtImp = mapper.mapper().convertValue(imp.getExt(), GridExtImp.class);
+        } catch (Exception e) {
+            throw new PreBidException(e.getMessage());
+        }
+
+        final GridExtImpData extImpData = gridExtImp != null ? gridExtImp.getData() : null;
+        final GridExtImpDataAdServer adServer = extImpData != null ? extImpData.getAdServer() : null;
+        final String adSlot = adServer != null ? adServer.getAdSlot() : null;
+        if (StringUtils.isNotEmpty(adSlot)) {
+
+            final GridExtImp modifiedGridExtImp = gridExtImp.toBuilder()
+                    .gpid(adSlot)
+                    .build();
+
+            return imp.toBuilder().ext(mapper.mapper().valueToTree(modifiedGridExtImp)).build();
+        }
+
         return imp;
     }
 

src/main/java/org/prebid/server/bidder/grid/model/GridExtImp.java

@@ -0,0 +1,19 @@
+package org.prebid.server.bidder.grid.model;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import lombok.Builder;
+import lombok.Value;
+import org.prebid.server.proto.openrtb.ext.request.ExtImpPrebid;
+
+@Builder(toBuilder = true)
+@Value
+public class GridExtImp {
+
+    ExtImpPrebid prebid;
+
+    JsonNode bidder;
+
+    GridExtImpData data;
+
+    String gpid;
+}

src/main/java/org/prebid/server/bidder/grid/model/GridExtImpData.java

@@ -0,0 +1,16 @@
+package org.prebid.server.bidder.grid.model;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Value;
+
+@AllArgsConstructor(staticName = "of")
+@Value
+public class GridExtImpData {
+
+    @JsonProperty("pbadslot")
+    String pbAdSlot;
+
+    @JsonProperty("adserver")
+    GridExtImpDataAdServer adServer;
+}

src/main/java/org/prebid/server/bidder/grid/model/GridExtImpDataAdServer.java

@@ -0,0 +1,15 @@
+package org.prebid.server.bidder.grid.model;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Value;
+
+@AllArgsConstructor(staticName = "of")
+@Value
+public class GridExtImpDataAdServer {
+
+    String name;
+
+    @JsonProperty("adslot")
+    String adSlot;
+}

src/test/java/org/prebid/server/bidder/grid/GridBidderTest.java

@@ -12,6 +12,9 @@
 import org.junit.Test;
 import org.prebid.server.VertxTest;
 import org.prebid.server.bidder.grid.model.ExtImpGrid;
+import org.prebid.server.bidder.grid.model.GridExtImp;
+import org.prebid.server.bidder.grid.model.GridExtImpData;
+import org.prebid.server.bidder.grid.model.GridExtImpDataAdServer;
 import org.prebid.server.bidder.model.BidderBid;
 import org.prebid.server.bidder.model.BidderError;
 import org.prebid.server.bidder.model.HttpCall;
@@ -231,6 +234,23 @@ public void makeBidsShouldReturnErrorIfImpIdsFromBidAndRequestWereNotMatchedAndI
                 .containsOnly(BidderError.badServerResponse("Failed to find impression for ID: 123"));
     }
 
+    @Test
+    public void modifyImpShouldChangeImpExt() {
+        // given
+        final GridExtImp gridExtImp = GridExtImp.builder()
+                .data(GridExtImpData.of(null, GridExtImpDataAdServer.of("name", "adslot")))
+                .build();
+
+        final Imp imp = Imp.builder().ext(mapper.valueToTree(gridExtImp)).build();
+
+        // when
+        final Imp modifiedImp = gridBidder.modifyImp(imp, ExtImpGrid.of(1));
+
+        // then
+        assertThat(mapper.convertValue(modifiedImp.getExt(), GridExtImp.class).getGpid())
+                .isEqualTo(gridExtImp.getData().getAdServer().getAdSlot());
+    }
+
     private static BidResponse givenBidResponse(UnaryOperator<BidResponse.BidResponseBuilder> bidResponseCustomizer,
                                                 UnaryOperator<Bid.BidBuilder> bidCustomizer) {
         return bidResponseCustomizer.apply(BidResponse.builder()

src/test/resources/org/prebid/server/it/openrtb2/grid/test-auction-grid-request.json

@@ -14,6 +14,12 @@
       "ext": {
         "grid": {
           "uid": 1
+        },
+        "data": {
+          "adserver": {
+            "name": "some_name",
+            "adslot": "some_slot"
+          }
         }
       }
     }

src/test/resources/org/prebid/server/it/openrtb2/grid/test-grid-bid-request-1.json

@@ -14,7 +14,14 @@
       "ext": {
         "bidder": {
           "uid": 1
-        }
+        },
+        "data": {
+          "adserver": {
+            "name": "some_name",
+            "adslot": "some_slot"
+          }
+        },
+        "gpid" : "some_slot"
       }
     }
   ],