V2NIMMessageAttachmentUtil.java

浏览该文件的文档.

package com.netease.nimlib.sdk.v2.utils;

import android.text.TextUtils;

import com.netease.nimlib.sdk.v2.message.attachment.V2NIMMessageAudioAttachment;
import com.netease.nimlib.sdk.v2.message.attachment.V2NIMMessageFileAttachment;
import com.netease.nimlib.sdk.v2.message.attachment.V2NIMMessageImageAttachment;
import com.netease.nimlib.sdk.v2.message.attachment.V2NIMMessageVideoAttachment;
import com.netease.nimlib.sdk.v2.storage.V2NIMStorageService;
import com.netease.nimlib.sdk.v2.storage.enums.V2NIMDownloadAttachmentType;
import com.netease.nimlib.util.MD5;
import com.netease.nimlib.util.StringUtil;
import com.netease.nimlib.util.storage.NimStorageType;
import com.netease.nimlib.util.storage.NimStorageUtil;

import java.io.File;

/**
 * 消息附件工具类
 */
public class V2NIMMessageAttachmentUtil {

    /**
     * 私有化构造器
     */
    private V2NIMMessageAttachmentUtil(){
    }
    /**
     * 获取文件路径
     * @param fileAttachment 文件附件
     * @return 文件路径，如果文件不存在，返回null
     */
    public static String getFileAttachmentPath(V2NIMMessageFileAttachment fileAttachment){
        if(fileAttachment == null){
            return null;
        }
        String path = getDownloadAttachmentDefaultStoragePath(fileAttachment);
        if(TextUtils.isEmpty(path) || !(new File(path).exists())) {
            path = getFileAttachmentStoragePath(fileAttachment);
        }
        if(TextUtils.isEmpty(path)){
            return null;
        }
        return new File(path).exists() ? path : null;
    }

    /**
     * 获取{@link V2NIMStorageService#downloadAttachment}的默认存储路径
     * @param fileAttachment 文件附件
     * @return 文件存储路径，一般不为null，fileAttachment为null时返回null
     */
    public static String getDownloadAttachmentDefaultStoragePath(V2NIMMessageFileAttachment fileAttachment) {
        if(fileAttachment == null){
            return null;
        }
        String path = fileAttachment.getPath();
        if (!TextUtils.isEmpty(path)) {
            return path;
        } else {
            String ext = extWithoutPointPrefixes(fileAttachment, V2NIMDownloadAttachmentType.V2NIM_DOWNLOAD_ATTACHMENT_TYPE_SOURCE);
            return getDefaultSavePath(fileAttachment, ext,V2NIMDownloadAttachmentType.V2NIM_DOWNLOAD_ATTACHMENT_TYPE_SOURCE);
        }
    }

    /**
     * 获取文件存储路径，和V1逻辑保持一致
     * @param fileAttachment 文件附件
     * @return 文件存储路径，一般不为null，fileAttachment为null时返回null
     */
    public static String getFileAttachmentStoragePath(V2NIMMessageFileAttachment fileAttachment){
        if(fileAttachment == null){
            return null;
        }
        String path = fileAttachment.getPath();
        if (!TextUtils.isEmpty(path)) {
            return path;
        } else {
            String fileName = getFileName(fileAttachment);
            if(TextUtils.isEmpty(fileName)){
                return null;
            }
            return NimStorageUtil.getWritePath(fileName, storageType(fileAttachment));
        }
    }

    /**
     * 获取文件缩略图路径，和V1逻辑保持一致
     * @param fileAttachment 文件附件
     * @return 文件缩略图路径，如果文件不存在，返回null
     */
    public static String getFileAttachmentThumbPath(V2NIMMessageFileAttachment fileAttachment){
        if(fileAttachment == null){
            return null;
        }
        String path = getDownloadAttachmentDefaultThumbStoragePath(fileAttachment);
        if(TextUtils.isEmpty(path) || !(new File(path).exists())) {
            path = getFileAttachmentThumbStoragePath(fileAttachment);
        }
        if(TextUtils.isEmpty(path)){
            return null;
        }
        return new File(path).exists() ? path : null;
    }

