using System;
using System.Collections.Generic;

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

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

        public DataColumnTimeOnly(string key, string header, int initialRowCount = 0)
            : base(key, header)
        {
            ValueFormat = "HH:mm:ss";

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

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

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

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

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

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

            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 TimeOnly timeOnlyValue)
            {
                Values[rowIndex] = timeOnlyValue;
                return;
            }

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

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