net.fortuna.ical4j.extensions.data.ExtendedContentHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ical4j-extensions Show documentation

Support for additional non-standard iCalendar objects

The newest version!

package net.fortuna.ical4j.extensions.data;

import net.fortuna.ical4j.data.ContentHandlerContext;
import net.fortuna.ical4j.data.DefaultContentHandler;
import net.fortuna.ical4j.extensions.model.property.WrTimezone;
import net.fortuna.ical4j.model.Calendar;
import net.fortuna.ical4j.model.Property;
import net.fortuna.ical4j.model.TimeZoneRegistry;

import java.util.Optional;
import java.util.function.Consumer;

public class ExtendedContentHandler extends DefaultContentHandler {

    private boolean inTimezone;

    public ExtendedContentHandler(Consumer consumer, TimeZoneRegistry tzRegistry) {
        super(consumer, tzRegistry, new ContentHandlerContext()
                .withPropertyFactorySupplier(new ExtendedPropertyFactorySupplier())
                .withParameterFactorySupplier(new ExtendedParameterFactorySupplier()));
    }

    public ExtendedContentHandler(Consumer consumer, TimeZoneRegistry tzRegistry, ContentHandlerContext context) {
        super(consumer, tzRegistry, context);
    }

    @Override
    public void startComponent(String name) {
        super.startComponent(name);
        if ("VTIMEZONE".equalsIgnoreCase(name)) {
            inTimezone = true;
        }
    }

    @Override
    public void endComponent(String name) {
        super.endComponent(name);
        if ("VTIMEZONE".equalsIgnoreCase(name)) {
            inTimezone = false;
        }
    }

    @Override
    public void startProperty(String name) {
        super.startProperty(name);
        // don't apply default timezone to calendar properties..
        if (getComponentBuilder() != null && !inTimezone) {
            Optional defaultTimezone = calendarProperties.stream().filter(p -> p instanceof WrTimezone).findFirst();
            defaultTimezone.ifPresent(property -> propertyBuilder.defaultTimeZone(((WrTimezone) property).toZoneId()));
        }
    }
}