spoon.support.util.ImmutableMapImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spoon-core Show documentation
Spoon is a tool for meta-programming, analysis and transformation of Java programs.
There is a newer version: 11.1.1-beta-14
/*
 * SPDX-License-Identifier: (MIT OR CECILL-C)
 *
 * Copyright (C) 2006-2023 INRIA and contributors
 *
 * Spoon is available either under the terms of the MIT License (see LICENSE-MIT.txt) or the Cecill-C License (see LICENSE-CECILL-C.txt). You as the user are entitled to choose the terms under which to adopt Spoon.
 */
package spoon.support.util;

import spoon.support.Internal;

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

/**
 *
 * Internal class only, not in the public API.
 *
 * Spoon implementation of {@link ImmutableMap}
 */
@Internal
public class ImmutableMapImpl implements ImmutableMap {

	protected final ImmutableMap parent;
	protected final Map map;

	public ImmutableMapImpl(Map map) {
		this(null, map);
	}
	private ImmutableMapImpl(ImmutableMap parent, Map map) {
		this.parent = parent;
		this.map = Collections.unmodifiableMap(map);
	}

	public ImmutableMapImpl(Map map, String parameterName, Object value) {
		this(null, map, parameterName, value);
	}

	private ImmutableMapImpl(ImmutableMap parent, Map map, String parameterName, Object value) {
		this.parent = null;
		Map copy = new HashMap<>(map.size() + 1);
		copy.putAll(map);
		copy.put(parameterName, value);
		this.map = Collections.unmodifiableMap(copy);
	}

	public ImmutableMapImpl() {
		this.parent = null;
		this.map = Collections.emptyMap();
	}

	@Override
	public ImmutableMapImpl checkpoint() {
		return new ImmutableMapImpl(this, Collections.emptyMap());
	}

	@Override
	public boolean hasValue(String parameterName) {
		if (map.containsKey(parameterName)) {
			return true;
		}
		if (parent != null) {
			return parent.hasValue(parameterName);
		}
		return false;
	}

	@Override
	public Object getValue(String parameterName) {
		Object v = map.get(parameterName);
		if (v == null && parent != null) {
			v = parent.getValue(parameterName);
		}
		return v;
	}

	@Override
	public ImmutableMap putValue(String parameterName, Object value) {
		return new ImmutableMapImpl(parent, map, parameterName, value);
	}

	@Override
	public String toString() {
		StringBuilder sb = new StringBuilder();
		appendMap(sb, map);
		if (parent != null) {
			sb.append("\nparent:\n");
			sb.append(parent);
		}
		return sb.toString();
	}

	private static void appendMap(StringBuilder sb, Map map) {
		List paramNames = new ArrayList<>(map.keySet());
		paramNames.sort((a, b) -> a.compareTo(b));
		for (String name : paramNames) {
			if (sb.length() > 0) {
				sb.append("\n");
			}
			sb.append(name).append('=').append(map.get(name));
		}
	}

	@Override
	public Map asMap() {
		if (parent != null) {
			Map merged = new HashMap<>();
			merged.putAll(parent.asMap());
			merged.putAll(map);
			return Collections.unmodifiableMap(merged);
		}
		return map;
	}

	@Override
	public Map getModifiedValues() {
		return map;
	}

	@Override
	public boolean equals(Object obj) {
		if (obj instanceof ImmutableMap) {
			obj = ((ImmutableMap) obj).asMap();
		}
		if (obj instanceof Map) {
			Map map = (Map) obj;
			return asMap().equals(map);
		}
		return false;
	}
}