MessageBuilder.java

浏览该文件的文档.

package com.netease.nimlib.sdk.msg;

import android.content.Context;
import android.net.Uri;
import android.text.TextUtils;

import com.netease.nimlib.BuildConfig;
import com.netease.nimlib.NimNosSceneKeyConstant;
import com.netease.nimlib.SDKCache;
import com.netease.nimlib.biz.constant.IAuthService;
import com.netease.nimlib.migration.MessageConvert;
import com.netease.nimlib.notifier.support26.annotation.NonNull;
import com.netease.nimlib.sdk.msg.attachment.AudioAttachment;
import com.netease.nimlib.sdk.msg.attachment.FileAttachment;
import com.netease.nimlib.sdk.msg.attachment.ImageAttachment;
import com.netease.nimlib.sdk.msg.attachment.LocationAttachment;
import com.netease.nimlib.sdk.msg.attachment.MsgAttachment;
import com.netease.nimlib.sdk.msg.attachment.VideoAttachment;
import com.netease.nimlib.sdk.msg.constant.AttachStatusEnum;
import com.netease.nimlib.sdk.msg.constant.MsgDirectionEnum;
import com.netease.nimlib.sdk.msg.constant.MsgStatusEnum;
import com.netease.nimlib.sdk.msg.constant.MsgTypeEnum;
import com.netease.nimlib.sdk.msg.constant.SessionTypeEnum;
import com.netease.nimlib.sdk.msg.model.CustomMessageConfig;
import com.netease.nimlib.sdk.msg.model.IMMessage;
import com.netease.nimlib.sdk.msg.model.MessageRobotInfo;
import com.netease.nimlib.sdk.robot.model.RobotAttachment;
import com.netease.nimlib.sdk.robot.model.RobotMsgType;
import com.netease.nimlib.sdk.util.UriUtils;
import com.netease.nimlib.session.IMMessageImpl;
import com.netease.nimlib.util.BitmapDecoder;
import com.netease.nimlib.util.StringUtil;
import com.netease.nimlib.util.TimeUtil;

