source: AnnotationManagerN4J/src/main/java/de/mpiwg/itgroup/annotations/neo4j/AnnotationStore.java @ 43:d1bef7952bec

/**
 * 
 */
package de.mpiwg.itgroup.annotations.neo4j;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Calendar;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.log4j.Logger;
import org.neo4j.graphdb.Direction;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Relationship;
import org.neo4j.graphdb.RelationshipType;
import org.neo4j.graphdb.Transaction;
import org.neo4j.graphdb.index.Index;
import org.neo4j.graphdb.index.IndexHits;

import de.mpiwg.itgroup.annotations.Actor;
import de.mpiwg.itgroup.annotations.Annotation;
import de.mpiwg.itgroup.annotations.Annotation.FragmentTypes;
import de.mpiwg.itgroup.annotations.Group;
import de.mpiwg.itgroup.annotations.Person;
import de.mpiwg.itgroup.annotations.Resource;
import de.mpiwg.itgroup.annotations.Tag;
import de.mpiwg.itgroup.annotations.Target;
import de.mpiwg.itgroup.annotations.Uri;

/**
 * Neo4J based Annotation store.
 * 
 * @author casties
 * 
 */
public class AnnotationStore {

    protected static Logger logger = Logger.getLogger(AnnotationStore.class);

    protected GraphDatabaseService graphDb;

    public static enum NodeTypes {
        ANNOTATION, PERSON, TARGET, GROUP, TAG, RESOURCE
    }

    // types of nodes that should not be automatically deleted.
    public Set<String> permanentNodeTypes = new HashSet<String>(Arrays.asList("PERSON", "GROUP", "TAG"));

    protected List<Index<Node>> nodeIndexes;

    public static enum RelationTypes implements RelationshipType {
        ANNOTATES, CREATED, PERMITS_ADMIN, PERMITS_DELETE, PERMITS_UPDATE, PERMITS_READ, MEMBER_OF, HAS_TAG, PART_OF
    }

    public static String ANNOTATION_URI_BASE = "http://entities.mpiwg-berlin.mpg.de/annotations/";

    public AnnotationStore(GraphDatabaseService graphDb) {
        super();
        this.graphDb = graphDb;
        nodeIndexes = new ArrayList<Index<Node>>(5);
        // List.set(enum.ordinal(), val) seems not to work.
        nodeIndexes.add(NodeTypes.ANNOTATION.ordinal(), graphDb.index().forNodes("annotations"));
        nodeIndexes.add(NodeTypes.PERSON.ordinal(), graphDb.index().forNodes("persons"));
        nodeIndexes.add(NodeTypes.TARGET.ordinal(), graphDb.index().forNodes("targets"));
        nodeIndexes.add(NodeTypes.GROUP.ordinal(), graphDb.index().forNodes("groups"));
        nodeIndexes.add(NodeTypes.TAG.ordinal(), graphDb.index().forNodes("tags"));
        nodeIndexes.add(NodeTypes.RESOURCE.ordinal(), graphDb.index().forNodes("resources"));
    }

    protected Index<Node> getNodeIndex(NodeTypes type) {
        return nodeIndexes.get(type.ordinal());
    }

    /**
     * @param userUri
     * @return
     */
    public Node getPersonNodeByUri(String userUri) {
        if (userUri == null) return null;
        Node person = getNodeIndex(NodeTypes.PERSON).get("uri", userUri).getSingle();
        return person;
    }

    /**
     * @param tagUri
     * @return
     */
    public Node getTagNodeByUri(String tagUri) {
        if (tagUri == null) return null;
        Node person = getNodeIndex(NodeTypes.TAG).get("uri", tagUri).getSingle();
        return person;
    }

    /**
     * @param resourceUri
     * @return
     */
    public Node getResourceNodeByUri(String resourceUri) {
        return getNodeFromIndex("uri", resourceUri, NodeTypes.RESOURCE);
    }

    /**
     * @param targetUri
     * @return
     */
    public Node getTargetNodeByUri(String targetUri) {
        return getNodeFromIndex("uri", targetUri, NodeTypes.RESOURCE);
    }

