using System;

using System.Collections.Generic;

using System.Linq;

using ArchUnitNET.Domain;

using ArchUnitNET.Domain.Extensions;

namespace ArchUnitNET.Fluent

{

internal class SystemTypeObjectProvider<T> : ISizedObjectProvider<T>

where T : IType

{

private readonly List<Type> _types;

public SystemTypeObjectProvider(IEnumerable<Type> types)

{

_types = types.ToList();

Description = string.Join(" or ", _types.Select(type => $"\"{type.FullName}\""));

}

public string Description { get; }

public int Count => _types.Count;

public IEnumerable<T> GetObjects(Architecture architecture)

{

return _types

.Select(architecture.GetITypeOfType)

.Select(

(type) =>

{

if (!(type is T result))

{

throw new ArgumentException($"Type {type} is not of type {typeof(T)}");

}

return result;

}

);

}

public string FormatDescription(

string emptyDescription,

string singleDescription,

string multipleDescription

)

{

switch (Count)

{

case 0:

return emptyDescription;

case 1:

return $"{singleDescription} {Description}";

}

return $"{multipleDescription} {Description}";

}

private bool Equals(SystemTypeObjectProvider<T> other)

{

return _types.SequenceEqual(other._types);

}

public override bool Equals(object obj)

{

if (ReferenceEquals(null, obj))

{

return false;

}

if (ReferenceEquals(this, obj))

{

return true;

}

return obj.GetType() == GetType() && Equals((SystemTypeObjectProvider<T>)obj);

}

public override int GetHashCode()

{

return _types != null

? _types.Aggregate(

0,

(current, type) => (current * 397) ^ (type?.GetHashCode() ?? 0)

)

: 0;

}

}

}