woocommerce/classes/woocommerce.class.php

266 lines
6.9 KiB
PHP

<?php
/**
* Contains the main functions for WooCommerce, stores variables, and handles error messages
*
* @class woocommerce
* @package WooCommerce
* @category Class
* @author WooThemes
*/
class woocommerce {
private static $_instance;
private static $_cache;
public static $errors = array();
public static $messages = array();
public static $attribute_taxonomies;
public static $plugin_url;
public static $plugin_path;
const SHOP_SMALL_W = '150';
const SHOP_SMALL_H = '150';
const SHOP_TINY_W = '36';
const SHOP_TINY_H = '36';
const SHOP_THUMBNAIL_W = '90';
const SHOP_THUMBNAIL_H = '90';
const SHOP_LARGE_W = '300';
const SHOP_LARGE_H = '300';
/** constructor */
function __construct () {
global $wpdb;
// Vars
self::$attribute_taxonomies = $wpdb->get_results("SELECT * FROM ".$wpdb->prefix."woocommerce_attribute_taxonomies;");
if (isset($_SESSION['errors'])) self::$errors = $_SESSION['errors'];
if (isset($_SESSION['messages'])) self::$messages = $_SESSION['messages'];
unset($_SESSION['messages']);
unset($_SESSION['errors']);
// Hooks
add_filter('wp_redirect', array(&$this, 'redirect'), 1, 2);
}
/** get */
public static function get() {
if (!isset(self::$_instance)) {
$c = __CLASS__;
self::$_instance = new $c;
}
return self::$_instance;
}
/**
* Get the plugin url
*
* @return string url
*/
public static function plugin_url() {
if(self::$plugin_url) return self::$plugin_url;
if (is_ssl()) :
return self::$plugin_url = str_replace('http://', 'https://', WP_PLUGIN_URL) . "/" . plugin_basename( dirname(dirname(__FILE__)));
else :
return self::$plugin_url = WP_PLUGIN_URL . "/" . plugin_basename( dirname(dirname(__FILE__)));
endif;
}
/**
* Get the plugin path
*
* @return string url
*/
public static function plugin_path() {
if(self::$plugin_path) return self::$plugin_path;
return self::$plugin_path = WP_PLUGIN_DIR . "/" . plugin_basename( dirname(dirname(__FILE__)));
}
/**
* Return the URL with https if SSL is on
*
* @return string url
*/
public static function force_ssl( $url ) {
if (is_ssl()) $url = str_replace('http:', 'https:', $url);
return $url;
}
/**
* Get a var
*
* Variable is filtered by woocommerce_get_var_{var name}
*
* @param string var
* @return string variable
*/
public static function get_var($var) {
$return = '';
switch ($var) :
case "version" : $return = WOOCOMMERCE_VERSION; break;
case "shop_small_w" : $return = self::SHOP_SMALL_W; break;
case "shop_small_h" : $return = self::SHOP_SMALL_H; break;
case "shop_tiny_w" : $return = self::SHOP_TINY_W; break;
case "shop_tiny_h" : $return = self::SHOP_TINY_H; break;
case "shop_thumbnail_w" : $return = self::SHOP_THUMBNAIL_W; break;
case "shop_thumbnail_h" : $return = self::SHOP_THUMBNAIL_H; break;
case "shop_large_w" : $return = self::SHOP_LARGE_W; break;
case "shop_large_h" : $return = self::SHOP_LARGE_H; break;
endswitch;
return apply_filters( 'woocommerce_get_var_'.$var, $return );
}
/**
* Add an error
*
* @param string error
*/
function add_error( $error ) { self::$errors[] = $error; }
/**
* Add a message
*
* @param string message
*/
function add_message( $message ) { self::$messages[] = $message; }
/** Clear messages and errors from the session data */
function clear_messages() {
self::$errors = self::$messages = array();
unset($_SESSION['messages']);
unset($_SESSION['errors']);
}
/**
* Get error count
*
* @return int
*/
function error_count() { return sizeof(self::$errors); }
/**
* Get message count
*
* @return int
*/
function message_count() { return sizeof(self::$messages); }
/**
* Output the errors and messages
*
* @return bool
*/
public static function show_messages() {
if (isset(self::$errors) && sizeof(self::$errors)>0) :
echo '<div class="woocommerce_error">'.self::$errors[0].'</div>';
self::clear_messages();
return true;
elseif (isset(self::$messages) && sizeof(self::$messages)>0) :
echo '<div class="woocommerce_message">'.self::$messages[0].'</div>';
self::clear_messages();
return true;
else :
return false;
endif;
}
public static function nonce_field ($action, $referer = true , $echo = true) {
$name = '_n';
$action = 'woocommerce-' . $action;
return wp_nonce_field($action, $name, $referer, $echo);
}
public static function nonce_url ($action, $url = '') {
$name = '_n';
$action = 'woocommerce-' . $action;
$url = add_query_arg( $name, wp_create_nonce( $action ), $url);
return $url;
}
/**
* Check a nonce and sets woocommerce error in case it is invalid
* To fail silently, set the error_message to an empty string
*
* @param string $name the nonce name
* @param string $action then nonce action
* @param string $method the http request method _POST, _GET or _REQUEST
* @param string $error_message custom error message, or false for default message, or an empty string to fail silently
*
* @return bool
*/
public static function verify_nonce($action, $method='_POST', $error_message = false) {
$name = '_n';
$action = 'woocommerce-' . $action;
if( $error_message === false ) $error_message = __('Action failed. Please refresh the page and retry.', 'woothemes');
if(!in_array($method, array('_GET', '_POST', '_REQUEST'))) $method = '_POST';
/*
$request = $GLOBALS[$method];
if ( isset($request[$name]) && wp_verify_nonce($request[$name], $action) ) return true;
*/
if ( isset($_REQUEST[$name]) && wp_verify_nonce($_REQUEST[$name], $action) ) return true;
if( $error_message ) woocommerce::add_error( $error_message );
return false;
}
/**
* Redirection hook which stores messages into session data
*
* @param location
* @param status
* @return location
*/
function redirect( $location, $status ) {
$_SESSION['errors'] = self::$errors;
$_SESSION['messages'] = self::$messages;
return $location;
}
static public function shortcode_wrapper ($function, $atts=array()) {
if( $content = woocommerce::cache_get( $function . '-shortcode', $atts ) ) return $content;
ob_start();
call_user_func($function, $atts);
return woocommerce::cache( $function . '-shortcode', ob_get_clean(), $atts);
}
/**
* Cache API
*/
public static function cache ( $id, $data, $args=array() ) {
if( ! isset(self::$_cache[ $id ]) ) self::$_cache[ $id ] = array();
if( empty($args) ) self::$_cache[ $id ][0] = $data;
else self::$_cache[ $id ][ serialize($args) ] = $data;
return $data;
}
public static function cache_get ( $id, $args=array() ) {
if( ! isset(self::$_cache[ $id ]) ) return null;
if( empty($args) && isset(self::$_cache[ $id ][0]) ) return self::$_cache[ $id ][0];
elseif ( isset(self::$_cache[ $id ][ serialize($args) ] ) ) return self::$_cache[ $id ][ serialize($args) ];
}
}