<?php  if ( ! defined('BASEPATH')) exit('No direct script access allowed');

class MY_URI extends CI_URI {

	// the segment the uri values as associative arrays
	public static $getvars;

	public static function get($var = FALSE)
	{
		return self::_get_vars($var);
	}

	private static function _return_var($var)
	{
		return ( ! empty(self::$getvars[$var])) ? self::$getvars[$var] : NULL;
	}

	private static function _get_vars($var = FALSE)
	{
		$CI = get_instance();

		if ($var AND self::$getvars)
			return self::_return_var($var);

		if ( ! $var AND self::$getvars)
			return self::$getvars;

		$use = ($CI->uri->segments == $CI->uri->rsegments) ? 'segments' : 'rsegments';

		$i = 2;
		foreach ($CI->uri->$use as $segment)
		{
			if ($segment == $CI->router->method)
				break;

			$i++;
		}

		self::$getvars = ($use == 'segments') ? $CI->uri->uri_to_assoc($i) : $CI->uri->ruri_to_assoc($i);

		return ($var) ? self::_return_var($var) : self::$getvars;

	}

}