Add cookie related implementation #156

src/android/src/main/java/com/RNFetchBlob/RNFetchBlob.java

@@ -3,6 +3,7 @@ package com.RNFetchBlob;
 import android.content.Intent;
 import android.net.Uri;
 
+import com.RNFetchBlob.Utils.RNFBCookieJar;
 import com.facebook.react.bridge.Callback;
 import com.facebook.react.bridge.Promise;
 import com.facebook.react.bridge.ReactApplicationContext;
@@ -10,6 +11,7 @@ import com.facebook.react.bridge.ReactContextBaseJavaModule;
 import com.facebook.react.bridge.ReactMethod;
 import com.facebook.react.bridge.ReadableArray;
 import com.facebook.react.bridge.ReadableMap;
+import com.facebook.react.bridge.WritableArray;
 
 import java.util.Map;
 import java.util.concurrent.LinkedBlockingQueue;
@@ -203,6 +205,20 @@ public class RNFetchBlob extends ReactContextBaseJavaModule {
 
     }
 
+    @ReactMethod
+    /**
+     * Get cookies belongs specific host.
+     * @param host String host name.
+     */
+    public void getCookies(String host, Promise promise) {
+        try {
+            WritableArray cookies = RNFBCookieJar.getCookies(host);
+            promise.resolve(cookies);
+        } catch(Exception err) {
+            promise.reject("RNFetchBlob.getCookies", err.getMessage());
+        }
+    }
+
     @ReactMethod
     /**
      * @param path Stream file path

src/android/src/main/java/com/RNFetchBlob/RNFetchBlobReq.java

@@ -11,6 +11,7 @@ import android.util.Base64;
 
 import com.RNFetchBlob.Response.RNFetchBlobDefaultResp;
 import com.RNFetchBlob.Response.RNFetchBlobFileResp;
+import com.RNFetchBlob.Utils.RNFBCookieJar;
 import com.facebook.react.bridge.Arguments;
 import com.facebook.react.bridge.Callback;
 import com.facebook.react.bridge.ReactApplicationContext;
@@ -25,6 +26,9 @@ import java.io.File;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.net.CookieHandler;
+import java.net.CookieManager;
+import java.net.CookiePolicy;
 import java.net.MalformedURLException;
 import java.net.SocketException;
 import java.net.SocketTimeoutException;
@@ -39,6 +43,7 @@ import java.util.concurrent.TimeUnit;
 
 import okhttp3.Call;
 import okhttp3.ConnectionPool;
+import okhttp3.CookieJar;
 import okhttp3.Headers;
 import okhttp3.Interceptor;
 import okhttp3.MediaType;
@@ -290,7 +295,11 @@ public class RNFetchBlobReq extends BroadcastReceiver implements Runnable {
                     break;
             }
 
+            // #156 fix cookie issue
+
+
             final Request req = builder.build();
+            clientBuilder.cookieJar(new RNFBCookieJar());
             clientBuilder.addNetworkInterceptor(new Interceptor() {
                 @Override
                 public Response intercept(Chain chain) throws IOException {

src/android/src/main/java/com/RNFetchBlob/Utils/RNFBCookieJar.java

@@ -0,0 +1,53 @@
+package com.RNFetchBlob.Utils;
+
+import com.facebook.react.bridge.Arguments;
+import com.facebook.react.bridge.ReadableMap;
+import com.facebook.react.bridge.WritableArray;
+import com.facebook.react.bridge.WritableMap;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+
+import okhttp3.Cookie;
+import okhttp3.CookieJar;
+import okhttp3.HttpUrl;
+
+/**
+ * Created by wkh237on 2016/10/14.
+ */
+
+
+
+public class RNFBCookieJar implements CookieJar {
+
+    static final HashMap<String, List<Cookie>> cookieStore = new HashMap<>();
+    private List<Cookie> cookies;
+
+    @Override
+    public void saveFromResponse(HttpUrl url, List<Cookie> cookies) {
+        cookieStore.put(url.host(), cookies);
+    }
+
+    @Override
+    public List<Cookie> loadForRequest(HttpUrl url) {
+        List<Cookie> cookies = cookieStore.get(url.host());
+        return cookies != null ? cookies : new ArrayList<Cookie>();
+    }
+
+    public static WritableArray getCookies(String host) {
+        HttpUrl url = HttpUrl.parse(host);
+        List<Cookie> cookies = null;
+        if(url != null) {
+            cookies = cookieStore.get(url.host());
+        }
+        WritableArray cookieList = Arguments.createArray();
+        if(cookies != null) {
+            for(Cookie c : cookies){
+                cookieList.pushString(c.toString());
+            }
+            return cookieList;
+        }
+        return null;
+    }
+}

