/*
 * Copyright 2017 Google
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#import <Foundation/Foundation.h>

#import "FIRFirestoreSwiftNameSupport.h"

NS_ASSUME_NONNULL_BEGIN

@class FIRDocumentReference;
@class FIRSetOptions;

/**
 * A write batch is used to perform multiple writes as a single atomic unit.
 *
 * A WriteBatch object can be acquired by calling [FIRFirestore batch]. It provides methods for
 * adding writes to the write batch. None of the writes will be committed (or visible locally)
 * until [FIRWriteBatch commit] is called.
 *
 * Unlike transactions, write batches are persisted offline and therefore are preferable when you
 * don't need to condition your writes on read data.
 */
FIR_SWIFT_NAME(WriteBatch)
@interface FIRWriteBatch : NSObject

/** :nodoc: */
- (id)init __attribute__((unavailable("FIRWriteBatch cannot be created directly.")));

/**
 * Writes to the document referred to by `document`. If the document doesn't yet exist,
 * this method creates it and then sets the data. If the document exists, this method overwrites
 * the document data with the new values.
 *
 * @param data An `NSDictionary` that contains the fields and data to write to the document.
 * @param document A reference to the document whose data should be overwritten.
 * @return This `FIRWriteBatch` instance. Used for chaining method calls.
 */
// clang-format off
- (FIRWriteBatch *)setData:(NSDictionary<NSString *, id> *)data
               forDocument:(FIRDocumentReference *)document FIR_SWIFT_NAME(setData(_:forDocument:));
// clang-format on

/**
 * Writes to the document referred to by `document`. If the document doesn't yet exist,
 * this method creates it and then sets the data. If you pass `FIRSetOptions`, the provided data
 * will be merged into an existing document.
 *
 * @param data An `NSDictionary` that contains the fields and data to write to the document.
 * @param document A reference to the document whose data should be overwritten.
 * @param options A `FIRSetOptions` used to configure the set behavior.
 * @return This `FIRWriteBatch` instance. Used for chaining method calls.
 */
// clang-format off
- (FIRWriteBatch *)setData:(NSDictionary<NSString *, id> *)data
               forDocument:(FIRDocumentReference *)document
                   options:(FIRSetOptions *)options
    FIR_SWIFT_NAME(setData(_:forDocument:options:));
// clang-format on

/**
 * Updates fields in the document referred to by `document`.
 * If document does not exist, the write batch will fail.
 *
 * @param fields An `NSDictionary` containing the fields (expressed as an `NSString` or
 * `FIRFieldPath`) and values with which to update the document.
 * @param document A reference to the document whose data should be updated.
 * @return This `FIRWriteBatch` instance. Used for chaining method calls.
 */
// clang-format off
- (FIRWriteBatch *)updateData:(NSDictionary<id, id> *)fields
                  forDocument:(FIRDocumentReference *)document
    FIR_SWIFT_NAME(updateData(_:forDocument:));
// clang-format on

/**
 * Deletes the document referred to by `document`.
 *
 * @param document A reference to the document that should be deleted.
 * @return This `FIRWriteBatch` instance. Used for chaining method calls.
 */
- (FIRWriteBatch *)deleteDocument:(FIRDocumentReference *)document
    FIR_SWIFT_NAME(deleteDocument(_:));

/**
 * Commits all of the writes in this write batch as a single atomic unit.
 *
 * @param completion A block to be called once all of the writes in the batch have been
 * successfully written to the backend as an atomic unit.
 */
- (void)commitWithCompletion:(void (^)(NSError *_Nullable error))completion;

@end

NS_ASSUME_NONNULL_END