import org.json.JSONException;
import org.json.JSONObject;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class MessageBuilder {

    public static IMMessage createTextMessage(String sessionId, SessionTypeEnum sessionType, String text) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.text.getValue());
        msg.setContent(text);
        return msg;
    }

    public static IMMessage createImageMessage(String sessionId, SessionTypeEnum sessionType, File file) {
        return createImageMessage(sessionId, sessionType, file, null, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createImageMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri) {
        return createImageMessage(context,sessionId, sessionType, uri, null, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createImageMessage(String sessionId, SessionTypeEnum sessionType, File file, String displayName) {
        return createImageMessage(sessionId, sessionType, file, displayName, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createImageMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri,String displayName) {
        return createImageMessage(context,sessionId, sessionType, uri, displayName, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createImageMessage(String sessionId, SessionTypeEnum sessionType, File file, String displayName, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.image.getValue());

        final ImageAttachment attachment = createImageAttachment(file, displayName, nosTokenSceneKey);
        msg.setAttachment(attachment);
        return msg;
    }

    public static IMMessage createImageMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, String displayName, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.image.getValue());
        final ImageAttachment attachment = createImageAttachment(context, uri, displayName, nosTokenSceneKey);
        msg.setAttachment(attachment);
        return msg;
    }

    public static IMMessage createAudioMessage(String sessionId, SessionTypeEnum sessionType, File file, long duration) {
        return createAudioMessage(sessionId, sessionType, file, duration, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createAudioMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, long duration) {
        return createAudioMessage(context,sessionId, sessionType, uri, duration, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createAudioMessage(String sessionId, SessionTypeEnum sessionType, File file, long duration, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.audio.getValue());
        if (duration > 0 && duration < 1000) {// 最低显示1秒
            duration = 1000;
        }
        final AudioAttachment attachment = createAudioAttachment(file, duration, nosTokenSceneKey);
        msg.setAttachment(attachment);
        return msg;
    }

    public static IMMessage createAudioMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, long duration, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.audio.getValue());
        if (duration > 0 && duration < 1000) {// 最低显示1秒
            duration = 1000;
        }
        final AudioAttachment attachment = createAudioAttachment(context, uri, duration, nosTokenSceneKey);
        msg.setAttachment(attachment);
        return msg;
    }

    public static IMMessage createLocationMessage(String sessionId, SessionTypeEnum sessionType, double lat, double lng, String addr) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.location.getValue());

        final LocationAttachment location = new LocationAttachment();
        location.setLatitude(lat);
        location.setLongitude(lng);
        location.setAddress(addr);
        msg.setAttachStatus(AttachStatusEnum.transferred);
        msg.setAttachment(location);

        return msg;
    }

    public static IMMessage createVideoMessage(String sessionId, SessionTypeEnum sessionType, File file, long duration, int width, int height, String displayName) {
        return createVideoMessage(sessionId, sessionType, file, duration, width, height, displayName, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    @SuppressWarnings("java:S107")
    public static IMMessage createVideoMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, long duration, int width, int height, String displayName) {
        return createVideoMessage(context,sessionId, sessionType, uri, duration, width, height, displayName, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createVideoMessage(String sessionId, SessionTypeEnum sessionType, File file, long duration, int width, int height, String displayName, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.video.getValue());

        final VideoAttachment attachment = createVideoAttachment(file, duration, width, height, displayName, nosTokenSceneKey);
        msg.setAttachment(attachment);

        BitmapDecoder.extractThumbnail(file.getPath(), attachment.getThumbPathForSave());

        return msg;
    }

    @SuppressWarnings("java:S107")
    public static IMMessage createVideoMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, long duration, int width, int height, String displayName, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.video.getValue());
        final VideoAttachment attachment = createVideoAttachment(context, uri, duration, width, height, displayName, nosTokenSceneKey);
        msg.setAttachment(attachment);

        BitmapDecoder.extractThumbnail(uri, attachment.getThumbPathForSave());

        return msg;
    }

    public static IMMessage createFileMessage(String sessionId, SessionTypeEnum sessionType, File file, String displayName) {
        return createFileMessage(sessionId, sessionType, file, displayName, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createFileMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, String displayName) {
        return createFileMessage(context,sessionId, sessionType, uri, displayName, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createFileMessage(String sessionId, SessionTypeEnum sessionType, File file, String displayName, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.file.getValue());

        final FileAttachment attachment = createFileAttachment(file, displayName, nosTokenSceneKey);
        msg.setAttachment(attachment);
        return msg;
    }

    public static IMMessage createFileMessage(Context context,String sessionId, SessionTypeEnum sessionType, Uri uri, String displayName, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.file.getValue());
        final FileAttachment attachment = createFileAttachment(context, uri, displayName, nosTokenSceneKey);
        msg.setAttachment(attachment);
        return msg;
    }

    public static IMMessage createTipMessage(String sessionId, SessionTypeEnum sessionType) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.tip.getValue());

        return msg;
    }

    public static IMMessage createRobotTextMessage(String sessionId, SessionTypeEnum sessionType, String text, MessageRobotInfo messageRobotInfo) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.text.getValue());
        msg.setContent(text);
        msg.setRobotInfo(messageRobotInfo);
        return msg;
    }

    public static IMMessage createRobotTipMessage(String sessionId, SessionTypeEnum sessionType, MessageRobotInfo messageRobotInfo) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.tip.getValue());
        msg.setRobotInfo(messageRobotInfo);
        return msg;
    }

    public static IMMessage createCustomMessage(String sessionId, SessionTypeEnum sessionType, MsgAttachment attachment) {
        return createCustomMessage(sessionId, sessionType, null, attachment, null);
    }

    public static IMMessage createCustomMessage(String sessionId, SessionTypeEnum sessionType, String content, MsgAttachment attachment) {
        return createCustomMessage(sessionId, sessionType, content, attachment, null);
    }

    public static IMMessage createCustomMessage(String sessionId, SessionTypeEnum sessionType, String content, MsgAttachment attachment, CustomMessageConfig config) {
        return createCustomMessage(sessionId, sessionType, content, attachment, config, NimNosSceneKeyConstant.NIM_DEFAULT_IM);
    }

    public static IMMessage createCustomMessage(String sessionId, SessionTypeEnum sessionType, String content, MsgAttachment attachment, CustomMessageConfig config, String nosTokenSceneKey) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.custom.getValue());
        msg.setContent(content);
        if (attachment != null && attachment instanceof FileAttachment) {
            ((FileAttachment) attachment).setNosTokenSceneKey(nosTokenSceneKey);
        }
        msg.setAttachment(attachment);
        msg.setConfig(config);
        return msg;
    }

    public static IMMessage createRobotMessage(String sessionId, SessionTypeEnum sessionType, String robotAccount, String text, String type, String content, String target, String params) {
        if (TextUtils.isEmpty(type) || TextUtils.isEmpty(robotAccount)) {
            throw new IllegalArgumentException("Invalid param, type and robot account should not be null");
        }

        if (type.equals(RobotMsgType.TEXT) && content == null) {
            throw new IllegalArgumentException("Invalid param, content should not be null");
        } else if (type.equals(RobotMsgType.LINK) && TextUtils.isEmpty(target)) {
            throw new IllegalArgumentException("Invalid param, target should not be null");
        }

        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.robot.getValue());

        // build attachment
        RobotAttachment robotAttachment = new RobotAttachment();
        robotAttachment.initSend(robotAccount, type, content, target, params);
        msg.setAttachment(robotAttachment);
        msg.setContent(text); // 文案显示用,便于全文检索等

        return msg;
    }

    public static IMMessage createNrtcNetcallMessage(String sessionId, SessionTypeEnum sessionType, MsgAttachment netCallAttachment) {
        IMMessageImpl msg = initSendMessage(sessionId, sessionType);
        msg.setMsgType(MsgTypeEnum.nrtc_netcall.getValue());
        msg.setAttachment(netCallAttachment);
        return msg;
    }

    public static IMMessage createEmptyMessage(String sessionId, SessionTypeEnum sessionType, long time) {
        IMMessageImpl msg = new IMMessageImpl();
        msg.setSessionId(sessionId);
        msg.setSessionType(sessionType);
        msg.setTime(time);
        return msg;
    }

    public static IMMessage createForwardMessage(IMMessage message, String sessionId, SessionTypeEnum sessionType) {
        if (message.getMsgType() == MsgTypeEnum.notification || message.getMsgType() == MsgTypeEnum.avchat || message.getMsgType() == MsgTypeEnum.robot) {
            return null;
        }
        IMMessageImpl forward = ((IMMessageImpl) message).deepClone();
        if (forward != null) {
            forward.setSessionId(sessionId);
            forward.setSessionType(sessionType);
            forward.setUuid(StringUtil.get32UUID());
            forward.setFromAccount(SDKCache.getAccount());
            forward.setDirect(MsgDirectionEnum.Out);
            forward.setStatus(MsgStatusEnum.sending);
            forward.setTime(TimeUtil.currentTimeMillis());
            forward.setServerId(0L);
            forward.setMessageId(0L);
            // clear team msg ack
            forward.setMsgAck(false);
            forward.setHasSendAck(false);
            forward.setTeamMsgAckCount(0);
            forward.setTeamMsgUnAckCount(0);

            // 接收到消息，但是附件没有下载成功的情况。不会再次上传附件，所以要将附件的发送状态直接改为def
            final MsgAttachment attachment = forward.getAttachment();
            if (attachment != null && attachment instanceof FileAttachment) {
                final FileAttachment media = (FileAttachment) attachment;
                if (!TextUtils.isEmpty(media.getUrl())) {
                    forward.setAttachStatus(AttachStatusEnum.def);
                }
            }
        }

        return forward;
    }

    private static IMMessageImpl initSendMessage(String toId, SessionTypeEnum sessionType) {
        IMMessageImpl msg = new IMMessageImpl();

        msg.setUuid(StringUtil.get32UUID());
        msg.setSessionId(toId);
        msg.setFromAccount(SDKCache.getAccount());
        msg.setDirect(MsgDirectionEnum.Out);
        msg.setStatus(MsgStatusEnum.sending);
        msg.setSessionType(sessionType);
        msg.setTime(TimeUtil.currentTimeMillis());

        return msg;
    }

    public static String createForwardMessageListFileDetail(List<? extends IMMessage> messages) {
        if (messages == null || messages.isEmpty()) {
            return "";
        }
        final String sessionId = messages.get(0).getSessionId();
        if (sessionId==null) {
            return "";
        }
        List<IMMessage> msgList = new ArrayList<>(messages.size());
        for (IMMessage msg : messages) {
            // 不是来自同一会话，不添加
            if (!sessionId.equals(msg.getSessionId())){
                continue;
            }
            // 不是可以转发的类型，不添加
            if (msg.getMsgType() == MsgTypeEnum.notification || msg.getMsgType() == MsgTypeEnum.avchat || msg.getMsgType() == MsgTypeEnum.robot) {
                continue;
            }
            msgList.add(msg);
        }
        return buildHeader(0, msgList.size()) + "\n" + buildBody(msgList);
    }

    private static String buildHeader(int version, int message_count) {
        JSONObject obj = new JSONObject();
        try {
            obj.put("version", version);
            //终端类型：Android
            obj.put("terminal", IAuthService.ClientType.Android);
            obj.put("sdk_version", BuildConfig.VERSION_NAME);
            obj.put("app_version", BuildConfig.VERSION_CODE);
            obj.put("message_count", message_count);
        } catch (JSONException e) {
            e.printStackTrace();
        }
        return obj.toString();
    }

    private static String buildBody(@NonNull List<IMMessage> msgList) {
        String enter = "\n";
        if (msgList.isEmpty()) {
            return "";
        }
        StringBuilder stringBuilder = new StringBuilder();
        for (IMMessage msg : msgList) {
            stringBuilder.append(enter).append(MessageConvert.convertMessageToJson(msg));
        }
        return stringBuilder.substring(enter.length());
    }


    public static IMMessage createFromJson(String json){
        if (TextUtils.isEmpty(json)) {
            return null;
        }
        return MessageConvert.convertJsonToMessage(json);
    }

    public static String convertMessageToJson(IMMessage msg) {
        if (msg == null) {
            return null;
        }
        return MessageConvert.convertMessageToJson(msg);
    }

    @androidx.annotation.NonNull
    public static ImageAttachment createImageAttachment(File file, String displayName, String nosTokenSceneKey) {
        final ImageAttachment attachment = new ImageAttachment();
        attachment.setPath(file.getPath());
        attachment.setSize(file.length());
        int[] dimension = BitmapDecoder.decodeBound(file);
        attachment.setWidth(dimension[0]);
        attachment.setHeight(dimension[1]);
        attachment.setDisplayName(displayName);
        attachment.setExtension(StringUtil.getExtension(file.getName()));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static ImageAttachment createImageAttachment(Context context, Uri uri, String displayName, String nosTokenSceneKey) {
        final ImageAttachment attachment = new ImageAttachment();
        attachment.setUri(uri);
        attachment.setSize(UriUtils.getFileSizeFromUri(context, uri));
        int[] dimension = BitmapDecoder.decodeBound(context, uri);
        attachment.setWidth(dimension[0]);
        attachment.setHeight(dimension[1]);
        attachment.setDisplayName(displayName);
        attachment.setExtension(UriUtils.getFileExtensionFromUri(context, uri));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static AudioAttachment createAudioAttachment(File file, long duration, String nosTokenSceneKey) {
        final AudioAttachment attachment = new AudioAttachment();
        attachment.setPath(file.getPath());
        attachment.setSize(file.length());
        attachment.setDuration(duration);// ms
        attachment.setExtension(StringUtil.getExtension(file.getName()));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static AudioAttachment createAudioAttachment(Context context, Uri uri, long duration, String nosTokenSceneKey) {
        final AudioAttachment attachment = new AudioAttachment();
        attachment.setUri(uri);
        attachment.setSize(UriUtils.getFileSizeFromUri(context, uri));
        attachment.setDuration(duration);// ms
        attachment.setExtension(UriUtils.getFileExtensionFromUri(context, uri));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static VideoAttachment createVideoAttachment(File file, long duration, int width, int height, String displayName, String nosTokenSceneKey) {
        final VideoAttachment attachment = new VideoAttachment();
        attachment.setPath(file.getPath());
        attachment.setSize(file.length());
        attachment.setDuration(duration);
        attachment.setWidth(width);
        attachment.setHeight(height);
        attachment.setDisplayName(displayName);
        attachment.setExtension(StringUtil.getExtension(file.getName()));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static VideoAttachment createVideoAttachment(Context context, Uri uri, long duration, int width, int height, String displayName,
        String nosTokenSceneKey) {
        final VideoAttachment attachment = new VideoAttachment();
        attachment.setUri(uri);
        attachment.setSize(UriUtils.getFileSizeFromUri(context, uri));
        attachment.setDuration(duration);
        attachment.setWidth(width);
        attachment.setHeight(height);
        attachment.setDisplayName(displayName);
        attachment.setExtension(UriUtils.getFileExtensionFromUri(context, uri));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static FileAttachment createFileAttachment(File file, String displayName, String nosTokenSceneKey) {
        final FileAttachment attachment = new FileAttachment();
        attachment.setPath(file.getPath());
        attachment.setSize(file.length());
        attachment.setDisplayName(displayName);
        attachment.setExtension(StringUtil.getExtension(file.getName()));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }

    @androidx.annotation.NonNull
    public static FileAttachment createFileAttachment(Context context, Uri uri, String displayName, String nosTokenSceneKey) {
        final FileAttachment attachment = new FileAttachment();
        attachment.setUri(uri);
        attachment.setSize(UriUtils.getFileSizeFromUri(context, uri));
        attachment.setDisplayName(displayName);
        attachment.setExtension(UriUtils.getFileExtensionFromUri(context, uri));
        attachment.setNosTokenSceneKey(nosTokenSceneKey);
        return attachment;
    }
}