using System;
using System.Collections.Generic;

namespace Sonex.Client.Data
{
    public sealed class DataColumnDateTime : DataColumnBase
    {
        public List<DateTime?> Values { get; } = new();

        public override Type DataType => typeof(DateTime);
        public override int RowCount => Values.Count;

        public DataColumnDateTime(string key, string header, int initialRowCount = 0)
            : base(key, header)
        {
            ValueFormat = "yyyy-MM-dd HH:mm:ss";

            for (int i = 0; i < initialRowCount; i++)
                Values.Add(null);
        }

        protected override string? FormatValue(int rowIndex)
        {
            DateTime? value = Values[rowIndex];
            if (value is null)
                return null;

            return value.Value.ToString(
                string.IsNullOrWhiteSpace(ValueFormat)
                    ? "yyyy-MM-dd HH:mm:ss"
                    : ValueFormat);
        }

        public override void SetValue(int rowIndex, string? value)
        {
            ValidateRowIndex(rowIndex);

            if (string.IsNullOrWhiteSpace(value))
            {
                Values[rowIndex] = null;
                return;
            }

            if (!DateTime.TryParse(value, out DateTime result))
                throw new FormatException($"Value '{value}' is not a valid DateTime.");

            Values[rowIndex] = result;
        }

        public override object? GetRawValue(int rowIndex)
        {
            ValidateRowIndex(rowIndex);
            return Values[rowIndex];
        }

        public override void SetRawValue(int rowIndex, object? value)
        {
            ValidateRowIndex(rowIndex);

            if (value is null)
            {
                Values[rowIndex] = null;
                return;
            }

            if (value is DateTime dateTimeValue)
            {
                Values[rowIndex] = dateTimeValue;
                return;
            }

            throw new InvalidCastException($"Column '{Key}' expects DateTime.");
        }

        public override void AddEmpty()
        {
            Values.Add(null);
        }
    }
}