package com.mobify.astro.dialogs;

import android.app.AlertDialog;
import android.app.Dialog;
import android.app.DialogFragment;
import android.os.Bundle;
import android.support.annotation.NonNull;

public class AlertFragment extends DialogFragment {

    protected AlertDialog.Builder alertBuilder;

    public void setAlertBuilder(@NonNull AlertDialog.Builder alertBuilder) {
        this.alertBuilder = alertBuilder;
    }

    @Override
    public Dialog onCreateDialog(Bundle savedInstanceState) {
        if (alertBuilder == null) {
            return null;
        }
        return alertBuilder.create();
    }
}