001/*
002 * The contents of this file are subject to the terms of the Common Development and
003 * Distribution License (the License). You may not use this file except in compliance with the
004 * License.
005 *
006 * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
007 * specific language governing permission and limitations under the License.
008 *
009 * When distributing Covered Software, include this CDDL Header Notice in each file and include
010 * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
011 * Header, with the fields enclosed by brackets [] replaced by your own identifying
012 * information: "Portions Copyright [year] [name of copyright owner]".
013 *
014 * Copyright 2015 ForgeRock AS.
015 *
016 */
017
018package org.forgerock.openig.resolver;
019
020import org.forgerock.http.protocol.Headers;
021
022/**
023 * Resolves {@link Headers} objects.
024 */
025public class HeadersResolver implements Resolver {
026
027    @Override
028    public Class<?> getKey() {
029        return Headers.class;
030    }
031
032    @Override
033    public Object get(Object object, Object element) {
034        if (object instanceof Headers) {
035            Headers headers = (Headers) object;
036            String key = element.toString();
037            if (headers.containsKey(key)) {
038                // We just do the getValues() on behalf of the script so we don't have to write .values in expressions
039                return headers.get(key).getValues();
040            }
041        }
042        return Resolver.UNRESOLVED;
043    }
044
045    @Override
046    public Object put(Object object, Object element, Object value) {
047        if (object instanceof Headers) {
048            Headers headers = (Headers) object;
049            String key = element.toString();
050            return headers.put(key, value);
051        }
052        return Resolver.UNRESOLVED;
053    }
054}