    /**
     * Returns the Node with the given key and value.
     * Key has to be indexed.
     * 
     * @param key
     * @param value
     * @param type
     * @return
     */
    public Node getNodeFromIndex(String key, String value, NodeTypes type) {
        if (key == null || value == null) return null;
        Node person = getNodeIndex(type).get(key, value).getSingle();
        return person;
    }

    /**
     * Returns list of Actors of given type (Group or Person).
     * Key has to be indexed.
     * 
     * @param key
     * @param query
     * @param type
     * @return
     */
    @SuppressWarnings("unchecked")
    public <T extends Actor> List<T> getActors(String key, String query, NodeTypes type) {
        ArrayList<T> actors = new ArrayList<T>();
        Index<Node> idx = getNodeIndex(type);
        if (key == null) {
            key = "uri";
            query = "*";
        }
        IndexHits<Node> actorNodes = idx.query(key, query);
        for (Node actorNode : actorNodes) {
            Actor actor = createActorFromNode(actorNode);
            actors.add((T) actor);
        }
        return actors;
    }
    
    /**
     * Returns list of groups. Key has to be indexed.
     * @param key
     * @param query
     * @return
     */
    public List<Group> getGroups(String key, String query) {
        List<Group> groups = getActors(key, query, NodeTypes.GROUP);
        return groups;
    }
    
    /**
     * Returns list of Persons. Key has to be indexed.
     * @param key
     * @param query
     * @return
     */
    public List<Person> getPersons(String key, String query) {
        List<Person> persons = getActors(key, query, NodeTypes.PERSON);
        return persons;
    }
    
    /**
     * Returns list of uri-like objects of given type (Target or Resource).
     * Key has to be indexed.
     * 
     * @param key
     * @param query
     * @param type
     * @return
     */
    @SuppressWarnings("unchecked")
    public <T extends Uri> List<T> getUris(String key, String query, NodeTypes type) {
        ArrayList<T> uris = new ArrayList<T>();
        Index<Node> idx = getNodeIndex(type);
        if (key == null) {
            key = "uri";
            query = "*";
        }
        IndexHits<Node> actorNodes = idx.query(key, query);
        for (Node actorNode : actorNodes) {
            Uri uri = createUriFromNode(actorNode);
            uris.add((T) uri);
        }
        return uris;
    }

    
    /**
     * Returns list of Targets. Key has to be indexed.
     * @param key
     * @param query
     * @return
     */
    public List<Target> getTargets(String key, String query) {
        List<Target> targets = getUris(key, query, NodeTypes.TARGET);
        return targets;
    }
    
    /**
     * Returns list of Resources. Key has to be indexed.
     * @param key
     * @param query
     * @return
     */
    public List<Resource> getResources(String key, String query) {
        List<Resource> targets = getUris(key, query, NodeTypes.RESOURCE);
        return targets;
    }
    
    /**
     * Returns List of Annotations.
     * Key has to be indexed.
     * 
     * @param key
     * @param query
     * @return
     */
    public List<Annotation> getAnnotations(String key, String query) {
        ArrayList<Annotation> annotations = new ArrayList<Annotation>();
        Index<Node> idx = getNodeIndex(NodeTypes.ANNOTATION);
        if (key == null) {
            key = "id";
            query = "*";
        }
        IndexHits<Node> annotNodes = idx.query(key, query);
        for (Node annotNode : annotNodes) {
            Annotation annotation = createAnnotationFromNode(annotNode);
            annotations.add(annotation);
        }
        return annotations;
    }

 
    
    /**
     * Returns List of Tags.
     * Key has to be indexed.
     * 
     * @param key
     * @param query
     * @return
     */
    public List<Tag> getTags(String key, String query) {
        ArrayList<Tag> tags = new ArrayList<Tag>();
        Index<Node> idx = getNodeIndex(NodeTypes.TAG);
        if (key == null) {
            key = "uri";
            query = "*";
        }
        IndexHits<Node> groupNodes = idx.query(key, query);
        for (Node groupNode : groupNodes) {
            Tag tag = createTagFromNode(groupNode);
            tags.add(tag);
        }
        return tags;
    }

