Normalize mod provider metadata

Author: realdoomsboygaming

Natives/installer/modpack/ModrinthAPI.m

@@ -141,6 +141,29 @@ - (void)saveModManagerMetadataRecords:(NSArray<NSDictionary *> *)records toPath:
     }
 }
 
+- (NSString *)modManagerSourceForDownloadURLs:(NSArray *)downloads {
+    for (NSString *download in downloads) {
+        if (![download isKindOfClass:NSString.class] || download.length == 0) {
+            continue;
+        }
+        NSURL *url = [NSURL URLWithString:download];
+        NSString *host = url.host.lowercaseString ?: @"";
+        NSString *urlString = download.lowercaseString;
+        if ([host containsString:@"modrinth.com"] || [urlString containsString:@"modrinth.com/"]) {
+            return @"modrinth";
+        }
+        if ([host containsString:@"curseforge.com"] ||
+            [host containsString:@"forgecdn.net"] ||
+            [host containsString:@"cursecdn.com"] ||
+            [urlString containsString:@"curseforge.com/"] ||
+            [urlString containsString:@"forgecdn.net/"] ||
+            [urlString containsString:@"cursecdn.com/"]) {
+            return @"curseforge";
+        }
+    }
+    return @"manual";
+}
+
 - (void)downloader:(MinecraftResourceDownloadTask *)downloader submitDownloadTasksFromPackage:(NSString *)packagePath toPath:(NSString *)destPath {
     NSError *error = nil;
     UZKArchive *archive = [[UZKArchive alloc] initWithPath:packagePath error:&error];
@@ -206,9 +229,11 @@ - (void)downloader:(MinecraftResourceDownloadTask *)downloader submitDownloadTas
         }
         NSString *fileName = relativePath.lastPathComponent;
         if ([relativePath hasPrefix:@"mods/"] && [fileName.lowercaseString hasSuffix:@".jar"]) {
+            NSString *source = [self modManagerSourceForDownloadURLs:downloads];
             [modManagerRecords addObject:@{
-                @"source": @"modrinth",
+                @"source": source,
                 @"fileName": fileName,
+                @"downloadUrl": url,
                 @"sha1": sha ?: @"",
                 @"size": [indexFile[@"fileSize"] respondsToSelector:@selector(unsignedLongLongValue)] ? indexFile[@"fileSize"] : @0,
                 @"enabled": @YES

Natives/installer/mods/ModManagerStore.m

@@ -151,6 +151,67 @@ - (BOOL)isSafeModFileName:(NSString *)fileName {
     return [lower hasSuffix:@".jar"];
 }
 
+- (NSString *)normalizedProviderSource:(id)value {
+    if (![value isKindOfClass:NSString.class]) {
+        return nil;
+    }
+    NSString *source = [[value stringByTrimmingCharactersInSet:NSCharacterSet.whitespaceAndNewlineCharacterSet] lowercaseString];
+    if ([source isEqualToString:@"modrinth"] ||
+        [source isEqualToString:@"curseforge"] ||
+        [source isEqualToString:@"manual"]) {
+        return source;
+    }
+    return nil;
+}
+
+- (NSString *)providerSourceForURLString:(id)value {
+    if (![value isKindOfClass:NSString.class] || [value length] == 0) {
+        return nil;
+    }
+    NSURL *url = [NSURL URLWithString:value];
+    NSString *host = url.host.lowercaseString ?: @"";
+    NSString *urlString = [value lowercaseString];
+    if ([host containsString:@"modrinth.com"] || [urlString containsString:@"modrinth.com/"]) {
+        return @"modrinth";
+    }
+    if ([host containsString:@"curseforge.com"] ||
+        [host containsString:@"forgecdn.net"] ||
+        [host containsString:@"cursecdn.com"] ||
+        [urlString containsString:@"curseforge.com/"] ||
+        [urlString containsString:@"forgecdn.net/"] ||
+        [urlString containsString:@"cursecdn.com/"]) {
+        return @"curseforge";
+    }
+    return nil;
+}
+
+- (NSString *)inferredProviderSourceForRecord:(NSDictionary *)record {
+    NSString *source = [self normalizedProviderSource:record[@"source"]];
+    NSString *downloadSource = [self providerSourceForURLString:record[@"downloadUrl"]];
+    NSString *manualSource = [self providerSourceForURLString:record[@"manualUrl"]];
+    NSString *urlSource = downloadSource ?: manualSource;
+    if (urlSource && (!source || [source isEqualToString:@"manual"] || ![urlSource isEqualToString:source])) {
+        return urlSource;
+    }
+    if (source) {
+        return source;
+    }
+
+    NSString *iconSource = [self providerSourceForURLString:record[@"iconUrl"]];
+    if (iconSource) {
+        return iconSource;
+    }
+
+    if (record[@"fileId"] || [record[@"projectId"] isKindOfClass:NSNumber.class]) {
+        return @"curseforge";
+    }
+    if ([record[@"versionId"] isKindOfClass:NSString.class] ||
+        [record[@"projectId"] isKindOfClass:NSString.class]) {
+        return @"modrinth";
+    }
+    return @"manual";
+}
+
 - (NSArray<NSString *> *)modFileNames {
     NSArray *contents = [NSFileManager.defaultManager contentsOfDirectoryAtPath:self.modsDir error:nil] ?: @[];
     NSMutableArray *files = [NSMutableArray new];
@@ -177,9 +238,7 @@ - (BOOL)isSafeModFileName:(NSString *)fileName {
         if (mod[@"title"] == nil) {
             mod[@"title"] = fileName.stringByDeletingPathExtension;
         }
-        if (mod[@"source"] == nil) {
-            mod[@"source"] = @"manual";
-        }
+        mod[@"source"] = [self inferredProviderSourceForRecord:mod];
         mod[@"fileName"] = fileName;
         mod[@"actualFileName"] = actualFileName;
         mod[@"path"] = [self.modsDir stringByAppendingPathComponent:actualFileName];
@@ -438,7 +497,7 @@ - (NSDictionary *)installPlanForFiles:(NSArray<NSDictionary *> *)files error:(NS
         }
 
         NSString *destination = [self.modsDir stringByAppendingPathComponent:fileName];
-        NSString *source = file[@"source"] ?: @"manual";
+        NSString *source = [self inferredProviderSourceForRecord:file];
         id projectId = file[@"projectId"];
         NSString *existingFileName = [self existingFileNameForSource:source projectId:projectId];
         if (existingFileName.length > 0 && ![existingFileName isEqualToString:fileName]) {
@@ -463,6 +522,7 @@ - (NSDictionary *)installPlanForFiles:(NSArray<NSDictionary *> *)files error:(NS
                 record[key] = value;
             }
         }
+        record[@"source"] = source;
         record[@"installedAt"] = installedAt;
         record[@"enabled"] = @YES;
         [records addObject:record];