de.belaso.mongolyn.ui.TaskDataHandler.java Source code

Introduction

Here is the source code for de.belaso.mongolyn.ui.TaskDataHandler.java

Source

/* 
   Copyright (C) 2011, Christian Trutz <christian.trutz@belaso.de>
    
   All rights reserved. This program and the accompanying materials
   are made available under the terms of the Eclipse Public License v1.0
   which accompanies this distribution, and is available at
   http://www.eclipse.org/legal/epl-v10.html
 */
package de.belaso.mongolyn.ui;

import java.util.Map;
import java.util.Set;

import org.bson.types.ObjectId;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.mylyn.tasks.core.ITaskMapping;
import org.eclipse.mylyn.tasks.core.RepositoryResponse;
import org.eclipse.mylyn.tasks.core.RepositoryResponse.ResponseKind;
import org.eclipse.mylyn.tasks.core.TaskRepository;
import org.eclipse.mylyn.tasks.core.data.AbstractTaskDataHandler;
import org.eclipse.mylyn.tasks.core.data.TaskAttribute;
import org.eclipse.mylyn.tasks.core.data.TaskAttributeMapper;
import org.eclipse.mylyn.tasks.core.data.TaskData;

import com.mongodb.BasicDBObject;
import com.mongodb.BasicDBObjectBuilder;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.MongoException;
import com.mongodb.WriteConcern;

/**
 * 
 * @author Christian Trutz
 * 
 */
public class TaskDataHandler extends AbstractTaskDataHandler {

    @Override
    public boolean initializeTaskData(TaskRepository repository, TaskData data, ITaskMapping initializationData,
            IProgressMonitor monitor) throws CoreException {
        TaskAttribute rootAttribute = data.getRoot();
        for (MongolynAttribute mongolynAttribute : MongolynAttribute.values()) {
            TaskAttribute newAttribute = rootAttribute.createAttribute(mongolynAttribute.getId());
            newAttribute.getMetaData().defaults().setKind(mongolynAttribute.getKind())
                    .setLabel(mongolynAttribute.getLabel()).setReadOnly(mongolynAttribute.isReadOnly())
                    .setType(mongolynAttribute.getType()).setDisabled(mongolynAttribute.isDisabled());
        }
        return true;
    }

    @Override
    public RepositoryResponse postTaskData(TaskRepository repository, TaskData taskData,
            Set<TaskAttribute> oldAttributes, IProgressMonitor monitor) throws CoreException {
        DBCollection dbCollection = MongolynUtils.getDBCollection(repository);
        BasicDBObjectBuilder bob = BasicDBObjectBuilder.start();
        for (Map.Entry<String, TaskAttribute> entry : taskData.getRoot().getAttributes().entrySet()) {
            String key = entry.getKey();
            TaskAttribute attribute = entry.getValue();
            String attributeValue = attribute.getValue();
            if (attributeValue != null) {
                bob.add(key.replace('.', '_'), attributeValue);
            }
        }
        DBObject dbObject = bob.get();
        try {
            if (taskData.isNew()) {
                dbCollection.insert(dbObject, WriteConcern.SAFE);
                return new RepositoryResponse(ResponseKind.TASK_CREATED, dbObject.get("_id").toString());
            } else {
                dbCollection.findAndModify(new BasicDBObject("_id", new ObjectId(taskData.getTaskId())), dbObject);
                return new RepositoryResponse(ResponseKind.TASK_UPDATED, taskData.getTaskId());
            }
        } catch (MongoException mongoException) {
            throw new CoreException(Activator.INSTANCE.getErrorStatus(mongoException));
        }
    }

    @Override
    public TaskAttributeMapper getAttributeMapper(TaskRepository repository) {
        return new TaskAttributeMapper(repository);
    }

}