using System;
using System.Runtime.InteropServices;

[assembly: TypeMapAssemblyTarget<MyTypeMapGroup>("TypeMapLib")]

[assembly: TypeMap<MyTypeMapGroup>("LocalEntry", typeof(LocalTarget))]
[assembly: TypeMapAssociation<MyTypeMapGroup>(typeof(LocalSource), typeof(LocalProxy))]

public class LocalTarget;
public class LocalSource;
public class LocalProxy;

class Program
{
    static int Main()
    {
        int failures = 0;

        // Root the types explicitly so the trimmer keeps them.
        // This ensures any failure is purely about the TypeMap attribute discovery,
        // not about the types themselves being trimmed.
        GC.KeepAlive(typeof(LocalTarget));
        GC.KeepAlive(typeof(LocalSource));
        GC.KeepAlive(typeof(LocalProxy));
        GC.KeepAlive(typeof(GreetingTarget));
        GC.KeepAlive(typeof(FarewellTarget));
        GC.KeepAlive(typeof(FarewellTrimTarget));
        GC.KeepAlive(typeof(GreetingSource));
        GC.KeepAlive(typeof(GreetingProxy));

        var externalMap = TypeMapping.GetOrCreateExternalTypeMapping<MyTypeMapGroup>();

        failures += Check(externalMap.TryGetValue("LocalEntry", out _),
            "External map: 'LocalEntry' (same-assembly)");

        failures += Check(externalMap.TryGetValue("Greeting", out _),
            "External map: 'Greeting' from TypeMapLib (CROSS-ASSEMBLY — fails without PR fix)");

        failures += Check(externalMap.TryGetValue("Farewell", out _),
            "External map: 'Farewell' from TypeMapLib (CROSS-ASSEMBLY — fails without PR fix)");

        var proxyMap = TypeMapping.GetOrCreateProxyTypeMapping<MyTypeMapGroup>();

        _ = new LocalSource();
        failures += Check(proxyMap.TryGetValue(typeof(LocalSource), out _),
            "Proxy map: LocalSource->LocalProxy (same-assembly)");

        _ = new GreetingSource();
        failures += Check(proxyMap.TryGetValue(typeof(GreetingSource), out _),
            "Proxy map: GreetingSource->GreetingProxy from TypeMapLib (CROSS-ASSEMBLY — fails without PR fix)");

        if (failures == 0)
            Console.WriteLine("\nPASS: All cross-assembly TypeMap entries preserved after trimming.");
        else
            Console.Error.WriteLine($"\nFAILED: {failures} check(s) failed — trimmer removed cross-assembly TypeMap data.");

        return failures;
    }

    static int Check(bool condition, string description)
    {
        if (condition)
        {
            Console.WriteLine($"  OK:   {description}");
            return 0;
        }
        else
        {
            Console.Error.WriteLine($"  FAIL: {description}");
            return 1;
        }
    }
}