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.model.AppenderRefModel;
032import ch.qos.logback.core.model.Model;
033import ch.qos.logback.core.model.processor.ModelHandlerBase;
034import ch.qos.logback.core.model.processor.ModelHandlerException;
035import ch.qos.logback.core.model.processor.ModelInterpretationContext;
036import ch.qos.logback.tyler.base.TylerModelInterpretationContext;
037import ch.qos.logback.tyler.base.util.VariableNameUtil;
038
039public class AppenderRefModelHandler extends ModelHandlerBase {
040
041    boolean inError = false;
042
043    public AppenderRefModelHandler(Context context) {
044        super(context);
045    }
046
047    static public ModelHandlerBase makeInstance(Context context, ModelInterpretationContext ic) {
048        return new AppenderRefModelHandler(context);
049    }
050
051    @Override
052    protected Class<? extends AppenderRefModel> getSupportedModelClass() {
053        return AppenderRefModel.class;
054    }
055
056    @Override
057    public void handle(ModelInterpretationContext mic, Model model) throws ModelHandlerException {
058        TylerModelInterpretationContext tmic = (TylerModelInterpretationContext) mic;
059
060        AppenderRefModel appenderRefModel = (AppenderRefModel) model;
061        Object o = mic.peekObject();
062
063        if(o instanceof String) {
064            String loggerName  = (String) o;
065            String variableName = VariableNameUtil.loggerNameToVariableName(loggerName);
066            addJavaStatement(tmic, variableName, appenderRefModel.getRef());
067        } else if(o instanceof ImplicitModelHandlerData) {
068            ImplicitModelHandlerData implicitModelHandlerData = (ImplicitModelHandlerData) o;
069            String variableName = implicitModelHandlerData.variableName;
070            addJavaStatement(tmic, variableName, appenderRefModel.getRef());
071        } else {
072            inError = true;
073            addError("Was expecting an object of type AppenderAttachableData");
074            return;
075        }
076
077
078    }
079
080    private void addJavaStatement(TylerModelInterpretationContext tmic, String variableName, String ref) {
081        String appenderVariableName = VariableNameUtil.appenderNameToVariableName(ref);
082        tmic.configureMethodSpecBuilder.addStatement("$N.addAppender($N)", variableName, appenderVariableName);
083
084
085    }
086}