org.openapitools.codegen.templating.mustache.TitlecaseLambda Maven / Gradle / Ivy
/*
* Copyright 2018 OpenAPI-Generator Contributors (https://openapi-generator.tech)
* Copyright 2018 SmartBear Software
*
* 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
*
* https://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.openapitools.codegen.templating.mustache;
import com.samskivert.mustache.Mustache;
import com.samskivert.mustache.Template;
import java.io.IOException;
import java.io.Writer;
import java.util.Locale;
/**
* Converts text in a fragment to title case.
*
* Register:
*
* additionalProperties.put("titlecase", new TitlecaseLambda());
*
*
* Use:
*
* {{#titlecase}}{{classname}}{{/titlecase}}
*
*/
public class TitlecaseLambda implements Mustache.Lambda {
private final String delimiter;
/**
* Constructs a new instance of {@link TitlecaseLambda}, which will convert all text
* in a space delimited string to title-case.
*/
public TitlecaseLambda() {
this(" ");
}
/**
* Constructs a new instance of {@link TitlecaseLambda}, splitting on the specified
* delimiter and converting each word to title-case.
*
* NOTE: passing {@code null} results in a title-casing the first word only.
*
* @param delimiter Provided to allow an override for the default space delimiter.
*/
public TitlecaseLambda(String delimiter) {
this.delimiter = delimiter;
}
private String titleCase(final String input) {
if(input == null || "".equals(input)){
return "";
}
String firstLetter = input.substring(0, 1).toUpperCase(Locale.ROOT);
if (input.length() == 1) {
return firstLetter;
}
return firstLetter + input.substring(1);
}
@Override
public void execute(Template.Fragment fragment, Writer writer) throws IOException {
String text = fragment.execute();
if (delimiter == null) {
writer.write(titleCase(text));
return;
}
// Split accepts regex. \Q and \E wrap the delimiter to create a literal regex,
// so things like "." and "|" aren't treated as their regex equivalents.
String[] parts = text.split("\\Q" + delimiter + "\\E");
for (int i = 0; i < parts.length; i++) {
String part = parts[i];
writer.write(titleCase(part));
if (i != parts.length - 1) {
writer.write(delimiter);
}
}
}
}