using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Reflection;
namespace Python.Runtime.Slots
{
internal static class MpLengthSlot
{
public static bool CanAssign(Type clrType)
{
if (typeof(ICollection).IsAssignableFrom(clrType))
{
return true;
}
if (clrType.GetInterfaces().Any(x => x.IsGenericType && x.GetGenericTypeDefinition() == typeof(ICollection<>)))
{
return true;
}
if (clrType.IsInterface && clrType.IsGenericType && clrType.GetGenericTypeDefinition() == typeof(ICollection<>))
{
return true;
}
return false;
}
///
/// Implements __len__ for classes that implement ICollection
/// (this includes any IList implementer or Array subclass)
///
internal static nint impl(BorrowedReference ob)
{
if (ManagedType.GetManagedObject(ob) is not CLRObject co)
{
Exceptions.RaiseTypeError("invalid object");
return -1;
}
// first look for ICollection implementation directly
if (co.inst is ICollection c)
{
return c.Count;
}
Type clrType = co.inst.GetType();
// now look for things that implement ICollection directly (non-explicitly)
PropertyInfo p = clrType.GetProperty("Count");
if (p != null && clrType.GetInterfaces().Any(x => x.IsGenericType && x.GetGenericTypeDefinition() == typeof(ICollection<>)))
{
return (int)p.GetValue(co.inst, null);
}
// finally look for things that implement the interface explicitly
var iface = clrType.GetInterfaces().FirstOrDefault(x => x.IsGenericType && x.GetGenericTypeDefinition() == typeof(ICollection<>));
if (iface != null)
{
p = iface.GetProperty(nameof(ICollection.Count));
return (int)p.GetValue(co.inst, null);
}
Exceptions.SetError(Exceptions.TypeError, $"object of type '{clrType.Name}' has no len()");
return -1;
}
}
}