examples/org.eclipse.ocl.examples.codegen/src/org/eclipse/ocl/examples/codegen/java/ImportUtils.java - ocl/org.eclipse.ocl - Git at Google

 /*******************************************************************************
  * Copyright (c) 2013, 2019 CEA LIST and others.
  * All rights reserved.   This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License v2.0
  * which accompanies this distribution, and is available at
  * http://www.eclipse.org/legal/epl-v20.html
  *
  * Contributors:
  *   E.D.Willink(CEA LIST) - Initial API and implementation
  *******************************************************************************/
 package org.eclipse.ocl.examples.codegen.java;

 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;

 import org.eclipse.emf.codegen.util.ImportManager;
 import org.eclipse.jdt.annotation.NonNull;
 import org.eclipse.jdt.annotation.Nullable;

 public class ImportUtils
 {
 	public static final String IMPORTS_MARKER = "<%**imports**%>";
 	public static final String IMPORTS_PREFIX = "<%";
 	public static final String IMPORTS_SUFFIX = "%>";

 	public static @NonNull String getAffixedName(@NonNull Class<?> javaClass) {
 		return IMPORTS_PREFIX + javaClass.getName() + IMPORTS_SUFFIX;
 	}

 	public static @NonNull String getAffixedName(@NonNull String className) {
 		assert !className.contains("@");
 		return IMPORTS_PREFIX + className + IMPORTS_SUFFIX;
 	}

 	@Deprecated /* @depreacted use ImportNameManager */
 	public static @NonNull Map<@NonNull String, @Nullable String> getLong2ShortImportNames(@NonNull Iterable<@NonNull String> allImports) {
 		ImportNameManager importManager = new JavaImportNameManager();
 		for (String longName : allImports) {
 			importManager.addImport(null, longName);
 		}
 		return importManager.getLong2ShortImportNames();
 	}

 	/*	public static @NonNull Map<@NonNull String, @Nullable String> getLong2ShortImportNames(@NonNull Iterable<String> allImports) {
 		Map<String, String> long2shortNames = new HashMap<String, String>();
 		Map<String, String> shortables = new HashMap<String, String>();
 		for (String longName : allImports) {
 			int index = longName.lastIndexOf(".");
 			String shortName = index >= 0 ? longName.substring(index+1) : longName;
 			if (shortables.containsKey(shortName)) {
 				String oldImport = shortables.get(shortName);
 				long2shortNames.put(oldImport, null);
 				long2shortNames.put(longName, null);
 				shortables.put(shortName, null);
 			}
 			else {
 				long2shortNames.put(longName, shortName);
 				shortables.put(shortName, longName);
 			}
 		}
 		Map<@NonNull String, @Nullable String> long2short = new HashMap<@NonNull String, @Nullable String>();
 		for (String longName : long2shortNames.keySet()) {
 			String shortName = long2shortNames.get(longName);
 			if (longName != null) {
 				long2short.put(longName, shortName != null ? shortName : null);
 			}
 		}
 		return long2short;
 	} */

 	/**
 	 * @deprecated add skipStartMarker
 	 */
 	@Deprecated
 	public static @NonNull String resolveImports(@NonNull String source, @NonNull Map<@NonNull String, @Nullable String> long2short) {
 		return resolveImports(source, long2short, false);
 	}

 	public static @NonNull String resolveImports(@NonNull String source, @NonNull Map<@NonNull String, @Nullable String> long2short, boolean skipStartMarker)
 	{
 		StringBuilder s = new StringBuilder();
 		int iStart = 0;
 		int iMax = source.length();
 		if (!skipStartMarker) {
 			iStart = source.indexOf(IMPORTS_MARKER);
 			if (iStart < 0) {
 				return source;
 			}
 			s.append(source, 0, iStart);
 			iStart += IMPORTS_MARKER.length();
 			List<String> sortedImports = new ArrayList<String>(long2short.keySet());
 			Collections.sort(sortedImports);
 			for (String anImport : sortedImports) {
 				String shortname = long2short.get(anImport);
 				if ((shortname != null) && (!anImport.startsWith("java.lang.") || !anImport.equals("java.lang." + shortname))) {
 					s.append("import " + anImport +";\n");
 				}
 			}
 		}
 		while (true) {
 			int iPrefix = source.indexOf(IMPORTS_PREFIX, iStart);
 			if (iPrefix < 0) {
 				break;
 			}
 			int iSuffix = source.indexOf(IMPORTS_SUFFIX, iPrefix);
 			if (iSuffix < 0) {
 				break;
 			}
 			s.append(source, iStart, iPrefix);
 			String longName = source.substring(iPrefix+IMPORTS_PREFIX.length(), iSuffix);


 			String annotatedName = longName;
 			//			@Nullable String longAnnotationName = null;
 			//			@Nullable String longTypeName = annotatedName;
 			int startIndex = annotatedName.indexOf("@");
 			int endIndex = annotatedName.indexOf(" ");
 			if ((0 <= startIndex) && (startIndex < endIndex)) {
 				//				longTypeName = annotatedName.substring(0, startIndex) + annotatedName.substring(endIndex).trim();
 				//				longAnnotationName = annotatedName.substring(startIndex+1, endIndex).trim();
 				//				if (importManager != null) {
 				//					importManager.addImport(longAnnotationName);
 				//				}
 			}
 			//			if (importManager != null) {
 			//				importManager.addImport(longTypeName);
 			///			}
 			/*			String shortTypeName = importManager != null ? importManager.getImportedName(longTypeName) : null;
 			String shortAnnotationName = (longAnnotationName != null) && (importManager != null) ? importManager.getImportedName(longAnnotationName) : null;
 			if (longAnnotationName == null) {
 				s.append(IMPORTS_PREFIX);
 				s.append(shortTypeName != null ? shortTypeName : longTypeName);
 				s.append(IMPORTS_SUFFIX);
 			}
 			else if ((shortTypeName != null) && !shortTypeName.equals(longTypeName)) {
 				s.append("@");
 				s.append(IMPORTS_PREFIX);
 				s.append(longAnnotationName);
 				s.append(IMPORTS_SUFFIX);
 				s.append(" ");
 				s.append(shortTypeName);
 			}
 			else {
 				s.append(annotatedName.substring(0, startIndex));
 				s.append("@");
 				s.append(IMPORTS_PREFIX);
 				s.append(longAnnotationName);
 				s.append(IMPORTS_SUFFIX);
 				s.append(" ");
 				s.append(annotatedName.substring(endIndex).trim());
 			} */


 			String shortname = long2short.get(longName);
 			s.append(shortname != null ? shortname : longName);
 			iStart = iSuffix + IMPORTS_SUFFIX.length();
 		}
 		s.append(source, iStart, iMax);
 		return s.toString();
 	}

 	/**
 	 * Rewrite double imports to suit the EMF generators. If importManager is null, as is the case
 	 * since it is not obvious how to re-use the ImportManager between the OCL pre-generate and the Ecore generate
 	 * sessions, an import such as <%x.y.@p.q z%> is changed to x.y.@<%p.q%> z so that the @p.q gets handler by
 	 * the Ecore ImportmManager. If importManager is non-null both imports are shortened.
 	 */
 	public static @NonNull String rewriteManagedImports(@NonNull String source, @Nullable ImportManager importManager)
 	{
 		int iMax = source.length();
 		int iStart = 0;
 		StringBuilder s = new StringBuilder();
 		while (true) {
 			int iPrefix = source.indexOf(IMPORTS_PREFIX, iStart);
 			if (iPrefix < 0) {
 				break;
 			}
 			int iSuffix = source.indexOf(IMPORTS_SUFFIX, iPrefix);
 			if (iSuffix < 0) {
 				break;
 			}
 			s.append(source, iStart, iPrefix);
 			String annotatedName = source.substring(iPrefix+IMPORTS_PREFIX.length(), iSuffix);
 			String longAnnotationName = null;
 			String longTypeName = annotatedName;
 			int startIndex = annotatedName.indexOf("@");
 			int endIndex = annotatedName.indexOf(" ");
 			if ((0 <= startIndex) && (startIndex < endIndex)) {
 				longTypeName = annotatedName.substring(0, startIndex) + annotatedName.substring(endIndex).trim();
 				longAnnotationName = annotatedName.substring(startIndex+1, endIndex).trim();
 				if (importManager != null) {
 					importManager.addImport(longAnnotationName);
 				}
 			}
 			if (importManager != null) {
 				importManager.addImport(longTypeName);
 			}
 			String shortTypeName = importManager != null ? importManager.getImportedName(longTypeName) : null;
 			//			String shortAnnotationName = (longAnnotationName != null) && (importManager != null) ? importManager.getImportedName(longAnnotationName) : null;
 			if (longAnnotationName == null) {
 				s.append(IMPORTS_PREFIX);
 				s.append(shortTypeName != null ? shortTypeName : longTypeName);
 				s.append(IMPORTS_SUFFIX);
 			}
 			else if ((shortTypeName != null) && !shortTypeName.equals(longTypeName)) {
 				s.append("@");
 				s.append(IMPORTS_PREFIX);
 				s.append(longAnnotationName);
 				s.append(IMPORTS_SUFFIX);
 				s.append(" ");
 				s.append(shortTypeName);
 			}
 			else {
 				s.append(annotatedName.substring(0, startIndex));
 				s.append("@");
 				s.append(IMPORTS_PREFIX);
 				s.append(longAnnotationName);
 				s.append(IMPORTS_SUFFIX);
 				s.append(" ");
 				s.append(annotatedName.substring(endIndex).trim());
 			}
 			iStart = iSuffix + IMPORTS_SUFFIX.length();
 		}
 		s.append(source, iStart, iMax);
 		return s.toString();
 	}
 }
	/*******************************************************************************
	* Copyright (c) 2013, 2019 CEA LIST and others.
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License v2.0
	* which accompanies this distribution, and is available at
	* http://www.eclipse.org/legal/epl-v20.html
	*
	* Contributors:
	* E.D.Willink(CEA LIST) - Initial API and implementation
	*******************************************************************************/
	package org.eclipse.ocl.examples.codegen.java;

	import java.util.ArrayList;
	import java.util.Collections;
	import java.util.List;
	import java.util.Map;

	import org.eclipse.emf.codegen.util.ImportManager;
	import org.eclipse.jdt.annotation.NonNull;
	import org.eclipse.jdt.annotation.Nullable;

	public class ImportUtils
	{
	public static final String IMPORTS_MARKER = "<%imports%>";
	public static final String IMPORTS_PREFIX = "<%";
	public static final String IMPORTS_SUFFIX = "%>";

	public static @NonNull String getAffixedName(@NonNull Class<?> javaClass) {
	return IMPORTS_PREFIX + javaClass.getName() + IMPORTS_SUFFIX;
	}

	public static @NonNull String getAffixedName(@NonNull String className) {
	assert !className.contains("@");
	return IMPORTS_PREFIX + className + IMPORTS_SUFFIX;
	}

	@Deprecated /* @depreacted use ImportNameManager */
	public static @NonNull Map<@NonNull String, @Nullable String> getLong2ShortImportNames(@NonNull Iterable<@NonNull String> allImports) {
	ImportNameManager importManager = new JavaImportNameManager();
	for (String longName : allImports) {
	importManager.addImport(null, longName);
	}
	return importManager.getLong2ShortImportNames();
	}

	/* public static @NonNull Map<@NonNull String, @Nullable String> getLong2ShortImportNames(@NonNull Iterable<String> allImports) {
	Map<String, String> long2shortNames = new HashMap<String, String>();
	Map<String, String> shortables = new HashMap<String, String>();
	for (String longName : allImports) {
	int index = longName.lastIndexOf(".");
	String shortName = index >= 0 ? longName.substring(index+1) : longName;
	if (shortables.containsKey(shortName)) {
	String oldImport = shortables.get(shortName);
	long2shortNames.put(oldImport, null);
	long2shortNames.put(longName, null);
	shortables.put(shortName, null);
	}
	else {
	long2shortNames.put(longName, shortName);
	shortables.put(shortName, longName);
	}
	}
	Map<@NonNull String, @Nullable String> long2short = new HashMap<@NonNull String, @Nullable String>();
	for (String longName : long2shortNames.keySet()) {
	String shortName = long2shortNames.get(longName);
	if (longName != null) {
	long2short.put(longName, shortName != null ? shortName : null);
	}
	}
	return long2short;
	} */

	/**
	* @deprecated add skipStartMarker
	*/
	@Deprecated
	public static @NonNull String resolveImports(@NonNull String source, @NonNull Map<@NonNull String, @Nullable String> long2short) {
	return resolveImports(source, long2short, false);
	}

	public static @NonNull String resolveImports(@NonNull String source, @NonNull Map<@NonNull String, @Nullable String> long2short, boolean skipStartMarker)
	{
	StringBuilder s = new StringBuilder();
	int iStart = 0;
	int iMax = source.length();
	if (!skipStartMarker) {
	iStart = source.indexOf(IMPORTS_MARKER);
	if (iStart < 0) {
	return source;
	}
	s.append(source, 0, iStart);
	iStart += IMPORTS_MARKER.length();
	List<String> sortedImports = new ArrayList<String>(long2short.keySet());
	Collections.sort(sortedImports);
	for (String anImport : sortedImports) {
	String shortname = long2short.get(anImport);
	if ((shortname != null) && (!anImport.startsWith("java.lang.") \|\| !anImport.equals("java.lang." + shortname))) {
	s.append("import " + anImport +";\n");
	}
	}
	}
	while (true) {
	int iPrefix = source.indexOf(IMPORTS_PREFIX, iStart);
	if (iPrefix < 0) {
	break;
	}
	int iSuffix = source.indexOf(IMPORTS_SUFFIX, iPrefix);
	if (iSuffix < 0) {
	break;
	}
	s.append(source, iStart, iPrefix);
	String longName = source.substring(iPrefix+IMPORTS_PREFIX.length(), iSuffix);


	String annotatedName = longName;
	// @Nullable String longAnnotationName = null;
	// @Nullable String longTypeName = annotatedName;
	int startIndex = annotatedName.indexOf("@");
	int endIndex = annotatedName.indexOf(" ");
	if ((0 <= startIndex) && (startIndex < endIndex)) {
	// longTypeName = annotatedName.substring(0, startIndex) + annotatedName.substring(endIndex).trim();
	// longAnnotationName = annotatedName.substring(startIndex+1, endIndex).trim();
	// if (importManager != null) {
	// importManager.addImport(longAnnotationName);
	// }
	}
	// if (importManager != null) {
	// importManager.addImport(longTypeName);
	/// }
	/* String shortTypeName = importManager != null ? importManager.getImportedName(longTypeName) : null;
	String shortAnnotationName = (longAnnotationName != null) && (importManager != null) ? importManager.getImportedName(longAnnotationName) : null;
	if (longAnnotationName == null) {
	s.append(IMPORTS_PREFIX);
	s.append(shortTypeName != null ? shortTypeName : longTypeName);
	s.append(IMPORTS_SUFFIX);
	}
	else if ((shortTypeName != null) && !shortTypeName.equals(longTypeName)) {
	s.append("@");
	s.append(IMPORTS_PREFIX);
	s.append(longAnnotationName);
	s.append(IMPORTS_SUFFIX);
	s.append(" ");
	s.append(shortTypeName);
	}
	else {
	s.append(annotatedName.substring(0, startIndex));
	s.append("@");
	s.append(IMPORTS_PREFIX);
	s.append(longAnnotationName);
	s.append(IMPORTS_SUFFIX);
	s.append(" ");
	s.append(annotatedName.substring(endIndex).trim());
	} */





	String shortname = long2short.get(longName);
	s.append(shortname != null ? shortname : longName);
	iStart = iSuffix + IMPORTS_SUFFIX.length();
	}
	s.append(source, iStart, iMax);
	return s.toString();
	}

	/**
	* Rewrite double imports to suit the EMF generators. If importManager is null, as is the case
	* since it is not obvious how to re-use the ImportManager between the OCL pre-generate and the Ecore generate
	* sessions, an import such as <%x.y.@p.q z%> is changed to x.y.@<%p.q%> z so that the @p.q gets handler by
	* the Ecore ImportmManager. If importManager is non-null both imports are shortened.
	*/
	public static @NonNull String rewriteManagedImports(@NonNull String source, @Nullable ImportManager importManager)
	{
	int iMax = source.length();
	int iStart = 0;
	StringBuilder s = new StringBuilder();
	while (true) {
	int iPrefix = source.indexOf(IMPORTS_PREFIX, iStart);
	if (iPrefix < 0) {
	break;
	}
	int iSuffix = source.indexOf(IMPORTS_SUFFIX, iPrefix);
	if (iSuffix < 0) {
	break;
	}
	s.append(source, iStart, iPrefix);
	String annotatedName = source.substring(iPrefix+IMPORTS_PREFIX.length(), iSuffix);
	String longAnnotationName = null;
	String longTypeName = annotatedName;
	int startIndex = annotatedName.indexOf("@");
	int endIndex = annotatedName.indexOf(" ");
	if ((0 <= startIndex) && (startIndex < endIndex)) {
	longTypeName = annotatedName.substring(0, startIndex) + annotatedName.substring(endIndex).trim();
	longAnnotationName = annotatedName.substring(startIndex+1, endIndex).trim();
	if (importManager != null) {
	importManager.addImport(longAnnotationName);
	}
	}
	if (importManager != null) {
	importManager.addImport(longTypeName);
	}
	String shortTypeName = importManager != null ? importManager.getImportedName(longTypeName) : null;
	// String shortAnnotationName = (longAnnotationName != null) && (importManager != null) ? importManager.getImportedName(longAnnotationName) : null;
	if (longAnnotationName == null) {
	s.append(IMPORTS_PREFIX);
	s.append(shortTypeName != null ? shortTypeName : longTypeName);
	s.append(IMPORTS_SUFFIX);
	}
	else if ((shortTypeName != null) && !shortTypeName.equals(longTypeName)) {
	s.append("@");
	s.append(IMPORTS_PREFIX);
	s.append(longAnnotationName);
	s.append(IMPORTS_SUFFIX);
	s.append(" ");
	s.append(shortTypeName);
	}
	else {
	s.append(annotatedName.substring(0, startIndex));
	s.append("@");
	s.append(IMPORTS_PREFIX);
	s.append(longAnnotationName);
	s.append(IMPORTS_SUFFIX);
	s.append(" ");
	s.append(annotatedName.substring(endIndex).trim());
	}
	iStart = iSuffix + IMPORTS_SUFFIX.length();
	}
	s.append(source, iStart, iMax);
	return s.toString();
	}
	}