WP-Query-Builder/Insertion.php

<?php
namespace DatabaseHelper;

namespace DatabaseHelper;
use DatabaseHelper\enums\ColumnTypes;
use InvalidArgumentException;
use wpdb;

global $wpdb;

class Insertion
{
    private Table $table;
    private array $currentRow = [];
    private array $batchRows = [];
    private bool $isReady = false;

    public function __construct(Table $table) {
        $this->table = $table;
    }

    public function data(string $col, mixed $val): Insertion {
        if (!isset($this->table->columns[$col]))
            throw new InvalidArgumentException("Column '$col' does not exist.");

        $columnType = $this->table->columns[$col]['colType'];
        $this->currentRow[$col] = $columnType->valCast($val);

        return $this;
    }

    public function done(): Insertion {
        if (!empty($this->currentRow)) {
            $this->batchRows[] = $this->currentRow;
            $this->currentRow = [];
        }
        $this->isReady = true;
        return $this;
    }

    public function insert(): void {
        global $wpdb;

        // Convert single to batch queries.
        if (!$this->isReady and !empty($this->currentRow)) {
            $this->batchRows[] = $this->currentRow;
            $this->currentRow = [];
            $this->isReady = true;
        }

        if (empty($this->batchRows))
            throw new InvalidArgumentException("No data set for insertion.");

        foreach ($this->batchRows as $row)
            if (!empty($row))
                $wpdb->insert($this->table->tableName, $row);
    }
}