org.nustaq.kontraktor.apputil.RegistrationMixin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of service-support Show documentation
service registry
There is a newer version: 5.2.0
package org.nustaq.kontraktor.apputil;

import org.nustaq.kontraktor.Actor;
import org.nustaq.kontraktor.Actors;
import org.nustaq.kontraktor.IPromise;
import org.nustaq.kontraktor.Promise;
import org.nustaq.kontraktor.annotations.CallerSideMethod;
import org.nustaq.kontraktor.annotations.Local;
import org.nustaq.kontraktor.remoting.http.undertow.builder.BldFourK;
import org.nustaq.kontraktor.util.Log;
import org.nustaq.reallive.api.Record;
import org.nustaq.serialization.coders.Unknown;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

public interface RegistrationMixin> extends LinkMapperMixin {

    static String UserTableName = "user";

    long MAX_AGE = TimeUnit.DAYS.toMillis(3);

    static void auto(BldFourK bld, Object registrationMixin) {
        LinkMapperMixin.auto(bld,registrationMixin);
    }

    /**
     * UI entry point
     * @param data
     * @return
     */
    default IPromise register(Unknown data) {
        Promise p = new Promise();
        RegistrationRecord registerform = new RegistrationRecord(data);
        registerform.creation(System.currentTimeMillis());
        registerform.name(registerform.getName().trim());
        String name = registerform.getName().trim().toLowerCase();
        getDClient().tbl(UserTableName).forEach(
            rec -> new UserRecord(rec).getName().equalsIgnoreCase(name),
            (rec,err) -> {
                if ( rec != null ) {
                    p.reject("Der Name '" + rec.get("name") + "' existiert bereits");
                } else {
                    if (!p.isSettled()) {
                        putRecord(registerform).then( (r,e) -> {
                            if ( r != null ) {
                                HashMap copiedMap = createData(data, registerform);
                                Mailer.get().sendTemplateChannelMail(
                                    "registration",
                                    registerform.getEmail(),
                                    "Bitte bestätigen sie Ihre Registrierung",
                                    "mail/opt-in.html",
                                    copiedMap);
                                p.resolve(registerform);
                            } else {
                                p.reject(e);
                            }
                        });
                    }
                }
            });
        return p;
    }

    static HashMap createData(Unknown data, RegistrationRecord registerform) {
        HashMap copiedMap = new HashMap<>();
        data.getFields().forEach( (k,v) -> copiedMap.put(k,v));
        copiedMap.put("confirm-link", LinkTableName +"/"+registerform.getKey());
        return copiedMap;
    }

    @CallerSideMethod @Local
    default String handleLinkSuccess(String linkId, Record linkRecord) {
        if ( linkRecord.getSafeString("type").equals("Registration") ) {
            RegistrationRecord regRec = new RegistrationRecord(linkRecord);
            if ( System.currentTimeMillis() - regRec.getCreation() > MAX_AGE ) {
                // outdated
                return applyTemplate(regRec, "html/registration-old.html");
            } else {
                //create user, check uniqueness should be done in create using atomic operations
                Record existing = getDClient().tbl(UserTableName).get(regRec.getEmail().toLowerCase()).await();
                if ( existing == null )
                    existing = getDClient().tbl(UserTableName).find( rec -> rec.getSafeString("name").equalsIgnoreCase(regRec.getName())).await();
                if ( existing != null ) {
                    return applyTemplate(regRec, "html/registration-exists.html");
                }
                createUserFromRegistrationRecord(regRec);
                return applyTemplate(regRec, "html/registration-success.html");
            }
        } else { // wrong type
            return applyTemplate(null, "html/registration-fail.html");
        }
    }

    @CallerSideMethod @Local
    default String handleLinkFailure(String linkId) {
        return applyTemplate(null, "html/registration-fail.html");
    }

    default IPromise createUserFromRegistrationRecord(RegistrationRecord rec) {
        UserRecord u = new UserRecord(rec.getEmail().toLowerCase());
        u.name(rec.getName())
            .email(rec.getEmail())
            .creation(System.currentTimeMillis())
            .pwd(rec.getPwd())
            .role("User");
        getDClient().tbl(UserTableName).setRecord(u.getRecord());
        return Actors.resolve(u);
    }

    static String applyTemplate(RegistrationRecord rec, String template) {
        try {
            Map data = rec != null ? rec.asMap() : new HashMap<>();
            return Mailer.applyTemplate(template, data,(k, v) -> ""+v);
        } catch (IOException e) {
            Log.Warn(RegistrationMixin.class,e);
            return "internal error "+e+"";
        }
    }

}