cog/Plugins/M3u/M3uContainer.m

//
//  M3uContainer.m
//  M3u
//
//  Created by Zaphod Beeblebrox on 10/8/07.
//  Copyright 2007 __MyCompanyName__. All rights reserved.
//

#import "M3uContainer.h"

#import "Logging.h"

@implementation M3uContainer

+ (NSArray *)fileTypes {
	return @[@"m3u", @"m3u8"];
}

+ (NSArray *)mimeTypes {
	return @[@"audio/x-mpegurl", @"audio/mpegurl"];
}

+ (float)priority {
	return 1.0f;
}

+ (NSURL *)urlForPath:(NSString *)path relativeTo:(NSString *)baseFilename {
	NSRange protocolRange = [path rangeOfString:@"://"];
	if(protocolRange.location != NSNotFound) {
		return [NSURL URLWithString:path];
	}

	NSMutableString *unixPath = [path mutableCopy];

	// Get the fragment
	NSString *fragment = @"";
	NSScanner *scanner = [NSScanner scannerWithString:unixPath];
	NSCharacterSet *characterSet = [NSCharacterSet characterSetWithCharactersInString:@"#1234567890"];
	while(![scanner isAtEnd]) {
		NSString *possibleFragment;
		[scanner scanUpToString:@"#" intoString:nil];

		if([scanner scanCharactersFromSet:characterSet intoString:&possibleFragment] && [scanner isAtEnd]) {
			fragment = possibleFragment;
			[unixPath deleteCharactersInRange:NSMakeRange([scanner scanLocation] - [possibleFragment length], [possibleFragment length])];
			break;
		}
	}
	DLog(@"Fragment: %@", fragment);

	if(![unixPath hasPrefix:@"/"]) {
		// Only relative paths would have windows backslashes.
		[unixPath replaceOccurrencesOfString:@"\\" withString:@"/" options:0 range:NSMakeRange(0, [unixPath length])];

		NSString *basePath = [[[baseFilename stringByStandardizingPath] stringByDeletingLastPathComponent] stringByAppendingString:@"/"];

		[unixPath insertString:basePath atIndex:0];
	}

	// Append the fragment
	NSURL *url = [NSURL URLWithString:[[[NSURL fileURLWithPath:unixPath] absoluteString] stringByAppendingString:fragment]];
	return url;
}

+ (NSArray *)urlsForContainerURL:(NSURL *)url {
	char *filecontents = nil;

	{
		id audioSourceClass = NSClassFromString(@"AudioSource");
		id<CogSource> source = [audioSourceClass audioSourceForURL:url];

		if(![source open:url])
			return @[];

		long size = 0;
		long bytesread = 0;

		do {
			filecontents = (char *)realloc(filecontents, size + 1024);
			bytesread = [source read:(filecontents + size) amount:1024];
			size += bytesread;
		} while(bytesread == 1024);

		filecontents = (char *)realloc(filecontents, size + 1);

		filecontents[size] = '\0';
	}

	// Handle macOS Classic and Windows line endings
	{
		char *contentsscan = filecontents;
		while(*contentsscan) {
			if(*contentsscan == '\r')
				*contentsscan = '\n';
			++contentsscan;
		}
	}

	DLog(@"Trying UTF8");
	NSStringEncoding encoding = NSUTF8StringEncoding;
	NSString *contents = [NSString stringWithCString:filecontents encoding:encoding];
	if(!contents) {
		DLog(@"Trying windows GB 18030 2000");
		contents = [NSString stringWithCString:filecontents encoding:CFStringConvertEncodingToNSStringEncoding(kCFStringEncodingGB_18030_2000)];
	}
	if(!contents) {
		DLog(@"Trying windows CP1251");
		contents = [NSString stringWithCString:filecontents encoding:NSWindowsCP1251StringEncoding];
	}
	if(!contents) {
		DLog(@"Trying latin1");
		contents = [NSString stringWithCString:filecontents encoding:NSISOLatin1StringEncoding];
	}
	free(filecontents);
	if(!contents) {
		ALog(@"Could not open file...%@ %@", url, contents);
		return @[];
	}

	NSMutableArray *entries = [NSMutableArray array];

	for(NSString *entry in [contents componentsSeparatedByString:@"\n"]) {
		NSString *_entry = [entry stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];

		if([_entry hasPrefix:@"#EXT-X-MEDIA-SEQUENCE"]) // Let FFmpeg handle HLS
			return @[];

		if([_entry hasPrefix:@"#"] || [_entry isEqualToString:@""]) // Ignore extra info
			continue;

		// Need to add basePath, and convert to URL
		[entries addObject:[self urlForPath:_entry relativeTo:[url path]]];
	}

	return entries;
}

@end