    /**
     * Returns List of Groups the person is member of.
     * 
     * @param person
     * @return
     */
    public List<Group> getGroupsForPersonNode(Node person) {
        ArrayList<Group> groups = new ArrayList<Group>();
        Iterable<Relationship> rels = person.getRelationships(RelationTypes.MEMBER_OF);
        for (Relationship rel : rels) {
            Node groupNode = rel.getEndNode();
            Actor group = createActorFromNode(groupNode);
            // make sure we're getting a group
            if (!(group instanceof Group)) {
                logger.error("target of MEMBER_OF is not GROUP! rel=" + rel);
                continue;
            }
            groups.add((Group) group);
        }
        return groups;
    }

    /**
     * Returns if person with uri is in Group group.
     * 
     * @param person
     * @param group
     * @return
     */
    public boolean isPersonInGroup(Person person, Group group) {
        Node pn = getPersonNodeByUri(person.getUriString());
        if (pn == null) return false;
        // optimized version of getGroupsForPersonNode
        Iterable<Relationship> rels = pn.getRelationships(RelationTypes.MEMBER_OF);
        for (Relationship rel : rels) {
            Node gn = rel.getEndNode();
            if (gn.getProperty("uri", "").equals(group.getUriString()) || gn.getProperty("id", "").equals(group.getId())) {
                return true;
            }
        }
        return false;
    }

    /**
     * Returns the members of the group.
     * 
     * @param group
     * @return
     */
    public List<Person> getMembersOfGroup(Group group) {
        ArrayList<Person> members = new ArrayList<Person>();
        Node gn = getActorNode(group);
        Iterable<Relationship> rels = gn.getRelationships(RelationTypes.MEMBER_OF);
        for (Relationship rel : rels) {
            Node memberNode = rel.getStartNode();
            Actor member = createActorFromNode(memberNode);
            // make sure we're getting a group
            if (!(member instanceof Person)) {
                logger.error("source of MEMBER_OF is not PERSON! rel=" + rel);
                continue;
            }
            members.add((Person) member);
        }
        return members;
    }

    /**
     * Add Person newMember to Group group.
     * 
     * @param group
     * @param member
     */
    public Person addGroupMember(Group group, Person member) {
        Node gn = getActorNode(group);
        Node pn = getActorNode(member);
        Person addedMember = null;
        if (gn != null && pn != null) {
            getOrCreateRelation(pn, RelationTypes.MEMBER_OF, gn);
            addedMember = member;
        }
        return addedMember;
    }

    /**
     * Delete Person oldMember from Group group.
     * 
     * @param group
     * @param member
     */
    public void deleteGroupMember(Group group, Person member) {
        Node gn = getActorNode(group);
        Node pn = getActorNode(member);
        Iterable<Relationship> rels = gn.getRelationships(RelationTypes.MEMBER_OF);
        for (Relationship rel : rels) {
            Node mn = rel.getStartNode();
            if (mn.equals(pn)) {
                Transaction tx = graphDb.beginTx();
                try {
                    rel.delete();
                    tx.success();
                } finally {
                    tx.finish();
                }
                // there should be only one
                break;
            }
        }
    }

    /**
     * Returns the stored Actor matching the given one.
     * 
     * @param actor
     * @return
     */
    public Actor getActor(Actor actor) {
        Node actorNode = getActorNode(actor);
        Actor storedActor = createActorFromNode(actorNode);
        return storedActor;
    }

    /**
     * Stores an Actor (Person or Group). Creates a new actor Node or update an
     * existing one.
     * 
     * @param actor
     * @return
     */
    public Actor storeActor(Actor actor) {
        Node actorNode = getOrCreateActorNode(actor);
        Transaction tx = graphDb.beginTx();
        try {
            // id
            String id = actor.getId();
            if (id != null) {
                actorNode.setProperty("id", id);
            }
            // name
            String name = actor.getName();
            if (name != null) {
                actorNode.setProperty("name", name);
            }
            // uri
            String uri = actor.getUri();
            if (uri != null) {
                actorNode.setProperty("uri", uri);
            }
            tx.success();
        } finally {
            tx.finish();
        }
        Actor storedActor = createActorFromNode(actorNode);
        return storedActor;
    }

