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.classic.Logger;
031import ch.qos.logback.classic.model.LoggerModel;
032import ch.qos.logback.core.Context;
033import ch.qos.logback.core.model.Model;
034import ch.qos.logback.core.model.processor.ModelHandlerBase;
035import ch.qos.logback.core.model.processor.ModelHandlerException;
036import ch.qos.logback.core.model.processor.ModelInterpretationContext;
037import ch.qos.logback.core.util.OptionHelper;
038import ch.qos.logback.tyler.base.TylerModelInterpretationContext;
039import ch.qos.logback.tyler.base.util.StringToVariableStament;
040import ch.qos.logback.tyler.base.util.VariableNameUtil;
041
042import static ch.qos.logback.classic.tyler.TylerConfiguratorBase.SETUP_LOGGER_METHOD_NAME;
043
044public class LoggerModelHandler  extends ModelHandlerBase {
045
046    boolean inError = false;
047    String loggerName;
048
049    public LoggerModelHandler(Context context) {
050        super(context);
051    }
052
053    protected Class<LoggerModel> getSupportedModelClass() {
054        return LoggerModel.class;
055    }
056
057    static public ModelHandlerBase makeInstance(Context context, ModelInterpretationContext mic) {
058        return new LoggerModelHandler(context);
059    }
060
061
062    @Override
063    public void handle(ModelInterpretationContext mic, Model model) throws ModelHandlerException {
064        LoggerModel loggerModel = (LoggerModel) model;
065        TylerModelInterpretationContext tmic = (TylerModelInterpretationContext) mic;
066
067        this.loggerName = loggerModel.getName();
068
069        String levelStr = loggerModel.getLevel();
070        String additivityStr = loggerModel.getAdditivity();
071        Boolean additivity = addtivityStringToBoolean(additivityStr);
072        addJavaStatement(tmic, loggerName, levelStr, additivity);
073        mic.pushObject(loggerName);
074    }
075
076
077    private Boolean addtivityStringToBoolean(String additivityStr) {
078        if (!OptionHelper.isNullOrEmptyOrAllSpaces(additivityStr)) {
079            return OptionHelper.toBoolean(additivityStr, true);
080        } else {
081            return null;
082        }
083    }
084
085    private void addJavaStatement(TylerModelInterpretationContext tmic, String loggerName, String levelStr, Boolean additivity) {
086        // Logger logger_XYZ = setupLogger(loggerName, level, levelString, additivity)
087        String loggerVariableName = VariableNameUtil.loggerNameToVariableName(loggerName);
088
089        String additivityStr = additivity == null ? "null" : "Boolean."+additivity.toString().toUpperCase();
090        boolean containsVariable = StringToVariableStament.containsVariable(levelStr);
091        String levelStrPart = containsVariable ? "subst($S)" : "$S";
092
093        tmic.configureMethodSpecBuilder.addStatement("$T $N = $N($S, "+levelStrPart+", $N)", Logger.class, loggerVariableName,
094                SETUP_LOGGER_METHOD_NAME, loggerName, levelStr, additivityStr );
095    }
096
097    @Override
098    public void postHandle(ModelInterpretationContext mic, Model model) throws ModelHandlerException {
099        if (inError) {
100            return;
101        }
102        Object o = mic.peekObject();
103        if (o != loggerName) {
104            addWarn("The object [" + o + "] on the top the of the stack is not the data pushed earlier");
105        } else {
106            mic.popObject();
107        }
108    }
109
110}