neilco / eventsource Goto Github PK

- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data
{
    __block NSString *eventString = [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];

    if ([eventString hasSuffix:@"\n\n"]) {
        dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_BACKGROUND, 0), ^{
            NSArray* eventArray = [eventString componentsSeparatedByString:@"\n\n"];
            for (NSString* sigleEventString in eventArray) {
                NSArray *components = [sigleEventString componentsSeparatedByString:@"\n"] ;
                Event *e = [Event new];
                e.readyState = kEventStateOpen;

                for (NSString *component in components) {
                    NSUInteger location = [component rangeOfString:@":"].location;
                    if (location != NSNotFound) {
                        NSString *dataPath = [component substringFromIndex:location+1];
                        dataPath = [dataPath stringByTrimmingCharactersInSet:[NSCharacterSet whitespaceAndNewlineCharacterSet]];
                        if ([component hasPrefix:@"id:"]) {
                            e.id = dataPath;
                        } else if ([component hasPrefix:@"event:"]) {
                            e.event = dataPath;
                        } else if ([component hasPrefix:@"data:"]) {
                            e.data = dataPath;
                        }

                    }
                }

                if (e.event != nil) {
                    NSArray *namedEventhandlers = listeners[e.event];
                    for (EventSourceEventHandler handler in namedEventhandlers) {
                        dispatch_async(dispatch_get_main_queue(), ^{
                            handler(e);
                        });
                    }
                }
            }
        });
    }
}

    Auth.auth().currentUser?.getIDTokenForcingRefresh(true, completion: { (token, error) in
    let server : String =  "https://project-XXXXX.firebaseio.com/.json?auth=\(token!)"

    let eventSource: EventSource = EventSource(url: server)
    eventSource.onOpen {
        // When opened
        debugPrint("eventSource open")
    }

    eventSource.onError { (error) in
        // When errors
        debugPrint("error = \(error?.localizedDescription)")
    }
    eventSource.onMessage { (id, event, data) in
         debugPrint("data = \(data)")
        // Here you get an event without event name!
    }

    eventSource.addEventListener("child_added") { (id, event, data) in
           debugPrint("data = \(data)")
        // Here you get an event 'event-name'
    }
})