FileMeta.java

package org.smartly.commons.io;

import org.json.JSONArray;
import org.json.JSONObject;
import org.smartly.commons.util.CollectionUtils;
import org.smartly.commons.util.JsonWrapper;

import java.io.InputStream;
import java.util.HashSet;
import java.util.Set;

/**
 * Simple data model to hold file meta info and content.
 * "Content" is only property not serialized in JSON structure.
 */
public class FileMeta {

    public static final String FILE_NAME = "name";
    public static final String FILE_SIZE = "size";
    public static final String FILE_TYPE = "type";
    public static final String ATTRIBUTES = "attributes";

    private final JSONObject _json;
    private InputStream _content;

    public FileMeta() {
        _json = new JSONObject();
        JsonWrapper.put(_json, ATTRIBUTES, new JSONObject());
    }

    @Override
    public String toString() {
        return _json.toString();
    }

    public JSONObject getJSON() {
        return _json;
    }

    public void put(final String name, final Object value) {
        JsonWrapper.put(_json, name, value);
    }

    public Object get(final String name) {
        return JsonWrapper.get(_json, name);
    }

    public String getName() {
        return JsonWrapper.getString(_json, FILE_NAME);
    }

    public void setName(String fileName) {
        JsonWrapper.put(_json, FILE_NAME, fileName);
    }

    public long getSize() {
        return JsonWrapper.getLong(_json, FILE_SIZE);
    }

    public void setSize(long fileSize) {
        JsonWrapper.put(_json, FILE_SIZE, fileSize);
    }

    public String getType() {
        return JsonWrapper.getString(_json, FILE_TYPE);
    }

    public void setType(String fileType) {
        JsonWrapper.put(_json, FILE_TYPE, fileType);
    }

    public JSONObject getAttributes() {
        return JsonWrapper.getJSON(_json, ATTRIBUTES);
    }

    public Set<String> getAttributeNames() {
        final Set<String> result = new HashSet<String>();
        final JSONObject attributes = this.getAttributes();
        final JSONArray names = attributes.names();
        CollectionUtils.forEach(names, new CollectionUtils.IterationCallback() {
            @Override
            public Object handle(Object item, int index, Object key) {
                if (item instanceof String) {
                    result.add((String) item);
                }
                return null;
            }
        });
        return result;
    }

    public void addAttribute(final String name, final Object value) {
        final JSONObject attributes = this.getAttributes();
        JsonWrapper.put(attributes, name, value);
    }

    public Object getAttribute(final String name) {
        final JSONObject attributes = this.getAttributes();
        return JsonWrapper.get(attributes, name);
    }

    public InputStream getContent() {
        return _content;
    }

    public void setContent(InputStream content) {
        _content = content;
    }


}