org.usergrid.services.users.activities.ActivitiesService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of usergrid-services Show documentation
Service layer for Usergrid system.
There is a newer version: 0.0.27.1
/*******************************************************************************
 * Copyright 2012 Apigee Corporation
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package org.usergrid.services.users.activities;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.usergrid.persistence.Entity;
import org.usergrid.persistence.EntityRef;
import org.usergrid.persistence.Results;
import org.usergrid.persistence.entities.Activity;
import org.usergrid.persistence.entities.Activity.ActivityObject;
import org.usergrid.persistence.entities.User;
import org.usergrid.services.ServiceContext;
import org.usergrid.services.ServicePayload;
import org.usergrid.services.ServiceResults;
import org.usergrid.services.generic.GenericCollectionService;

public class ActivitiesService extends GenericCollectionService {

	private static final Logger logger = LoggerFactory
			.getLogger(ActivitiesService.class);

	public ActivitiesService() {
		super();
		logger.info("/users/*/activities");
	}

	@SuppressWarnings("unchecked")
	@Override
	public ServiceResults postCollection(ServiceContext context)
			throws Exception {

		ServicePayload payload = context.getPayload();

		Entity user = em.get(context.getOwner());

		Object actor = payload.getProperty(Activity.PROPERTY_ACTOR);

		if (actor instanceof Map) {
			handleDynamicPayload((Map) actor, user, payload);
		} else if (actor instanceof ActivityObject) {
			handleDynamicPayload((ActivityObject) actor, user, payload);
		} else if (actor == null) {
			handleDynamicPayload((ActivityObject) actor, user, payload);
		}

		ServiceResults results = super.postCollection(context);

		distribute(context.getOwner(), results.getEntity());
		return results;
	}

	/**
	 * Invoked when our actor is a map
	 * 
	 * @param actor
	 * @param user
	 * @param payload
	 */
	private void handleDynamicPayload(Map actor, Entity user,
			ServicePayload payload) {

		// create a new actor object
		if (actor == null) {
			actor = new HashMap();
			payload.setProperty(Activity.PROPERTY_ACTOR, actor);
		}

		if (user != null) {
			if (actor.get(User.PROPERTY_UUID) == null && user.getUuid() != null) {
				actor.put(User.PROPERTY_UUID, user.getUuid().toString());
			}

			if (actor.get(User.PROPERTY_EMAIL) == null
					&& user.getProperty(User.PROPERTY_EMAIL) != null) {
				actor.put(User.PROPERTY_EMAIL,
						user.getProperty(User.PROPERTY_EMAIL).toString());
			}
		}
	}

	/**
	 * Invoked to set values when our actor is an activity object
	 * 
	 * @param actor
	 * @param user
	 * @param payload
	 */
	private void handleDynamicPayload(ActivityObject actor, Entity user,
			ServicePayload payload) {

		// create a new actor object
		if (actor == null) {
			actor = new ActivityObject();
			payload.setProperty(Activity.PROPERTY_ACTOR, actor);
		}

		if (user != null) {
			if (actor.getId() == null && user.getUuid() != null) {
				actor.setUuid(user.getUuid());
				// TODO TN should this also populate id?
			}

			if (actor.getDynamicProperties().get(User.PROPERTY_EMAIL) == null
					&& user.getProperty(User.PROPERTY_EMAIL) != null) {
				actor.getDynamicProperties().put(User.PROPERTY_EMAIL,
						user.getProperty(User.PROPERTY_EMAIL).toString());
			}
		}

	}

	@Override
	public ServiceResults postItemById(ServiceContext context, UUID id)
			throws Exception {

		ServiceResults results = super.postItemById(context, id);

		distribute(context.getOwner(), results.getEntity());
		return results;
	}

	public void distribute(EntityRef user, Entity activity) throws Exception {
		if (activity == null) {
			return;
		}
		em.addToCollection(user, "feed", activity);
		Results r = em.getConnectingEntities(user.getUuid(), "following",
				User.ENTITY_TYPE, Results.Level.REFS);
		List refs = r.getRefs();
		if (refs != null) {
			em.addToCollections(refs, "feed", activity);
		}
	}

}