org.elasticsearch.client.security.PutPrivilegesResponse Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch-rest-high-level-client Show documentation
Elasticsearch subproject :client:rest-high-level
There is a newer version: 8.0.0-alpha2
Show newest version
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */

package org.elasticsearch.client.security;

import org.elasticsearch.common.ParsingException;
import org.elasticsearch.common.Strings;
import org.elasticsearch.xcontent.XContentParser;

import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

/**
 * Response when creating/updating one or more application privileges to the
 * security index.
 */
public final class PutPrivilegesResponse {

    /*
     * Map of application name to a map of privilege name to boolean denoting
     * created or update status.
     */
    private final Map> applicationPrivilegesCreatedOrUpdated;

    public PutPrivilegesResponse(final Map> applicationPrivilegesCreatedOrUpdated) {
        this.applicationPrivilegesCreatedOrUpdated = Collections.unmodifiableMap(applicationPrivilegesCreatedOrUpdated);
    }

    /**
     * Get response status for the request to create or update application
     * privileges.
     *
     * @param applicationName application name as specified in the request
     * @param privilegeName privilege name as specified in the request
     * @return {@code true} if the privilege was created, {@code false} if the
     * privilege was updated
     * @throws IllegalArgumentException thrown for unknown application name or
     * privilege name.
     */
    public boolean wasCreated(final String applicationName, final String privilegeName) {
        if (Strings.hasText(applicationName) == false) {
            throw new IllegalArgumentException("application name is required");
        }
        if (Strings.hasText(privilegeName) == false) {
            throw new IllegalArgumentException("privilege name is required");
        }
        if (applicationPrivilegesCreatedOrUpdated.get(applicationName) == null
            || applicationPrivilegesCreatedOrUpdated.get(applicationName).get(privilegeName) == null) {
            throw new IllegalArgumentException("application name or privilege name not found in the response");
        }
        return applicationPrivilegesCreatedOrUpdated.get(applicationName).get(privilegeName);
    }

    @SuppressWarnings("unchecked")
    public static PutPrivilegesResponse fromXContent(final XContentParser parser) throws IOException {
        final Map> applicationPrivilegesCreatedOrUpdated = new HashMap<>();
        XContentParser.Token token = parser.currentToken();
        if (token == null) {
            token = parser.nextToken();
        }
        final Map appNameToPrivStatus = parser.map();
        for (Entry entry : appNameToPrivStatus.entrySet()) {
            if (entry.getValue() instanceof Map) {
                final Map privilegeToStatus = applicationPrivilegesCreatedOrUpdated.computeIfAbsent(
                    entry.getKey(),
                    (a) -> new HashMap<>()
                );
                final Map createdOrUpdated = (Map) entry.getValue();
                for (String privilegeName : createdOrUpdated.keySet()) {
                    if (createdOrUpdated.get(privilegeName) instanceof Map) {
                        final Map statusMap = (Map) createdOrUpdated.get(privilegeName);
                        final Object status = statusMap.get("created");
                        if (status instanceof Boolean) {
                            privilegeToStatus.put(privilegeName, (Boolean) status);
                        } else {
                            throw new ParsingException(parser.getTokenLocation(), "Failed to parse object, unexpected structure");
                        }
                    } else {
                        throw new ParsingException(parser.getTokenLocation(), "Failed to parse object, unexpected structure");
                    }
                }
            } else {
                throw new ParsingException(parser.getTokenLocation(), "Failed to parse object, unexpected structure");
            }
        }
        return new PutPrivilegesResponse(applicationPrivilegesCreatedOrUpdated);
    }
}