001/*
002 * Copyright (c) 2024 QOS.ch Sarl (Switzerland)
003 * All rights reserved.
004 *
005 * Permission is hereby granted, free  of charge, to any person obtaining
006 * a  copy  of this  software  and  associated  documentation files  (the
007 * "Software"), to  deal in  the Software without  restriction, including
008 * without limitation  the rights to  use, copy, modify,  merge, publish,
009 * distribute,  sublicense, and/or sell  copies of  the Software,  and to
010 * permit persons to whom the Software  is furnished to do so, subject to
011 * the following conditions:
012 *
013 * The  above  copyright  notice  and  this permission  notice  shall  be
014 * included in all copies or substantial portions of the Software.
015 *
016 * THE  SOFTWARE IS  PROVIDED  "AS  IS", WITHOUT  WARRANTY  OF ANY  KIND,
017 * EXPRESS OR  IMPLIED, INCLUDING  BUT NOT LIMITED  TO THE  WARRANTIES OF
018 * MERCHANTABILITY,    FITNESS    FOR    A   PARTICULAR    PURPOSE    AND
019 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
020 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
021 * OF CONTRACT, TORT OR OTHERWISE,  ARISING FROM, OUT OF OR IN CONNECTION
022 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
023 *
024 *
025 *
026 */
027
028package ch.qos.logback.tyler.base.handler;
029
030import ch.qos.logback.core.Context;
031import ch.qos.logback.core.joran.action.ActionUtil;
032import ch.qos.logback.core.model.ComponentModel;
033import ch.qos.logback.core.model.DefineModel;
034import ch.qos.logback.core.model.processor.ModelHandlerBase;
035import ch.qos.logback.core.model.processor.ModelInterpretationContext;
036import ch.qos.logback.core.spi.PropertyDefiner;
037import ch.qos.logback.tyler.base.util.VariableNameUtil;
038import com.squareup.javapoet.MethodSpec;
039
040public class DefineModelHandler extends ComponentModelHandler {
041
042    static String PROPERTY_VALUE_VARIABLE_NAME = "propertyValue";
043    static String SCOPE_VARIABLE_NAME = "scope";
044
045    public DefineModelHandler(Context context) {
046        super(context);
047    }
048
049    static public ModelHandlerBase makeInstance(Context context, ModelInterpretationContext mic) {
050        return new DefineModelHandler(context);
051    }
052
053    @Override
054    protected Class<DefineModel> getSupportedModelClass() {
055        return DefineModel.class;
056    }
057
058
059    @Override
060    String getTargetType() {
061        return PropertyDefiner.class.getSimpleName();
062    }
063
064    @Override
065    protected void addAdditionalJavaStatement(MethodSpec.Builder methodSpec, ComponentModel componentModel) {
066        DefineModel defineModel = (DefineModel) componentModel;
067        String componentClassName = componentModel.getClassName();
068        String variableName = VariableNameUtil.fullyQualifiedClassNameToVariableName(componentClassName);
069        String propertyName = defineModel.getName();
070        String scopeStr = defineModel.getScopeStr();
071
072        methodSpec.addStatement("$T $N  = $T.stringToScope($S)", ActionUtil.Scope.class,  SCOPE_VARIABLE_NAME,  ActionUtil.class, scopeStr);
073        methodSpec.addStatement("String propertyValue = $N.getPropertyValue()", variableName);
074        methodSpec.beginControlFlow("if(propertyValue != null)", variableName);
075        methodSpec.addStatement("addInfo(\"Setting property '$N' to '\"+$N+\"' in scope \"+$N)", propertyName, PROPERTY_VALUE_VARIABLE_NAME, SCOPE_VARIABLE_NAME);
076        methodSpec.addStatement("$T.setProperty(this, $S, $N, $N)", ActionUtil.class, propertyName, PROPERTY_VALUE_VARIABLE_NAME, SCOPE_VARIABLE_NAME);
077        methodSpec.endControlFlow();
078
079    }
080
081
082}