/*
 * Copyright 2011-2015 B2i Healthcare Pte Ltd, http://b2i.sg
 * 
 * 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 com.b2international.snowowl.snomed.datastore.index.update;

import java.util.Collection;

import com.b2international.snowowl.core.exceptions.CycleDetectedException;
import com.b2international.snowowl.snomed.SnomedConstants.Concepts;
import com.b2international.snowowl.snomed.datastore.index.entry.SnomedConceptDocument;
import com.b2international.snowowl.snomed.datastore.taxonomy.ISnomedTaxonomyBuilder;

/**
 * @since 4.3
 */
public class IconIdUpdater {
	
	private final ISnomedTaxonomyBuilder inferredTaxonomy;
	private final ISnomedTaxonomyBuilder statedTaxonomy;
	private final Collection<String> availableImages;
	
	public IconIdUpdater(ISnomedTaxonomyBuilder inferredTaxonomy, ISnomedTaxonomyBuilder statedTaxonomy, Collection<String> availableImages) {
		this.inferredTaxonomy = inferredTaxonomy;
		this.statedTaxonomy = statedTaxonomy;
		this.availableImages = availableImages;
	}
	
	public void update(String id, boolean active, SnomedConceptDocument.Builder doc) {
		doc.iconId(getIconId(id, active));
	}
	
	protected String getIconId(String conceptId, boolean active) {
		String iconId = null;
		if (active) {
			// TODO do not set the iconId to the one in the taxonomyBuilder if the concept is inactive
			// we may have to need a ref to the concept itself or at least know the active flag
			if (inferredTaxonomy.containsNode(conceptId)) {
				iconId = getParentIcon(conceptId, inferredTaxonomy);
			}
			// try to use the stated form to get the ID
			if (iconId == null) {
				iconId = getParentIcon(conceptId, statedTaxonomy);
			}
		}
		// default icon is the ROOT if neither the inferred nor the stated tree provided an ICON, usually this means that the concept is inactive
		return iconId == null ? Concepts.ROOT_CONCEPT : iconId;
	}
	
	private String getParentIcon(String componentId, ISnomedTaxonomyBuilder taxonomyBuilder) {
		if (componentId == null) {
			return null;
		}
		if (taxonomyBuilder.getAllAncestorNodeIds(componentId).contains(Long.valueOf(componentId))) {
			throw new CycleDetectedException("Concept " + componentId + " would introduce a cycle in the ISA graph (loop).");
		}
		return getParentFrom(componentId, taxonomyBuilder);
	}
	
	private String getParentFrom(final String conceptId, ISnomedTaxonomyBuilder taxonomyBuilder) {
		if (taxonomyBuilder.getAncestorNodeIds(conceptId).size() == 0) {
			return null;
		}
		if (this.availableImages.contains(conceptId)) {
			return conceptId;
		}
		return getParentFrom(Long.toString(taxonomyBuilder.getAncestorNodeIds(conceptId).iterator().next()), taxonomyBuilder);
	}
}

@@ -1937,60 +1937,8 @@
 lag%0A
-%09%09%09if (inferredTaxonomy.containsNode(conceptId)) %7B%0A%09
 %09%09%09i
@@ -1989,21 +1989,16 @@
 onomy);%0A
-%09%09%09%7D%0A
 %09%09%09// tr
@@ -2441,24 +2441,70 @@
 ntId == null
+ %7C%7C !taxonomyBuilder.containsNode(componentId)
 ) %7B%0A%09%09%09retur

Check in IconIdUpdater whether the concept exists in a taxonomy