MediaModel.java

/*****************************************************************************************
Infosistema - OpenBaas
Copyright(C) 2002-2014 Infosistema, S.A.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
www.infosistema.com
info@openbaas.com
Av. José Gomes Ferreira, 11 3rd floor, s.34
Miraflores
1495-139 Algés Portugal
****************************************************************************************/
package infosistema.openbaas.dataaccess.models;

import infosistema.openbaas.data.enums.ModelEnum;
import infosistema.openbaas.data.enums.OperatorEnum;
import infosistema.openbaas.utils.Log;

import java.util.List;
import java.util.Map;

import org.codehaus.jettison.json.JSONException;
import org.codehaus.jettison.json.JSONObject;



import com.mongodb.BasicDBObject;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;

public class MediaModel extends ModelAbstract {

	// *** CONTRUCTORS *** //

	public MediaModel() {
	}


	// *** PRIVATE *** //

	private static BasicDBObject dataProjection = null; 	
	private static BasicDBObject dataProjectionMetadata = null; 	

	
	// *** PROTECTED *** //
	
	@Override
	protected DBCollection getCollection(String appId) {
		return super.getCollection(String.format(APP_MEDIA_COLL_FORMAT, appId));
	}
	
	@Override
	protected BasicDBObject getDataProjection(boolean getMetadata, List<String> toShow, List<String> toHide) {
		if (getMetadata) {
			if (dataProjectionMetadata == null) {
				dataProjectionMetadata = super.getDataProjection(new BasicDBObject(), true);
				dataProjectionMetadata.append(_TYPE, 0);
			}
			return dataProjectionMetadata;
		} else {
			if (dataProjection == null) {
				dataProjection = super.getDataProjection(new BasicDBObject(), false);
				dataProjection.append(_TYPE, 0);
			}
			return dataProjection;
		}
	}


	// *** CONSTANTS *** //

	protected static final String _TYPE = "_type";
	
	
	// *** KEYS *** //

	public static final String APP_MEDIA_COLL_FORMAT = "app%sdata";
	private static final String TYPE_QUERY_FORMAT = "{" + _TYPE + ": \"%s\"";

	private String getTypeQuery(ModelEnum type) {
		if (type == null) return "";
		return String.format(TYPE_QUERY_FORMAT, type.toString());
	}


	// *** CREATE *** //
	
	public JSONObject createMedia(String appId, String userId, ModelEnum type, String objId, Map<String, String> mediaFields, Map<String, String> extraMetadata) {
		try{
			if (type == null) {
				Log.error("", this, "createMedia", "Media as no type.");
				return null;
			}
			JSONObject data = getJSonObject(mediaFields);
			data.put(_ID, objId);
			data.put(_TYPE, type.toString());
			JSONObject metadata = getMetadaJSONObject(getMetadataCreate(userId, extraMetadata));
			JSONObject geolocation = getGeolocation(metadata);
			Map<?, ?> metaMap = convertJsonToMap(metadata);
			Map<?, ?>  metaGeo = convertJsonToMap(geolocation);
			if(metadata!=null){
				data.put(_METADATA, metaMap);
				//jedis.hset(userKey, _METADATA, metadata.toString());
			}
			if(geolocation!=null){
				data.put(_GEO, metaGeo);
				//jedis.hset(userKey, _METADATA, metadata.toString());
			}
			return super.insert(appId, data, metadata, geolocation);
		} catch (Exception e) {
			Log.error("", this, "createMedia", "An error ocorred.", e);
		}
		return null;
	}
	
	// *** UPDATE *** //

	// *** GET LIST *** //
	
	public List<DBObject> getMedia(String appId, ModelEnum type, Double latitude, Double longitude, Double radius, JSONObject query, String orderType, String orderBy, List<String> toShow) throws Exception {
		JSONObject finalQuery = new JSONObject();
		if (type != null) {
			finalQuery.append(OperatorEnum.oper.toString(), OperatorEnum.and.toString());
			finalQuery.append(OperatorEnum.op1.toString(), getTypeQuery(type));
			finalQuery.append(OperatorEnum.op2.toString(), query.toString());
		} else {
			finalQuery = query;
		}
		return super.getDocuments(appId, null, null, latitude, longitude, radius, finalQuery, orderType, orderBy, toShow);
	}

	
	// *** GET *** //

	public JSONObject getMedia(String appId, ModelEnum type, String objId, boolean getMetadata) {
		//CACHE
		try {
			return super.getDocument(appId, objId, getMetadata, null, null);
		} catch (JSONException e) {
			Log.error("", this, "getMedia", "Error getting Media.", e);
		}
		return null;
	}

	/**
	 * Returns the directory of the specified 'id' file.
	 * 
	 * @param appId
	 * @param id
	 * @param folderType
	 * @param requestType
	 * @return
	 */
	public String getMediaField(String appId, ModelEnum type, String objId, String field) {
		//CACHE
		try {
			return getMedia(appId, type, objId, false).get(field).toString();
		} catch (Exception e) {
			return null;
		}
	}


	// *** DELETE *** //
	
	public Boolean deleteMedia(String appId, ModelEnum type, String objId) {
		//CACHE
		return super.deleteDocument(appId, objId);		
	}

	
	// *** EXISTS *** //
	
	public Boolean mediaExists(String appId, ModelEnum type, String objId) {
		//CACHE
		return super.existsNode(appId, objId);
	}

	

	// *** OTHERS *** //

}