org.openapitools.codegen.mustache.CamelCaseLambda 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
*
* 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.openapitools.codegen.mustache;
import com.google.common.base.CaseFormat;
import com.samskivert.mustache.Mustache;
import com.samskivert.mustache.Template;
import org.openapitools.codegen.CodegenConfig;
import org.openapitools.codegen.DefaultCodegen;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.text.WordUtils;
import java.io.IOException;
import java.io.Writer;
/**
* Converts text in a fragment to camelCase.
*
* Register:
*
* additionalProperties.put("camelcase", new CamelCaseLambda());
*
*
* Use:
*
* {{#camelcase}}{{name}}{{/camelcase}}
*
*/
public class CamelCaseLambda implements Mustache.Lambda {
private CodegenConfig generator = null;
private Boolean escapeParam = false;
public CamelCaseLambda() {
}
public CamelCaseLambda generator(final CodegenConfig generator) {
this.generator = generator;
return this;
}
public CamelCaseLambda escapeAsParamName(final Boolean escape) {
this.escapeParam = escape;
return this;
}
@Override
public void execute(Template.Fragment fragment, Writer writer) throws IOException {
String text = DefaultCodegen.camelize(fragment.execute(), true);
if (generator != null) {
text = generator.sanitizeName(text);
if (generator.reservedWords().contains(text)) {
// Escaping must be done *after* camelize, because generators may escape using characters removed by camelize function.
text = generator.escapeReservedWord(text);
}
if (escapeParam) {
// NOTE: many generators call escapeReservedWord in toParamName, but we can't assume that's always the case.
// Here, we'll have to accept that we may be duplicating some work.
text = generator.toParamName(text);
}
}
writer.write(text);
}
}