com.dbay.apns4j.model.Payload.java Source code

Introduction

Here is the source code for com.dbay.apns4j.model.Payload.java

Source

/*
 * Copyright 2013 DiscoveryBay Inc.
 *
 * 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.dbay.apns4j.model;

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

import org.json.simple.JSONArray;
import org.json.simple.JSONObject;

/**
 * @author RamosLi
 * For more details, view the following site:
 *   https://developer.apple.com/library/ios/documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/Chapters/ApplePushService.html
 */
public class Payload {
    private static final String APS = "aps";
    private Map<String, Object> params;
    private String alert;
    private Integer badge;
    private String sound = "default.caf";
    private Integer contentAvailable;

    private String alertBody;
    private String alertActionLocKey;
    private String alertLocKey;
    private String[] alertLocArgs;
    private String alertLaunchImage;

    public Map<String, Object> getParams() {
        return params;
    }

    public void setParams(Map<String, Object> params) {
        this.params = params;
    }

    public void addParam(String key, Object obj) {
        if (params == null) {
            params = new HashMap<String, Object>();
        }
        if (APS.equalsIgnoreCase(key)) {
            throw new IllegalArgumentException("the key can't be aps");
        }
        params.put(key, obj);
    }

    public String getAlert() {
        return alert;
    }

    public void setAlert(String alert) {
        this.alert = alert;
    }

    public Integer getBadge() {
        return badge;
    }

    public void setBadge(Integer badge) {
        this.badge = badge;
    }

    public String getSound() {
        return sound;
    }

    public void setSound(String sound) {
        this.sound = sound;
    }

    @SuppressWarnings("unchecked")
    @Override
    public String toString() {
        JSONObject object = new JSONObject();
        JSONObject apsObj = new JSONObject();
        if (getAlert() != null) {
            apsObj.put("alert", getAlert());
        } else {
            if (getAlertBody() != null || getAlertLocKey() != null) {
                JSONObject alertObj = new JSONObject();
                putIntoJson("body", getAlertBody(), alertObj);
                putIntoJson("action-loc-key", getAlertActionLocKey(), alertObj);
                putIntoJson("loc-key", getAlertLocKey(), alertObj);
                putIntoJson("launch-image", getAlertLaunchImage(), alertObj);
                if (getAlertLocArgs() != null) {
                    JSONArray array = new JSONArray();
                    for (String str : getAlertLocArgs()) {
                        array.add(str);
                    }
                    alertObj.put("loc-args", array);
                }
                apsObj.put("alert", alertObj);
            }
        }

        if (getBadge() != null) {
            apsObj.put("badge", getBadge().intValue());
        }
        putIntoJson("sound", getSound(), apsObj);

        if (getContentAvailable() != null) {
            apsObj.put("content-available", getContentAvailable().intValue());
        }

        object.put(APS, apsObj);
        if (getParams() != null) {
            for (Entry<String, Object> e : getParams().entrySet()) {
                object.put(e.getKey(), e.getValue());
            }
        }
        return object.toString();
    }

    @SuppressWarnings("unchecked")
    private void putIntoJson(String key, String value, JSONObject obj) {
        if (value != null) {
            obj.put(key, value);
        }
    }

    public static void main(String[] args) {
        Payload payload = new Payload();
        payload.setAlert("How are you?");
        payload.setBadge(1);
        payload.setSound("a");
        payload.addParam("para1", "1231dfasfwer");
        payload.addParam("number", 12312312312L);
        System.out.println(payload.toString());
    }

    public String getAlertBody() {
        return alertBody;
    }

    public void setAlertBody(String alertBody) {
        this.alertBody = alertBody;
    }

    public String getAlertActionLocKey() {
        return alertActionLocKey;
    }

    public void setAlertActionLocKey(String alertActionLocKey) {
        this.alertActionLocKey = alertActionLocKey;
    }

    public String getAlertLocKey() {
        return alertLocKey;
    }

    public void setAlertLocKey(String alertLocKey) {
        this.alertLocKey = alertLocKey;
    }

    public String getAlertLaunchImage() {
        return alertLaunchImage;
    }

    public void setAlertLaunchImage(String alertLaunchImage) {
        this.alertLaunchImage = alertLaunchImage;
    }

    public String[] getAlertLocArgs() {
        return alertLocArgs;
    }

    public void setAlertLocArgs(String[] alertLocArgs) {
        this.alertLocArgs = alertLocArgs;
    }

    public Integer getContentAvailable() {
        return contentAvailable;
    }

    public void setContentAvailable(Integer contentAvailable) {
        this.contentAvailable = contentAvailable;
    }
}