<?php

// Todo: use the JSON API
class Redirection_Api {
	public $endpoints = array(
		'load_settings',
		'save_settings',
		'get_logs',
		'log_action',
		'delete_all',
		'delete_plugin',
		'get_redirect',
		'set_redirect',
		'redirect_action',
		'get_group',
		'set_group',
		'group_action',
		'import_data',
		'export_data',
		'ping',
		'plugin_status',
		'get_importers',
	);

	public function __construct() {
		global $wpdb;

		$wpdb->hide_errors();

		foreach ( $this->endpoints as $point ) {
			add_action( 'wp_ajax_red_'.$point, array( $this, 'check_auth' ), 9 );
			add_action( 'wp_ajax_red_'.$point, array( $this, 'ajax_'.$point ), 10 );
		}
	}

	private function addDatabaseError( $response ) {
		global $wpdb;

		if ( isset( $response['error'] ) && isset( $wpdb->last_error ) && $wpdb->last_error ) {
			$response['error']['wpdb'] = $wpdb->last_error;
		}

		return $response;
	}

	private function getError( $message, $line ) {
		return array(
			'error' => array(
				'message' => $message,
				'code' => $line,
			)
		);
	}

	public function check_auth( $params ) {
		if ( check_ajax_referer( 'wp_rest', false, false ) === false ) {
			$error = $this->getError( 'Unable to perform action - bad nonce "wp_rest"', __LINE__ );
			$error['error']['action'] = 'reload';
			wp_die( $this->output_ajax_response( $error ) );
		}

		if ( $this->user_has_access() === false ) {
			wp_die( $this->output_ajax_response( $this->getError( 'No permissions to perform action', __LINE__ ) ) );
		}
	}

	private function get_params( $params = array() ) {
		if ( empty( $params ) && isset( $_POST['data'] ) ) {
			$params = json_decode( wp_unslash( $_POST['data'] ), true );
		}

		return $params;
	}

	public function ajax_ping() {
		return $this->output_ajax_response( array( 'nonce' => wp_create_nonce( 'wp_rest' ) ) );
	}

	public function ajax_import_data( $params ) {
		$params = $this->get_params( $params );
		$upload = isset( $_FILES[ 'file' ] ) ? $_FILES[ 'file' ] : false;
		$group_id = isset( $params['group'] ) ? intval( $params['group'], 10 ) : 0;

		$result = $this->getError( 'Invalid file', __LINE__ );
		if ( $upload && is_uploaded_file( $upload['tmp_name'] ) ) {
			$result = $this->getError( 'Invalid group', __LINE__ );

			$count = Red_FileIO::import( $group_id, $upload );
			if ( $count !== false ) {
				$result = array(
					'imported' => $count,
				);
			}
		}

		return $this->output_ajax_response( $result );
	}

	public function ajax_export_data( $params ) {
		$params = $this->get_params( $params );
		$moduleId = isset( $params['module'] ) ? intval( $params['module'], 10 ) : false;
		$format = 'json';

		if ( isset( $params['format'] ) && in_array( $params['format'], array( 'csv', 'apache', 'nginx', 'json' ) ) ) {
			$format = $params['format'];
		}

		$result = $this->getError( 'Invalid module', __LINE__ );

		$export = Red_FileIO::export( $moduleId, $format );
		if ( $export !== false ) {
			$result = array(
				'data' => $export['data'],
				'total' => $export['total'],
			);
		}

		return $this->output_ajax_response( $result );
	}

	public function ajax_group_action( $params ) {
		$params = $this->get_params( $params );

		$action = false;
		$items = array();
		if ( isset( $params['items'] ) ) {
			$items = array_map( 'intval', explode( ',', $params['items'] ) );
		}

		if ( isset( $params['bulk'] ) && in_array( $params['bulk'], array( 'delete', 'enable', 'disable' ) ) ) {
			$action = $params['bulk'];

			foreach ( $items as $item ) {
				$group = Red_Group::get( intval( $item, 10 ) );

				if ( $group ) {
					if ( $action === 'delete' ) {
						$group->delete();
					} else if ( $action === 'disable' ) {
						$group->disable();
					} else if ( $action === 'enable' ) {
						$group->enable();
					}
				}
			}
		}

		return $this->output_ajax_response( Red_Group::get_filtered( $params ) );
	}