    /**
     * Deletes the given Actor.
     * 
     * @param actor
     */
    public void deleteActor(Actor actor) {
        String uri = actor.getUriString();
        Index<Node> idx;
        if (actor.isGroup()) {
            idx = getNodeIndex(NodeTypes.GROUP);
        } else {
            idx = getNodeIndex(NodeTypes.PERSON);
        }
        Node actorNode = idx.get("uri", uri).getSingle();
        if (actorNode != null) {
            // delete relations
            Transaction tx = graphDb.beginTx();
            try {
                for (Relationship rel : actorNode.getRelationships()) {
                    rel.delete();
                }
                if (!actorNode.hasRelationship()) {
                    // this shouldn't happen
                    deleteNode(actorNode);
                } else {
                    logger.error("deleteActor: unable to delete: Node still has relations.");
                }
                tx.success();
            } finally {
                tx.finish();
            }
        }
    }

    /**
     * Returns the Annotation with the given id.
     * 
     * @param id
     * @return
     */
    public Annotation getAnnotationById(String id) {
        Node annotNode = getNodeIndex(NodeTypes.ANNOTATION).get("id", id).getSingle();
        Annotation annot = createAnnotationFromNode(annotNode);
        return annot;
    }

    /**
     * Returns an Annotation object from an annotation-Node.
     * 
     * @param annotNode
     * @return
     */
    public Annotation createAnnotationFromNode(Node annotNode) {
        Annotation annot = new Annotation();
        annot.setUri((String) annotNode.getProperty("id", null));
        annot.setBodyText((String) annotNode.getProperty("bodyText", null));
        annot.setBodyUri((String) annotNode.getProperty("bodyUri", null));
        /*
         * get annotation target from relation
         */
        Relationship targetRel = getRelation(annotNode, RelationTypes.ANNOTATES, null);
        if (targetRel != null) {
            Node target = targetRel.getEndNode();
            annot.setTargetBaseUri((String) target.getProperty("uri", null));
        } else {
            logger.error("annotation " + annotNode + " has no target node!");
        }
        annot.setTargetFragment((String) annotNode.getProperty("targetFragment", null));
        String ft = (String) annotNode.getProperty("fragmentType", null);
        if (ft != null) {
            annot.setFragmentType(FragmentTypes.valueOf(ft));
        }
        /*
         * get creator from relation
         */
        Relationship creatorRel = getRelation(annotNode, RelationTypes.CREATED, null);
        if (creatorRel != null) {
            Node creatorNode = creatorRel.getStartNode();
            Actor creator = createActorFromNode(creatorNode);
            annot.setCreator(creator);
        } else {
            logger.error("annotation " + annotNode + " has no creator node!");
        }
        /*
         * get creation date
         */
        annot.setCreated((String) annotNode.getProperty("created", null));
        /*
         * get permissions
         */
        Relationship adminRel = getRelation(annotNode, RelationTypes.PERMITS_ADMIN, null);
        if (adminRel != null) {
            Node adminNode = adminRel.getEndNode();
            Actor admin = createActorFromNode(adminNode);
            annot.setAdminPermission(admin);
        }
        Relationship deleteRel = getRelation(annotNode, RelationTypes.PERMITS_DELETE, null);
        if (deleteRel != null) {
            Node deleteNode = deleteRel.getEndNode();
            Actor delete = createActorFromNode(deleteNode);
            annot.setDeletePermission(delete);
        }
        Relationship updateRel = getRelation(annotNode, RelationTypes.PERMITS_UPDATE, null);
        if (updateRel != null) {
            Node updateNode = updateRel.getEndNode();
            Actor update = createActorFromNode(updateNode);
            annot.setUpdatePermission(update);
        }
        Relationship readRel = getRelation(annotNode, RelationTypes.PERMITS_READ, null);
        if (readRel != null) {
            Node readNode = readRel.getEndNode();
            Actor read = createActorFromNode(readNode);
            annot.setReadPermission(read);
        }
        /*
         * get tags
         */
        Set<String> tags = new HashSet<String>();
        for (Relationship rel : annotNode.getRelationships(RelationTypes.HAS_TAG)) {
            String tag = (String) rel.getEndNode().getProperty("name", null);
            if (tag != null) {
                tags.add(tag);
            }
        }
        annot.setTags(tags);

        return annot;
    }

