cog/Plugins/M3u/M3uContainer.m

144 lines
3.9 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 == nil || [_entry length] < 1) continue;
if([_entry hasPrefix:@"#EXT-X-MEDIA-SEQUENCE"]) // Let FFmpeg handle HLS
return @[];
if([_entry hasPrefix:@"#"]) // Ignore extra info
continue;
// Need to add basePath, and convert to URL
NSURL *fileUrl = [self urlForPath:_entry relativeTo:[url path]];
if(fileUrl)
[entries addObject:fileUrl];
}
return entries;
}
@end