src/main/java/com/google/android/libraries/feed/feedrequestmanager/UploadableActionsRequestBuilder.java

// Copyright 2019 The Feed Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.android.libraries.feed.feedrequestmanager;

import com.google.android.libraries.feed.api.common.ActionPropertiesWithId;
import com.google.android.libraries.feed.api.protocoladapter.ProtocolAdapter;
import com.google.android.libraries.feed.common.Result;
import com.google.android.libraries.feed.internalapi.actionmanager.ActionReader;
import com.google.search.now.feed.client.StreamDataProto.StreamUploadableAction;
import com.google.search.now.wire.feed.ActionPayloadProto.ActionPayload;
import com.google.search.now.wire.feed.ActionPropertiesProto.ActionProperties;
import com.google.search.now.wire.feed.ActionRequestProto.ActionRequest;
import com.google.search.now.wire.feed.ConsistencyTokenProto.ConsistencyToken;
import com.google.search.now.wire.feed.ContentIdProto.ContentId;
import com.google.search.now.wire.feed.FeedActionProto.FeedAction;
import com.google.search.now.wire.feed.FeedActionRequestProto.FeedActionRequest;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

// A class that creates an ActionsRequest for uploading actions
final class UploadableActionsRequestBuilder {
  private Set<StreamUploadableAction> uploadableActions;
  private ConsistencyToken token;
  private final ActionReader actionReader;
  private final ProtocolAdapter protocolAdapter;

  UploadableActionsRequestBuilder(ActionReader actionReader, ProtocolAdapter protocolAdapter) {
    this.actionReader = actionReader;
    this.protocolAdapter = protocolAdapter;
  }

  UploadableActionsRequestBuilder setConsistencyToken(ConsistencyToken token) {
    this.token = token;
    return this;
  }

  boolean hasConsistencyToken() {
    return token != null;
  }

  UploadableActionsRequestBuilder setActions(Set<StreamUploadableAction> uploadableActions) {
    this.uploadableActions = uploadableActions;
    return this;
  }

  public ActionRequest build() {
    ActionRequest.Builder requestBuilder =
        ActionRequest.newBuilder()
            .setRequestVersion(ActionRequest.RequestVersion.FEED_UPLOAD_ACTION);
    FeedActionRequest.Builder feedActionRequestBuilder = FeedActionRequest.newBuilder();
    if (uploadableActions != null) {
      Map<String, ActionProperties> actionPropertyMap = new HashMap<>();
      List<String> contentIds = new ArrayList<>(uploadableActions.size());
      for (StreamUploadableAction action : uploadableActions) {
        contentIds.add(action.getFeatureContentId());
      }
      List<ActionPropertiesWithId> actionProperties = getActionPropertiesForIds(contentIds);
      for (ActionPropertiesWithId property : actionProperties) {
        actionPropertyMap.put(
            property.contentId,
            ActionProperties.newBuilder().setActionData(property.actionData).build());
      }
      for (StreamUploadableAction action : uploadableActions) {
        String contentId = action.getFeatureContentId();
        FeedAction.Builder feedAction = FeedAction.newBuilder();
        ActionPayload payload = action.getPayload();
        feedAction.setActionPayload(payload);
        ActionProperties properties = actionPropertyMap.get(contentId);
        if (properties != null) {
          feedAction.setActionProperties(properties);
        }
        Result<ContentId> contentIdResult = protocolAdapter.getWireContentId(contentId);
        if (contentIdResult.isSuccessful()) {
          feedAction.setContentId(contentIdResult.getValue());
        }
        feedActionRequestBuilder.addFeedAction(feedAction);
      }
    }
    if (hasConsistencyToken()) {
      feedActionRequestBuilder.setConsistencyToken(token);
    }
    requestBuilder.setExtension(
        FeedActionRequest.feedActionRequest, feedActionRequestBuilder.build());

    return requestBuilder.build();
  }

  private List<ActionPropertiesWithId> getActionPropertiesForIds(List<String> ids) {
    Result<List<ActionPropertiesWithId>> actionResult = actionReader.getActionProperties(ids);
    if (actionResult.isSuccessful()) {
      return actionResult.getValue();
    }
    return new ArrayList<>();
  }
}