src/index.js

@@ -24,6 +24,7 @@ import base64 from 'base-64'
 import polyfill from './polyfill'
 import android from './android'
 import ios from './ios'
+import net from './net'
 import JSONStream from './json-stream'
 const {
   RNFetchBlobSession,
@@ -41,10 +42,9 @@ const {
   cp
 } = fs
 
-
 const Blob = polyfill.Blob
 const emitter = DeviceEventEmitter
-const RNFetchBlob:RNFetchBlobNative = NativeModules.RNFetchBlob
+const RNFetchBlob= NativeModules.RNFetchBlob
 
 // register message channel event handler.
 emitter.addListener("RNFetchBlobMessage", (e) => {
@@ -549,6 +549,7 @@ export default {
   session,
   fs,
   wrap,
+  net,
   polyfill,
   JSONStream
 }

src/ios/RNFetchBlob/RNFetchBlob.m

@@ -91,6 +91,7 @@ RCT_EXPORT_METHOD(fetchBlobForm:(NSDictionary *)options
 
 }
 
+
 // Fetch blob data request
 RCT_EXPORT_METHOD(fetchBlob:(NSDictionary *)options
                   taskId:(NSString *)taskId
@@ -473,8 +474,11 @@ RCT_EXPORT_METHOD(previewDocument:(NSString*)uri scheme:(NSString *)scheme resol
     return window.rootViewController;
 }
 
-
-#pragma mark RNFetchBlob private methods
+# pragma mark - getCookies
+RCT_EXPORT_METHOD(getCookies:(NSString *)url resolver:(RCTPromiseResolveBlock)resolve rejecter:(RCTPromiseRejectBlock)reject
+{
+    resolve([RNFetchBlobNetwork getCookies:url]);
+})
 
 
 @end

src/ios/RNFetchBlobNetwork.h

@@ -44,6 +44,7 @@ typedef void(^DataTaskCompletionHander) (NSData * _Nullable resp, NSURLResponse
 - (void) sendRequest:(NSDictionary  * _Nullable )options contentLength:(long)contentLength bridge:(RCTBridge * _Nullable)bridgeRef taskId:(NSString * _Nullable)taskId withRequest:(NSURLRequest * _Nullable)req callback:(_Nullable RCTResponseSenderBlock) callback;
 + (void) enableProgressReport:(NSString *) taskId config:(RNFetchBlobProgress *)config;
 + (void) enableUploadProgress:(NSString *) taskId config:(RNFetchBlobProgress *)config;
++ (NSArray *) getCookies:(NSString *) url;
 
 
 

src/ios/RNFetchBlobNetwork.m

@@ -28,6 +28,7 @@
 
 NSMapTable * taskTable;
 NSMapTable * expirationTable;
+NSMapTable * cookiesTable;
 NSMutableDictionary * progressTable;
 NSMutableDictionary * uploadProgressTable;
 
@@ -93,9 +94,55 @@ NSOperationQueue *taskQueue;
     {
         uploadProgressTable = [[NSMutableDictionary alloc] init];
     }
+    if(cookiesTable == nil)
+    {
+        cookiesTable = [[NSMapTable alloc] init];
+    }
     return self;
 }
 
++ (NSArray *) getCookies:(NSString *) url
+{
+    NSString * hostname = [[NSURL URLWithString:url] host];
+    NSMutableArray * cookies = [NSMutableArray new];
+    NSArray * list = [cookiesTable objectForKey:hostname];
+    for(NSHTTPCookie * cookie in list)
+    {
+        NSMutableString * cookieStr = [[NSMutableString alloc] init];
+        [cookieStr appendString:cookie.name];
+        [cookieStr appendString:@"="];
+        [cookieStr appendString:cookie.value];
+        
+        if(cookie.expiresDate == nil) {
+            [cookieStr appendString:@"; max-age=0"];
+        }
+        else {
+            [cookieStr appendString:@"; expires="];
+            NSDateFormatter *dateFormatter = [[NSDateFormatter alloc] init];
+            [dateFormatter setDateFormat:@"EEE, dd MM yyyy HH:mm:ss ZZZ"];
+            NSString *strDate = [dateFormatter stringFromDate:cookie.expiresDate];
+            [cookieStr appendString:strDate];
+        }
+        
+        
+        [cookieStr appendString:@"; domain="];
+        [cookieStr appendString:hostname];
+        [cookieStr appendString:@"; path="];
+        [cookieStr appendString:cookie.path];
+        
+        
+        if (cookie.isSecure) {
+            [cookieStr appendString:@"; secure"];
+        }
+        
+        if (cookie.isHTTPOnly) {
+            [cookieStr appendString:@"; httponly"];
+        }
+        [cookies addObject:cookieStr];
+    }
+    return cookies;
+}
+
 + (void) enableProgressReport:(NSString *) taskId config:(RNFetchBlobProgress *)config
 {
     [progressTable setValue:config forKey:taskId];
@@ -253,6 +300,8 @@ NSOperationQueue *taskQueue;
 
 #pragma mark NSURLSession delegate methods
 
+
+#pragma mark - Received Response
 // set expected content length on response received
 - (void) URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveResponse:(NSURLResponse *)response completionHandler:(void (^)(NSURLSessionResponseDisposition))completionHandler
 {
@@ -332,12 +381,23 @@ NSOperationQueue *taskQueue;
                      @"status": [NSString stringWithFormat:@"%d", statusCode ]
                     };
 
+#pragma mark - handling cookies
+        // # 153 get cookies
+        if(response.URL != nil)
+        {
+            NSArray<NSHTTPCookie *> * cookies = [NSHTTPCookie cookiesWithResponseHeaderFields: headers forURL:response.URL];
+            if(cookies != nil && [cookies count] > 0) {
+                [cookiesTable setObject:cookies forKey:response.URL.host];
+            }
+        }
+        
         [self.bridge.eventDispatcher
          sendDeviceEventWithName: EVENT_STATE_CHANGE
          body:respInfo
         ];
         headers = nil;
         respInfo = nil;
+
     }
     else
         NSLog(@"oops");

src/net.js

@@ -0,0 +1,26 @@
+// Copyright 2016 wkh237@github. All rights reserved.
+// Use of this source code is governed by a MIT-style license that can be
+// found in the LICENSE file.
+// @flow
+
+import {
+  NativeModules,
+  DeviceEventEmitter,
+  Platform,
+  NativeAppEventEmitter,
+} from 'react-native'
+
+const RNFetchBlob = NativeModules.RNFetchBlob
+
+/**
+ * Get cookie according to the given url.
+ * @param  {string} url HTTP URL string.
+ * @return {Promise<Array<String>>}     Cookies of a specific domain.
+ */
+function getCookies(url:string):Promise<Array<String>> {
+  return RNFetchBlob.getCookies(url)
+}
+
+export default {
+  getCookies
+}

test/test-0.10.0.js

@@ -26,65 +26,82 @@ const dirs = RNFetchBlob.fs.dirs
 let prefix = ((Platform.OS === 'android') ? 'file://' : '')
 let begin = Date.now()
 
-describe('json stream via HTTP', (report, done) => {
+// describe('json stream via HTTP', (report, done) => {
+//
+//   let count = 0
+//   JSONStream(`${TEST_SERVER_URL}/public/json-dummy.json`).node('name', (name) => {
+//     count++
+//     if(Date.now() - begin < 100)
+//     return
+//     begin = Date.now()
+//     report(<Info key="report" uid="100">
+//       <Text>{count} records</Text>
+//     </Info>)
+//     done()
+//   })
+//
+// })
+//
+// describe('json stream via fs', (report, done) => {
+//
+//   let fetch2 = new RNFetchBlob.polyfill.Fetch({
+//     auto : true
+//   })
+//   let res = null
+//   let count = 0
+//
+//   RNFetchBlob.config({
+//     fileCache : true
+//   })
+//   .fetch('GET',`${TEST_SERVER_URL}/public/json-dummy.json`)
+//   .then((resp) => {
+//     res = resp
+//     JSONStream({
+//       url : RNFetchBlob.wrap(res.path()),
+//       headers : { bufferSize : 10240 }
+//     }).node('name', (name) => {
+//       count++
+//       if(Date.now() - begin < 100)
+//       return
+//       begin = Date.now()
+//       report(<Info key="report" uid="100">
+//         <Text>{count} records</Text>
+//       </Info>)
+//       done()
+//     })
+//   })
+// })
+//
+// describe('issue #102', (report, done) => {
+//   let tmp = null
+//   RNFetchBlob.config({ fileCache: true, appendExt : 'png' })
+//     .fetch('GET', `${TEST_SERVER_URL}/public/github.png`)
+//     .then((res) => {
+//       tmp = res
+//       RNFetchBlob.ios.previewDocument(res.path())
+//       return RNFetchBlob.fetch('POST', `${TEST_SERVER_URL}/upload-form`, {},
+//       [{ name : String(1), data : RNFetchBlob.wrap(res.path()), filename: '#102-test-image.png' }])
+//     })
+//     .then((res) =>  tmp.flush())
+//     .then(() => {
+//       done()
+//     })
+//
+// })
 
-  let count = 0
-  JSONStream(`${TEST_SERVER_URL}/public/json-dummy.json`).node('name', (name) => {
-    count++
-    if(Date.now() - begin < 100)
-    return
-    begin = Date.now()
-    report(<Info key="report" uid="100">
-      <Text>{count} records</Text>
-    </Info>)
-    done()
-  })
-
-})
 
-describe('json stream via fs', (report, done) => {
-
-  let fetch2 = new RNFetchBlob.polyfill.Fetch({
-    auto : true
-  })
-  let res = null
-  let count = 0
+describe('cookie test', (report, done) => {
 
-  RNFetchBlob.config({
-    fileCache : true
+  RNFetchBlob.fetch('GET', `${TEST_SERVER_URL}/cookie`)
+  .then((res) => {
+    return RNFetchBlob.fetch('GET', `${TEST_SERVER_URL}/xhr-header`)
   })
-  .fetch('GET',`${TEST_SERVER_URL}/public/json-dummy.json`)
-  .then((resp) => {
-    res = resp
-    JSONStream({
-      url : RNFetchBlob.wrap(res.path()),
-      headers : { bufferSize : 10240 }
-    }).node('name', (name) => {
-      count++
-      if(Date.now() - begin < 100)
-      return
-      begin = Date.now()
-      report(<Info key="report" uid="100">
-        <Text>{count} records</Text>
-      </Info>)
-      done()
+  .then((res) => {
+    console.log(res)
+    RNFetchBlob.net.getCookies(`${TEST_SERVER_URL}`)
+    .then((cookies) => {
+      console.log(cookies)
     })
   })
-})
-
-describe('issue #102', (report, done) => {
-  let tmp = null
-  RNFetchBlob.config({ fileCache: true, appendExt : 'png' })
-    .fetch('GET', `${TEST_SERVER_URL}/public/github.png`)
-    .then((res) => {
-      tmp = res
-      RNFetchBlob.ios.previewDocument(res.path())
-      return RNFetchBlob.fetch('POST', `${TEST_SERVER_URL}/upload-form`, {},
-      [{ name : String(1), data : RNFetchBlob.wrap(res.path()), filename: '#102-test-image.png' }])
-    })
-    .then((res) =>  tmp.flush())
-    .then(() => {
-      done()
-    })
 
 })