    /**
     * Returns an Actor object from a node.
     * 
     * @param actorNode
     * @return
     */
    protected Actor createActorFromNode(Node actorNode) {
        if (actorNode == null) return null;
        String id = (String) actorNode.getProperty("id", null);
        String uri = (String) actorNode.getProperty("uri", null);
        String name = (String) actorNode.getProperty("name", null);
        String type = (String) actorNode.getProperty("TYPE", null);
        if (type != null && type.equals("PERSON")) {
            return new Person(id, uri, name);
        } else if (type != null && type.equals("GROUP")) {
            return new Group(id, uri, name);
        }
        return null;
    }

    public Tag createTagFromNode(Node tagNode) {
        if (tagNode == null) return null;
        String name = (String) tagNode.getProperty("name", null);
        String uri = (String) tagNode.getProperty("uri", null);
        String id = (String) tagNode.getProperty("id", null);

        return new Tag(id, uri, name);

    }

    /**
     * @param resourceNode
     * @return
     */
    public Resource createResourceFromNode(Node resourceNode) {
        return (Resource) createUriFromNode(resourceNode);
    }
    
    /**
     * @param targetNode
     * @return
     */
    public Target createTargetFromNode(Node targetNode) {
        return (Target) createUriFromNode(targetNode);
    }
    
    
    protected Uri createUriFromNode(Node uriNode) {
        if (uriNode == null) return null;
        String uri = (String) uriNode.getProperty("uri", null);
        String type = (String) uriNode.getProperty("TYPE", null);
        if (type != null && type.equals("TARGET")) {
            return new Target(uri);
        } else if (type != null && type.equals("RESOURCE")) {
            return new Resource(uri);
        }
        return null;
    }

    /**
     * Store a new annotation in the store or update an existing one. Returns
     * the stored annotation.
     * 
     * @param annot
     * @return
     */
    public Annotation storeAnnotation(Annotation annot) {
        Node annotNode = null;
        Transaction tx = graphDb.beginTx();
        try {
            /*
             * create or get the annotation
             */
            String id = annot.getUri();
            if (id == null) {
                id = createRessourceURI("annot:");
            }
            annotNode = getOrCreateAnnotationNode(id);

            /*
             * the annotation body
             */
            String bodyText = annot.getBodyText();
            if (bodyText != null) {
                annotNode.setProperty("bodyText", bodyText);
            }
            String bodyUri = annot.getBodyUri();
            if (bodyUri != null) {
                annotNode.setProperty("bodyUri", bodyUri);
            }

            /*
             * the annotation target
             */
            String targetBaseUri = annot.getTargetBaseUri();
            Node targetNode = null;
            if (targetBaseUri != null) {
                targetNode = getOrCreateTargetNode(targetBaseUri, NodeTypes.TARGET);
                getOrCreateRelation(annotNode, RelationTypes.ANNOTATES, targetNode);
            }

            /*
             * The fragment part of the annotation target.
             */
            String targetFragment = annot.getTargetFragment();
            FragmentTypes fragmentType = annot.getFragmentType();
            if (targetFragment != null) {
                annotNode.setProperty("targetFragment", targetFragment);
                annotNode.setProperty("fragmentType", fragmentType.name());
            }

            /*
             * the annotation resource
             */
            String resourceUri = annot.getResourceUri();
            if (resourceUri != null) {
                Node resource = getOrCreateTargetNode(resourceUri, NodeTypes.RESOURCE);
                getOrCreateRelation(annotNode, RelationTypes.ANNOTATES, resource);
                getOrCreateRelation(targetNode, RelationTypes.PART_OF, resource);
            }

            /*
             * The creator of this annotation.
             */
            Actor creator = annot.getCreator();
            if (creator != null) {
                Node creatorNode = getOrCreateActorNode(creator);
                getOrCreateRelation(creatorNode, RelationTypes.CREATED, annotNode);
            }

            /*
             * The creation date of this annotation.
             */
            String created = annot.getCreated();
            if (created != null) {
                annotNode.setProperty("created", created);
            }

            /*
             * Permissions for this annotation.
             */
            setPermissionRelation(annotNode, RelationTypes.PERMITS_ADMIN, annot.getAdminPermission());
            setPermissionRelation(annotNode, RelationTypes.PERMITS_DELETE, annot.getDeletePermission());
            setPermissionRelation(annotNode, RelationTypes.PERMITS_UPDATE, annot.getUpdatePermission());
            setPermissionRelation(annotNode, RelationTypes.PERMITS_READ, annot.getReadPermission());

            /*
             * Tags on this annotation.
             */
            Set<String> newTags = annot.getTags();
            // we ignore existing tags if tags == null
            if (newTags != null) {
                List<Relationship> oldHasTags = new ArrayList<Relationship>();
                for (Relationship rel : annotNode.getRelationships(RelationTypes.HAS_TAG)) {
                    oldHasTags.add(rel);
                }
                // adjust to new tags
                if (newTags.isEmpty()) {
                    // remove old tags
                    if (!oldHasTags.isEmpty()) {
                        for (Relationship rel : oldHasTags) {
                            rel.delete();
                            // TODO: should we delete orphan nodes too?
                        }
                    }
                } else {
                    if (!oldHasTags.isEmpty()) {
                        // adjust old tags
                        for (Relationship rel : oldHasTags) {
                            String oldTag = (String) rel.getEndNode().getProperty("name", null);
                            if (newTags.contains(oldTag)) {
                                // tag exists
                                newTags.remove(oldTag);
                            } else {
                                // tag exists no longer
                                rel.delete();
                                // TODO: should we delete orphan nodes too?
                            }
                        }
                    }
                    if (!newTags.isEmpty()) {
                        // still tags to add
                        for (String tag : newTags) {
                            // create new tag
                            Node tagNode = getOrCreateTagNode(new Tag(null, null, tag));
                            getOrCreateRelation(annotNode, RelationTypes.HAS_TAG, tagNode);
                        }
                    }

                }
            }
            tx.success();
        } finally {
            tx.finish();
        }

        // re-read and return annotation
        Annotation storedAnnot = createAnnotationFromNode(annotNode);
        return storedAnnot;
    }

