package com.vaadin.tests.design;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;

import org.junit.Test;

import com.vaadin.ui.Component;
import com.vaadin.ui.Label;
import com.vaadin.ui.declarative.Design;
import com.vaadin.ui.declarative.Design.ComponentFactory;
import com.vaadin.ui.declarative.Design.ComponentMapper;
import com.vaadin.ui.declarative.DesignContext;

public class ComponentMapperTest {
    private static final ComponentMapper defaultMapper = Design
            .getComponentMapper();

    private static final ThreadLocal<ComponentMapper> currentMapper = new ThreadLocal<>();

    static {
        Design.setComponentMapper(new ComponentMapper() {
            @Override
            public Component tagToComponent(String tag,
                    ComponentFactory componentFactory, DesignContext context) {
                return getActualMapper().tagToComponent(tag, componentFactory,
                        context);
            }

            @Override
            public String componentToTag(Component component,
                    DesignContext context) {
                return getActualMapper().componentToTag(component, context);
            }

            private ComponentMapper getActualMapper() {
                ComponentMapper mapper = currentMapper.get();
                if (mapper == null) {
                    mapper = defaultMapper;
                }
                return mapper;
            }
        });
    }

    private final class CustomComponentMapper
            extends Design.DefaultComponentMapper {
        @Override
        public Component tagToComponent(String tag,
                ComponentFactory componentFactory, DesignContext context) {
            if (tag.startsWith("custom-")) {
                ComponentWithCustomTagName component = (ComponentWithCustomTagName) componentFactory
                        .createComponent(
                                ComponentWithCustomTagName.class.getName(),
                                context);
                component.tagName = tag;
                return component;
            } else {
                return super.tagToComponent(tag, componentFactory, context);
            }
        }

        @Override
        public String componentToTag(Component component,
                DesignContext context) {
            if (component instanceof ComponentWithCustomTagName) {
                ComponentWithCustomTagName withCustomTagName = (ComponentWithCustomTagName) component;
                return withCustomTagName.tagName;
            } else {
                return super.componentToTag(component, context);
            }
        }
    }

    public static class ComponentWithCustomTagName extends Label {
        private String tagName;
    }

    @Test
    public void testCustomComponentMapperRead() {
        currentMapper.set(new CustomComponentMapper());

        Component component = Design
                .read(new ByteArrayInputStream("<custom-foobar />".getBytes()));

        assertTrue(
                "<custom-foobar> should resolve "
                        + ComponentWithCustomTagName.class.getSimpleName(),
                component instanceof ComponentWithCustomTagName);
        assertEquals("custom-foobar",
                ((ComponentWithCustomTagName) component).tagName);
    }

    @Test
    public void testCustomComponentMapperWrite() throws IOException {
        currentMapper.set(new CustomComponentMapper());

        ComponentWithCustomTagName component = new ComponentWithCustomTagName();
        component.tagName = "custom-special";

        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        Design.write(component, bos);
        String writtenDesign = new String(bos.toByteArray());

        assertTrue(
                "Written design should contain \"<custom-special\", but instead got "
                        + writtenDesign,
                writtenDesign.contains("<custom-special"));
    }

    public void cleanup() {
        currentMapper.remove();
    }
}