<?php

function ends_with($str, $key) {
    return strrpos($str, $key) == strlen($str) - strlen($key);
}

function ctype_alnum_underscore($str) {
    return ctype_alnum(str_replace('_', '', $str));
}

function &array_ensure_item_has_array(&$array, $key) {
    if (!array_key_exists($key, $array))
        $array[$key] = array();
    return $array[$key];
}

function array_get($array, $key, $default = NULL) {
    if (!array_key_exists($key, $array))
        return $default;
    return $array[$key];
}

function array_set_default(&$array, $key, $default) {
    if (!array_key_exists($key, $array))
        $array[$key] = $default;
}

$_config = NULL;

define('CONFIG_DIR', realpath(dirname(__FILE__) . '/../../'));

function config($key, $default = NULL) {
    global $_config;
    if (!$_config) {
        $file_path = getenv('ORG_WEBKIT_PERF_CONFIG_PATH');
        if (!$file_path)
            $file_path = CONFIG_DIR . '/config.json';
        $_config = json_decode(file_get_contents($file_path), true);
    }
    return array_get($_config, $key, $default);
}

function config_path($key, $path) {
    return CONFIG_DIR . '/' . config($key) . '/' . $path;
}

function generate_data_file($filename, $content) {
    if (!assert(ctype_alnum(str_replace(array('-', '_', '.'), '', $filename))))
        return FALSE;
    return file_put_contents(config_path('dataDirectory', $filename), $content, LOCK_EX);
}

if (config('debug')) {
    error_reporting(E_ALL | E_STRICT);
    ini_set('display_errors', 'On');
} else
    error_reporting(E_ERROR);

date_default_timezone_set('UTC');

class Database
{
    private $connection = false;

    function __destruct() {
        if ($this->connection)
            pg_close($this->connection);
        $this->connection = false;
    }

    static function is_true($value) {
        return $value == 't';
    }

    static function to_database_boolean($value) {
        return $value ? 't' : 'f';
    }

    static function to_js_time($time_str) {
        $timestamp_in_s = strtotime($time_str);
        $dot_index = strrpos($time_str, '.');
        if ($dot_index !== FALSE)
            $timestamp_in_s += floatval(substr($time_str, $dot_index));
        return intval($timestamp_in_s * 1000);
    }

    static function escape_for_like($string) {
        return str_replace(array('\\', '_', '%'), array('\\\\', '\\_', '\\%'), $string);
    }

    function connect() {
        $databaseConfig = config('database');
        $this->connection = @pg_connect('host=' . $databaseConfig['host'] . ' port=' . $databaseConfig['port']
            . ' dbname=' . $databaseConfig['name'] . ' user=' . $databaseConfig['username'] . ' password=' . $databaseConfig['password']);
        return $this->connection ? true : false;
    }

    private function prefixed_column_names($columns, $prefix = NULL) {
        if (!$prefix || !$columns)
            return join(', ', $columns);
        return $prefix . '_' . join(', ' . $prefix . '_', $columns);
    }

    private function prefixed_name($column, $prefix = NULL) {
        return $prefix ? $prefix . '_' . $column : $column;
    }

    private function prepare_params($params, &$placeholders, &$values, &$null_columns = NULL) {
        $column_names = array();

        $i = count($values) + 1;
        foreach (array_keys($params) as $name) {
            $current_value = $params[$name];
            if ($current_value === NULL && $null_columns !== NULL) {
                array_push($null_columns, $name);
                continue;
            }
            assert(ctype_alnum_underscore($name));
            array_push($column_names, $name);
            array_push($placeholders, '$' . $i);
            if (is_bool($current_value))
                $current_value = $this->to_database_boolean($current_value);
            array_push($values, $current_value);
            $i++;
        }

        return $column_names;
    }

    private function select_conditions_with_null_columns($prefix, $column_names, $placeholders, $null_columns) {
        $column_names = $this->prefixed_column_names($column_names, $prefix);
        $placeholders = join(', ', $placeholders);

        if (!$column_names && !$placeholders)
            $column_names = $placeholders = '1';
        $query = "($column_names) = ($placeholders)";
        foreach ($null_columns as $column_name)
            $query .= ' AND ' . $this->prefixed_name($column_name, $prefix) . ' IS NULL';
        return $query;
    }

    function insert_row($table, $prefix, $params, $returning = 'id') {
        $placeholders = array();
        $values = array();
        $column_names = $this->prepare_params($params, $placeholders, $values);

        assert(!$prefix || ctype_alnum_underscore($prefix));
        $column_names = $this->prefixed_column_names($column_names, $prefix);
        $placeholders = join(', ', $placeholders);

        $value_query = $column_names ? "($column_names) VALUES ($placeholders)" : ' VALUES (default)';
        if ($returning) {
            $returning_column_name = $this->prefixed_name($returning, $prefix);
            $rows = $this->query_and_fetch_all("INSERT INTO $table $value_query RETURNING $returning_column_name", $values);
            return $rows ? $rows[0][$returning_column_name] : NULL;
        }

        return $this->query_and_get_affected_rows("INSERT INTO $table $value_query", $values) == 1;
    }

    function select_or_insert_row($table, $prefix, $select_params, $insert_params = NULL, $returning = 'id') {
        return $this->_select_update_or_insert_row($table, $prefix, $select_params, $insert_params, $returning, FALSE, TRUE);
    }

    function update_or_insert_row($table, $prefix, $select_params, $insert_params = NULL, $returning = 'id') {
        return $this->_select_update_or_insert_row($table, $prefix, $select_params, $insert_params, $returning, TRUE, TRUE);
    }

