Hack23/cia

/*
 * Copyright 2010-2021 James Pether Sörling
 *
 * 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.
 *
 *    $Id$
 *  $HeadURL$
*/
package com.hack23.cia.web.impl.ui.application;

import com.vaadin.server.UICreateEvent;
import com.vaadin.server.VaadinSession;
import com.vaadin.spring.internal.SpringViewDisplayRegistrationBean;
import com.vaadin.spring.internal.UIID;
import com.vaadin.spring.server.SpringUIProvider;
import com.vaadin.ui.UI;
import com.vaadin.util.CurrentInstance;

/**
 * The Class CustomSpringUIProvider.
 */
public class CustomSpringUIProvider extends SpringUIProvider {

    /** The Constant serialVersionUID. */
    private static final long serialVersionUID = 1L;

    /** The spring view display registration bean. */
    private final SpringViewDisplayRegistrationBean springViewDisplayRegistrationBean = new SpringViewDisplayRegistrationBean();

    /**
     * Instantiates a new custom spring UI provider.
     *
     * @param vaadinSession the vaadin session
     */
    public CustomSpringUIProvider(final VaadinSession vaadinSession) {
        super(vaadinSession);
    }

    @Override
    public UI createInstance(final UICreateEvent event) {
        final Class<UIID> key = UIID.class;
        final UIID identifier = new UIID(event);
        CurrentInstance.set(key, identifier);
        try {
            logger.debug("Creating a new UI bean of class [{}] with identifier [{}]",
                    event.getUIClass().getCanonicalName(), identifier);
            final UI ui = getWebApplicationContext().getBean(event.getUIClass());

            getSpringViewDisplayRegistrationBean().setBeanClass(event.getUIClass());

            configureNavigator(ui);
            return ui;
        } finally {
            CurrentInstance.set(key, null);
        }
    }

    @Override
    protected Object findSpringViewDisplay(final UI ui) {
        return getSpringViewDisplayRegistrationBean().getSpringViewDisplay(getWebApplicationContext());
    }

    /**
     * Gets the spring view display registration bean.
     *
     * @return the spring view display registration bean
     */
    private SpringViewDisplayRegistrationBean getSpringViewDisplayRegistrationBean() {
        return springViewDisplayRegistrationBean;

    }
}