package com.cordova.plugins.cookiemaster;

import org.apache.cordova.CallbackContext;
import org.apache.cordova.CordovaPlugin;
import org.apache.cordova.PluginResult;
import org.json.JSONObject;
import org.json.JSONArray;
import org.json.JSONException;

import android.util.Log;

import org.json.JSONStringer;

import java.net.HttpCookie;
import java.util.Calendar;
import java.util.Date;
import java.util.Iterator;

import android.webkit.CookieManager;

public class CookieMaster extends CordovaPlugin {

    private final String TAG = "CookieMasterPlugin";
    public static final String ACTION_GET_COOKIES = "getCookies";
    public static final String ACTION_SET_COOKIES = "setCookies";
    public static final String ACTION_GET_COOKIE_VALUE = "getCookieValue";
    public static final String ACTION_SET_COOKIE_VALUE = "setCookieValue";
    public static final String ACTION_CLEAR_COOKIES = "clearCookies";

    @Override
    public boolean execute(String action, JSONArray args, final CallbackContext callbackContext) throws JSONException {

        if (ACTION_GET_COOKIES.equals(action)) {
            final String url = args.getString(0);

            cordova.getThreadPool().execute(new Runnable() {
                public void run() {
                    try {
                        CookieManager cookieManager = CookieManager.getInstance();
                        String[] cookies = cookieManager.getCookie(url).split("; ");
                        JSONStringer json = new JSONStringer();
                        json.object();
                        for (String c : cookies) {
                            String[] cookie = c.split("=");
                            json.key(cookie[0].trim());
                            json.value(cookie[1].trim());
                        }
                        json.endObject();

                        if (json != null) {
                            PluginResult res = new PluginResult(PluginResult.Status.OK, json.toString());
                            callbackContext.sendPluginResult(res);
                        } else {
                            callbackContext.error("Cookie not found!");
                        }
                    } catch (Exception e) {
                        Log.e(TAG, "Exception: " + e.getMessage());
                        callbackContext.error(e.getMessage());
                    }
                }
            });
            return true;

        }
        else if (ACTION_SET_COOKIES.equals(action)) {
           final String url = args.getString(0);
           final JSONObject jsonCookies = args.getJSONObject(1);

           cordova.getThreadPool().execute(new Runnable() {
               public void run() {

                   Iterator<String> keys = jsonCookies.keys();
                   Date now = new Date();
                   Calendar expiry = Calendar.getInstance();
                   expiry.setTime(now);
                   expiry.add(Calendar.MONTH, 3);

                   try {
                       while (keys.hasNext()) {
                           String key = keys.next();
                           String cookieValue = null;

                           cookieValue = jsonCookies.getString(key);
                           HttpCookie cookie = new HttpCookie(key, cookieValue);

                           String cookieString = cookie.toString().replace("\"", "");
                           cookieString += ";expires="+expiry.getTime().toGMTString();
                           CookieManager cookieManager = CookieManager.getInstance();
                           cookieManager.setCookie(url, cookieString);
                       }
                       PluginResult res = new PluginResult(PluginResult.Status.OK, "Successfully added cookie");
                       callbackContext.sendPluginResult(res);

                   } catch (JSONException e) {
                       Log.e(TAG, "Exception: " + e.getMessage());
                       callbackContext.error("Cannot set cookie. Invalid format.");
                   }
               }
           });
           return true;

       }
        else if (ACTION_GET_COOKIE_VALUE.equals(action)) {
            final String url = args.getString(0);
            final String cookieName = args.getString(1);

            cordova.getThreadPool().execute(new Runnable() {
                public void run() {
                    try {
                        CookieManager cookieManager = CookieManager.getInstance();
                        String[] cookies = cookieManager.getCookie(url).split("; ");
                        String cookieValue = "";

                        for (int i = 0; i < cookies.length; i++) {
                            if (cookies[i].contains(cookieName + "=")) {
                                cookieValue = cookies[i].split("=")[1].trim();
                                break;
                            }
                        }

                        JSONObject json = null;
                        if (cookieValue != "") {
                            json = new JSONObject("{cookieValue:\"" + cookieValue + "\"}");
                        }
                        if (json != null) {
                            PluginResult res = new PluginResult(PluginResult.Status.OK, json);
                            callbackContext.sendPluginResult(res);
                        } else {
                            callbackContext.error("Cookie not found!");
                        }
                    } catch (Exception e) {
                        Log.e(TAG, "Exception: " + e.getMessage());
                        callbackContext.error(e.getMessage());
                    }
                }
            });
            return true;

        }
         else if (ACTION_SET_COOKIE_VALUE.equals(action)) {
            final String url = args.getString(0);
            final String cookieName = args.getString(1);
            final String cookieValue = args.getString(2);

            cordova.getThreadPool().execute(new Runnable() {
                public void run() {
                    try {
                        HttpCookie cookie = new HttpCookie(cookieName, cookieValue);

                        String cookieString = cookie.toString().replace("\"", "");
                        CookieManager cookieManager = CookieManager.getInstance();
                        cookieManager.setCookie(url, cookieString);

                        PluginResult res = new PluginResult(PluginResult.Status.OK, "Successfully added cookie");
                        callbackContext.sendPluginResult(res);
                    } catch (Exception e) {
                        Log.e(TAG, "Exception: " + e.getMessage());
                        callbackContext.error(e.getMessage());
                    }
                }
            });
            return true;

        }
        else if (ACTION_CLEAR_COOKIES.equals(action)) {
            CookieManager.getInstance().removeAllCookie();
            callbackContext.success();
            return true;

        }

        callbackContext.error("Invalid action: "+action);
        return false;

    }
}