package com.box.sdk;

import com.box.sdk.BoxGroupMembership.Permission;
import com.eclipsesource.json.Json;
import com.eclipsesource.json.JsonArray;
import com.eclipsesource.json.JsonObject;
import com.eclipsesource.json.JsonValue;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Map;

/**
 * Represents a set of Box users.
 *
 * <p>Unless otherwise noted, the methods in this class can throw an unchecked {@link BoxAPIException} (unchecked
 * meaning that the compiler won't force you to handle it) if an error occurs. If you wish to implement custom error
 * handling for errors related to the Box REST API, you should capture this exception explicitly.</p>
 */
@BoxResourceType("group")
public class BoxGroup extends BoxCollaborator {

    /**
     * @see #getAllGroups(BoxAPIConnection, String...)
     */
    public static final URLTemplate GROUPS_URL_TEMPLATE = new URLTemplate("groups");

    /**
     * @see #getInfo(String...)
     */
    public static final URLTemplate GROUP_URL_TEMPLATE = new URLTemplate("groups/%s");

    /**
     * @see #getMemberships()
     */
    public static final URLTemplate MEMBERSHIPS_URL_TEMPLATE = new URLTemplate("groups/%s/memberships");

    /**
     * @see #addMembership(BoxUser)
     */
    public static final URLTemplate ADD_MEMBERSHIP_URL_TEMPLATE = new URLTemplate("group_memberships");

    /**
     * @see #getCollaborations()
     */
    public static final URLTemplate COLLABORATIONS_URL_TEMPLATE = new URLTemplate("groups/%s/collaborations");

    /**
     * Constructs a BoxGroup for a group with a given ID.
     *
     * @param api the API connection to be used by the group.
     * @param id  the ID of the group.
     */
    public BoxGroup(BoxAPIConnection api, String id) {
        super(api, id);
    }

    /**
     * Creates a new group with a specified name.
     *
     * @param api  the API connection to be used by the group.
     * @param name the name of the new group.
     * @return info about the created group.
     */
    public static BoxGroup.Info createGroup(BoxAPIConnection api, String name) {
        return createGroup(api, name, null, null, null, null, null);
    }

    /**
     * Creates a new group with a specified name.
     *
     * @param api                    the API connection to be used by the group.
     * @param name                   the name of the new group.
     * @param provenance             the provenance of the new group
     * @param externalSyncIdentifier the external_sync_identifier of the new group
     * @param description            the description of the new group
     * @param invitabilityLevel      the invitibility_level of the new group
     * @param memberViewabilityLevel the member_viewability_level of the new group
     * @return info about the created group.
     */
    public static BoxGroup.Info createGroup(BoxAPIConnection api, String name, String provenance,
                                            String externalSyncIdentifier, String description,
                                            String invitabilityLevel, String memberViewabilityLevel) {
        JsonObject requestJSON = new JsonObject();
        requestJSON.add("name", name);

        if (provenance != null) {
            requestJSON.add("provenance", provenance);
        }
        if (externalSyncIdentifier != null) {
            requestJSON.add("external_sync_identifier", externalSyncIdentifier);
        }
        if (description != null) {
            requestJSON.add("description", description);
        }
        if (invitabilityLevel != null) {
            requestJSON.add("invitability_level", invitabilityLevel);
        }
        if (memberViewabilityLevel != null) {
            requestJSON.add("member_viewability_level", memberViewabilityLevel);
        }

        URL url = GROUPS_URL_TEMPLATE.build(api.getBaseURL());
        BoxJSONRequest request = new BoxJSONRequest(api, url, "POST");
        request.setBody(requestJSON.toString());
        try (BoxJSONResponse response = request.send()) {
            JsonObject responseJSON = Json.parse(response.getJSON()).asObject();

            BoxGroup group = new BoxGroup(api, responseJSON.get("id").asString());
            return group.new Info(responseJSON);
        }
    }

    /**
     * Gets an iterable of all the groups in the enterprise.
     *
     * @param api the API connection to be used when retrieving the groups.
     * @return an iterable containing info about all the groups.
     */
    public static Iterable<BoxGroup.Info> getAllGroups(final BoxAPIConnection api) {
        return () -> {
            URL url = GROUPS_URL_TEMPLATE.build(api.getBaseURL());
            return new BoxGroupIterator(api, url);
        };
    }

    /**
     * Gets an iterable of all the groups in the enterprise.
     *
     * @param api    the API connection to be used when retrieving the groups.
     * @param fields the fields to retrieve.
     * @return an iterable containing info about all the groups.
     */
    public static Iterable<BoxGroup.Info> getAllGroups(final BoxAPIConnection api, String... fields) {
        final QueryStringBuilder builder = new QueryStringBuilder();
        if (fields.length > 0) {
            builder.appendParam("fields", fields);
        }
        return () -> {
            URL url = GROUPS_URL_TEMPLATE.buildWithQuery(api.getBaseURL(), builder.toString());
            return new BoxGroupIterator(api, url);
        };
    }

    /**
     * Gets an iterable of all the groups in the enterprise that are starting with the given name string.
     *
     * @param api    the API connection to be used when retrieving the groups.
     * @param name   the name prefix of the groups. If the groups need to searched by full name that has spaces,
     *               then the parameter string should have been wrapped with "".
     * @param fields the fields to retrieve.
     * @return an iterable containing info about all the groups.
     */
    public static Iterable<BoxGroup.Info> getAllGroupsByName(
        final BoxAPIConnection api, String name, String... fields
    ) {
        final QueryStringBuilder builder = new QueryStringBuilder();
        if (name == null || name.trim().isEmpty()) {
            throw new BoxAPIException("Searching groups by name requires a non NULL or non empty name");
        } else {
            builder.appendParam("filter_term", name);
            if (fields != null && fields.length > 0) {
                builder.appendParam("fields", fields);
            }
        }

        return () -> {
            URL url = GROUPS_URL_TEMPLATE.buildWithQuery(api.getBaseURL(), builder.toString());
            return new BoxGroupIterator(api, url);
        };
    }

    /**
     * Gets information about this group.
     *
     * @param fields the fields to retrieve.
     * @return info about this group.
     */
    public Info getInfo(String... fields) {
        URL url = GROUP_URL_TEMPLATE.build(this.getAPI().getBaseURL(), this.getID());
        if (fields.length > 0) {
            QueryStringBuilder builder = new QueryStringBuilder().appendParam("fields", fields);
            url = GROUP_URL_TEMPLATE.buildWithQuery(this.getAPI().getBaseURL(), builder.toString(), this.getID());
        }
        BoxJSONRequest request = new BoxJSONRequest(this.getAPI(), url, "GET");
        try (BoxJSONResponse response = request.send()) {
            JsonObject responseJSON = Json.parse(response.getJSON()).asObject();
            return new Info(responseJSON);
        }
    }

    /**
     * Gets information about all of the group memberships for this group.
     * Does not support paging.
     *
     * @return a collection of information about the group memberships for this group.
     */
    public Collection<BoxGroupMembership.Info> getMemberships() {
        final BoxAPIConnection api = this.getAPI();
        final String groupID = this.getID();

        Iterable<BoxGroupMembership.Info> iter = () -> {
            URL url = MEMBERSHIPS_URL_TEMPLATE.build(api.getBaseURL(), groupID);
            return new BoxGroupMembershipIterator(api, url);
        };

        // We need to iterate all results because this method must return a Collection. This logic should be removed in
        // the next major version, and instead return the Iterable directly.
        Collection<BoxGroupMembership.Info> memberships = new ArrayList<>();
        for (BoxGroupMembership.Info membership : iter) {
            memberships.add(membership);
        }
        return memberships;
    }

    /**
     * Gets information about all of the group memberships for this group as iterable with paging support.
     *
     * @param fields the fields to retrieve.
     * @return an iterable with information about the group memberships for this group.
     */
    public Iterable<BoxGroupMembership.Info> getAllMemberships(String... fields) {
        final QueryStringBuilder builder = new QueryStringBuilder();
        if (fields.length > 0) {
            builder.appendParam("fields", fields);
        }
        return () -> {
            URL url = MEMBERSHIPS_URL_TEMPLATE.buildWithQuery(
                BoxGroup.this.getAPI().getBaseURL(), builder.toString(), BoxGroup.this.getID());
            return new BoxGroupMembershipIterator(BoxGroup.this.getAPI(), url);
        };
    }

    /**
     * Adds a member to this group with the default role.
     *
     * @param user the member to be added to this group.
     * @return info about the new group membership.
     */
    public BoxGroupMembership.Info addMembership(BoxUser user) {
        return this.addMembership(user, null, null);
    }

    /**
     * Adds a member to this group with the specified role.
     *
     * @param user the member to be added to this group.
     * @param role the role of the user in this group. Can be null to assign the default role.
     * @return info about the new group membership.
     */
    public BoxGroupMembership.Info addMembership(BoxUser user, BoxGroupMembership.GroupRole role) {
        return this.addMembership(user, role, null);
    }

    /**
     * Adds a member to this group with the specified role.
     *
     * @param user                    the member to be added to this group.
     * @param role                    the role of the user in this group. Can be null to assign the default role.
     * @param configurablePermissions the configurable permission of the user as a group admin.
     *                                Can be null to give all group admin permissions.
     * @return info about the new group membership.
     */
    public BoxGroupMembership.Info addMembership(BoxUser user, BoxGroupMembership.GroupRole role,
                                                 Map<BoxGroupMembership.Permission, Boolean> configurablePermissions) {
        BoxAPIConnection api = this.getAPI();

        JsonObject requestJSON = new JsonObject();
        requestJSON.add("user", new JsonObject().add("id", user.getID()));
        requestJSON.add("group", new JsonObject().add("id", this.getID()));
        if (role != null) {
            requestJSON.add("role", role.toJSONString());
        }

        if (configurablePermissions != null) {
            JsonObject configurablePermissionJson = new JsonObject();
            for (Permission attrKey : configurablePermissions.keySet()) {
                configurablePermissionJson.set(attrKey.toJSONValue(), configurablePermissions.get(attrKey));
            }
            requestJSON.add("configurable_permissions", configurablePermissionJson);
        }

        URL url = ADD_MEMBERSHIP_URL_TEMPLATE.build(api.getBaseURL());
        BoxJSONRequest request = new BoxJSONRequest(api, url, "POST");
        request.setBody(requestJSON.toString());
        try (BoxJSONResponse response = request.send()) {
            JsonObject responseJSON = Json.parse(response.getJSON()).asObject();

            BoxGroupMembership membership = new BoxGroupMembership(api, responseJSON.get("id").asString());
            return membership.new Info(responseJSON);
        }
    }

    /**
     * Gets information about all of the collaborations for this group.
     *
     * @return a collection of information about the collaborations for this group.
     */
    public Collection<BoxCollaboration.Info> getCollaborations() {
        BoxAPIConnection api = this.getAPI();
        URL url = COLLABORATIONS_URL_TEMPLATE.build(api.getBaseURL(), this.getID());

        BoxJSONRequest request = new BoxJSONRequest(api, url, "GET");
        try (BoxJSONResponse response = request.send()) {
            JsonObject responseJSON = Json.parse(response.getJSON()).asObject();

            int entriesCount = responseJSON.get("total_count").asInt();
            Collection<BoxCollaboration.Info> collaborations = new ArrayList<>(entriesCount);
            JsonArray entries = responseJSON.get("entries").asArray();
            for (JsonValue entry : entries) {
                JsonObject entryObject = entry.asObject();
                BoxCollaboration collaboration = new BoxCollaboration(api, entryObject.get("id").asString());
                BoxCollaboration.Info info = collaboration.new Info(entryObject);
                collaborations.add(info);
            }

            return collaborations;
        }
    }

    /**
     * Gets information about all of the collaborations for this group.
     *
     * @param fields the optional fields to retrieve.
     * @return An iterable of BoxCollaboration.Info instances associated with the item.
     */
    public Iterable<BoxCollaboration.Info> getAllCollaborations(String... fields) {
        final BoxAPIConnection api = this.getAPI();
        final QueryStringBuilder builder = new QueryStringBuilder();
        if (fields.length > 0) {
            builder.appendParam("fields", fields);
        }
        return () -> {
            URL url = COLLABORATIONS_URL_TEMPLATE.buildWithQuery(api.getBaseURL(), builder.toString(),
                BoxGroup.this.getID());
            return new BoxCollaborationIterator(api, url);
        };
    }

    /**
     * Deletes this group.
     */
    public void delete() {
        URL url = GROUP_URL_TEMPLATE.build(this.getAPI().getBaseURL(), this.getID());
        BoxAPIRequest request = new BoxAPIRequest(this.getAPI(), url, "DELETE");
        request.send().close();
    }

    /**
     * Updates the information about this group with any info fields that have been modified locally.
     *
     * @param info the updated info.
     */
    public void updateInfo(BoxGroup.Info info) {
        URL url = GROUP_URL_TEMPLATE.build(this.getAPI().getBaseURL(), this.getID());
        BoxJSONRequest request = new BoxJSONRequest(this.getAPI(), url, "PUT");
        request.setBody(info.getPendingChanges());
        try (BoxJSONResponse response = request.send()) {
            JsonObject jsonObject = Json.parse(response.getJSON()).asObject();
            info.update(jsonObject);
        }
    }

    /**
     * Contains information about a BoxGroup.
     */
    public class Info extends BoxCollaborator.Info {

        /**
         * @see #getProvenance()
         */
        private String provenance;

        /**
         * @see #getExternalSyncIdentifier()
         */
        private String externalSyncIdentifier;

        /**
         * @see #getDescription()
         */
        private String description;

        /**
         * @see #getInvitabilityLevel()
         */
        private String invitabilityLevel;

        /**
         * @see #getMemberViewabilityLevel()
         */
        private String memberViewabilityLevel;

        /**
         * Constructs an empty Info object.
         */
        public Info() {
            super();
        }

        /**
         * Constructs an Info object by parsing information from a JSON string.
         *
         * @param json the JSON string to parse.
         */
        public Info(String json) {
            super(json);
        }

        /**
         * Constructs an Info object using an already parsed JSON object.
         *
         * @param jsonObject the parsed JSON object.
         */
        Info(JsonObject jsonObject) {
            super(jsonObject);
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public BoxGroup getResource() {
            return BoxGroup.this;
        }

        /**
         * {@inheritDoc}
         */
        @Override
        protected void parseJSONMember(JsonObject.Member member) {
            super.parseJSONMember(member);

            String memberName = member.getName();
            JsonValue value = member.getValue();
            try {
                switch (memberName) {
                    case "description":
                        this.description = value.asString();
                        break;
                    case "external_sync_identifier":
                        this.externalSyncIdentifier = value.asString();
                        break;
                    case "invitability_level":
                        this.invitabilityLevel = value.asString();
                        break;
                    case "member_viewability_level":
                        this.memberViewabilityLevel = value.asString();
                        break;
                    case "provenance":
                        this.provenance = value.asString();
                        break;
                    default:
                        break;
                }
            } catch (Exception e) {
                throw new BoxDeserializationException(memberName, value.toString(), e);
            }
        }

        /**
         * Gets the description for the group.
         *
         * @return the description for the group.
         */
        public String getDescription() {
            return this.description;
        }

        /**
         * Sets the description for the group.
         *
         * @param description the description for the group.
         */
        public void setDescription(String description) {
            this.description = description;
            addPendingChange("description", description);
        }

        /**
         * Gets the external_sync_identifier for the group.
         *
         * @return the external_sync_identifier for the group.
         */
        public String getExternalSyncIdentifier() {
            return this.externalSyncIdentifier;
        }

        /**
         * Sets the external_sync_identifier for the group.
         *
         * @param externalSyncIdentifier the external_sync_identifier for the group.
         */
        public void setExternalSyncIdentifier(String externalSyncIdentifier) {
            this.externalSyncIdentifier = externalSyncIdentifier;
            addPendingChange("external_sync_identifier", externalSyncIdentifier);
        }

        /**
         * Gets the invitability_level for the group.
         *
         * @return the invitability_level for the group.
         */
        public String getInvitabilityLevel() {
            return this.invitabilityLevel;
        }

        /**
         * Sets the invitability_level for the group.
         *
         * @param invitabilityLevel the invitability_level for the group.
         */
        public void setInvitabilityLevel(String invitabilityLevel) {
            this.invitabilityLevel = invitabilityLevel;
            addPendingChange("invitability_level", invitabilityLevel);
        }

        /**
         * Gets the member_viewability_level for the group.
         *
         * @return the member_viewability_level for the group.
         */
        public String getMemberViewabilityLevel() {
            return this.memberViewabilityLevel;
        }

        /**
         * Sets the member_viewability_level for the group.
         *
         * @param memberViewabilityLevel the member_viewability_level for the group.
         */
        public void setMemberViewabilityLevel(String memberViewabilityLevel) {
            this.memberViewabilityLevel = memberViewabilityLevel;
            addPendingChange("member_viewability_level", memberViewabilityLevel);
        }

        /**
         * Gets the provenance for the group.
         *
         * @return the provenance for the group.
         */
        public String getProvenance() {
            return this.provenance;
        }

        /**
         * Sets the provenance for the group.
         *
         * @param provenance the provenance for the group.
         */
        public void setProvenance(String provenance) {
            this.provenance = provenance;
            addPendingChange("provenance", provenance);
        }
    }
}