FileAttachment.java

浏览该文件的文档.

package com.netease.nimlib.sdk.msg.attachment;

import android.net.Uri;
import android.text.TextUtils;
import com.netease.nimlib.NimNosSceneKeyConstant;
import com.netease.nimlib.SDKCache;
import com.netease.nimlib.sdk.util.UriUtils;
import com.netease.nimlib.util.JSONHelper;
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;
import org.json.JSONObject;

public class FileAttachment implements MsgAttachment {

    protected String path;

    protected long size;

    protected String md5;

    protected String url;

    protected String displayName;

    protected String extension;

    private long expire;

    protected String nosTokenSceneKey = NimNosSceneKeyConstant.NIM_DEFAULT_IM;

    protected boolean forceUpload = false;
    private   boolean     isUri         = false;

    public FileAttachment() {

    }

    public FileAttachment(String attach) {
        fromJson(attach);
    }

    public String getPath() {
        String path = getPathForSave();
        return new File(path).exists() ? path : null;
    }

    public String getPathForSave() {
        if (!isUri && !TextUtils.isEmpty(path)) {
            return path;
        } else {
            return NimStorageUtil.getWritePath(getFileName(), storageType());
        }
    }

    public String getThumbPath() {
        String path = getThumbPathForSave();
        return new File(path).exists() ? path : null;
    }

    public String getThumbPathForSave() {
        String fileName = getFileName();
        if (!TextUtils.isEmpty(fileName)) {
            int dotIndex = fileName.lastIndexOf(".");
            // 如果找到了 '.' 而且最后一个字符不是 '.'，则需要进行裁剪
            if (dotIndex >= 0 && dotIndex < fileName.length() - 1) {
                fileName = fileName.substring(0, dotIndex);
            }
        }
        return NimStorageUtil.getWritePath(fileName, NimStorageType.TYPE_THUMB_IMAGE);
    }

    public void setPath(String path) {
        this.path = path;
        this.isUri = UriUtils.isFileOrContentUri(path);
    }

    public boolean setUri(Uri uri) {
        if(uri == null)
        {
            return false;
        }

        if(!UriUtils.isFileOrContentUri(uri))
        {
            return false;
        }
        isUri = true;
        this.path = uri.toString();
        return true;
    }

    public Uri getUri() {
        if(!isUri)
        {
            return null;
        }
        return UriUtils.string2Uri(path);
    }



    public long getSize() {
        return size;
    }

    public void setSize(long size) {
        this.size = size;
    }

    public String getMd5() {
        return md5;
    }

    public void setMd5(String md5) {
        this.md5 = md5;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }

    public String getExtension() {
        return extension;
    }

    public void setExtension(String extension) {
        this.extension = extension;
    }

    public String getFileName() {
        if(isUri)
        {
            return UriUtils.getFileNameFromUri(SDKCache.getContext(),UriUtils.string2Uri(path));
        }
        else if (!TextUtils.isEmpty(path)) {
            return StringUtil.nameOfPath(path);
        } else {
            if (TextUtils.isEmpty(md5)) {
                return MD5.getStringMD5(url);
            } else {
                return md5;
            }
        }
    }

    public String getDisplayName() {
        return displayName;
    }

    public void setDisplayName(String displayName) {
        this.displayName = displayName;
    }


    public String getNosTokenSceneKey() {
        return nosTokenSceneKey;
    }

    public void setNosTokenSceneKey(String nosTokenSceneKey) {
        if (TextUtils.isEmpty(nosTokenSceneKey)) {
            return;
        }
        this.nosTokenSceneKey = nosTokenSceneKey;
    }

    public long getExpire() {
        return expire;
    }

    public boolean isForceUpload() {
        return forceUpload;
    }

    public void setForceUpload(boolean forceUpload) {
        this.forceUpload = forceUpload;
    }

    protected NimStorageType storageType() {
        return NimStorageType.TYPE_FILE;
    }

    protected void save(JSONObject json) {

    }

    protected void load(JSONObject json) {

    }

    private static final String KEY_PATH = "path";
    private static final String KEY_NAME = "name";
    private static final String KEY_SIZE = "size";
    private static final String KEY_MD5 = "md5";
    private static final String KEY_URL = "url";
    private static final String KEY_EXT = "ext";
    private static final String KEY_SCENE = "sen";
    private static final String KEY_FORCE_UPLOAD = "force_upload";
    private static final String KEY_EXPIRE = "expire";


    @Override
    public String toJson(boolean send) {
        JSONObject object = new JSONObject();
        try {
            if (!send && !TextUtils.isEmpty(path)) {
                object.put(KEY_PATH, path);
            }

            if (!TextUtils.isEmpty(md5)) {
                object.put(KEY_MD5, md5);
            }

            if (!TextUtils.isEmpty(displayName)) {
                object.put(KEY_NAME, displayName);
            }

            object.put(KEY_URL, url);
            object.put(KEY_SIZE, size);


            if (!TextUtils.isEmpty(extension)) {
                object.put(KEY_EXT, extension);
            }
            if (!TextUtils.isEmpty(nosTokenSceneKey)) {
                object.put(KEY_SCENE, nosTokenSceneKey);
            }

            if (expire > 0) {
                object.put(KEY_EXPIRE, expire);
            }

            object.put(KEY_FORCE_UPLOAD, forceUpload);
            save(object);

        } catch (Exception e) {
            e.printStackTrace();
        }

        return object.toString();
    }

    private void fromJson(String attach) {
        JSONObject json = JSONHelper.parse(attach);
        if(json == null)
        {
            return;
        }
        path = JSONHelper.getString(json, KEY_PATH);
        Uri uri = UriUtils.string2Uri(path);
        if(UriUtils.isFileOrContentUri(uri))
        {
            isUri = true;
        }
        md5 = JSONHelper.getString(json, KEY_MD5);
        url = JSONHelper.getString(json, KEY_URL);
        displayName = JSONHelper.getString(json, KEY_NAME);
        size = JSONHelper.getLong(json, KEY_SIZE);
        extension = JSONHelper.getString(json, KEY_EXT);
        setNosTokenSceneKey(JSONHelper.getString(json, KEY_SCENE));
        forceUpload = JSONHelper.getBoolean(json, KEY_FORCE_UPLOAD);
        expire = JSONHelper.getLong(json, KEY_EXPIRE);
        load(json);
    }
}