	public function ajax_get_group( $params ) {
		$params = $this->get_params( $params );

		return $this->output_ajax_response( Red_Group::get_filtered( $params ) );
	}

	public function ajax_set_group( $params ) {
		$params = $this->get_params( $params );

		$groupId = 0;
		$name = '';
		$moduleId = 0;
		if ( isset( $params['id'] ) ) {
			$groupId = intval( $params['id'], 10 );
		}

		$result = $this->getError( 'Invalid group or parameters', __LINE__ );
		if ( $groupId > 0 ) {
			$group = Red_Group::get( $groupId );

			if ( $group && isset( $params['name'] ) && isset( $params['moduleId'] ) ) {
				if ( $group->update( $params ) ) {
					$result = array( 'item' => $group->to_json() );
				}
			}
		} else {
			$group = Red_Group::create( isset( $params['name'] ) ? $params['name'] : '', isset( $params['moduleId'] ) ? $params['moduleId'] : 0 );

			if ( $group ) {
				$result = Red_Group::get_filtered( $params );
			}
		}

		return $this->output_ajax_response( $result );
	}

	public function ajax_get_redirect( $params ) {
		$params = $this->get_params( $params );

		return $this->output_ajax_response( Red_Item::get_filtered( $params ) );
	}

	public function ajax_set_redirect( $params ) {
		$params = $this->get_params( $params );

		$redirectId = 0;
		if ( isset( $params['id'] ) ) {
			$redirectId = intval( $params['id'], 10 );
		}

		$result = $this->getError( 'Invalid redirect details', __LINE__ );
		if ( $redirectId === 0 ) {
			$redirect = Red_Item::create( $params );

			if ( is_wp_error( $redirect ) ) {
				$result = $this->getError( $redirect->get_error_message(), __LINE__ );
			} else {
				$result = Red_Item::get_filtered( $params );
			}
		} else {
			$redirect = Red_Item::get_by_id( $redirectId );

			if ( $redirect ) {
				$result = $redirect->update( $params );

				if ( is_wp_error( $result ) ) {
					$result = $this->getError( $result->get_error_message(), __LINE__ );
				} else {
					$result = array( 'item' => $redirect->to_json() );
				}
			}
		}

		return $this->output_ajax_response( $result );
	}

	public function ajax_redirect_action( $params ) {
		$params = $this->get_params( $params );

		$action = false;
		$items = array();

		if ( isset( $params['items'] ) ) {
			$items = array_map( 'intval', explode( ',', $params['items'] ) );
		}

		if ( isset( $params['bulk'] ) && in_array( $params['bulk'], array( 'delete', 'enable', 'disable', 'reset' ) ) ) {
			$action = $params['bulk'];

			foreach ( $items as $item ) {
				$redirect = Red_Item::get_by_id( intval( $item, 10 ) );

				if ( $redirect ) {
					if ( $action === 'delete' ) {
						$redirect->delete();
					} else if ( $action === 'disable' ) {
						$redirect->disable();
					} else if ( $action === 'enable' ) {
						$redirect->enable();
					} else if ( $action === 'reset' ) {
						$redirect->reset();
					}
				}
			}
		}

		return $this->output_ajax_response( Red_Item::get_filtered( $params ) );
	}

	public function ajax_delete_plugin() {
		$plugin = Redirection_Admin::init();
		$plugin->plugin_uninstall();

		$current = get_option( 'active_plugins' );
		array_splice( $current, array_search( basename( dirname( REDIRECTION_FILE ) ).'/'.basename( REDIRECTION_FILE ), $current ), 1 );
		update_option( 'active_plugins', $current );

		return $this->output_ajax_response( array( 'location' => admin_url().'plugins.php' ) );
	}

	public function ajax_load_settings() {
		return $this->output_ajax_response( array(
			'settings' => red_get_options(),
			'groups' => $this->groups_to_json( Red_Group::get_for_select() ),
			'installed' => get_home_path(),
		) );
	}

	public function ajax_save_settings( $settings = array() ) {
		red_set_options( $this->get_params( $settings ) );

		return $this->ajax_load_settings();
	}

