com.botscrew.messengercdk.service.impl.handler.BotFrameworkReferralHandler Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2018 BotsCrew
 *
 * 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.botscrew.messengercdk.service.impl.handler;

import com.botscrew.botframework.container.ReferralContainer;
import com.botscrew.botframework.domain.argument.ArgumentType;
import com.botscrew.botframework.domain.argument.kit.ArgumentKit;
import com.botscrew.botframework.domain.argument.kit.SimpleArgumentKit;
import com.botscrew.botframework.domain.argument.wrapper.SimpleArgumentWrapper;
import com.botscrew.messengercdk.model.MessengerUser;
import com.botscrew.messengercdk.model.incomming.EventType;
import com.botscrew.messengercdk.model.incomming.Messaging;
import com.botscrew.messengercdk.model.incomming.Referral;
import com.botscrew.messengercdk.service.EventHandler;
import lombok.RequiredArgsConstructor;

/**
 * Default handler for `Referral` event
 */
@RequiredArgsConstructor
public class BotFrameworkReferralHandler implements EventHandler {
    private final ReferralContainer referralContainer;

    @Override
    public EventType getHandlingEventType() {
        return EventType.REFERRAL;
    }

    @Override
    public void handle(MessengerUser messengerUser, Messaging messaging) {
        Referral referral = messaging.getReferral() != null
                ? messaging.getReferral()
                : messaging.getPostback().getReferral();

        ArgumentKit kit = new SimpleArgumentKit();
        kit.put(ArgumentType.ORIGINAL_RESPONSE, new SimpleArgumentWrapper(referral));
        if (messaging.getPostback() != null) {
            kit.put(ArgumentType.POSTBACK,
                    new SimpleArgumentWrapper(messaging.getPostback().getPayload()));
        }
        referralContainer.process(messengerUser, referral.getValue(), kit);
    }
}