    /**
     * 获取{@link V2NIMStorageService#downloadAttachment}的缩略图默认存储路径
     * @param fileAttachment 文件附件
     * @return 文件存储路径，一般不为null，fileAttachment为null时返回null
     */
    public static String getDownloadAttachmentDefaultThumbStoragePath(V2NIMMessageFileAttachment fileAttachment) {
        if(fileAttachment == null){
            return null;
        }
        String ext = extWithoutPointPrefixes(fileAttachment, V2NIMDownloadAttachmentType.V2NIM_DOWNLOAD_ATTACHMENT_TYPE_THUMBNAIL);
        return getDefaultSavePath(fileAttachment, ext,V2NIMDownloadAttachmentType.V2NIM_DOWNLOAD_ATTACHMENT_TYPE_THUMBNAIL);
    }

    /**
     * 获取文件缩略图存储路径,和V1逻辑保持一致
     * @param fileAttachment 文件附件
     * @return 文件缩略图存储路径,一般不为null，fileAttachment为null时返回null
     */
    public static String getFileAttachmentThumbStoragePath(V2NIMMessageFileAttachment fileAttachment){
        if(fileAttachment == null){
            return null;
        }
        String fileName = getFileName(fileAttachment);
        if (!TextUtils.isEmpty(fileName)) {
            int dotIndex = fileName.lastIndexOf(".");
            // 如果找到了 '.' 而且最后一个字符不是 '.'，则需要进行裁剪
            if (dotIndex >= 0 && dotIndex < fileName.length() - 1) {
                fileName = fileName.substring(0, dotIndex);
            }
        }else{
            return null;
        }
        return NimStorageUtil.getWritePath(fileName, NimStorageType.TYPE_THUMB_IMAGE);
    }

    private static String getFileName(V2NIMMessageFileAttachment fileAttachment){
        if(fileAttachment == null){
            return null;
        }
        String md5 = fileAttachment.getMd5();
        if (TextUtils.isEmpty(md5)) {
            md5 = MD5.getStringMD5(fileAttachment.getUrl());
        }
        return md5 != null ? md5 : "";
    }

    private static NimStorageType storageType(V2NIMMessageFileAttachment fileAttachment){
        if(fileAttachment instanceof V2NIMMessageAudioAttachment){
            return NimStorageType.TYPE_AUDIO;
        }else if(fileAttachment instanceof V2NIMMessageImageAttachment){
            return NimStorageType.TYPE_IMAGE;
        }else if(fileAttachment instanceof V2NIMMessageVideoAttachment){
            return NimStorageType.TYPE_VIDEO;
        }else{
            return NimStorageType.TYPE_FILE;
        }
    }

        private static String extWithoutPointPrefixes(V2NIMMessageFileAttachment attachment,V2NIMDownloadAttachmentType downloadType)
        {
            String ext;
            switch (downloadType) {
                // 封面图用png
                case V2NIM_DOWNLOAD_ATTACHMENT_TYPE_THUMBNAIL:
                case V2NIM_DOWNLOAD_ATTACHMENT_TYPE_VIDEO_COVER:
                    ext = "png";
                    break;
                // 原图使用自带后缀
                case V2NIM_DOWNLOAD_ATTACHMENT_TYPE_SOURCE:
                default:
                    ext = attachment == null ? "" : attachment.getExt();
                    break;
            }

            //去掉点号前缀
            if (StringUtil.isEmpty(ext))
            {
                return "";
            }
            if (ext.startsWith("."))
            {
                ext = ext.substring(1);
            }
            return ext;
        }

    private static String getDefaultSavePath(V2NIMMessageFileAttachment fileAttachment, String ext,V2NIMDownloadAttachmentType downloadType) {
        if (fileAttachment == null) {
            return null;
        }
        String md5 = fileAttachment.getMd5();
        if (StringUtil.isEmpty(md5)) {
            return null;
        }
        String fileName = "" + (md5) + "." + (ext);
        if (downloadType != V2NIMDownloadAttachmentType.V2NIM_DOWNLOAD_ATTACHMENT_TYPE_SOURCE) {
            fileName = "thumb_" + (fileName);
        }
        return NimStorageUtil.getWritePath(fileName, storageType(fileAttachment));
    }
}