/*
   Separated List Adapter
   Taken from http://jsharkey.org/blog/2008/08/18/separating-lists-with-headers-in-android-09/

   Copyright Jeff Sharkey

   This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
   If a copy of the MPL was not distributed with this file, You can obtain one at
   http://mozilla.org/MPL/2.0/.
*/

package com.freerdp.freerdpcore.utils;

import android.content.Context;
import android.view.View;
import android.view.ViewGroup;
import android.widget.Adapter;
import android.widget.ArrayAdapter;
import android.widget.BaseAdapter;

import com.freerdp.freerdpcore.R;

import java.util.LinkedHashMap;
import java.util.Map;

public class SeparatedListAdapter extends BaseAdapter
{

	public final static int TYPE_SECTION_HEADER = 0;
	public final Map<String, Adapter> sections = new LinkedHashMap<>();
	public final ArrayAdapter<String> headers;

	public SeparatedListAdapter(Context context)
	{
		headers = new ArrayAdapter<>(context, R.layout.list_header);
	}

	public void addSection(String section, Adapter adapter)
	{
		this.headers.add(section);
		this.sections.put(section, adapter);
	}

	public void setSectionTitle(int section, String title)
	{
		String oldTitle = this.headers.getItem(section);

		// remove/add to headers array
		this.headers.remove(oldTitle);
		this.headers.insert(title, section);

		// remove/add to section map
		Adapter adapter = this.sections.get(oldTitle);
		this.sections.remove(oldTitle);
		this.sections.put(title, adapter);
	}

	public Object getItem(int position)
	{
		for (int i = 0; i < headers.getCount(); i++)
		{
			String section = headers.getItem(i);
			Adapter adapter = sections.get(section);

			// ignore empty sections
			if (adapter.getCount() > 0)
			{
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position == 0)
					return section;
				if (position < size)
					return adapter.getItem(position - 1);

				// otherwise jump into next section
				position -= size;
			}
		}
		return null;
	}

	public int getCount()
	{
		// total together all sections, plus one for each section header (except if the section is
		// empty)
		int total = 0;
		for (Adapter adapter : this.sections.values())
			total += ((adapter.getCount() > 0) ? adapter.getCount() + 1 : 0);
		return total;
	}

	public int getViewTypeCount()
	{
		// assume that headers count as one, then total all sections
		int total = 1;
		for (Adapter adapter : this.sections.values())
			total += adapter.getViewTypeCount();
		return total;
	}

	public int getItemViewType(int position)
	{
		int type = 1;
		for (int i = 0; i < headers.getCount(); i++)
		{
			String section = headers.getItem(i);
			Adapter adapter = sections.get(section);

			// skip empty sections
			if (adapter.getCount() > 0)
			{
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position == 0)
					return TYPE_SECTION_HEADER;
				if (position < size)
					return type + adapter.getItemViewType(position - 1);

				// otherwise jump into next section
				position -= size;
				type += adapter.getViewTypeCount();
			}
		}
		return -1;
	}

	public boolean areAllItemsSelectable()
	{
		return false;
	}

	public boolean isEnabled(int position)
	{
		return (getItemViewType(position) != TYPE_SECTION_HEADER);
	}

	@Override public View getView(int position, View convertView, ViewGroup parent)
	{
		int sectionnum = 0;
		for (int i = 0; i < headers.getCount(); i++)
		{
			String section = headers.getItem(i);
			Adapter adapter = sections.get(section);

			// skip empty sections
			if (adapter.getCount() > 0)
			{
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position == 0)
					return headers.getView(sectionnum, convertView, parent);
				if (position < size)
					return adapter.getView(position - 1, null, parent);

				// otherwise jump into next section
				position -= size;
			}
			sectionnum++;
		}
		return null;
	}

	@Override public long getItemId(int position)
	{
		for (int i = 0; i < headers.getCount(); i++)
		{
			String section = headers.getItem(i);
			Adapter adapter = sections.get(section);
			if (adapter.getCount() > 0)
			{
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position < size)
					return adapter.getItemId(position - 1);

				// otherwise jump into next section
				position -= size;
			}
		}
		return -1;
	}

	public String getSectionForPosition(int position)
	{
		int curPos = 0;
		for (int i = 0; i < headers.getCount(); i++)
		{
			String section = headers.getItem(i);
			Adapter adapter = sections.get(section);
			if (adapter.getCount() > 0)
			{
				int size = adapter.getCount() + 1;

				// check if position inside this section
				if (position >= curPos && position < (curPos + size))
					return section;

				// otherwise jump into next section
				curPos += size;
			}
		}
		return null;
	}
}