    /**
     * Deletes the annotation with the given id.
     * 
     * @param id
     */
    public void deleteAnnotationById(String id) {
        Node annotNode = getNodeIndex(NodeTypes.ANNOTATION).get("id", id).getSingle();
        if (annotNode != null) {
            // delete related objects
            Transaction tx = graphDb.beginTx();
            try {
                for (Relationship rel : annotNode.getRelationships()) {
                    // delete relation and the related node if it has no other
                    // relations and is not permanent
                    Node other = rel.getOtherNode(annotNode);
                    rel.delete();
                    if (!(other.hasRelationship() || permanentNodeTypes.contains(other.getProperty("TYPE", null)))) {
                        deleteNode(other);
                    }
                }
                if (!annotNode.hasRelationship()) {
                    deleteNode(annotNode);
                } else {
                    logger.error("deleteById: unable to delete: Node still has relations.");
                }
                tx.success();
            } finally {
                tx.finish();
            }
        }
    }

    /**
     * Returns all annotations with the given uri and/or user.
     * 
     * @param uri
     * @param userUri
     * @param limit
     * @param offset
     * @return
     */
    public List<Annotation> searchAnnotationByUriUser(String targetUri, String userUri, String limit, String offset) {
        List<Annotation> annotations = new ArrayList<Annotation>();
        if (targetUri != null) {
            // there should be only one
            Node target = getNodeIndex(NodeTypes.TARGET).get("uri", targetUri).getSingle();
            if (target != null) {
                Iterable<Relationship> relations = target.getRelationships(RelationTypes.ANNOTATES);
                for (Relationship relation : relations) {
                    Node ann = relation.getStartNode();
                    if (ann.getProperty("TYPE", "").equals("ANNOTATION")) {
                        Annotation annot = createAnnotationFromNode(ann);
                        annotations.add(annot);
                    } else {
                        logger.error("ANNOTATES relation does not start with ANNOTATION: " + ann);
                    }
                }
            }
        }
        if (userUri != null) {
            // there should be only one
            Node person = getPersonNodeByUri(userUri);
            if (person != null) {
                Iterable<Relationship> relations = person.getRelationships(RelationTypes.CREATED);
                for (Relationship relation : relations) {
                    Node ann = relation.getEndNode();
                    if (ann.getProperty("TYPE", "").equals("ANNOTATION")) {
                        Annotation annot = createAnnotationFromNode(ann);
                        annotations.add(annot);
                    } else {
                        logger.error("CREATED relation does not end with ANNOTATION: " + ann);
                    }
                }
            }
        }
        // TODO: if both uri and user are given we should intersect
        return annotations;
    }

