connector-bugzilla-rest/org.eclipse.mylyn.bugzilla.rest.core/src/org/eclipse/mylyn/internal/bugzilla/rest/core/BugzillaRestPutUpdateTask.java - mylyn/org.eclipse.mylyn.tasks - Git at Google

 /*******************************************************************************
  * Copyright (c) 2015 Frank Becker and others.
  * 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
  *
  * Contributors:
  *     Frank Becker - initial API and implementation
  *******************************************************************************/

 package org.eclipse.mylyn.internal.bugzilla.rest.core;

 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.util.Arrays;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;

 import org.eclipse.core.runtime.Assert;
 import org.eclipse.jdt.annotation.NonNull;
 import org.eclipse.mylyn.commons.repositories.http.core.CommonHttpClient;
 import org.eclipse.mylyn.internal.bugzilla.rest.core.response.data.PutUpdateResult;
 import org.eclipse.mylyn.tasks.core.data.TaskAttribute;
 import org.eclipse.mylyn.tasks.core.data.TaskData;

 import com.google.common.base.Joiner;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
 import com.google.gson.Gson;
 import com.google.gson.reflect.TypeToken;
 import com.google.gson.stream.JsonWriter;

 public class BugzillaRestPutUpdateTask extends BugzillaRestPutRequest<PutUpdateResult> {
 	private final TaskData taskData;

 	ImmutableList<String> legalUpdateAttributes = new ImmutableList.Builder<String>()
 			.add(BugzillaRestTaskSchema.getDefault().PRODUCT.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().COMPONENT.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().SUMMARY.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().VERSION.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().DESCRIPTION.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().OS.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().PLATFORM.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().PRIORITY.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().SEVERITY.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().ALIAS.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().ASSIGNED_TO.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().QA_CONTACT.getKey())
 			.add(TaskAttribute.OPERATION)
 			.add(BugzillaRestTaskSchema.getDefault().TARGET_MILESTONE.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().NEW_COMMENT.getKey())
 			.add("resolutionInput") //$NON-NLS-1$
 			.add(BugzillaRestTaskSchema.getDefault().RESOLUTION.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().DUPE_OF.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().BLOCKS.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().DEPENDS_ON.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().KEYWORDS.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().RESET_QA_CONTACT.getKey())
 			.add(BugzillaRestTaskSchema.getDefault().RESET_ASSIGNED_TO.getKey())
 			.build();

 	public BugzillaRestPutUpdateTask(CommonHttpClient client, @NonNull TaskData taskData,
 			@NonNull Set<TaskAttribute> oldAttributes) {
 		super(client, "/bug/" + taskData.getTaskId(), false, oldAttributes); //$NON-NLS-1$
 		Assert.isNotNull(taskData);
 		this.taskData = taskData;
 	}

 	public static String convert(String str) {
 		str = str.replace("\"", "\\\"").replace("\n", "\\\n"); //$NON-NLS-1$//$NON-NLS-2$ //$NON-NLS-3$//$NON-NLS-4$
 		StringBuffer ostr = new StringBuffer();
 		for (int i = 0; i < str.length(); i++) {
 			char ch = str.charAt(i);
 			if ((ch >= 0x0020) && (ch <= 0x007e)) {
 				ostr.append(ch);
 			} else {
 				ostr.append("\\u"); //$NON-NLS-1$
 				String hex = Integer.toHexString(str.charAt(i) & 0xFFFF);
 				for (int j = 0; j < 4 - hex.length(); j++) {
 					ostr.append("0"); //$NON-NLS-1$
 				}
 				ostr.append(hex.toLowerCase());
 			}
 		}
 		return (new String(ostr));
 	}

 	@Override
 	protected PutUpdateResult parseFromJson(InputStreamReader in) {
 		TypeToken<PutUpdateResult> type = new TypeToken<PutUpdateResult>() {
 		};
 		return new Gson().fromJson(in, type.getType());
 	}

 	@Override
 	public void addPrefixToOutput(JsonWriter out) throws IOException {
 		// not needed
 	}

 	@Override
 	public void addAttributeToOutput(JsonWriter out, TaskAttribute element) throws IOException {
 		TaskAttribute taskAttribute = taskData.getRoot().getAttribute(element.getId());
 		String id = taskAttribute.getId();
 		String value = BugzillaRestGsonUtil.convertString2GSonString(taskAttribute.getValue());
 		if ((legalUpdateAttributes.contains(id) || id.startsWith("cf_")) && value != null) { //$NON-NLS-1$
 			id = BugzillaRestTaskSchema.getFieldNameFromAttributeName(id);
 			if (id.equals("status")) { //$NON-NLS-1$
 				if (value != null && value.equals(TaskAttribute.PREFIX_OPERATION + "default")) { //$NON-NLS-1$
 					return;
 				}
 				if (value.equals("duplicate")) { //$NON-NLS-1$
 					TaskAttribute res = element.getParentAttribute()
 							.getAttribute(BugzillaRestTaskSchema.getDefault().RESOLUTION.getKey());
 					if (!taskAttributes.contains(res)) {
 						out.name("resolution").value("DUPLICATE"); //$NON-NLS-1$ //$NON-NLS-2$
 					} else {
 						TaskAttribute res1 = taskData.getRoot()
 								.getAttribute(BugzillaRestTaskSchema.getDefault().RESOLUTION.getKey());
 						res1.setValue("DUPLICATE"); //$NON-NLS-1$
 					}
 					value = "RESOLVED"; //$NON-NLS-1$
 				}
 			}
 			if (taskAttribute.getMetaData().getType() != null
 					&& taskAttribute.getMetaData().getType().equals(TaskAttribute.TYPE_MULTI_SELECT)) {
 				Iterable<String> taskIdsTemp = Iterables.transform(taskAttribute.getValues(), function);
 				Joiner joiner = Joiner.on(",").skipNulls(); //$NON-NLS-1$
 				value = joiner.join(taskIdsTemp);
 			}
 			if (id.equals(BugzillaRestTaskSchema.getDefault().NEW_COMMENT.getKey())) {
 				out.name("comment").beginObject(); //$NON-NLS-1$
 				out.name("body").value(value); //$NON-NLS-1$
 				out.endObject();
 			} else if (id.equals(BugzillaRestTaskSchema.getDefault().BLOCKS.getKey())
 					|| id.equals(BugzillaRestTaskSchema.getDefault().DEPENDS_ON.getKey())) {
 				Set<String> setOld;
 				if (element.getValues().size() > 1) {
 					setOld = new HashSet<String>(element.getValues());
 				} else {
 					setOld = new HashSet<String>(Arrays.asList(element.getValue().split("\\s*,\\s*"))); //$NON-NLS-1$
 				}
 				Set<String> setNew;
 				if (taskAttribute.getValues().size() > 1) {
 					setNew = new HashSet<String>(taskAttribute.getValues());
 				} else {
 					setNew = new HashSet<String>(Arrays.asList(taskAttribute.getValue().split("\\s*,\\s*"))); //$NON-NLS-1$
 				}
 				BugzillaRestGsonUtil.getDefault().buildAddRemoveIntegerHash(out, id, setOld, setNew);
 			} else if (id.equals(BugzillaRestTaskSchema.getDefault().KEYWORDS.getKey())) {
 				Set<String> setOld = new HashSet<String>(element.getValues());
 				Set<String> setNew = new HashSet<String>(taskAttribute.getValues());
 				BugzillaRestGsonUtil.getDefault().buildAddRemoveHash(out, id, setOld, setNew);
 			} else {
 				out.name(id).value(value);
 				if (id.equals("description")) { //$NON-NLS-1$
 					TaskAttribute descriptionpri = taskAttribute
 							.getAttribute(BugzillaRestTaskSchema.getDefault().COMMENT_ISPRIVATE.getKey());
 					Boolean descriptionprivalue = (descriptionpri != null)
 							? (descriptionpri.getValue().equals("1")) //$NON-NLS-1$
 							: false;
 					out.name("comment_is_private").value(Boolean.toString(descriptionprivalue)); //$NON-NLS-1$
 				}
 			}
 		}
 	}

 	@Override
 	public void addSuffixToOutput(JsonWriter out) throws IOException {
 		TaskAttribute cc = taskData.getRoot().getAttribute(BugzillaRestTaskSchema.getDefault().CC.getKey());
 		TaskAttribute addCC = taskData.getRoot().getAttribute(BugzillaRestTaskSchema.getDefault().ADD_CC.getKey());
 		TaskAttribute removeCC = taskData.getRoot()
 				.getAttribute(BugzillaRestTaskSchema.getDefault().REMOVE_CC.getKey());
 		TaskAttribute addSelfCC = taskData.getRoot()
 				.getAttribute(BugzillaRestTaskSchema.getDefault().ADD_SELF_CC.getKey());
 		if (Boolean.valueOf(addSelfCC.getValue())) {
 			String userName = addSelfCC.getMetaData().getValue("UserName"); //$NON-NLS-1$
 			if (userName != null) {
 				if (removeCC.getValues().contains(userName)) {
 					removeCC.removeValue(userName);
 				} else if (!cc.getValues().contains(userName)) {
 					List<String> addCCList = Arrays.asList(addCC.getValue().split("\\s*,\\s*")); //$NON-NLS-1$
 					if (!addCCList.contains(userName)) {
 						addCC.setValue(userName + (addCCList.size() > 0 && !addCCList.get(0).equals("") ? ", " : "") //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
 								+ addCC.getValue());
 					}
 				}
 			}
 		}
 		if (addCC.getValues().size() > 0 || removeCC.getValues().size() > 0) {
 			Set<String> setOld = new HashSet<String>(removeCC.getValues());
 			HashSet<String> setNew = new HashSet<String>(Arrays.asList(addCC.getValue().split("\\s*,\\s*"))); //$NON-NLS-1$
 			BugzillaRestGsonUtil.getDefault().buildAddRemoveHash(out, "cc", setOld, setNew); //$NON-NLS-1$
 		}
 		BugzillaRestGsonUtil.buildFlags(out, taskAttributes.getTaskAttributes(), taskData.getRoot());
 	}

 }
	/*******************************************************************************
	* Copyright (c) 2015 Frank Becker and others.
	* 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
	*
	* Contributors:
	* Frank Becker - initial API and implementation
	*******************************************************************************/

	package org.eclipse.mylyn.internal.bugzilla.rest.core;

	import java.io.IOException;
	import java.io.InputStreamReader;
	import java.util.Arrays;
	import java.util.HashSet;
	import java.util.List;
	import java.util.Set;

	import org.eclipse.core.runtime.Assert;
	import org.eclipse.jdt.annotation.NonNull;
	import org.eclipse.mylyn.commons.repositories.http.core.CommonHttpClient;
	import org.eclipse.mylyn.internal.bugzilla.rest.core.response.data.PutUpdateResult;
	import org.eclipse.mylyn.tasks.core.data.TaskAttribute;
	import org.eclipse.mylyn.tasks.core.data.TaskData;

	import com.google.common.base.Joiner;
	import com.google.common.collect.ImmutableList;
	import com.google.common.collect.Iterables;
	import com.google.gson.Gson;
	import com.google.gson.reflect.TypeToken;
	import com.google.gson.stream.JsonWriter;

	public class BugzillaRestPutUpdateTask extends BugzillaRestPutRequest<PutUpdateResult> {
	private final TaskData taskData;

	ImmutableList<String> legalUpdateAttributes = new ImmutableList.Builder<String>()
	.add(BugzillaRestTaskSchema.getDefault().PRODUCT.getKey())
	.add(BugzillaRestTaskSchema.getDefault().COMPONENT.getKey())
	.add(BugzillaRestTaskSchema.getDefault().SUMMARY.getKey())
	.add(BugzillaRestTaskSchema.getDefault().VERSION.getKey())
	.add(BugzillaRestTaskSchema.getDefault().DESCRIPTION.getKey())
	.add(BugzillaRestTaskSchema.getDefault().OS.getKey())
	.add(BugzillaRestTaskSchema.getDefault().PLATFORM.getKey())
	.add(BugzillaRestTaskSchema.getDefault().PRIORITY.getKey())
	.add(BugzillaRestTaskSchema.getDefault().SEVERITY.getKey())
	.add(BugzillaRestTaskSchema.getDefault().ALIAS.getKey())
	.add(BugzillaRestTaskSchema.getDefault().ASSIGNED_TO.getKey())
	.add(BugzillaRestTaskSchema.getDefault().QA_CONTACT.getKey())
	.add(TaskAttribute.OPERATION)
	.add(BugzillaRestTaskSchema.getDefault().TARGET_MILESTONE.getKey())
	.add(BugzillaRestTaskSchema.getDefault().NEW_COMMENT.getKey())
	.add("resolutionInput") //$NON-NLS-1$
	.add(BugzillaRestTaskSchema.getDefault().RESOLUTION.getKey())
	.add(BugzillaRestTaskSchema.getDefault().DUPE_OF.getKey())
	.add(BugzillaRestTaskSchema.getDefault().BLOCKS.getKey())
	.add(BugzillaRestTaskSchema.getDefault().DEPENDS_ON.getKey())
	.add(BugzillaRestTaskSchema.getDefault().KEYWORDS.getKey())
	.add(BugzillaRestTaskSchema.getDefault().RESET_QA_CONTACT.getKey())
	.add(BugzillaRestTaskSchema.getDefault().RESET_ASSIGNED_TO.getKey())
	.build();

	public BugzillaRestPutUpdateTask(CommonHttpClient client, @NonNull TaskData taskData,
	@NonNull Set<TaskAttribute> oldAttributes) {
	super(client, "/bug/" + taskData.getTaskId(), false, oldAttributes); //$NON-NLS-1$
	Assert.isNotNull(taskData);
	this.taskData = taskData;
	}

	public static String convert(String str) {
	str = str.replace("\"", "\\\"").replace("\n", "\\\n"); //$NON-NLS-1$//$NON-NLS-2$ //$NON-NLS-3$//$NON-NLS-4$
	StringBuffer ostr = new StringBuffer();
	for (int i = 0; i < str.length(); i++) {
	char ch = str.charAt(i);
	if ((ch >= 0x0020) && (ch <= 0x007e)) {
	ostr.append(ch);
	} else {
	ostr.append("\\u"); //$NON-NLS-1$
	String hex = Integer.toHexString(str.charAt(i) & 0xFFFF);
	for (int j = 0; j < 4 - hex.length(); j++) {
	ostr.append("0"); //$NON-NLS-1$
	}
	ostr.append(hex.toLowerCase());
	}
	}
	return (new String(ostr));
	}

	@Override
	protected PutUpdateResult parseFromJson(InputStreamReader in) {
	TypeToken<PutUpdateResult> type = new TypeToken<PutUpdateResult>() {
	};
	return new Gson().fromJson(in, type.getType());
	}

	@Override
	public void addPrefixToOutput(JsonWriter out) throws IOException {
	// not needed
	}

	@Override
	public void addAttributeToOutput(JsonWriter out, TaskAttribute element) throws IOException {
	TaskAttribute taskAttribute = taskData.getRoot().getAttribute(element.getId());
	String id = taskAttribute.getId();
	String value = BugzillaRestGsonUtil.convertString2GSonString(taskAttribute.getValue());
	if ((legalUpdateAttributes.contains(id) \|\| id.startsWith("cf_")) && value != null) { //$NON-NLS-1$
	id = BugzillaRestTaskSchema.getFieldNameFromAttributeName(id);
	if (id.equals("status")) { //$NON-NLS-1$
	if (value != null && value.equals(TaskAttribute.PREFIX_OPERATION + "default")) { //$NON-NLS-1$
	return;
	}
	if (value.equals("duplicate")) { //$NON-NLS-1$
	TaskAttribute res = element.getParentAttribute()
	.getAttribute(BugzillaRestTaskSchema.getDefault().RESOLUTION.getKey());
	if (!taskAttributes.contains(res)) {
	out.name("resolution").value("DUPLICATE"); //$NON-NLS-1$ //$NON-NLS-2$
	} else {
	TaskAttribute res1 = taskData.getRoot()
	.getAttribute(BugzillaRestTaskSchema.getDefault().RESOLUTION.getKey());
	res1.setValue("DUPLICATE"); //$NON-NLS-1$
	}
	value = "RESOLVED"; //$NON-NLS-1$
	}
	}
	if (taskAttribute.getMetaData().getType() != null
	&& taskAttribute.getMetaData().getType().equals(TaskAttribute.TYPE_MULTI_SELECT)) {
	Iterable<String> taskIdsTemp = Iterables.transform(taskAttribute.getValues(), function);
	Joiner joiner = Joiner.on(",").skipNulls(); //$NON-NLS-1$
	value = joiner.join(taskIdsTemp);
	}
	if (id.equals(BugzillaRestTaskSchema.getDefault().NEW_COMMENT.getKey())) {
	out.name("comment").beginObject(); //$NON-NLS-1$
	out.name("body").value(value); //$NON-NLS-1$
	out.endObject();
	} else if (id.equals(BugzillaRestTaskSchema.getDefault().BLOCKS.getKey())
	\|\| id.equals(BugzillaRestTaskSchema.getDefault().DEPENDS_ON.getKey())) {
	Set<String> setOld;
	if (element.getValues().size() > 1) {
	setOld = new HashSet<String>(element.getValues());
	} else {
	setOld = new HashSet<String>(Arrays.asList(element.getValue().split("\\s,\\s"))); //$NON-NLS-1$
	}
	Set<String> setNew;
	if (taskAttribute.getValues().size() > 1) {
	setNew = new HashSet<String>(taskAttribute.getValues());
	} else {
	setNew = new HashSet<String>(Arrays.asList(taskAttribute.getValue().split("\\s,\\s"))); //$NON-NLS-1$
	}
	BugzillaRestGsonUtil.getDefault().buildAddRemoveIntegerHash(out, id, setOld, setNew);
	} else if (id.equals(BugzillaRestTaskSchema.getDefault().KEYWORDS.getKey())) {
	Set<String> setOld = new HashSet<String>(element.getValues());
	Set<String> setNew = new HashSet<String>(taskAttribute.getValues());
	BugzillaRestGsonUtil.getDefault().buildAddRemoveHash(out, id, setOld, setNew);
	} else {
	out.name(id).value(value);
	if (id.equals("description")) { //$NON-NLS-1$
	TaskAttribute descriptionpri = taskAttribute
	.getAttribute(BugzillaRestTaskSchema.getDefault().COMMENT_ISPRIVATE.getKey());
	Boolean descriptionprivalue = (descriptionpri != null)
	? (descriptionpri.getValue().equals("1")) //$NON-NLS-1$
	: false;
	out.name("comment_is_private").value(Boolean.toString(descriptionprivalue)); //$NON-NLS-1$
	}
	}
	}
	}

	@Override
	public void addSuffixToOutput(JsonWriter out) throws IOException {
	TaskAttribute cc = taskData.getRoot().getAttribute(BugzillaRestTaskSchema.getDefault().CC.getKey());
	TaskAttribute addCC = taskData.getRoot().getAttribute(BugzillaRestTaskSchema.getDefault().ADD_CC.getKey());
	TaskAttribute removeCC = taskData.getRoot()
	.getAttribute(BugzillaRestTaskSchema.getDefault().REMOVE_CC.getKey());
	TaskAttribute addSelfCC = taskData.getRoot()
	.getAttribute(BugzillaRestTaskSchema.getDefault().ADD_SELF_CC.getKey());
	if (Boolean.valueOf(addSelfCC.getValue())) {
	String userName = addSelfCC.getMetaData().getValue("UserName"); //$NON-NLS-1$
	if (userName != null) {
	if (removeCC.getValues().contains(userName)) {
	removeCC.removeValue(userName);
	} else if (!cc.getValues().contains(userName)) {
	List<String> addCCList = Arrays.asList(addCC.getValue().split("\\s,\\s")); //$NON-NLS-1$
	if (!addCCList.contains(userName)) {
	addCC.setValue(userName + (addCCList.size() > 0 && !addCCList.get(0).equals("") ? ", " : "") //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
	+ addCC.getValue());
	}
	}
	}
	}
	if (addCC.getValues().size() > 0 \|\| removeCC.getValues().size() > 0) {
	Set<String> setOld = new HashSet<String>(removeCC.getValues());
	HashSet<String> setNew = new HashSet<String>(Arrays.asList(addCC.getValue().split("\\s,\\s"))); //$NON-NLS-1$
	BugzillaRestGsonUtil.getDefault().buildAddRemoveHash(out, "cc", setOld, setNew); //$NON-NLS-1$
	}
	BugzillaRestGsonUtil.buildFlags(out, taskAttributes.getTaskAttributes(), taskData.getRoot());
	}

	}