	public function ajax_get_logs( $params ) {
		$params = $this->get_params( $params );
		$result = $this->get_logs( $params );

		return $this->output_ajax_response( $result );
	}

	public function ajax_log_action( $params ) {
		$params = $this->get_params( $params );

		// Do the action
		if ( isset( $params['bulk'] ) && isset( $params['items'] ) && $params['bulk'] === 'delete' ) {
			$items = explode( ',', $params['items'] );

			if ( $this->get_log_type( $params ) === 'log' ) {
				array_map( array( 'RE_Log', 'delete' ), $items );
			} else {
				array_map( array( 'RE_404', 'delete' ), $items );
			}
		}

		$result = $this->get_logs( $params );

		return $this->output_ajax_response( $result );
	}

	public function ajax_delete_all( $params ) {
		$params = $this->get_params( $params );
		$filter = '';
		$filterBy = '';
		if ( isset( $params['filter'] ) ) {
			$filter = $params['filter'];
		}

		if ( isset( $params['filterBy'] ) && in_array( $params['filterBy'], array( 'url', 'ip', 'url-exact' ), true ) ) {
			$filterBy = $params['filterBy'];
			unset( $params['filter'] );
			unset( $params['filterBy'] );
		}

		if ( isset( $params['logType'] ) ) {
			if ( $params['logType'] === 'log' ) {
				RE_Log::delete_all( $filterBy, $filter );
			} else {
				RE_404::delete_all( $filterBy, $filter );
			}
		}

		$result = $this->get_logs( $params );

		return $this->output_ajax_response( $result );
	}

	public function ajax_plugin_status( $params ) {
		$params = $this->get_params( $params );

		$fixit = false;
		if ( isset( $params['fixIt'] ) && $params['fixIt'] ) {
			$fixit = true;
		}

		include_once dirname( REDIRECTION_FILE ).'/models/fixer.php';

		$fixer = new Red_Fixer();
		$result = $fixer->get_status();

		if ( $fixit ) {
			$result = $fixer->fix( $result );
		}

		return $this->output_ajax_response( $result );
	}

	public function ajax_get_importers( $params ) {
		include_once dirname( __FILE__ ).'/models/importer.php';

		$params = $this->get_params( $params );

		if ( isset( $params['plugin'] ) ) {
			$groups = Red_Group::get_all();
			$result = array( 'imported' => Red_Plugin_Importer::import( $params['plugin'], $groups[ 0 ]['id'] ) );
		} else {
			$result = array( 'importers' => Red_Plugin_Importer::get_plugins() );
		}

		return $this->output_ajax_response( $result );
	}

	private function get_log_type( $params ) {
		$type = 'log';

		if ( isset( $params['logType'] ) && in_array( $params['logType'], array( 'log', '404' ), true ) ) {
			$type = $params['logType'];
		}

		return $type;
	}

	private function get_logs( array $params ) {
		$type = $this->get_log_type( $params );

		if ( $type === 'log' ) {
			return RE_Filter_Log::get( 'redirection_logs', 'RE_Log', $params );
		} else if ( $type === '404' ) {
			if ( isset( $params['filterBy'] ) && isset( $params['filter'] ) && $params['filterBy'] === 'ip' ) {
				$params['filter'] = ip2long( $params['filter'] );
			}

			return RE_Filter_Log::get( 'redirection_404', 'RE_404', $params );
		}

		return array( 'items' => array(), 'total' => 0 );
	}

	private function groups_to_json( $groups, $depth = 0 ) {
		$items = array();

		foreach ( $groups as $text => $value ) {
			if ( is_array( $value ) && $depth === 0 ) {
				$items[] = (object)array( 'text' => $text, 'value' => $this->groups_to_json( $value, 1 ) );
			} else {
				$items[] = (object)array( 'text' => $value, 'value' => $text );
			}
		}

		return $items;
	}

	private function output_ajax_response( $data ) {
		$data = $this->addDatabaseError( $data );
		$result = wp_json_encode( $data );

		if ( defined( 'DOING_AJAX' ) ) {
			header( 'Content-Type: application/json' );
			echo $result;
			wp_die();
		}

		return $result;
	}

	private function user_has_access() {
		return current_user_can( apply_filters( 'redirection_role', 'administrator' ) );
	}
}