/*
 * Project:  NextGIS Mobile
 * Purpose:  Mobile GIS for Android.
 * Author:   Dmitry Baryshnikov (aka Bishop), [email protected]
 * Author:   NikitaFeodonit, [email protected]
 * Author:   Stanislav Petriakov, [email protected]
 * *****************************************************************************
 * Copyright (c) 2012-2017 NextGIS, [email protected]
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.nextgis.maplib.datasource.ngw;

import android.os.Parcel;
import android.os.Parcelable;

import com.nextgis.maplib.util.Constants;
import com.nextgis.maplib.util.HttpResponse;
import com.nextgis.maplib.util.NGWUtil;
import com.nextgis.maplib.util.NetworkUtil;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

public class ResourceGroup extends Resource {
    protected List<Resource> mChildren;
    protected boolean mChildrenLoaded;

    public ResourceGroup(long remoteId, Connection connection) {
        super(remoteId, connection);
        mChildren = new ArrayList<>();
        mChildrenLoaded = false;
    }

    private ResourceGroup(JSONObject json, Connection connection) {
        super(json, connection);
        mChildren = new ArrayList<>();
        mChildrenLoaded = false;
    }

    public void loadChildren() {
        if (mChildrenLoaded)
            return;

        try {
            String sURL = NGWUtil.getResourceChildrenUrl(mConnection.getURL(), mRemoteId);
            HttpResponse response =
                    NetworkUtil.get(sURL, mConnection.getLogin(), mConnection.getPassword(), false);
            if (!response.isOk())
                return;

            JSONArray children = new JSONArray(response.getResponseBody());
            for (int i = 0; i < children.length(); i++)
                addResource(children.getJSONObject(i));

            Collections.sort(mChildren, new Comparator<Resource>() {
                @Override
                public int compare(Resource left, Resource right) {
                    return left.getName().compareToIgnoreCase(right.getName());
                }
            });

            Collections.sort(mChildren, new Comparator<Resource>() {
                @Override
                public int compare(Resource left, Resource right) {
                    int b1 = left.getType() == Connection.NGWResourceTypeResourceGroup ? 1 : 0;
                    int b2 = right.getType() == Connection.NGWResourceTypeResourceGroup ? 1 : 0;
                    return b2 - b1;
                }
            });

            mChildrenLoaded = true;
        } catch (IOException | JSONException e) {
            e.printStackTrace();
        }
    }

    protected void addResource(JSONObject data) {
        int type = getType(data);
        Resource resource = null;
        switch (type) {
            case Connection.NGWResourceTypeResourceGroup:
                resource = new ResourceGroup(data, mConnection);
                break;
            case Connection.NGWResourceTypePostgisLayer:
                if (mConnection.getNgwVersionMajor() < Constants.NGW_v3)
                    break;
            case Connection.NGWResourceTypeVectorLayer:
            case Connection.NGWResourceTypeRasterLayer:
                LayerWithStyles layer = new LayerWithStyles(data, mConnection);
                layer.fillExtent();
                layer.fillStyles();
                resource = layer;
                break;
            case Connection.NGWResourceTypeWMSClient:
                resource = new LayerWithStyles(data, mConnection);
                break;
            case Connection.NGWResourceTypeLookupTable:
                resource = new ResourceWithoutChildren(data, mConnection);
                break;
            case Connection.NGWResourceTypeWebMap:
                resource = new WebMap(data, mConnection);
                break;
        }

        if (null != resource) {
            resource.setParent(this);
            resource.fillPermissions();
            mChildren.add(resource);
        }
    }

    protected int getType(JSONObject data) {
        try {
            String sType = data.getJSONObject("resource").getString("cls");
            return mConnection.getType(sType);
        } catch (JSONException e) {
            return Connection.NGWResourceTypeNone;
        }
    }

    @Override
    public void writeToParcel(Parcel parcel, int i) {
        super.writeToParcel(parcel, i);
        parcel.writeByte(mChildrenLoaded ? (byte) 1 : (byte) 0);
        parcel.writeInt(mChildren.size());

        for (Resource resource : mChildren) {
            parcel.writeInt(resource.getType());
            parcel.writeParcelable(resource, i);
        }
    }

    public static final Parcelable.Creator<ResourceGroup> CREATOR =
            new Parcelable.Creator<ResourceGroup>() {
                public ResourceGroup createFromParcel(Parcel in) {
                    return new ResourceGroup(in);
                }

                public ResourceGroup[] newArray(int size) {
                    return new ResourceGroup[size];
                }
            };

    private ResourceGroup(Parcel in) {
        super(in);
        mChildrenLoaded = in.readByte() == 1;
        mChildren = new ArrayList<>();
        int count = in.readInt();

        for (int i = 0; i < count; i++) {
            int type = in.readInt();
            switch (type) {
                case Connection.NGWResourceTypeResourceGroup:
                    ResourceGroup resourceGroup = in.readParcelable(ResourceGroup.class.getClassLoader());
                    resourceGroup.setParent(this);
                    mChildren.add(resourceGroup);
                    break;
                case Connection.NGWResourceTypePostgisLayer:
                    if (mConnection.getNgwVersionMajor() < Constants.NGW_v3)
                        break;
                case Connection.NGWResourceTypeRasterLayer:
                case Connection.NGWResourceTypeVectorLayer:
                case Connection.NGWResourceTypeWMSClient:
                    LayerWithStyles layer = in.readParcelable(LayerWithStyles.class.getClassLoader());
                    layer.setParent(this);
                    mChildren.add(layer);
                    break;
                case Connection.NGWResourceTypeLookupTable:
                    ResourceWithoutChildren resourceWoChildren = in.readParcelable(ResourceWithoutChildren.class.getClassLoader());
                    resourceWoChildren.setParent(this);
                    mChildren.add(resourceWoChildren);
                case Connection.NGWResourceTypeWebMap:
                    WebMap webMap = in.readParcelable(WebMap.class.getClassLoader());
                    webMap.setParent(this);
                    mChildren.add(webMap);
                    break;
            }
        }
    }

    @Override
    public void setConnection(Connection connection) {
        super.setConnection(connection);
        for (Resource resource : mChildren)
            resource.setConnection(connection);
    }

    @Override
    public INGWResource getResourceById(int id) {
        INGWResource ret = super.getResourceById(id);
        if (null != ret)
            return ret;

        for (Resource resource : mChildren) {
            ret = resource.getResourceById(id);
            if (null != ret)
                return ret;
        }

        return super.getResourceById(id);
    }

    @Override
    public int getChildrenCount() {
        if (null == mChildren)
            return 0;

        return mChildren.size();
    }

    @Override
    public INGWResource getChild(int i) {
        if (null == mChildren)
            return null;

        return mChildren.get(i);
    }

    public boolean isChildrenLoaded() {
        return mChildrenLoaded;
    }

    public void setLoadChildren(boolean loadChildren) {
        mChildrenLoaded = loadChildren;
    }
}