RequestBatch.java

package com.novoda.downloadmanager.lib;

import android.content.ContentValues;

import com.novoda.downloadmanager.notifications.NotificationVisibility;

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

public class RequestBatch {

    private final BatchInfo batchInfo;
    private final List<Request> requests;

    RequestBatch(BatchInfo batchInfo, List<Request> requests) {
        this.batchInfo = batchInfo;
        this.requests = requests;
    }

    public String getTitle() {
        return batchInfo.getTitle();
    }

    public String getDescription() {
        return batchInfo.getDescription();
    }

    public String getBigPictureUrl() {
        return batchInfo.getBigPictureUrl();
    }

    public String getExtraData() {
        return batchInfo.getExtraData();
    }

    List<Request> getRequests() {
        return requests;
    }

    public void addRequest(Request request) {
        requests.add(request);
    }

    ContentValues toContentValues() {
        ContentValues values = new ContentValues();
        values.put(DownloadContract.Batches.COLUMN_TITLE, batchInfo.getTitle());
        values.put(DownloadContract.Batches.COLUMN_DESCRIPTION, batchInfo.getDescription());
        values.put(DownloadContract.Batches.COLUMN_BIG_PICTURE, batchInfo.getBigPictureUrl());
        values.put(DownloadContract.Batches.COLUMN_VISIBILITY, batchInfo.getVisibility());
        values.put(DownloadContract.Batches.COLUMN_EXTRA_DATA, batchInfo.getExtraData());
        return values;
    }

    public static class Builder {

        private String title;
        private String description;
        private String bigPictureUrl;
        @NotificationVisibility.Value
        private int visibility;
        private String extraData;

        public Builder withTitle(String title) {
            this.title = title;
            return this;
        }

        public Builder withDescription(String description) {
            this.description = description;
            return this;
        }

        public Builder withBigPictureUrl(String bigPictureUrl) {
            this.bigPictureUrl = bigPictureUrl;
            return this;
        }

        public Builder withVisibility(@NotificationVisibility.Value int visibility) {
            this.visibility = visibility;
            return this;
        }

        public Builder withExtraData(String extraData) {
            this.extraData = extraData;
            return this;
        }

        public RequestBatch build() {
            BatchInfo batchInfo = new BatchInfo(title, description, bigPictureUrl, visibility, extraData);
            return new RequestBatch(batchInfo, new ArrayList<Request>());
        }

    }
}