    /**
     * Returns Relationship of type from Node start to Node end. Creates one if
     * it doesn't exist.
     * 
     * @param start
     * @param type
     * @param end
     * @return
     */
    protected Relationship getOrCreateRelation(Node start, RelationshipType type, Node end) {
        if (start == null || end == null) return null;
        if (start.hasRelationship()) {
            // there are relations
            Iterable<Relationship> rels = start.getRelationships(type, Direction.OUTGOING);
            for (Relationship rel : rels) {
                if (rel.getEndNode().equals(end)) {
                    // relation exists
                    return rel;
                }
            }
        }
        // create new one
        Relationship rel;
        Transaction tx = graphDb.beginTx();
        try {
            rel = start.createRelationshipTo(end, type);
            tx.success();
        } finally {
            tx.finish();
        }
        return rel;
    }

    protected Node getOrCreateAnnotationNode(String id) {
        Index<Node> idx = getNodeIndex(NodeTypes.ANNOTATION);
        IndexHits<Node> annotations = idx.get("id", id);
        Node annotation = annotations.getSingle();
        if (annotation == null) {
            // does not exist yet
            Transaction tx = graphDb.beginTx();
            try {
                annotation = graphDb.createNode();
                annotation.setProperty("TYPE", NodeTypes.ANNOTATION.name());
                annotation.setProperty("id", id);
                idx.add(annotation, "id", id);
                tx.success();
            } finally {
                tx.finish();
            }
        }
        return annotation;
    }

    protected Node getOrCreateTargetNode(String uri, NodeTypes type) {
        Index<Node> idx = getNodeIndex(type);
        IndexHits<Node> targets = idx.get("uri", uri);
        Node target = targets.getSingle();
        if (target == null) {
            // does not exist yet
            Transaction tx = graphDb.beginTx();
            try {
                target = graphDb.createNode();
                target.setProperty("TYPE", type.name());
                target.setProperty("uri", uri);
                idx.add(target, "uri", uri);
                tx.success();
            } finally {
                tx.finish();
            }
        }
        return target;
    }

    protected Node getActorNode(Actor actor) {
        // Person/Group is identified by URI or id
        String uri = actor.getUriString();
        Index<Node> idx;
        if (actor.isGroup()) {
            idx = getNodeIndex(NodeTypes.GROUP);
        } else {
            idx = getNodeIndex(NodeTypes.PERSON);
        }
        IndexHits<Node> persons = idx.get("uri", uri);
        Node person = persons.getSingle();
        return person;
    }

    protected Node getOrCreateActorNode(Actor actor) {
        // Person/Group is identified by URI or id
        String uri = actor.getUriString();
        String name = actor.getName();
        String id = actor.getId();
        Index<Node> idx;
        if (actor.isGroup()) {
            idx = getNodeIndex(NodeTypes.GROUP);
        } else {
            idx = getNodeIndex(NodeTypes.PERSON);
        }
        IndexHits<Node> persons = idx.get("uri", uri);
        Node person = persons.getSingle();
        if (person == null) {
            // does not exist yet
            Transaction tx = graphDb.beginTx();
            try {
                person = graphDb.createNode();
                if (actor.isGroup()) {
                    person.setProperty("TYPE", NodeTypes.GROUP.name());
                } else {
                    person.setProperty("TYPE", NodeTypes.PERSON.name());
                }
                person.setProperty("uri", uri);
                idx.add(person, "uri", uri);
                if (name != null) {
                    person.setProperty("name", name);
                }
                if (id != null) {
                    person.setProperty("id", id);
                }
                tx.success();
            } finally {
                tx.finish();
            }
        }
        return person;
    }

