/*
 * Copyright 2010, Haiku, Inc. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Adrien Destugues <pulkomandy@gmail.com>
 *              Oliver Tappe <zooey@hirschkaefer.de>
 */


#include <unicode/uversion.h>
#include <TimeUnitFormat.h>

#include <new>

#include <unicode/format.h>
#include <unicode/locid.h>
#include <unicode/tmutfmt.h>
#include <unicode/utypes.h>
#include <ICUWrapper.h>

#include <Language.h>
#include <Locale.h>
#include <LocaleRoster.h>


U_NAMESPACE_USE


// maps our unit element to the corresponding ICU unit
static const TimeUnit::UTimeUnitFields skUnitMap[] = {
        TimeUnit::UTIMEUNIT_YEAR,
        TimeUnit::UTIMEUNIT_MONTH,
        TimeUnit::UTIMEUNIT_WEEK,
        TimeUnit::UTIMEUNIT_DAY,
        TimeUnit::UTIMEUNIT_HOUR,
        TimeUnit::UTIMEUNIT_MINUTE,
        TimeUnit::UTIMEUNIT_SECOND,
};

//maps our unit style to the corresponding ICU unit
static const UTimeUnitFormatStyle kTimeUnitStyleToICU[] = {
        UTMUTFMT_ABBREVIATED_STYLE,
        UTMUTFMT_FULL_STYLE,
};


BTimeUnitFormat::BTimeUnitFormat(const time_unit_style style)
        : Inherited()
{
        Locale icuLocale(fLanguage.Code());
        UErrorCode icuStatus = U_ZERO_ERROR;
        if (style != B_TIME_UNIT_ABBREVIATED && style != B_TIME_UNIT_FULL) {
                fFormatter = NULL;
                fInitStatus = B_BAD_VALUE;
                return;
        }

        fFormatter = new TimeUnitFormat(icuLocale, kTimeUnitStyleToICU[style],
                icuStatus);
        if (fFormatter == NULL) {
                fInitStatus = B_NO_MEMORY;
                return;
        }

        if (!U_SUCCESS(icuStatus))
                fInitStatus = B_ERROR;
}


BTimeUnitFormat::BTimeUnitFormat(const BLanguage& language,
        const BFormattingConventions& conventions,
        const time_unit_style style)
        : Inherited(language, conventions)
{
        Locale icuLocale(fLanguage.Code());
        UErrorCode icuStatus = U_ZERO_ERROR;
        if (style != B_TIME_UNIT_ABBREVIATED && style != B_TIME_UNIT_FULL) {
                fFormatter = NULL;
                fInitStatus = B_BAD_VALUE;
                return;
        }

        fFormatter = new TimeUnitFormat(icuLocale, kTimeUnitStyleToICU[style],
                icuStatus);
        if (fFormatter == NULL) {
                fInitStatus = B_NO_MEMORY;
                return;
        }

        if (!U_SUCCESS(icuStatus))
                fInitStatus = B_ERROR;
}


BTimeUnitFormat::BTimeUnitFormat(const BTimeUnitFormat& other)
        :
        Inherited(other),
        fFormatter(other.fFormatter != NULL
                ? new TimeUnitFormat(*other.fFormatter) : NULL)
{
        if (fFormatter == NULL && other.fFormatter != NULL)
                fInitStatus = B_NO_MEMORY;
}


BTimeUnitFormat::~BTimeUnitFormat()
{
        delete fFormatter;
}


status_t
BTimeUnitFormat::Format(BString& buffer, const int32 value,
        const time_unit_element unit) const
{
        if (unit < 0 || unit > B_TIME_UNIT_LAST)
                return B_BAD_VALUE;

        if (fFormatter == NULL)
                return B_NO_INIT;

        UErrorCode icuStatus = U_ZERO_ERROR;
        TimeUnitAmount* timeUnitAmount
                = new TimeUnitAmount((double)value, skUnitMap[unit], icuStatus);
        if (timeUnitAmount == NULL)
                return B_NO_MEMORY;
        if (!U_SUCCESS(icuStatus)) {
                delete timeUnitAmount;
                return B_ERROR;
        }

        Formattable formattable;
        formattable.adoptObject(timeUnitAmount);
        FieldPosition pos(FieldPosition::DONT_CARE);
        UnicodeString unicodeResult;
        fFormatter->format(formattable, unicodeResult, pos, icuStatus);
        if (!U_SUCCESS(icuStatus))
                return B_ERROR;

        BStringByteSink byteSink(&buffer);
        unicodeResult.toUTF8(byteSink);

        return B_OK;
}