001/*
002 * VM-Operator
003 * Copyright (C) 2023 Michael N. Lipp
004 * 
005 * This program is free software: you can redistribute it and/or modify
006 * it under the terms of the GNU Affero General Public License as
007 * published by the Free Software Foundation, either version 3 of the
008 * License, or (at your option) any later version.
009 *
010 * This program is distributed in the hope that it will be useful,
011 * but WITHOUT ANY WARRANTY; without even the implied warranty of
012 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
013 * GNU Affero General Public License for more details.
014 *
015 * You should have received a copy of the GNU Affero General Public License
016 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
017 */
018
019package org.jdrupes.vmoperator.util;
020
021import freemarker.template.DefaultObjectWrapper;
022import freemarker.template.TemplateModel;
023import freemarker.template.TemplateModelException;
024import freemarker.template.Version;
025
026/**
027 * Provides an object wrapper that handles {@link Dto}s.
028 */
029public class ExtendedObjectWrapper extends DefaultObjectWrapper {
030
031    /**
032     * Instantiates a new extended object wrapper.
033     *
034     * @param incompatibleImprovements the incompatible improvements
035     */
036    public ExtendedObjectWrapper(Version incompatibleImprovements) {
037        super(incompatibleImprovements);
038    }
039
040    @Override
041    protected TemplateModel handleUnknownType(final Object obj)
042            throws TemplateModelException {
043        if (obj instanceof Dto dto) {
044            return new DtoTemplateModel(this, dto);
045        }
046        return super.handleUnknownType(obj);
047    }
048}