com.hp.autonomy.types.idol.content.ScheduleAdapter.java Source code

Introduction

Here is the source code for com.hp.autonomy.types.idol.content.ScheduleAdapter.java

Source

/*
 * Copyright 2016 Hewlett Packard Enterprise Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.types.idol.content;

import biweekly.Biweekly;
import biweekly.ICalendar;
import biweekly.component.VEvent;
import biweekly.property.DateEnd;
import biweekly.property.DateStart;
import biweekly.util.ICalDate;
import biweekly.util.Recurrence;
import org.apache.commons.lang3.StringUtils;
import org.joda.time.DateTime;

import javax.xml.bind.annotation.adapters.XmlAdapter;
import java.util.regex.Pattern;

@SuppressWarnings("WeakerAccess")
public class ScheduleAdapter extends XmlAdapter<String, Schedule> {
    private static final Pattern UNMARSHAL_PATTERN = Pattern.compile("\\+\\\\\\\\");
    private static final Pattern MARSHAL_PATTERN = Pattern.compile("\r\n");

    @Override
    public Schedule unmarshal(final String scheduleAsString) {
        if (StringUtils.isBlank(scheduleAsString)) {
            return null;
        }

        final String correctCalendar = UNMARSHAL_PATTERN.matcher(scheduleAsString).replaceAll("\r\n");
        final ICalendar iCalendar = Biweekly.parse(correctCalendar).first();

        String productId = null;
        if (iCalendar.getProductId() != null) {
            productId = iCalendar.getProductId().getValue();
        }

        final VEvent event = iCalendar.getEvents().get(0);
        final DateTime startDate = new DateTime(event.getDateStart().getValue().getTime());
        final DateTime endDate = new DateTime(event.getDateEnd().getValue().getTime());

        DateTime until = null;
        Recurrence.Frequency frequency = null;

        if (event.getRecurrenceRule() != null) {
            frequency = event.getRecurrenceRule().getValue().getFrequency();

            if (event.getRecurrenceRule().getValue().getUntil() != null) {
                until = new DateTime(event.getRecurrenceRule().getValue().getUntil().getRawComponents().toDate());
            }
        }

        return new Schedule.Builder().setProductId(productId).setStartDate(startDate).setEndDate(endDate)
                .setUntil(until).setFrequency(frequency).build();
    }

    @Override
    public String marshal(final Schedule schedule) {
        final ICalendar iCalendar = toICalendar(schedule);
        // we need the replace because IDOL is doing it wrong
        // we need 8 slashes because the replacement string is not really a literal
        return MARSHAL_PATTERN.matcher(iCalendar.write()).replaceAll("+\\\\\\\\");
    }

    public ICalendar toICalendar(final Schedule schedule) {
        final ICalendar iCalendar = new ICalendar();

        final VEvent event = new VEvent();
        event.setDateStart(new DateStart(schedule.getStartDate().toDate(), true));
        event.setDateEnd(new DateEnd(schedule.getEndDate().toDate(), true));

        final Recurrence.Frequency frequency = schedule.getFrequency();
        if (frequency != null) {
            final Recurrence.Builder builder = new Recurrence.Builder(frequency);

            final DateTime until = schedule.getUntil();
            if (until != null) {
                builder.until(new ICalDate(until.toDate()));
            }

            event.setRecurrenceRule(builder.build());
        }

        iCalendar.addEvent(event);
        iCalendar.setProductId(schedule.getProductId());

        return iCalendar;
    }
}