cog/Plugins/M3u/M3uContainer.m

147 lines
4.1 KiB
Objective-C

//
// 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