xref: /external/robolectric/src/main/java/com/xtremelabs/robolectric/shadows/ShadowTabHost.java

package com.xtremelabs.robolectric.shadows;

import android.app.Activity;
import android.view.View;
import android.widget.TabHost;
import com.xtremelabs.robolectric.Robolectric;
import com.xtremelabs.robolectric.internal.Implementation;
import com.xtremelabs.robolectric.internal.Implements;
import com.xtremelabs.robolectric.internal.RealObject;

import java.util.ArrayList;
import java.util.List;

import static com.xtremelabs.robolectric.Robolectric.shadowOf;

@SuppressWarnings({"UnusedDeclaration"})
@Implements(TabHost.class)
public class ShadowTabHost extends ShadowFrameLayout {
    private List<TabHost.TabSpec> tabSpecs = new ArrayList<TabHost.TabSpec>();
    private TabHost.OnTabChangeListener listener;
    private TabHost.TabSpec currentTab;

    @RealObject
    TabHost realObject;

    @Implementation
    public android.widget.TabHost.TabSpec newTabSpec(java.lang.String tag) {
        TabHost.TabSpec realTabSpec = Robolectric.newInstanceOf(TabHost.TabSpec.class);
        shadowOf(realTabSpec).setTag(tag);
        return realTabSpec;
    }

    @Implementation
    public void addTab(android.widget.TabHost.TabSpec tabSpec) {
        tabSpecs.add(tabSpec);
        View indicatorAsView = shadowOf(tabSpec).getIndicatorAsView();
        if (indicatorAsView != null) {
            realObject.addView(indicatorAsView);
        }
    }

    @Implementation
    public void setCurrentTab(int index) {
        currentTab = tabSpecs.get(index);
        if (listener != null) {
            listener.onTabChanged(currentTab.getTag());
        }
    }

    @Implementation
    public void setCurrentTabByTag(String tag) {
        for (TabHost.TabSpec tabSpec : tabSpecs) {
            if (tag.equals(tabSpec.getTag())) {
                currentTab = tabSpec;
            }
        }
        if (listener != null) {
            listener.onTabChanged(currentTab.getTag());
        }
    }

    @Implementation
    public String getCurrentTabTag() {
    	return currentTab.getTag();
    }

    @Implementation
    public void setOnTabChangedListener(android.widget.TabHost.OnTabChangeListener listener) {
        this.listener = listener;
    }
    @Implementation
    public View getCurrentView() {
    	ShadowTabSpec ts = Robolectric.shadowOf(currentTab);
    	View v = ts.getContentView(); //just get the view if it was built by a TabContentFactory
    	if (v==null) {
    		//otherwise findViewById
	    	int viewId = ts.getContentViewId();
	    	if (getContext() instanceof Activity) {
	    		/** wasn't quite sure how to find the view here, simply using
	    		findViewById() wasn't enough, I had to get it though the
	    		Activity context.**/
	    		v = ((Activity) getContext()).findViewById(viewId);
	    	} else {
	    	//	throw new RuntimeException("getContext is not an Activity, unable to findViewById without an Activity context");
	    		return null;
	    	}
    	}
    	return v;
    }

    public TabHost.TabSpec getSpecByTag(String tag) {
        for (TabHost.TabSpec tabSpec : tabSpecs) {
            if (tag.equals(tabSpec.getTag())) {
                return tabSpec;
            }
        }
        return null;
    }
}