Add cookie related implementation #156

Author: wkh237

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

@@ -3,13 +3,15 @@
 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;
 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 void run() {
 
     }
 
+    @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 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.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 okhttp3.Call;
 import okhttp3.ConnectionPool;
+import okhttp3.CookieJar;
 import okhttp3.Headers;
 import okhttp3.Interceptor;
 import okhttp3.MediaType;
@@ -290,7 +295,11 @@ else if(cType.isEmpty()) {
                     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 @@ - (NSDictionary *)constantsToExport
 
 }
 
+
 // Fetch blob data request
 RCT_EXPORT_METHOD(fetchBlob:(NSDictionary *)options
                   taskId:(NSString *)taskId
@@ -473,8 +474,11 @@ - (UIViewController *) documentInteractionControllerViewControllerForPreview: (U
     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 @@ - (id)init {
     {
         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 @@ + (void) emitExpiredTasks
 
 #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 @@ - (void) URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dat
                      @"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
+}