ok now it works

2024-10-22 21:11:33 +07:00
parent 8f7fef8857
commit bf3eaedf62
46 changed files with 1150 additions and 244 deletions
--- a/CringePlugins/Resolver/PackageResolver.cs
+++ b/CringePlugins/Resolver/PackageResolver.cs
@@ -0,0 +1,148 @@
+using System.Collections.Immutable;
+using System.IO.Compression;
+using NuGet;
+using NuGet.Frameworks;
+using NuGet.Models;
+using NuGet.Versioning;
+
+namespace CringePlugins.Resolver;
+
+public class PackageResolver(NuGetFramework runtimeFramework, ImmutableArray<PackageReference> references, PackageSourceMapping packageSources)
+{
+    public async Task<ImmutableHashSet<ResolvedPackage>> ResolveAsync()
+    {
+        var order = 0;
+        var packages = new SortedSet<Package>();
+
+        foreach (var reference in references)
+        {
+            var client = await packageSources.GetClientAsync(reference.Id);
+
+            var registrationRoot = await client.GetPackageRegistrationRootAsync(reference.Id);
+
+            NuGetVersion? version = null;
+            foreach (var page in registrationRoot.Items)
+            {
+                version = reference.Range.FindBestMatch(page.Items.Where(b => b.CatalogEntry.PackageTypes is [{ Name: "CringePlugin" }]).Select(b => b.CatalogEntry.Version));
+                if (version is not null)
+                    break;
+            }
+            
+            if (version is null)
+                throw new Exception($"Unable to find version for package {reference.Id}");
+            
+            var package = new Package(order, reference.Id, version, []); // todo resolve dependencies
+            
+            if (packages.Add(package))
+                continue;
+            
+            if (!packages.TryGetValue(package, out var existingPackage))
+                throw new Exception($"Duplicate package {package.Id}");
+            
+            if (package.Version < existingPackage.Version)
+                throw new Exception($"Package reference {package.Id} has lower version {package.Version} than already resolved {existingPackage.Version}");
+            
+            if (package.Version == existingPackage.Version)
+                continue;
+
+            packages.Remove(existingPackage);
+            packages.Add(package with
+            {
+                Order = ++order
+            });
+        }
+
+        var set = ImmutableHashSet<ResolvedPackage>.Empty.ToBuilder();
+        foreach (var package in packages)
+        {
+            var client = await packageSources.GetClientAsync(package.Id);
+            
+            var (catalogEntryUrl, catalogEntry) = await client.GetPackageRegistrationAsync(package.Id, package.Version);
+
+            catalogEntry ??= await client.GetPackageCatalogEntryAsync(catalogEntryUrl);
+
+            var nearestGroup = NuGetFrameworkUtility.GetNearest(catalogEntry.DependencyGroups, runtimeFramework,
+                g => g.TargetFramework);
+            
+            if (nearestGroup is null)
+                throw new Exception($"Unable to find compatible dependency group for package {package.Id}");
+
+            set.Add(new RemotePackage(package, nearestGroup.TargetFramework, client, catalogEntry));
+        }
+
+        return set.ToImmutable();
+    }
+
+    public async Task<ImmutableHashSet<CachedPackage>> DownloadPackagesAsync(DirectoryInfo baseDirectory,
+        IReadOnlySet<ResolvedPackage> resolvedPackages, IProgress<float>? progress = null)
+    {
+        var packages = ImmutableHashSet<CachedPackage>.Empty.ToBuilder();
+        
+        var i = 0f;
+        foreach (var package in resolvedPackages)
+        {
+            switch (package)
+            {
+                case RemotePackage remotePackage:
+                {
+                    var dir = new DirectoryInfo(Path.Join(baseDirectory.FullName, package.Package.Id, package.Package.Version.ToString()));
+                    if (!dir.Exists)
+                    {
+                        dir.Create();
+
+                        await using var stream = await remotePackage.Client.GetPackageContentStreamAsync(remotePackage.Package.Id, remotePackage.Package.Version);
+                        using var memStream = new MemoryStream();
+                        await stream.CopyToAsync(memStream);
+                        memStream.Position = 0;
+                        using var archive = new ZipArchive(memStream, ZipArchiveMode.Read);
+                        archive.ExtractToDirectory(dir.FullName);
+                    }
+                
+                    packages.Add(new CachedPackage(package.Package, package.ResolvedFramework, dir, package.Entry));
+                    break;
+                }
+                case CachedPackage cachedPackage:
+                    packages.Add(cachedPackage);
+                    break;
+            }
+
+            progress?.Report(i++ / resolvedPackages.Count);
+        }
+        
+        return packages.ToImmutable();
+    }
+}
+
+public record CachedPackage(Package Package, NuGetFramework ResolvedFramework, DirectoryInfo Directory, CatalogEntry Entry) : ResolvedPackage(Package, ResolvedFramework, Entry);
+public record RemotePackage(Package Package, NuGetFramework ResolvedFramework, NuGetClient Client, CatalogEntry Entry) : ResolvedPackage(Package, ResolvedFramework, Entry);
+
+public abstract record ResolvedPackage(Package Package, NuGetFramework ResolvedFramework, CatalogEntry Entry);
+
+public record Package(int Order, string Id, NuGetVersion Version, ImmutableArray<string> Dependencies) : IComparable<Package>, IComparable
+{
+    public int CompareTo(Package? other)
+    {
+        if (ReferenceEquals(this, other)) return 0;
+        if (other is null) return 1;
+        var orderComparison = Order.CompareTo(other.Order);
+        if (orderComparison != 0) return orderComparison;
+        return string.Compare(Id, other.Id, StringComparison.OrdinalIgnoreCase);
+    }
+
+    public int CompareTo(object? obj)
+    {
+        if (obj is null) return 1;
+        if (ReferenceEquals(this, obj)) return 0;
+        return obj is Package other ? CompareTo(other) : throw new ArgumentException($"Object must be of type {nameof(Package)}");
+    }
+
+    public override int GetHashCode() => StringComparer.OrdinalIgnoreCase.GetHashCode(Id);
+
+    public virtual bool Equals(Package? other)
+    {
+        if (other is null) return false;
+        return Id.Equals(other.Id, StringComparison.OrdinalIgnoreCase);
+    }
+}
+
+public record PackageReference(string Id, VersionRange Range);