BulkWriteCommand delete methods

Author: jmikola

src/MongoDB/BulkWriteCommand.c

@@ -25,6 +25,7 @@
 #include "php_phongo.h"
 #include "phongo_bson_encode.h"
 #include "phongo_error.h"
+#include "phongo_util.h"
 #include "BulkWriteCommand_arginfo.h"
 
 #include "MongoDB/WriteConcern.h"
@@ -137,24 +138,184 @@ static PHP_METHOD(MongoDB_Driver_BulkWriteCommand, count)
 	RETURN_LONG(intern->num_ops);
 }
 
+static bool phongo_bwc_parse_hint(zval* zhint, bson_value_t* bhint)
+{
+	if (Z_TYPE_P(zhint) != IS_STRING && Z_TYPE_P(zhint) != IS_OBJECT && Z_TYPE_P(zhint) != IS_ARRAY) {
+		phongo_throw_exception(PHONGO_ERROR_INVALID_ARGUMENT, "Expected \"hint\" option to be string, array, or object, %s given", zend_get_type_by_const(Z_TYPE_P(zhint)));
+		return false;
+	}
+
+	phongo_zval_to_bson_value(zhint, bhint);
+
+	if (EG(exception)) {
+		return false;
+	}
+
+	// Catch the edge case where the option yields a BSON array
+	if (bhint->value_type != BSON_TYPE_UTF8 && bhint->value_type != BSON_TYPE_DOCUMENT) {
+		phongo_throw_exception(PHONGO_ERROR_INVALID_ARGUMENT, "Expected \"hint\" option to yield string or document but got \"%s\"", php_phongo_bson_type_to_string(bhint->value_type));
+		return false;
+	}
+
+	return true;
+}
+
+static bool phongo_bwc_parse_document(zval* zdocument, bson_t* bdocument, const char* key)
+{
+	if (Z_TYPE_P(zdocument) != IS_OBJECT && Z_TYPE_P(zdocument) != IS_ARRAY) {
+		phongo_throw_exception(PHONGO_ERROR_INVALID_ARGUMENT, "Expected \"%s\" option to be array or object, %s given", key, zend_get_type_by_const(Z_TYPE_P(zdocument)));
+		return false;
+	}
+
+	php_phongo_zval_to_bson(zdocument, PHONGO_BSON_NONE, bdocument, NULL);
+
+	if (EG(exception)) {
+		return false;
+	}
+
+	return true;
+}
+
 static PHP_METHOD(MongoDB_Driver_BulkWriteCommand, deleteMany)
 {
-	php_phongo_bulkwritecommand_t* intern;
+	php_phongo_bulkwritecommand_t*     intern;
+	char*                              ns;
+	size_t                             ns_len;
+	zval*                              zfilter;
+	zval*                              zoptions = NULL;
+	bson_t                             bfilter  = BSON_INITIALIZER;
+	mongoc_bulkwrite_deletemanyopts_t* opts     = NULL;
+	bson_error_t                       error    = { 0 };
 
 	intern = Z_BULKWRITECOMMAND_OBJ_P(getThis());
 
-	// TODO: implementation
-	PHONGO_PARSE_PARAMETERS_NONE();
+	PHONGO_PARSE_PARAMETERS_START(2, 3)
+	Z_PARAM_STRING(ns, ns_len)
+	Z_PARAM_ARRAY_OR_OBJECT(zfilter)
+	Z_PARAM_OPTIONAL
+	Z_PARAM_ARRAY_OR_NULL(zoptions)
+	PHONGO_PARSE_PARAMETERS_END();
+
+	if (strlen(ns) != ns_len) {
+		phongo_throw_exception(PHONGO_ERROR_INVALID_ARGUMENT, "Namespace string should not contain null bytes");
+		return;
+	}
+
+	php_phongo_zval_to_bson(zfilter, PHONGO_BSON_NONE, &bfilter, NULL);
+
+	if (EG(exception)) {
+		goto cleanup;
+	}
+
+	if (zoptions) {
+		opts = mongoc_bulkwrite_deletemanyopts_new();
+
+		if (php_array_existsc(zoptions, "hint")) {
+			bson_value_t bhint = { 0 };
+
+			if (!phongo_bwc_parse_hint(php_array_fetchc_deref(zoptions, "hint"), &bhint)) {
+				bson_value_destroy(&bhint);
+				goto cleanup;
+			}
+
+			mongoc_bulkwrite_deletemanyopts_set_hint(opts, &bhint);
+			bson_value_destroy(&bhint);
+		}
+
+		if (php_array_existsc(zoptions, "collation")) {
+			bson_t bcollation = BSON_INITIALIZER;
+
+			if (!phongo_bwc_parse_document(php_array_fetchc_deref(zoptions, "collation"), &bcollation, "collation")) {
+				bson_destroy(&bcollation);
+				goto cleanup;
+			}
+
+			mongoc_bulkwrite_deletemanyopts_set_collation(opts, &bcollation);
+			bson_destroy(&bcollation);
+		}
+	}
+
+	if (!mongoc_bulkwrite_append_deletemany(intern->bw, ns, &bfilter, opts, &error)) {
+		phongo_throw_exception_from_bson_error_t(&error);
+		goto cleanup;
+	}
+
+	intern->num_ops++;
+
+cleanup:
+	bson_destroy(&bfilter);
+	mongoc_bulkwrite_deletemanyopts_destroy(opts);
 }
 
 static PHP_METHOD(MongoDB_Driver_BulkWriteCommand, deleteOne)
 {
-	php_phongo_bulkwritecommand_t* intern;
+	php_phongo_bulkwritecommand_t*    intern;
+	char*                             ns;
+	size_t                            ns_len;
+	zval*                             zfilter;
+	zval*                             zoptions = NULL;
+	bson_t                            bfilter  = BSON_INITIALIZER;
+	mongoc_bulkwrite_deleteoneopts_t* opts     = NULL;
+	bson_error_t                      error    = { 0 };
 
 	intern = Z_BULKWRITECOMMAND_OBJ_P(getThis());
 
-	// TODO: implementation
-	PHONGO_PARSE_PARAMETERS_NONE();
+	PHONGO_PARSE_PARAMETERS_START(2, 3)
+	Z_PARAM_STRING(ns, ns_len)
+	Z_PARAM_ARRAY_OR_OBJECT(zfilter)
+	Z_PARAM_OPTIONAL
+	Z_PARAM_ARRAY_OR_NULL(zoptions)
+	PHONGO_PARSE_PARAMETERS_END();
+
+	if (strlen(ns) != ns_len) {
+		phongo_throw_exception(PHONGO_ERROR_INVALID_ARGUMENT, "Namespace string should not contain null bytes");
+		return;
+	}
+
+	php_phongo_zval_to_bson(zfilter, PHONGO_BSON_NONE, &bfilter, NULL);
+
+	if (EG(exception)) {
+		goto cleanup;
+	}
+
+	if (zoptions) {
+		opts = mongoc_bulkwrite_deleteoneopts_new();
+
+		if (php_array_existsc(zoptions, "hint")) {
+			bson_value_t bhint = { 0 };
+
+			if (!phongo_bwc_parse_hint(php_array_fetchc_deref(zoptions, "hint"), &bhint)) {
+				bson_value_destroy(&bhint);
+				goto cleanup;
+			}
+
+			mongoc_bulkwrite_deleteoneopts_set_hint(opts, &bhint);
+			bson_value_destroy(&bhint);
+		}
+
+		if (php_array_existsc(zoptions, "collation")) {
+			bson_t bcollation = BSON_INITIALIZER;
+
+			if (!phongo_bwc_parse_document(php_array_fetchc_deref(zoptions, "collation"), &bcollation, "collation")) {
+				bson_destroy(&bcollation);
+				goto cleanup;
+			}
+
+			mongoc_bulkwrite_deleteoneopts_set_collation(opts, &bcollation);
+			bson_destroy(&bcollation);
+		}
+	}
+
+	if (!mongoc_bulkwrite_append_deleteone(intern->bw, ns, &bfilter, opts, &error)) {
+		phongo_throw_exception_from_bson_error_t(&error);
+		goto cleanup;
+	}
+
+	intern->num_ops++;
+
+cleanup:
+	bson_destroy(&bfilter);
+	mongoc_bulkwrite_deleteoneopts_destroy(opts);
 }
 
 static PHP_METHOD(MongoDB_Driver_BulkWriteCommand, execute)