﻿/*
 * 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.
 */

namespace DotPulsar.Internal.Extensions
{
    using System;
    using System.Text;
    using Metadata = PulsarApi.MessageMetadata;

    public static class MessageMetadataExtensions
    {
        // Deliver at
        public static DateTime GetDeliverAtTimeAsDateTime(this Metadata metadata)
            => metadata.GetDeliverAtTimeAsDateTimeOffset().UtcDateTime;

        public static void SetDeliverAtTime(this Metadata metadata, DateTime timestamp)
            => metadata.SetDeliverAtTime(new DateTimeOffset(timestamp));

        public static DateTimeOffset GetDeliverAtTimeAsDateTimeOffset(this Metadata metadata)
            => DateTimeOffset.FromUnixTimeMilliseconds(metadata.DeliverAtTime);

        public static void SetDeliverAtTime(this Metadata metadata, DateTimeOffset timestamp)
            => metadata.DeliverAtTime = timestamp.ToUnixTimeMilliseconds();

        // Event time
        public static DateTime GetEventTimeAsDateTime(this Metadata metadata)
            => metadata.GetEventTimeAsDateTimeOffset().UtcDateTime;

        public static void SetEventTime(this Metadata metadata, DateTime timestamp)
            => metadata.SetEventTime(new DateTimeOffset(timestamp));

        public static DateTimeOffset GetEventTimeAsDateTimeOffset(this Metadata metadata)
            => DateTimeOffset.FromUnixTimeMilliseconds((long) metadata.EventTime);

        public static void SetEventTime(this Metadata metadata, DateTimeOffset timestamp)
            => metadata.EventTime = (ulong) timestamp.ToUnixTimeMilliseconds();

        // Key
        public static byte[]? GetKeyAsBytes(this Metadata metadata)
        {
            if (metadata.PartitionKey is null)
                return null;

            if (metadata.PartitionKeyB64Encoded)
                return Convert.FromBase64String(metadata.PartitionKey);

            return Encoding.UTF8.GetBytes(metadata.PartitionKey);
        }

        public static void SetKey(this Metadata metadata, string? key)
        {
            metadata.PartitionKey = key;
            metadata.PartitionKeyB64Encoded = false;
        }

        public static void SetKey(this Metadata metadata, byte[]? key)
        {
            if (key is null)
                return;

            metadata.PartitionKey = Convert.ToBase64String(key);
            metadata.PartitionKeyB64Encoded = true;
        }
    }
}