com.jjlharrison.jollyday.util.XMLUtil.java Source code

Introduction

Here is the source code for com.jjlharrison.jollyday.util.XMLUtil.java

Source

/**
 * Copyright 2010 Sven Diedrichsen
 *
 * 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.
 *
 * @author sven
 * @version $Id: $
 */
package com.jjlharrison.jollyday.util;

import java.io.IOException;
import java.io.InputStream;
import java.util.logging.Logger;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;

import com.jjlharrison.jollyday.holidaytype.LocalizedHolidayType;
import org.joda.time.DateTimeConstants;

import com.jjlharrison.jollyday.HolidayType;
import com.jjlharrison.config.Configuration;
import com.jjlharrison.config.Month;
import com.jjlharrison.config.ObjectFactory;
import com.jjlharrison.config.Weekday;

public class XMLUtil {

    /**
     * the package name to search for the generated java classes.
     */
    public static final String PACKAGE = "com.jjlharrison.config";

    private static Logger LOG = Logger.getLogger(XMLUtil.class.getName());

    private JAXBContextCreator contextCreator = new JAXBContextCreator();
    private ClassLoadingUtil classLoadingUtil = new ClassLoadingUtil();

    /**
     * Unmarshalls the configuration from the stream. Uses <code>JAXB</code> for
     * this.
     *
     * @param stream
     *            a {@link java.io.InputStream} object.
     * @return The unmarshalled configuration.
     * @throws java.io.IOException
     *             Could not close the provided stream.
     */
    public Configuration unmarshallConfiguration(InputStream stream) throws IOException {
        if (stream == null) {
            throw new IllegalArgumentException("Stream is NULL. Cannot read XML.");
        }
        try {
            JAXBContext ctx = null;
            try {
                ctx = contextCreator.create(XMLUtil.PACKAGE, classLoadingUtil.getClassloader());
            } catch (JAXBException e) {
                LOG.warning(
                        "Could not create JAXB context using the current threads context classloader. Defaulting to ObjectFactory classloader.");
                ctx = null;
            }
            if (ctx == null) {
                ctx = contextCreator.create(XMLUtil.PACKAGE, ObjectFactory.class.getClassLoader());
            }
            Unmarshaller um = ctx.createUnmarshaller();
            @SuppressWarnings("unchecked")
            JAXBElement<Configuration> el = (JAXBElement<Configuration>) um.unmarshal(stream);
            return el.getValue();
        } catch (JAXBException ue) {
            throw new IllegalStateException("Cannot parse holidays XML file.", ue);
        } finally {
            stream.close();
        }
    }

    /**
     * Returns the <code>DateTimeConstants</code> value for the given weekday.
     *
     * @param weekday
     *            a {@link com.jjlharrison.config.Weekday} object.
     * @return DateTimeConstants value.
     */
    public final int getWeekday(Weekday weekday) {
        switch (weekday) {
        case MONDAY:
            return DateTimeConstants.MONDAY;
        case TUESDAY:
            return DateTimeConstants.TUESDAY;
        case WEDNESDAY:
            return DateTimeConstants.WEDNESDAY;
        case THURSDAY:
            return DateTimeConstants.THURSDAY;
        case FRIDAY:
            return DateTimeConstants.FRIDAY;
        case SATURDAY:
            return DateTimeConstants.SATURDAY;
        case SUNDAY:
            return DateTimeConstants.SUNDAY;
        default:
            throw new IllegalArgumentException("Unknown weekday " + weekday);
        }
    }

    /**
     * Returns the <code>DateTimeConstants</code> value for the given month.
     *
     * @param month
     *            a {@link com.jjlharrison.config.Month} object.
     * @return DateTimeConstants value.
     */
    public int getMonth(Month month) {
        switch (month) {
        case JANUARY:
            return DateTimeConstants.JANUARY;
        case FEBRUARY:
            return DateTimeConstants.FEBRUARY;
        case MARCH:
            return DateTimeConstants.MARCH;
        case APRIL:
            return DateTimeConstants.APRIL;
        case MAY:
            return DateTimeConstants.MAY;
        case JUNE:
            return DateTimeConstants.JUNE;
        case JULY:
            return DateTimeConstants.JULY;
        case AUGUST:
            return DateTimeConstants.AUGUST;
        case SEPTEMBER:
            return DateTimeConstants.SEPTEMBER;
        case OCTOBER:
            return DateTimeConstants.OCTOBER;
        case NOVEMBER:
            return DateTimeConstants.NOVEMBER;
        case DECEMBER:
            return DateTimeConstants.DECEMBER;
        default:
            throw new IllegalArgumentException("Unknown month " + month);
        }
    }

    /**
     * Gets the type.
     *
     * @param type
     *            the type of holiday in the config
     * @return the type of holiday
     */
    public HolidayType getType(com.jjlharrison.config.HolidayType type) {
        switch (type) {
        case OFFICIAL_HOLIDAY:
            return LocalizedHolidayType.OFFICIAL_HOLIDAY;
        case UNOFFICIAL_HOLIDAY:
            return LocalizedHolidayType.UNOFFICIAL_HOLIDAY;
        default:
            throw new IllegalArgumentException("Unknown type " + type);
        }
    }

    public class JAXBContextCreator {
        public JAXBContext create(String packageName, ClassLoader classLoader) throws JAXBException {
            return JAXBContext.newInstance(packageName, classLoader);
        }
    }

}