de.taimos.dvalin.notification.push.APNSMessage Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dvalin-notification-core Show documentation

There is a newer version: 1.37

Show newest version

package de.taimos.dvalin.notification.push;

import java.util.HashMap;
import java.util.Map;

public class APNSMessage extends PushMessage {
    
    private final Platform type;
    
    private final String message;
    private String title;
    private int badge;
    
    private final Map customData = new HashMap<>();
    
    public APNSMessage(String message) {
        this(Platform.APNS, message);
    }
    
    public APNSMessage(Platform type, String message) {
        this.type = type;
        this.message = message;
    }
    
    public void setTitle(String title) {
        this.title = title;
    }
    
    public void setBadge(int badge) {
        this.badge = badge;
    }
    
    public void addCustomData(String key, Object value) {
        this.customData.put(key, value);
    }
    
    @Override
    protected Map getPayload() {
        Map map = new HashMap<>();
        map.putAll(this.customData);
        
        if (this.title != null) {
            Map alert = new HashMap<>();
            alert.put("body", this.message);
            alert.put("title", this.title);
            map.put("alert", alert);
        } else {
            map.put("alert", this.message);
        }
        map.put("badge", this.badge);
        return map;
    }
    
    @Override
    public Platform getType() {
        return this.type;
    }
}