BartRunnerAndroid/src/com/dougkeen/bart/activities/AddRouteDialogFragment.java

package com.dougkeen.bart.activities;

import android.content.ContentValues;
import android.view.View;
import android.widget.CheckBox;

import com.dougkeen.bart.R;
import com.dougkeen.bart.data.RoutesColumns;
import com.dougkeen.bart.model.Constants;
import com.dougkeen.bart.model.Station;

public class AddRouteDialogFragment extends AbstractRouteSelectionFragment {
	public AddRouteDialogFragment(String title) {
		super(title);
	}

	@Override
	public void onStart() {
		super.onStart();
		final View checkbox = getDialog().findViewById(R.id.return_checkbox);
		checkbox.setVisibility(View.VISIBLE);
	}

	@Override
	protected void onOkButtonClick(Station origin, Station destination) {
		ContentValues values = new ContentValues();
		values.put(RoutesColumns.FROM_STATION.string, origin.abbreviation);
		values.put(RoutesColumns.TO_STATION.string, destination.abbreviation);

		getActivity().getContentResolver().insert(
				Constants.FAVORITE_CONTENT_URI, values);

		if (((CheckBox) getDialog().findViewById(R.id.return_checkbox))
				.isChecked()) {
			values = new ContentValues();
			values.put(RoutesColumns.FROM_STATION.string,
					destination.abbreviation);
			values.put(RoutesColumns.TO_STATION.string, origin.abbreviation);

			getActivity().getContentResolver().insert(
					Constants.FAVORITE_CONTENT_URI, values);
		}

		dismiss();
	}

}