<?php

namespace CodesVault\Howdyqb\Statement;

use CodesVault\Howdyqb\Api\CreateInterface;
use CodesVault\Howdyqb\SqlGenerator;
use CodesVault\Howdyqb\Utilities;

class Create implements CreateInterface
{
    protected $db;
    public $sql = [];
    protected $params = [];
    protected $column_name;
    protected $wpdb_object;
    protected $table_name;

    public function __construct($db, string $table_name)
    {
        $this->db = $db;
        $this->table_name = $table_name;

        $this->start();
        $this->sql['table_name'] = $this->get_table_name();
    }

    public function column(string $column_name): self
    {
        $this->column_name = $column_name;
        $this->sql['columns'][$column_name] = [];
        return $this;
    }

    public function int(int $size = 255): self
    {
        $this->sql['columns'][$this->column_name]['int'] = "INT($size)";
        return $this;
    }

    public function bigInt(int $size = 255): self
    {
        $this->sql['columns'][$this->column_name]['bigInt'] = "BIGINT($size)";
        return $this;
    }

    public function double(): self
    {
        $this->sql['columns'][$this->column_name]['double'] = "DOUBLE";
        return $this;
    }

    public function boolean(): self
    {
        $this->sql['columns'][$this->column_name]['boolean'] = "BOOLEAN";
        return $this;
    }

    public function string(int $size = 255): self
    {
        $this->sql['columns'][$this->column_name]['string'] = "VARCHAR($size)";
        return $this;
    }

    public function text(int $size = 10000): self
    {
        $this->sql['columns'][$this->column_name]['text'] = "TEXT($size)";
        return $this;
    }

    public function longText(): self
    {
        $this->sql['columns'][$this->column_name]['longText'] = "LONGTEXT";
        return $this;
    }

	public function json(): self
    {
        $this->sql['columns'][$this->column_name]['json'] = "JSON";
        return $this;
    }

    public function required(): self
    {
        $this->sql['columns'][$this->column_name]['required'] = "NOT NULL";
        return $this;
    }

    public function nullable(): self
    {
        return $this->default('NULL');
    }

    public function primary($columns = []): self
    {
        if (! empty($columns)) {
            $this->sql['primary'] = "PRIMARY KEY (" . implode(',', $columns) . ")";
            return $this;
        }
        $this->sql['columns'][$this->column_name]['primary'] = "PRIMARY KEY";
        return $this;
    }

    public function index(array $columns): self
    {
        $this->sql['index'] = "INDEX (" . implode(',', $columns) . ")";
        return $this;
    }

    public function date(): self
    {
        $this->sql['columns'][$this->column_name]['date'] = "DATE";
        return $this;
    }

    public function dateTime(): self
    {
        $this->sql['columns'][$this->column_name]['dateTime'] = "DATETIME";
        return $this;
    }

	public function timestamp($default = null, $on_update = null): self
	{
		$this->sql['columns'][$this->column_name]['timestamp'] = "TIMESTAMP";

		if ($default === 'now') {
			$this->sql['columns'][$this->column_name]['default'] = "DEFAULT CURRENT_TIMESTAMP";
		} elseif ($default && $default !== 'now') {
			$this->sql['columns'][$this->column_name]['default'] = "DEFAULT " . $default;
		}

		if ($on_update === 'current') {
			$this->sql['columns'][$this->column_name]['onUpdate'] = "ON UPDATE CURRENT_TIMESTAMP";
		} elseif ($on_update && $on_update !== 'current') {
			$this->sql['columns'][$this->column_name]['onUpdate'] = "ON UPDATE " . $on_update;
		}

		return $this;
	}

    public function unsigned(): self
    {
        $this->sql['columns'][$this->column_name]['unsigned'] = "UNSIGNED";
        return $this;
    }

    public function autoIncrement(): self
    {
        $this->sql['columns'][$this->column_name]['autoIncrement'] = "AUTO_INCREMENT";
        return $this;
    }

    public function default($value): self
    {
        $val = is_string($value) ? "'$value'" : $value;
        $this->sql['columns'][$this->column_name]['default'] = "DEFAULT $val";
        return $this;
    }

    public function foreignKey(string $column, string $ref_table_column, ?string $on_delete = null): self
    {
		$ref_table_column = explode('.', $ref_table_column);
        $table_name = Utilities::get_db_configs()->prefix . $ref_table_column[0];
        $this->sql['foreignKey'] = "FOREIGN KEY ($column) REFERENCES $table_name ($ref_table_column[1])";

		if ($on_delete) {
			$this->sql['foreignKey'] .= " ON DELETE " . strtoupper($on_delete);
		}
        return $this;
    }

    public function onDelete(string $action): self
    {
        $this->sql['onDelete'] = "ON DELETE " . strtoupper($action);
        return $this;
    }

	public function enum(array $allowed): self
	{
		$list = '';
		foreach ($allowed as $value) {
			if (gettype($value) === 'string') {
				$list .= "'$value', ";
			} else {
				$list .= $value . ", ";
			}
		}

		$list = substr(trim($list), 0, -1);

		$this->sql['columns'][$this->column_name]['enum'] = "ENUM(" . $list . ")";
		return $this;
	}

	public function decimal($precision = 8, $scale = 2): self
	{
		$this->sql['columns'][$this->column_name]['decimal'] = "DECIMAL($precision, $scale)";
		return $this;
	}

	public function float(): self
	{
		$this->sql['columns'][$this->column_name]['float'] = "FLOAT";
		return $this;
	}

	public function unique(): self
	{
		if (! isset($this->sql['unique'])) {
			$this->sql['unique'] = [];
		}
		array_push($this->sql['unique'], $this->column_name);
		return $this;
	}

    protected function start()
    {
        $this->sql['start'] = 'CREATE TABLE IF NOT EXISTS';
    }

    protected function get_table_name()
    {
        return Utilities::get_db_configs()->prefix . $this->table_name;
    }

    // get only sql query string
    public function getSql()
    {
        $this->start();
        $query = [
            'query' => SqlGenerator::create($this->sql),
        ];
        return $query;
    }

    private function driver_execute($sql)
    {
        $driver = $this->db;
        if (class_exists('wpdb') && $driver instanceof \wpdb) {
            return $driver->query($sql);
        }

		try {
			return $driver->exec($sql);
        } catch (\PDOException $exception) {
            Utilities::throughException($exception);
        }
    }

    public function execute()
    {
        $this->start();
        $query = SqlGenerator::create($this->sql);

        $this->driver_execute($query);
    }
}
