org.dmfs.httpclient.headers.UpdatedHeaders Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-client-headers Show documentation
HTTP header implementations for http-client-essentials.
There is a newer version: 1.22.1
/*
 * Copyright (C) 2016 Marten Gajda 
 *
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.dmfs.httpclient.headers;

import java.util.Iterator;
import java.util.List;

import org.dmfs.iterators.AbstractFilteredIterator.IteratorFilter;
import org.dmfs.iterators.FilteredIterator;
import org.dmfs.iterators.SerialIterator;
import org.dmfs.iterators.SingletonIterator;


/**
 * {@link Headers} that add or replace a {@link Header}.
 * 
 * @author Marten Gajda 
 */
public final class UpdatedHeaders implements Headers
{
	private final Headers mHeaders;
	private final Header mNewHeader;


	/**
	 * Create new {@link Headers} that update a single header of other {@link Headers}.
	 * 
	 * @param headers
	 *            The original headers.
	 * @param updatedHeader
	 *            The updated header.
	 */
	public UpdatedHeaders(Headers headers, Header updatedHeader)
	{
		mHeaders = headers;
		mNewHeader = updatedHeader;
	}


	@Override
	public Iterator> iterator()
	{
		return new SerialIterator>(new FilteredIterator>(mHeaders.iterator(), new IteratorFilter>()
		{
			@Override
			public boolean iterate(Header element)
			{
				return !element.type().equals(mNewHeader.type());
			}
		}), new SingletonIterator>(mNewHeader));
	}


	@Override
	public boolean contains(HeaderType headerType)
	{
		return mNewHeader.type().equals(headerType) || mHeaders.contains(headerType);
	}


	@SuppressWarnings("unchecked")
	@Override
	public  Header header(SingletonHeaderType headerType)
	{
		return mNewHeader.type().equals(headerType) ? (Header) mNewHeader : mHeaders.header(headerType);
	}


	@SuppressWarnings("unchecked")
	@Override
	public  Header> header(ListHeaderType headerType)
	{
		return mNewHeader.type().equals(headerType) ? (Header>) mNewHeader : mHeaders.header(headerType);
	}


	@Override
	public  Headers withHeader(Header header)
	{
		if (mNewHeader.type().equals(header.type()))
		{
			return new UpdatedHeaders(mHeaders, header);
		}
		return new UpdatedHeaders(this, header);
	}


	@Override
	public  Headers withoutHeaderType(HeaderType headerType)
	{
		if (mNewHeader.type().equals(headerType))
		{
			return mHeaders.withoutHeaderType(headerType);
		}
		if (mHeaders.contains(headerType))
		{
			return new UpdatedHeaders(mHeaders.withoutHeaderType(headerType), mNewHeader);
		}
		return this;
	}
}