UndoBarView.java

package com.jensdriller.libs.undobar;

import android.annotation.TargetApi;
import android.content.Context;
import android.os.Build;
import android.os.Parcel;
import android.os.Parcelable;
import android.util.AttributeSet;
import android.widget.TextView;

class UndoBarView extends MaxWidthRelativeLayout {

	public UndoBarView(Context context) {
		super(context);
	}

	public UndoBarView(Context context, AttributeSet attrs) {
		super(context, attrs);
	}

	public UndoBarView(Context context, AttributeSet attrs, int defStyle) {
		super(context, attrs, defStyle);
	}

    @TargetApi(Build.VERSION_CODES.LOLLIPOP)
    public UndoBarView(Context context, AttributeSet attrs, int defStyleAttr, int defStyleRes) {
        super(context, attrs, defStyleAttr, defStyleRes);
    }

    private TextView mMessage;
	private TextView mButton;

	@Override
	protected void onFinishInflate() {
		super.onFinishInflate();
		
		mMessage = (TextView) findViewById(R.id.message);
		mButton = (TextView) findViewById(R.id.button);
	}

	void setMessage(CharSequence message) {
		mMessage.setText(message);
	}

    void setButtonLabel(int buttonLabelResId) {
        mButton.setText(buttonLabelResId);
    }

    void setUndoColor(int color) {
        mButton.setTextColor(color);
    }

	void setOnUndoClickListener(OnClickListener onClickListener) {
		mButton.setOnClickListener(onClickListener);
	}

	@Override
	public Parcelable onSaveInstanceState() {
		Parcelable superState = super.onSaveInstanceState();
		SavedState ss = new SavedState(superState);
		ss.message = mMessage.getText().toString();
		return ss;
	}

	@Override
	public void onRestoreInstanceState(Parcelable state) {
		if (!(state instanceof SavedState)) {
			super.onRestoreInstanceState(state);
			return;
		}

		SavedState ss = (SavedState) state;
		super.onRestoreInstanceState(ss.getSuperState());
		setMessage(ss.message);
	}

	@SuppressWarnings("NullableProblems")
    private static class SavedState extends BaseSavedState {

		String message;

		SavedState(Parcelable superState) {
			super(superState);
		}

		@Override
		public void writeToParcel(Parcel dest, int flags) {
			super.writeToParcel(dest, flags);
			dest.writeString(message);
		}

		private SavedState(Parcel in) {
			super(in);
			message = in.readString();
		}

		public static final Parcelable.Creator<SavedState> CREATOR = new Parcelable.Creator<SavedState>() {

			@Override
			public SavedState createFromParcel(Parcel in) {
				return new SavedState(in);
			}

			@Override
			public SavedState[] newArray(int size) {
				return new SavedState[size];
			}
		};
	}
}