From 92f6800a0149723f5f291382ee21faaad0e3ad9b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lucas=20Mez=C3=AAncio?= <lucas.mezencio@prodemge.gov.br>
Date: Fri, 19 Aug 2016 14:09:04 -0300
Subject: [PATCH] Add z/OS Connector and Processor

---
 src/Connectors/ODBCZOSConnector.php      | 40 +++++++++++++++++++
 src/Query/Processors/DB2ZOSProcessor.php | 49 ++++++++++++++++++++++++
 2 files changed, 89 insertions(+)
 create mode 100644 src/Connectors/ODBCZOSConnector.php
 create mode 100644 src/Query/Processors/DB2ZOSProcessor.php

diff --git a/src/Connectors/ODBCZOSConnector.php b/src/Connectors/ODBCZOSConnector.php
new file mode 100644
index 0000000..0a36d6a
--- /dev/null
+++ b/src/Connectors/ODBCZOSConnector.php
@@ -0,0 +1,40 @@
+<?php
+
+namespace Cooperl\Database\DB2\Connectors;
+
+/**
+ * Class ODBCZOSConnector
+ *
+ * @package Cooperl\Database\DB2\Connectors
+ */
+class ODBCZOSConnector extends ODBCConnector
+{
+    /**
+     * @param array $config
+     *
+     * @return string
+     */
+    protected function getDsn(array $config)
+    {
+        $dsnParts = [
+            'odbc:DRIVER={IBM DB2 ODBC DRIVER}',
+            'Database=%s',
+            'Hostname=%s',
+            'Port=%s',
+            'Protocol=TCPIP',
+            'Uid=%s',
+            'Pwd=%s',
+            '', // Just to add a semicolon to the end of string
+        ];
+
+        $dsnConfig = [
+            $config['database'],
+            $config['host'],
+            $config['port'],
+            $config['username'],
+            $config['password'],
+        ];
+
+        return sprintf(implode(';', $dsnParts), ...$dsnConfig);
+    }
+}
diff --git a/src/Query/Processors/DB2ZOSProcessor.php b/src/Query/Processors/DB2ZOSProcessor.php
new file mode 100644
index 0000000..59ae6cc
--- /dev/null
+++ b/src/Query/Processors/DB2ZOSProcessor.php
@@ -0,0 +1,49 @@
+<?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();
+            $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;
+        }
+    }
+}