using System;
using System.Collections.Generic;

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

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

        public DataColumnShort(string key, string header, int initialRowCount = 0)
            : base(key, header)
        {
            for (int i = 0; i < initialRowCount; i++)
                Values.Add(null);
        }

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

            return string.IsNullOrWhiteSpace(ValueFormat)
                ? value.Value.ToString()
                : value.Value.ToString(ValueFormat);
        }

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

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

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

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

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

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