io.imunity.vaadin.endpoint.common.plugins.credentials.sms.MobileNumberConfirmationDialog Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of unity-server-vaadin-endpoint-common Show documentation
Common Vaadin endpoint components
The newest version!
/*
 * Copyright (c) 2021 Bixbit - Krzysztof Benedyczak. All rights reserved.
 * See LICENCE.txt file for licensing information.
 */

package io.imunity.vaadin.endpoint.common.plugins.credentials.sms;

import com.vaadin.flow.component.Component;
import com.vaadin.flow.component.confirmdialog.ConfirmDialog;
import com.vaadin.flow.component.formlayout.FormLayout;
import com.vaadin.flow.component.html.Span;
import com.vaadin.flow.component.orderedlayout.FlexComponent;
import com.vaadin.flow.component.orderedlayout.VerticalLayout;
import com.vaadin.flow.component.textfield.TextField;
import io.imunity.vaadin.elements.NotificationPresenter;
import io.imunity.vaadin.endpoint.common.forms.components.CaptchaComponent;
import org.apache.logging.log4j.Logger;
import pl.edu.icm.unity.base.confirmation.ConfirmationInfo;
import pl.edu.icm.unity.base.confirmation.MobileNumberConfirmationConfiguration;
import pl.edu.icm.unity.base.exceptions.EngineException;
import pl.edu.icm.unity.base.exceptions.WrongArgumentException;
import pl.edu.icm.unity.base.message.MessageSource;
import pl.edu.icm.unity.base.utils.Log;
import pl.edu.icm.unity.engine.api.authn.InvocationContext;
import pl.edu.icm.unity.engine.api.confirmation.MobileNumberConfirmationManager;
import pl.edu.icm.unity.engine.api.confirmation.SMSCode;

public class MobileNumberConfirmationDialog extends ConfirmDialog
{
	private static final Logger log = Log.getLogger(Log.U_SERVER_WEB, MobileNumberConfirmationDialog.class);

	private final MessageSource msg;
	private final Callback callback;
	private TextField field;
	private CaptchaComponent captcha;
	private final MobileNumberConfirmationManager mobileConfirmationMan;
	private final NotificationPresenter notificationPresenter;
	private final MobileNumberConfirmationConfiguration confirmationConfiguration;
	private SMSCode code;
	private final String mobileToConfirm;
	private final ConfirmationInfo confirmationInfo;
	private final Component captchaComponent;
	private final Component confirmCodeComponent;
	private boolean capchaVerified = false;
	
	public MobileNumberConfirmationDialog(String mobileToConfirm, ConfirmationInfo confirmatioInfo,
	                                      MessageSource msg,
	                                      MobileNumberConfirmationManager mobileConfirmationMan,
	                                      MobileNumberConfirmationConfiguration confirmationConfiguration,
	                                      Callback callback, NotificationPresenter notificationPresenter)
	{
		setHeader(msg.getMessage("MobileNumberConfirmationDialog.caption"));
		this.msg = msg;
		this.callback = callback;
		this.mobileConfirmationMan = mobileConfirmationMan;
		this.confirmationConfiguration = confirmationConfiguration;
		this.mobileToConfirm = mobileToConfirm;
		this.confirmationInfo = confirmatioInfo;
		this.notificationPresenter = notificationPresenter;

		this.captchaComponent = getCapchaComponent();
		this.captchaComponent.setVisible(false);
		this.confirmCodeComponent = getConfirmCodeComponent();
		this.confirmCodeComponent.setVisible(false);
		setWidth("36em");
		setHeight("25em");
		addConfirmListener(event -> onConfirm());
		addRejectListener(event -> close());
		setCancelable(true);
		setConfirmText("OK");
		add(getContents());
	}
	
	
	private Component getCapchaComponent()
	{
		Span infoLabel = new Span(msg.getMessage("MobileNumberConfirmationDialog.capchaInfo"));
		infoLabel.setSizeFull();	
		captcha = new CaptchaComponent(msg, 4, false);		
		VerticalLayout wrapper = new VerticalLayout();
		wrapper.setMargin(false);
		VerticalLayout component = captcha.getAsComponent();
		wrapper.add(infoLabel, component);
		return wrapper;
	}
	
	private Component getConfirmCodeComponent()
	{
		field = new TextField();
		field.setLabel(msg.getMessage("MobileNumberConfirmationDialog.code"));
		field.setRequiredIndicatorVisible(true);
		Span infoLabel = new Span(msg.getMessage("MobileNumberConfirmationDialog.confirmInfo",
				mobileToConfirm));
		infoLabel.setSizeFull();

		FormLayout mainForm = new FormLayout();
		mainForm.add(field);
		VerticalLayout wrapper = new VerticalLayout();
		wrapper.setAlignItems(FlexComponent.Alignment.START);
		wrapper.setMargin(false);
		wrapper.add(infoLabel, mainForm);

		return wrapper;
	}
	

	protected Component getContents()
	{	
		if (isCapchaNeeded())
		{
			captchaComponent.setVisible(true);
		} else 
		{
			capchaVerified = true;
			sendVerificationCode();
			confirmCodeComponent.setVisible(true);
		}
		VerticalLayout main = new VerticalLayout();
		main.setMargin(false);
		main.setPadding(false);
		main.add(captchaComponent, confirmCodeComponent);
		return main;
	}
	
	private void sendVerificationCode()
	{
		
		try
		{
			code = mobileConfirmationMan.sendConfirmationRequest(confirmationConfiguration,
					mobileToConfirm, confirmationInfo);

		} catch (EngineException er)
		{
			notificationPresenter.showError(msg.getMessage(
					"MobileNumberConfirmationDialog.confirmationSendError",
					mobileToConfirm), er.getMessage());
			log.error("Cannot send confirmation request message", er);
		}
	}

	protected void onConfirm()
	{

		if (!capchaVerified)
		{
			try
			{
				captcha.verify();
				capchaVerified = true;
				sendVerificationCode();
				captchaComponent.setVisible(false);
				confirmCodeComponent.setVisible(true);
				open();
				return;
			} catch (WrongArgumentException e)
			{
				open();
				captcha.setInvalid();
				return;
			}
		}

		
		if (code != null && field.getValue().equals(code.getValue()))
		{
			if (System.currentTimeMillis() > code.getValidTo())
			{
				open();
				setError((msg.getMessage("MobileNumberConfirmationDialog.invalidCode")));
				return;
			}
			confirmationInfo.confirm();
			callback.onConfirm();
		} else
		{
			open();
			setError(msg.getMessage("MobileNumberConfirmationDialog.incorrectCode"));
		}
	}

	private void setError(String msg)
	{
		field.setErrorMessage(msg);
		field.setInvalid(true);
	}

	public interface Callback
	{
		void onConfirm();
	}

	private boolean isCapchaNeeded()
	{
		return InvocationContext.getCurrent().getLoginSession() == null;
	}

}