Duration.java

/*
 * Copyright (c) 2010, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 * 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.
 */

package org.wso2.carbon.humantask.core.utils;

import java.math.BigDecimal;
import java.util.Calendar;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * XML Schema Duration
 */
public class Duration {
    // TODO: Fix this with Java.time.Duration from java 8.
    private static final Pattern PATTERN =
            Pattern.compile("P(\\d+Y)?(\\d+M)?(\\d+D)?(T(\\d+H)?(\\d+M)?((\\d+\\.?\\d*|\\.\\d+)S)?)?");

    private static final int YEAR_MG = 1;
    private static final int MONTH_MG = 2;
    private static final int DAY_MG = 3;
    private static final int HOUR_MG = 5;
    private static final int MIN_MG = 6;
    private static final int SEC_MG = 8;

    private int years;
    private int months;
    private int days;
    private int hours;
    private int minutes;
    private BigDecimal seconds;

    private boolean addition = true;

    public Duration(String duration) {
        String tDuration = duration;
        if (tDuration.startsWith("-")) {
            addition = false;
            tDuration = tDuration.substring(1);
        }
        Matcher m = PATTERN.matcher(tDuration);
        if (m.matches()) {
            years = parseInt(m.group(YEAR_MG));
            months = parseInt(m.group(MONTH_MG));
            days = parseInt(m.group(DAY_MG));
            hours = parseInt(m.group(HOUR_MG));
            minutes = parseInt(m.group(MIN_MG));
            seconds = m.group(SEC_MG) == null
                    ? null : new BigDecimal(m.group(SEC_MG));
        } else {
            throw new IllegalArgumentException("Invalid duration: " + tDuration);
        }
    }

    public void addTo(Calendar calendar) {

        calendar.add(Calendar.YEAR, addition ? years : 0 - years);
        calendar.add(Calendar.MONTH, addition ? months : 0 - months);
        calendar.add(Calendar.DAY_OF_MONTH, addition ? days : 0 - days);
        calendar.add(Calendar.HOUR, addition ? hours : 0 - hours);
        calendar.add(Calendar.MINUTE, addition ? minutes : 0 - minutes);
        calendar.add(Calendar.SECOND, (seconds == null)
                ? 0 : addition ? seconds.intValue() : 0 - seconds.intValue());

        if (seconds != null) {
            BigDecimal fraction = seconds.subtract(seconds.setScale(0, BigDecimal.ROUND_DOWN));
            int millisec = fraction.movePointRight(3).intValue();
            calendar.add(Calendar.MILLISECOND, addition ? millisec : 0 - millisec);
        }
    }

    private static int parseInt(String value) {
        if (value == null) {
            return 0;
        } else {
            return Integer.parseInt(value.substring(0, value.length() - 1));
        }
    }

    public static boolean isValidExpression(String exp) {
        String tExp = exp;
        if (tExp.startsWith("-")) {
            tExp = tExp.substring(1);
        }
        Matcher m = PATTERN.matcher(tExp);
        return m.matches();
    }
}