    protected Node getOrCreateTagNode(Tag inTag) {
        Index<Node> idx = getNodeIndex(NodeTypes.TAG);
        String tagname = inTag.getName();
        IndexHits<Node> tags = idx.get("name", tagname);
        Node tag = tags.getSingle();
        if (tag == null) {
            // does not exist yet
            Transaction tx = graphDb.beginTx();
            try {
                tag = graphDb.createNode();
                tag.setProperty("TYPE", NodeTypes.TAG.name());
                tag.setProperty("name", tagname);
                idx.add(tag, "name", tagname);

                tag.setProperty("id", inTag.getId());
                tag.setProperty("uri", inTag.getUri());
                idx.add(tag, "uri", inTag.getUri());

                tx.success();
            } finally {
                tx.finish();
            }
        }
        return tag;
    }

    /**
     * Create or update permissions relations for an annotation.
     * 
     * @param annotNode
     * @param type
     * @param annot
     */
    protected void setPermissionRelation(Node annotNode, RelationTypes type, Actor actor) {
        Node newActorNode = null;
        if (actor != null) {
            newActorNode = getOrCreateActorNode(actor);
        }
        Relationship rel = getRelation(annotNode, type, null);
        if (rel != null) {
            // relation exists
            Node oldActorNode = rel.getEndNode();
            if (!oldActorNode.equals(newActorNode)) {
                // new admin is different
                Transaction tx = graphDb.beginTx();
                try {
                    rel.delete();
                    tx.success();
                } finally {
                    tx.finish();
                }
                if (newActorNode != null) {
                    rel = getOrCreateRelation(annotNode, type, newActorNode);
                }
            }
        } else {
            // no relation yet
            if (newActorNode != null) {
                rel = getOrCreateRelation(annotNode, type, newActorNode);
            }
        }
    }

    /**
     * Unindexes and deletes given Node if it has no relations.
     * 
     * @param node
     */
    protected void deleteNode(Node node) {
        Transaction tx = graphDb.beginTx();
        try {
            if (node.hasRelationship()) {
                logger.error("deleteNode: unable to delete: Node still has relations.");
            } else {
                String ts = (String) node.getProperty("TYPE", null);
                try {
                    NodeTypes type = NodeTypes.valueOf(ts);
                    getNodeIndex(type).remove(node);
                } catch (Exception e) {
                    logger.error("deleteNode: unable to get TYPE of node: " + node);
                }
                node.delete();
            }
            tx.success();
        } finally {
            tx.finish();
        }
    }

    /**
     * returns the (first) Relationship of RelationTypes type from Node start.
     * 
     * @param start
     * @param type
     * @param direction
     * @return
     */
    protected Relationship getRelation(Node start, RelationTypes type, Direction direction) {
        Iterable<Relationship> rels;
        if (direction == null) {
            // ignore direction
            rels = start.getRelationships(type);
        } else {
            rels = start.getRelationships(type, direction);
        }
        for (Relationship rel : rels) {
            // just the first one
            return rel;
        }
        return null;
    }

    /**
     * Erzeuge eine urn aus der aktuellen Zeit in millis
     * 
     * @return
     */
    private String createRessourceURI(String prefix) {

        Calendar cal = Calendar.getInstance();

        long time = cal.getTimeInMillis();

        return String.format("%s%s%s", ANNOTATION_URI_BASE, prefix, time);

    }

    public List<Annotation> getAnnotationsByTag(String tagUri) {

        ArrayList<Annotation> ret = new ArrayList<Annotation>();
        Node tag = getTagNodeByUri(tagUri);

        Iterable<Relationship> rels = tag.getRelationships(Direction.INCOMING, RelationTypes.HAS_TAG);

        for (Relationship rel : rels) {
            Node node = rel.getStartNode();
            ret.add(createAnnotationFromNode(node));

        }
        return ret;
    }

    public List<Annotation> getAnnotationsByResource(String resourceUri) {
        ArrayList<Annotation> ret = new ArrayList<Annotation>();
        Node tag = getNodeFromIndex("uri", resourceUri, NodeTypes.RESOURCE);
        Iterable<Relationship> rels = tag.getRelationships(Direction.INCOMING, RelationTypes.ANNOTATES);
        for (Relationship rel : rels) {
            Node an = rel.getStartNode();
            Node rn = rel.getEndNode();
            if (rn.getProperty("TYPE", "").equals("RESOURCE")) {
                logger.error("getAnnotationsByResource got ANNOTATES != RESOURCE");
            }
            ret.add(createAnnotationFromNode(an));
        }
        return ret;
    }

}