<?php

namespace Cooperl\Database\DB2\Query\Processors;

use Illuminate\Database\Query\Builder;
use Illuminate\Database\Query\Processors\Processor;
use Cooperl\Database\DB2\Query\Grammars\DB2Grammar;

/**
 * Class DB2ZOSProcessor
 *
 * @package Cooperl\Database\DB2\Query\Processors
 */
class DB2ZOSProcessor extends Processor
{
    /**
     * Process an "insert get ID" query.
     *
     * @param  \Illuminate\Database\Query\Builder $query
     * @param  string $sql
     * @param  array $values
     * @param  string $sequence
     *
     * @return int/array
     */
    public function processInsertGetId(Builder $query, $sql, $values, $sequence = null)
    {
        $sequenceStr = $sequence ?: 'id';

        if (is_array($sequence)) {
            $grammar = new DB2Grammar("z");
            $sequenceStr = $grammar->columnize($sequence);
        }

        $sqlStr = 'select %s from final table (%s)';

        $finalSql = sprintf($sqlStr, $sequenceStr, $sql);
        $results = $query->getConnection()->select($finalSql, $values);

        if (is_array($sequence)) {
            return array_values((array) $results[0]);
        } else {
            $result = (array) $results[0];
            $id = $result[strtolower($sequenceStr)];

            return is_numeric($id) ? (int) $id : $id;
        }
    }
}