package com.basic.security.model;

import android.text.TextUtils;
import android.util.Base64;

import com.basic.security.utils.Base64Util;
import com.basic.security.utils.Constants;
import com.couchbase.lite.Blob;
import com.couchbase.lite.MutableDocument;

import org.apache.commons.io.IOUtils;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

public class ModelAdapter {

    public MutableDocument couchbaseModel;
    public Map<String, Object> model;

    public ModelAdapter() {
        if (Constants.useCouchbase) {
            couchbaseModel = new MutableDocument();
        } else {
            model = new HashMap<>();
        }
    }

    public ModelAdapter(MutableDocument mutableDocument) {
        if (Constants.useCouchbase) {
            this.couchbaseModel = mutableDocument;
        } else {
            this.model = new HashMap<>();
        }
    }

    public ModelAdapter(String id) {
        if (Constants.useCouchbase) {
            this.couchbaseModel = new MutableDocument(id);
        } else {
            this.model = new HashMap<>();
            this.model.put("id", id);
        }
    }

    public static void main(String[] args) {
        byte[] bytes = new byte[]{1, 2, 3, 4, 5, 6};
        String str = Base64.encodeToString(bytes, Base64.NO_WRAP);
        byte[] bytes1 = Base64.decode(str, Base64.NO_WRAP);
        System.out.println("hello");
    }

    public static String getAttachmentPath(String id, String key, String table) {
        return Constants.attachmentPath + table + "_" + key + "_" + id;
    }

    public String getString(String key) {
        String value = "";
        if (Constants.useCouchbase) {
            try {
                value = couchbaseModel.getString(key);
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            try {
                value = (String) model.get(key);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        if (value == null) {
            value = "";
        }
        return value;
    }

    public ModelAdapter setString(String key, String value) {
        if (Constants.useCouchbase) {
            try {
                couchbaseModel.setString(key, value);
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            try {
                model.put(key, value);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return this;
    }

    public String getId() {
        if (Constants.useCouchbase) {
            try {
                return couchbaseModel.getId();
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            try {
                String id = (String) model.get("id");
                if (id == null) {
                    id = UUID.randomUUID().toString();
                    model.put("id", id);
                }
                return id;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return "";
    }

    public byte[] getBlob(String key) {
        if (Constants.useCouchbase) {
            try {
                Blob blob = couchbaseModel.getBlob(key);
                if (blob != null) {
                    return blob.getContent();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            try {
                String table = (String) model.get("table");
                if (table == null) {
                    table = "";
                }
                if ("camera_image_feature".equals(key)) {
                    return Base64Util.featureBase642Bytes((String) this.model.get(key));
                } else {
                    String value = (String) model.get(key);
                    if (!TextUtils.isEmpty(value)) {
                        File attachFile = new File(getAttachmentPath(getId(), key, table));
                        if (attachFile.exists()) {
                            return IOUtils.toByteArray(new FileInputStream(attachFile));
                        }
                    }
                }
            } catch (IOException e) {
                System.out.println("ModelAdapter:getBlob exception=" + e.getLocalizedMessage());
            }
        }
        return null;
    }

    public void setBlob(String key, byte[] content) {
        if (content != null) {
            if (Constants.useCouchbase) {
                Blob blob = new Blob("image/jpeg", content);
                couchbaseModel.setBlob(key, blob);
            } else {
                try {
                    String table = (String) model.get("table");
                    if (table == null) {
                        table = "";
                    }
                    if ("camera_image_feature".equals(key)) {
                        this.model.put(key, Base64Util.featureBytes2Base64(content));
//                        String key2 = key + "_path";
//                        String filePath= getAttachmentPath(getId(), key2, table);
//                        IOUtils.write(content, new FileOutputStream(filePath));
//                        this.sqliteModel.put(key2, filePath);
                    } else {
                        String filePath = getAttachmentPath(getId(), key, table);
                        File security_attachment_dir = new File(Constants.attachmentPath);
                        if (!security_attachment_dir.exists()) {
                            security_attachment_dir.mkdirs();
                        }
                        IOUtils.write(content, new FileOutputStream(filePath));
                        this.model.put(key, filePath);
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    public Map<String, Object> toMap() {
        if (Constants.useCouchbase) {
            return couchbaseModel.toMap();
        } else {
            return model;
        }
    }
}