<?php
/**
 * Database manager
 */

// If this file is called directly, abort.
if ( ! defined( 'WPINC' ) ) {
	die;
}

if ( ! class_exists( 'Jet_Engine_DB' ) ) {

	/**
	 * Define Jet_Engine_DB class
	 */
	class Jet_Engine_DB {

		public $query_cache = array();

		/**
		 * Return table name by key
		 *
		 * @param  string $table table key.
		 * @return string
		 */
		public static function tables( $table = null, $return = 'all' ) {

			global $wpdb;

			$prefix = 'jet_';

			$tables = array(
				'post_types' => array(
					'name'        => $wpdb->prefix . $prefix . 'post_types',
					'export_name' => $prefix . 'post_types',
					'query'       => "
						id bigint(20) NOT NULL AUTO_INCREMENT,
						slug text,
						status text,
						labels longtext,
						args longtext,
						meta_fields longtext,
						PRIMARY KEY (id)
					",
				),
				'taxonomies' => array(
					'name'        => $wpdb->prefix . $prefix . 'taxonomies',
					'export_name' => $prefix . 'taxonomies',
					'query'       => "
						id bigint(20) NOT NULL AUTO_INCREMENT,
						slug text,
						object_type text,
						status text,
						labels longtext,
						args longtext,
						meta_fields longtext,
						PRIMARY KEY (id)
					",
				)
			);

			if ( ! $table && 'all' === $return ) {
				return $tables;
			}

			switch ( $return ) {
				case 'all':
					return isset( $tables[ $table ] ) ? $tables[ $table ] : false;

				case 'name':
					return isset( $tables[ $table ] ) ? $tables[ $table ]['name'] : false;

				case 'query':
					return isset( $tables[ $table ] ) ? $tables[ $table ]['query'] : false;
			}

			return false;

		}

		/**
		 * Create all tables on activation
		 *
		 * @return [type] [description]
		 */
		public static function create_all_tables() {

			global $wpdb;

			$charset_collate = $wpdb->get_charset_collate();

			foreach ( self::tables() as $table ) {

				$table_name  = $table['name'];
				$table_query = $table['query'];

				if ( $table_name !== $wpdb->get_var( "SHOW TABLES LIKE '$table_name'" ) ) {
					$sql = "CREATE TABLE $table_name (
						$table_query
					) $charset_collate;";

					require_once( ABSPATH . 'wp-admin/includes/upgrade.php' );

					dbDelta( $sql );
				}

			}

		}

		/**
		 * Insert or update row into table
		 *
		 * @param  array  $data [description]
		 * @return [type]       [description]
		 */
		public function update( $table, $data = array(), $format = array() ) {

			$prepared_data = array();

			foreach ( $data as $key => $value ) {
				if ( is_array( $value ) ) {
					$prepared_data[ $key ] = maybe_serialize( $value );
				} else {
					$prepared_data[ $key ] = $value;
				}
			}

			global $wpdb;

			$table_name = $this->tables( $table, 'name' );
			$result     = false;

			if ( ! isset( $prepared_data['id'] ) ) {

				$inserted = $wpdb->insert( $table_name, $prepared_data, $format );

				if ( $inserted ) {
					$result = $wpdb->insert_id;
				}

			} else {

				$where        = array( 'id' => $prepared_data['id'] );
				$where_format = array( '%d' );
				$wpdb->update( $table_name, $prepared_data, $where, $format, $where_format );
				$result = $prepared_data['id'];

			}

			return $result;
		}

		public function delete( $table, $where = array(), $format = array() ) {
			global $wpdb;
			$table_name = $this->tables( $table, 'name' );
			$wpdb->delete( $table_name, $where, $format );
		}

		/**
		 * Check if table is exists
		 *
		 * @param  string  $table Table name.
		 * @return boolean
		 */
		public function is_table_exists( $table = null ) {

			global $wpdb;

			$table_name = $this->tables( $table, 'name' );

			if ( ! $table_name ) {
				return false;
			}

			return ( strtolower( $table_name ) === strtolower( $wpdb->get_var( "SHOW TABLES LIKE '$table_name'" ) ) );
		}

		/**
		 * Create table if allowed
		 * @return [type] [description]
		 */
		public function create_table( $table = null ) {

			global $wpdb;

			$charset_collate = $wpdb->get_charset_collate();
			$table_data      = $this->tables( $table );

			if ( ! $table_data ) {
				return;
			}

			$table_name  = $table_data['name'];
			$table_query = $table_data['query'];

			$sql = "CREATE TABLE $table_name (
				$table_query
			) $charset_collate;";

			require_once( ABSPATH . 'wp-admin/includes/upgrade.php' );

			dbDelta( $sql );

		}

		/**
		 * Returns total rows count in requested table
		 *
		 * @param  string   $table  Table name.
		 * @return int
		 */
		public function count( $table, $args = array() ) {

			global $wpdb;

			$table_name = $this->tables( $table, 'name' );
			$query      = "SELECT COUNT(*) FROM $table_name";
			$query     .= $this->get_where_statement( $args );

			return $wpdb->get_var( $query );

		}

		/**
		 * Build where statement for SQL query
		 *
		 * @return [type] [description]
		 */
		public function get_where_statement( $args = array() ) {

			if ( empty( $args ) ) {
				return;
			}

			$query = ' WHERE ';
			$glue  = '';

			foreach ( $args as $key => $value ) {

				$query .= $glue;

				if ( ! is_array( $value ) ) {
					$query .= sprintf( '`%1$s` = \'%2$s\'', esc_sql( $key ), esc_sql( $value ) );
				} else {
					$value  = array_map( 'esc_sql', $value );
					$query .= sprintf( '`%1$s` IN (%2$s)', esc_sql( $key ), implode( ',' , $value ) );
				}

				$glue   = ' AND ';

			}

			return $query;

		}

		/**
		 * Query data from passed table by passed args
		 *
		 * @param  string   $table  Table name.
		 * @param  array    $args   Args array.
		 * @param  callable $filter Callback to filter results.
		 * @return array
		 */
		public function query( $table = null, $args = array(), $filter = null, $from_cache = true ) {

			$result = array();

			if ( $from_cache && isset( $this->query_cache[ $table ] ) ) {
				$result = $this->query_cached( $table, $args, $filter );
			} elseif ( $from_cache ) {
				$result = $this->query_raw( $table, $args, $filter );
			} else {
				global $wpdb;

				$table_name = $this->tables( $table, 'name' );
				$query      = "SELECT * FROM $table_name";
				$query     .= $this->get_where_statement( $args );
				$query     .= " ORDER BY id DESC";
				$raw        = $wpdb->get_results( $query, ARRAY_A );

				if ( ! $raw ) {
					$raw = array();
				}

				if ( ! $filter ) {
					$result = $raw;
				} else {
					$result = array_map( $filter, $raw );
				}
			}

			return $result;

		}

		/**
		 * Query raw results from table
		 *
		 * @param  [type] $table  [description]
		 * @param  array  $args   [description]
		 * @param  [type] $filter [description]
		 * @return [type]         [description]
		 */
		public function query_raw( $table = null, $args = array(), $filter = null ) {

			global $wpdb;

			$table_name = $this->tables( $table, 'name' );
			$raw        = $wpdb->get_results( "SELECT * FROM $table_name ORDER BY id DESC", ARRAY_A );

			if ( ! $raw ) {
				$raw = array();
			}

			$this->query_cache[ $table ] = $raw;

			return $this->query_cached( $table, $args, $filter );

		}

		/**
		 * Query by cached results
		 *
		 * @param  array  $args   [description]
		 * @param  [type] $filter [description]
		 * @return [type]         [description]
		 */
		public function query_cached( $table = null, $args = array(), $filter = null ) {

			if ( empty( $this->query_cache[ $table ] ) ) {
				return array();
			}

			$result = array();

			foreach ( $this->query_cache[ $table ] as $row ) {

				if ( empty( $args ) ) {

					if ( ! $filter ) {
						$result[] = $row;
					} else {
						$result[] = call_user_func( $filter, $row );