All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.springframework.boot.ansi.AnsiPropertySource Maven / Gradle / Ivy

There is a newer version: 3.3.3
Show newest version
/*
 * Copyright 2012-2015 the original author or authors.
 *
 * 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 org.springframework.boot.ansi;

import java.util.ArrayList;
import java.util.Collections;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;

import org.springframework.core.env.PropertyResolver;
import org.springframework.core.env.PropertySource;
import org.springframework.util.StringUtils;

/**
 * {@link PropertyResolver} for {@link AnsiStyle}, {@link AnsiColor} and
 * {@link AnsiBackground} elements. Supports properties of the form {@code AnsiStyle.BOLD}
 * , {@code AnsiColor.RED} or {@code AnsiBackground.GREEN}. Also supports a prefix of
 * {@code Ansi.} which is an aggregation of everything (with background colors prefixed
 * {@code BG_}).
 *
 * @author Phillip Webb
 * @since 1.3.0
 */
public class AnsiPropertySource extends PropertySource {

	private static final Iterable> MAPPED_ENUMS;

	static {
		List> enums = new ArrayList>();
		enums.add(new MappedEnum("AnsiStyle.", AnsiStyle.class));
		enums.add(new MappedEnum("AnsiColor.", AnsiColor.class));
		enums.add(
				new MappedEnum("AnsiBackground.", AnsiBackground.class));
		enums.add(new MappedEnum("Ansi.", AnsiStyle.class));
		enums.add(new MappedEnum("Ansi.", AnsiColor.class));
		enums.add(new MappedEnum("Ansi.BG_", AnsiBackground.class));
		MAPPED_ENUMS = Collections.unmodifiableList(enums);
	}

	private final boolean encode;

	/**
	 * Create a new {@link AnsiPropertySource} instance.
	 * @param name the name of the property source
	 * @param encode if the output should be encoded
	 */
	public AnsiPropertySource(String name, boolean encode) {
		super(name);
		this.encode = encode;
	}

	@Override
	public Object getProperty(String name) {
		if (StringUtils.hasLength(name)) {
			for (MappedEnum mappedEnum : MAPPED_ENUMS) {
				if (name.startsWith(mappedEnum.getPrefix())) {
					String enumName = name.substring(mappedEnum.getPrefix().length());
					for (Enum ansiEnum : mappedEnum.getEnums()) {
						if (ansiEnum.name().equals(enumName)) {
							if (this.encode) {
								return AnsiOutput.encode((AnsiElement) ansiEnum);
							}
							return ansiEnum;
						}
					}
				}
			}
		}
		return null;
	}

	/**
	 * Mapping between an enum and the pseudo property source.
	 */
	private static class MappedEnum> {

		private final String prefix;

		private final Set enums;

		MappedEnum(String prefix, Class enumType) {
			this.prefix = prefix;
			this.enums = EnumSet.allOf(enumType);

		}

		public String getPrefix() {
			return this.prefix;
		}

		public Set getEnums() {
			return this.enums;
		}

	}

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy