package com.box.sdk;

import com.eclipsesource.json.Json;
import com.eclipsesource.json.JsonObject;
import com.eclipsesource.json.JsonValue;
import java.net.URL;
import java.util.Date;

/**
 * Represents a legal hold policy. Legal Hold Policy information describes the basic characteristics of the Policy, such
 * as name, description, and filter dates.
 *
 * @see <a href="https://developer.box.com/reference/resources/legal-hold-policy/">Box legal holds</a>
 *
 * <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("legal_hold")
public class BoxLegalHoldPolicy extends BoxResource {
    /**
     * Legal Hold URL Template.
     */
    public static final URLTemplate LEGAL_HOLD_URL_TEMPLATE = new URLTemplate("legal_hold_policies/%s");
    /**
     * All Legal Hold URL Template.
     */
    public static final URLTemplate ALL_LEGAL_HOLD_URL_TEMPLATE = new URLTemplate("legal_hold_policies");
    /**
     * Legal Hold Assignments URL Template.
     */
    public static final URLTemplate LEGAL_HOLD_ASSIGNMENTS_URL_TEMPLATE
        = new URLTemplate("legal_hold_policies/%s/assignments");
    /**
     * List of File Version Holds URL Template.
     */
    public static final URLTemplate LIST_OF_FILE_VERSION_HOLDS_URL_TEMPLATE
        = new URLTemplate("file_version_legal_holds");
    private static final int DEFAULT_LIMIT = 100;

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

    /**
     * Creates a new Legal Hold Policy.
     *
     * @param api  the API connection to be used by the resource.
     * @param name the name of Legal Hold Policy.
     * @return information about the Legal Hold Policy created.
     */
    public static BoxLegalHoldPolicy.Info create(BoxAPIConnection api, String name) {
        return createOngoing(api, name, null);
    }

    /**
     * Creates a new Legal Hold Policy.
     *
     * @param api             the API connection to be used by the resource.
     * @param name            the name of Legal Hold Policy.
     * @param description     the description of Legal Hold Policy.
     * @param filterStartedAt optional date filter applies to Custodian assignments only.
     * @param filterEndedAt   optional date filter applies to Custodian assignments only.
     * @return information about the Legal Hold Policy created.
     */
    public static BoxLegalHoldPolicy.Info create(BoxAPIConnection api, String name, String description,
                                                 Date filterStartedAt, Date filterEndedAt) {
        URL url = ALL_LEGAL_HOLD_URL_TEMPLATE.build(api.getBaseURL());
        BoxJSONRequest request = new BoxJSONRequest(api, url, "POST");
        JsonObject requestJSON = new JsonObject()
            .add("policy_name", name);
        if (description != null) {
            requestJSON.add("description", description);
        }
        if (filterStartedAt != null) {
            requestJSON.add("filter_started_at", BoxDateFormat.format(filterStartedAt));
        }
        if (filterEndedAt != null) {
            requestJSON.add("filter_ended_at", BoxDateFormat.format(filterEndedAt));
        }
        request.setBody(requestJSON.toString());
        try (BoxJSONResponse response = request.send()) {
            JsonObject responseJSON = Json.parse(response.getJSON()).asObject();
            BoxLegalHoldPolicy createdPolicy = new BoxLegalHoldPolicy(api, responseJSON.get("id").asString());
            return createdPolicy.new Info(responseJSON);
        }
    }

    /**
     * Creates a new ongoing Legal Hold Policy.
     *
     * @param api         the API connection to be used by the resource.
     * @param name        the name of Legal Hold Policy.
     * @param description the description of Legal Hold Policy.
     * @return information about the Legal Hold Policy created.
     */
    public static BoxLegalHoldPolicy.Info createOngoing(BoxAPIConnection api, String name, String description) {
        return createOngoing(api, name, description, null);
    }


    /**
     * Creates a new ongoing Legal Hold Policy.
     *
     * @param api         the API connection to be used by the resource.
     * @param name        the name of Legal Hold Policy.
     * @param description the description of Legal Hold Policy.
     * @param filterStartedAt optional date filter applies to Custodian assignments only.
     * @return information about the Legal Hold Policy created.
     */
    public static BoxLegalHoldPolicy.Info createOngoing(
            BoxAPIConnection api, String name, String description, Date filterStartedAt
    ) {
        URL url = ALL_LEGAL_HOLD_URL_TEMPLATE.build(api.getBaseURL());
        BoxJSONRequest request = new BoxJSONRequest(api, url, "POST");
        JsonObject requestJSON = new JsonObject()
                .add("policy_name", name)
                .add("is_ongoing", true);
        if (description != null) {
            requestJSON.add("description", description);
        }
        if (filterStartedAt != null) {
            requestJSON.add("filter_started_at", BoxDateFormat.format(filterStartedAt));
        }
        request.setBody(requestJSON.toString());
        try (BoxJSONResponse response = request.send()) {
            JsonObject responseJSON = Json.parse(response.getJSON()).asObject();
            BoxLegalHoldPolicy createdPolicy = new BoxLegalHoldPolicy(api, responseJSON.get("id").asString());
            return createdPolicy.new Info(responseJSON);
        }
    }

    /**
     * Retrieves a list of Legal Hold Policies that belong to your Enterprise as an Iterable.
     *
     * @param api api the API connection to be used by the resource.
     * @return the Iterable of Legal Hold Policies in your Enterprise.
     */
    public static Iterable<BoxLegalHoldPolicy.Info> getAll(final BoxAPIConnection api) {
        return getAll(api, null, DEFAULT_LIMIT);
    }

    /**
     * Retrieves a list of Legal Hold Policies that belong to your Enterprise as an Iterable.
     *
     * @param api        api the API connection to be used by the resource.
     * @param policyName case insensitive prefix-match filter on Policy name.
     * @param limit      the limit of retrieved entries per page.
     * @param fields     the optional fields to retrieve.
     * @return the Iterable of Legal Hold Policies in your Enterprise that match the filter parameters.
     */
    public static Iterable<BoxLegalHoldPolicy.Info> getAll(
        final BoxAPIConnection api, String policyName, int limit, String... fields) {
        QueryStringBuilder builder = new QueryStringBuilder();
        if (policyName != null) {
            builder.appendParam("policy_name", policyName);
        }
        if (fields.length > 0) {
            builder.appendParam("fields", fields);
        }
        return new BoxResourceIterable<BoxLegalHoldPolicy.Info>(api,
            ALL_LEGAL_HOLD_URL_TEMPLATE.buildWithQuery(api.getBaseURL(), builder.toString()),
            limit) {

            @Override
            protected BoxLegalHoldPolicy.Info factory(JsonObject jsonObject) {
                BoxLegalHoldPolicy policy = new BoxLegalHoldPolicy(api, jsonObject.get("id").asString());
                return policy.new Info(jsonObject);
            }
        };
    }

    /**
     * Gets information about the Legal Hold.
     *
     * @param fields the fields to retrieve.
     * @return information about this legal hold policy.
     */
    public Info getInfo(String... fields) {
        QueryStringBuilder builder = new QueryStringBuilder();
        if (fields.length > 0) {
            builder.appendParam("fields", fields);
        }
        URL url = LEGAL_HOLD_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);
        }
    }

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

    /**
     * Updates the information about this retention policy with modified locally info.
     * Only policy_name, description and release_notes can be modified.
     *
     * @param info the updated info.
     */
    public void updateInfo(BoxLegalHoldPolicy.Info info) {
        URL url = LEGAL_HOLD_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 responseJSON = Json.parse(response.getJSON()).asObject();
            info.update(responseJSON);
        }
    }

    /**
     * Assigns this legal holds policy to the given box resource.
     * Currently only {@link BoxFile}, {@link BoxFileVersion}, {@link BoxFolder} and {@link BoxUser} are supported.
     *
     * @param resource the box resource to assign legal hold policy to.
     * @return info about created legal hold policy assignment.
     */
    public BoxLegalHoldAssignment.Info assignTo(BoxResource resource) {
        return BoxLegalHoldAssignment.create(
            this.getAPI(), this.getID(), BoxResource.getResourceType(resource.getClass()), resource.getID());
    }

    /**
     * Returns iterable containing assignments for this single legal hold policy.
     *
     * @param fields the fields to retrieve.
     * @return an iterable containing assignments for this single legal hold policy.
     */
    public Iterable<BoxLegalHoldAssignment.Info> getAssignments(String... fields) {
        return this.getAssignments(null, null, DEFAULT_LIMIT, fields);
    }

    /**
     * Returns iterable containing assignments for this single legal hold policy.
     * Parameters can be used to filter retrieved assignments.
     *
     * @param type   filter assignments of this type only.
     *               Can be "file_version", "file", "folder", "user" or null if no type filter is necessary.
     * @param id     filter assignments to this ID only. Can be null if no id filter is necessary.
     * @param limit  the limit of entries per page. Default limit is 100.
     * @param fields the fields to retrieve.
     * @return an iterable containing assignments for this single legal hold policy.
     */
    public Iterable<BoxLegalHoldAssignment.Info> getAssignments(String type, String id, int limit, String... fields) {
        QueryStringBuilder builder = new QueryStringBuilder();
        if (type != null) {
            builder.appendParam("assign_to_type", type);
        }
        if (id != null) {
            builder.appendParam("assign_to_id", id);
        }
        if (fields.length > 0) {
            builder.appendParam("fields", fields);
        }
        return new BoxResourceIterable<BoxLegalHoldAssignment.Info>(
            this.getAPI(), LEGAL_HOLD_ASSIGNMENTS_URL_TEMPLATE.buildWithQuery(
            this.getAPI().getBaseURL(), builder.toString(), this.getID()), limit) {

            @Override
            protected BoxLegalHoldAssignment.Info factory(JsonObject jsonObject) {
                BoxLegalHoldAssignment assignment = new BoxLegalHoldAssignment(
                    BoxLegalHoldPolicy.this.getAPI(), jsonObject.get("id").asString());
                return assignment.new Info(jsonObject);
            }
        };
    }

    /**
     * Returns iterable with all non-deleted file version legal holds for this legal hold policy.
     *
     * @param fields the fields to retrieve.
     * @return an iterable containing file version legal holds info.
     */
    public Iterable<BoxFileVersionLegalHold.Info> getFileVersionHolds(String... fields) {
        return this.getFileVersionHolds(DEFAULT_LIMIT, fields);
    }

    /**
     * Returns iterable with all non-deleted file version legal holds for this legal hold policy.
     *
     * @param limit  the limit of entries per response. The default value is 100.
     * @param fields the fields to retrieve.
     * @return an iterable containing file version legal holds info.
     */
    public Iterable<BoxFileVersionLegalHold.Info> getFileVersionHolds(int limit, String... fields) {
        QueryStringBuilder queryString = new QueryStringBuilder().appendParam("policy_id", this.getID());
        if (fields.length > 0) {
            queryString.appendParam("fields", fields);
        }
        URL url = LIST_OF_FILE_VERSION_HOLDS_URL_TEMPLATE.buildWithQuery(getAPI().getBaseURL(), queryString.toString());
        return new BoxResourceIterable<BoxFileVersionLegalHold.Info>(getAPI(), url, limit) {

            @Override
            protected BoxFileVersionLegalHold.Info factory(JsonObject jsonObject) {
                BoxFileVersionLegalHold assignment
                    = new BoxFileVersionLegalHold(getAPI(), jsonObject.get("id").asString());
                return assignment.new Info(jsonObject);
            }

        };
    }

    /**
     * Contains information about the legal hold policy.
     */
    public class Info extends BoxResource.Info {

        /**
         * @see #getPolicyName()
         */
        private String policyName;

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

        /**
         * @see #getStatus()
         */
        private String status;

        /**
         * @see #getAssignmentCountUser()
         */
        private int assignmentCountUser;

        /**
         * @see #getAssignmentCountFolder()
         */
        private int assignmentCountFolder;

        /**
         * @see #getAssignmentCountFile()
         */
        private int assignmentCountFile;

        /**
         * @see #getAssignmentCountFileVersion()
         */
        private int assignmentCountFileVersion;

        /**
         * @see #getCreatedAt()
         */
        private BoxUser.Info createdBy;

        /**
         * @see #getCreatedAt()
         */
        private Date createdAt;

        /**
         * @see #getModifiedAt()
         */
        private Date modifiedAt;

        /**
         * @see #getDeletedAt()
         */
        private Date deletedAt;

        /**
         * @see #getFilterStartedAt()
         */
        private Date filterStartedAt;

        /**
         * @see #getFilterEndedAt()
         */
        private Date filterEndedAt;

        /**
         * @see #getReleaseNotes()
         */
        private String releaseNotes;

        /**
         * @see #getIsOngoing()
         */
        private Boolean isOngoing;

        /**
         * 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 BoxResource getResource() {
            return BoxLegalHoldPolicy.this;
        }

        /**
         * @return the name of the policy.
         */
        public String getPolicyName() {
            return this.policyName;
        }

        /**
         * Sets the policy's name.
         *
         * @param policyName the policy's new name.
         */
        public void setPolicyName(String policyName) {
            this.policyName = policyName;
            this.addPendingChange("policy_name", policyName);
        }

        /**
         * @return the description of the policy.
         */
        public String getDescription() {
            return this.description;
        }

        /**
         * Sets the policy's description.
         *
         * @param description the policy's new description.
         */
        public void setDescription(String description) {
            this.description = description;
            this.addPendingChange("description", description);
        }

        /**
         * Status can be "active", "applying", "releasing" or "released".
         *
         * @return the status of the policy.
         */
        public String getStatus() {
            return this.status;
        }

        /**
         * @return count of users this policy assigned to.
         */
        public int getAssignmentCountUser() {
            return this.assignmentCountUser;
        }

        /**
         * @return count of folders this policy assigned to.
         */
        public int getAssignmentCountFolder() {
            return this.assignmentCountFolder;
        }

        /**
         * @return count of files this policy assigned to.
         */
        public int getAssignmentCountFile() {
            return this.assignmentCountFile;
        }

        /**
         * @return count of file versions this policy assigned to.
         */
        public int getAssignmentCountFileVersion() {
            return this.assignmentCountFileVersion;
        }

        /**
         * @return info about the user who created this policy.
         */
        public BoxUser.Info getCreatedBy() {
            return this.createdBy;
        }

        /**
         * @return time the policy was created.
         */
        public Date getCreatedAt() {
            return this.createdAt;
        }

        /**
         * @return time the policy was modified.
         */
        public Date getModifiedAt() {
            return this.modifiedAt;
        }

        /**
         * @return time that the policy release request was sent.
         */
        public Date getDeletedAt() {
            return this.deletedAt;
        }

        /**
         * @return optional date filter applies to Custodian assignments only.
         */
        public Date getFilterStartedAt() {
            return this.filterStartedAt;
        }

        /**
         * @return optional date filter applies to Custodian assignments only.
         */
        public Date getFilterEndedAt() {
            return this.filterEndedAt;
        }

        /**
         * @return notes around why the policy was released.
         */
        public String getReleaseNotes() {
            return this.releaseNotes;
        }

        /**
         * Sets the policy's release notes.
         *
         * @param releaseNotes the policy's new release notes.
         */
        public void setReleaseNotes(String releaseNotes) {
            this.releaseNotes = releaseNotes;
            this.addPendingChange("release_notes", releaseNotes);
        }

        /**
         * @return boolean indicating whether the policy will continue applying to files based on events, indefinitely.
         */
        public Boolean getIsOngoing() {
            return this.isOngoing;
        }

        /**
         * {@inheritDoc}
         */
        @Override
        void parseJSONMember(JsonObject.Member member) {
            super.parseJSONMember(member);
            String memberName = member.getName();
            JsonValue value = member.getValue();
            try {
                if (memberName.equals("policy_name")) {
                    this.policyName = value.asString();
                } else if (memberName.equals("description")) {
                    this.description = value.asString();
                } else if (memberName.equals("status")) {
                    this.status = value.asString();
                } else if (memberName.equals("release_notes")) {
                    this.releaseNotes = value.asString();
                } else if (memberName.equals("assignment_counts")) {
                    JsonObject countsJSON = value.asObject();
                    this.assignmentCountUser = countsJSON.get("user").asInt();
                    this.assignmentCountFolder = countsJSON.get("folder").asInt();
                    this.assignmentCountFile = countsJSON.get("file").asInt();
                    this.assignmentCountFileVersion = countsJSON.get("file_version").asInt();
                } else if (memberName.equals("created_by")) {
                    JsonObject userJSON = value.asObject();
                    if (this.createdBy == null) {
                        String userID = userJSON.get("id").asString();
                        BoxUser user = new BoxUser(getAPI(), userID);
                        this.createdBy = user.new Info(userJSON);
                    } else {
                        this.createdBy.update(userJSON);
                    }
                } else if (memberName.equals("created_at")) {
                    this.createdAt = BoxDateFormat.parse(value.asString());
                } else if (memberName.equals("modified_at")) {
                    this.modifiedAt = BoxDateFormat.parse(value.asString());
                } else if (memberName.equals("deleted_at")) {
                    this.deletedAt = BoxDateFormat.parse(value.asString());
                } else if (memberName.equals("filter_started_at")) {
                    this.filterStartedAt = BoxDateFormat.parse(value.asString());
                } else if (memberName.equals("filter_ended_at")) {
                    this.filterEndedAt = BoxDateFormat.parse(value.asString());
                } else if (memberName.equals("is_ongoing")) {
                    this.isOngoing = value.asBoolean();
                }
            } catch (Exception e) {
                throw new BoxDeserializationException(memberName, value.toString(), e);
            }
        }
    }
}