    function update_row($table, $prefix, $select_params, $update_params, $returning = 'id') {
        return $this->_select_update_or_insert_row($table, $prefix, $select_params, $update_params, $returning, TRUE, FALSE);
    }

    private function _select_update_or_insert_row($table, $prefix, $select_params, $insert_params, $returning, $should_update, $should_insert) {
        $values = array();

        $select_placeholders = array();
        $select_null_columns = array();
        $select_column_names = $this->prepare_params($select_params, $select_placeholders, $values, $select_null_columns);
        $select_values = array_slice($values, 0);

        if ($insert_params === NULL)
            $insert_params = $select_params;
        $insert_placeholders = array();
        $insert_column_names = $this->prepare_params($insert_params, $insert_placeholders, $values);

        assert(!!$returning);
        assert(!$prefix || ctype_alnum_underscore($prefix));
        $returning_column_name = $returning == '*' ? '*' : $this->prefixed_name($returning, $prefix);

        $condition = $this->select_conditions_with_null_columns($prefix, $select_column_names, $select_placeholders, $select_null_columns);
        $query = "SELECT $returning_column_name FROM $table WHERE $condition";

        $insert_column_names = $this->prefixed_column_names($insert_column_names, $prefix);
        $insert_placeholders = join(', ', $insert_placeholders);

        // http://stackoverflow.com/questions/1109061/insert-on-duplicate-update-in-postgresql
        $rows = NULL;
        if ($should_update) {
            $rows = $this->query_and_fetch_all("UPDATE $table SET ($insert_column_names) = ($insert_placeholders)
                WHERE $condition RETURNING $returning_column_name", $values);
        }
        if (!$rows && $should_insert) {
            $rows = $this->query_and_fetch_all("INSERT INTO $table ($insert_column_names) SELECT $insert_placeholders
                WHERE NOT EXISTS ($query) RETURNING $returning_column_name", $values);
        }
        if (!$should_update && !$rows)
            $rows = $this->query_and_fetch_all($query, $select_values);

        return $rows ? ($returning == '*' ? $rows[0] : $rows[0][$returning_column_name]) : NULL;
    }

    function select_first_row($table, $prefix, $params, $order_by = NULL) {
        return $this->select_first_or_last_row($table, $prefix, $params, $order_by, FALSE);
    }

    function select_last_row($table, $prefix, $params, $order_by = NULL) {
        return $this->select_first_or_last_row($table, $prefix, $params, $order_by, TRUE);
    }

    private function select_first_or_last_row($table, $prefix, $params, $order_by, $descending_order) {
        $rows = $this->select_rows($table, $prefix, $params, $order_by, $descending_order, 0, 1);
        return $rows ? $rows[0] : NULL;
    }

    function select_rows($table, $prefix, $params,
        $order_by = NULL, $descending_order = FALSE, $offset = NULL, $limit = NULL) {

        $placeholders = array();
        $values = array();
        $null_columns = array();
        $column_names = $this->prepare_params($params, $placeholders, $values, $null_columns);
        $condition = $this->select_conditions_with_null_columns($prefix, $column_names, $placeholders, $null_columns);

        $query = "SELECT * FROM $table WHERE $condition";

        if ($order_by) {
            if (!is_array($order_by))
                $order_by = array($order_by);

            $order_columns = array();
            foreach ($order_by as $order_key) {
                assert(ctype_alnum_underscore($order_key));
                $order_column = $this->prefixed_name($order_key, $prefix) . ' ' . ($descending_order? 'DESC' : 'ASC');
                array_push($order_columns, $order_column);
            }
            $query .= ' ORDER BY ' . join(', ', $order_columns);
        }
        if ($offset !== NULL)
            $query .= ' OFFSET ' . intval($offset);
        if ($limit !== NULL)
            $query .= ' LIMIT ' . intval($limit);

        return $this->query_and_fetch_all($query, $values);
    }

    function query_and_get_affected_rows($query, $params = array()) {
        if (!$this->connection)
            return FALSE;
        $result = pg_query_params($this->connection, $query, $params);
        if (!$result)
            return FALSE;
        return pg_affected_rows($result);
    }

    function query_and_fetch_all($query, $params = array()) {
        if (!$this->connection)
            return NULL;
        $result = pg_query_params($this->connection, $query, $params);
        if (!$result)
            return NULL;
        if (pg_num_rows($result) == 0)
            return array();
        return pg_fetch_all($result);
    }

    function query($query, $params = array()) {
        if (!$this->connection)
            return FALSE;
        return pg_query_params($this->connection, $query, $params);
    }

    function fetch_next_row($result) {
        return pg_fetch_assoc($result);
    }

    function fetch_table($table_name, $column_to_be_ordered_by = null) {
        if (!$this->connection || !ctype_alnum_underscore($table_name) || ($column_to_be_ordered_by && !ctype_alnum_underscore($column_to_be_ordered_by)))
            return false;
        $clauses = '';
        if ($column_to_be_ordered_by)
            $clauses .= 'ORDER BY ' . $column_to_be_ordered_by;
        return $this->query_and_fetch_all("SELECT * FROM $table_name $clauses");
    }

    function begin_transaction() {
        return $this->connection and pg_query($this->connection, "BEGIN");
    }

    function commit_transaction() {
        return $this->connection and pg_query($this->connection, 'COMMIT');
    }

    function rollback_transaction() {
        return $this->connection and pg_query($this->connection, 'ROLLBACK');
    }

}

?>