woocommerce/admin/woocommerce-admin-install.php

<?php
/**
 * WooCommerce Install
 *
 * Plugin install script which adds default pages, taxonomies, and database tables to WordPress. Runs on activation and upgrade.
 *
 * @author 		WooThemes
 * @category 	Admin
 * @package 	WooCommerce/Admin/Install
 * @version     1.6.4
 */

/**
 * Runs the installer.
 *
 * @access public
 * @return void
 */
function do_install_woocommerce() {
	global $woocommerce_settings, $woocommerce;

	// Do install
	woocommerce_default_options();
	woocommerce_tables_install();
	woocommerce_init_roles();
	
	// Setup default permalinks
	$permalinks 	= get_option( 'woocommerce_permalinks' );
	$shop_page_id 	= woocommerce_get_page_id( 'shop' );
	if ( empty( $permalinks ) && $shop_page_id > 0 ) {
	
		$base_slug 		= $shop_page_id > 0 && get_page( $shop_page_id ) ? get_page_uri( $shop_page_id ) : 'shop';
		
		$category_base 	= get_option('woocommerce_prepend_shop_page_to_urls') == "yes" ? trailingslashit( $base_slug ) : '';
		$category_slug 	= get_option('woocommerce_product_category_slug') ? get_option('woocommerce_product_category_slug') : _x( 'product-category', 'slug', 'woocommerce' );
		$tag_slug 		= get_option('woocommerce_product_tag_slug') ? get_option('woocommerce_product_tag_slug') : _x( 'product-tag', 'slug', 'woocommerce' );
		
		if ( 'yes' == get_option('woocommerce_prepend_shop_page_to_products') ) {
			$product_base = trailingslashit( $base_slug );
		} else {
			if ( ( $product_slug = get_option('woocommerce_product_slug') ) !== false && ! empty( $product_slug ) ) {
				$product_base = trailingslashit( $product_slug );
			} else {
				$product_base = trailingslashit( _x('product', 'slug', 'woocommerce') );
			}
		}
		
		if ( get_option('woocommerce_prepend_category_to_products') == 'yes' ) 
			$product_base .= trailingslashit('%product_cat%');
	
		$permalinks = array(
			'product_base' 		=> untrailingslashit( $product_base ),
			'category_base' 	=> untrailingslashit( $category_base . $category_slug ),
			'attribute_base' 	=> untrailingslashit( $category_base ),
			'tag_base' 			=> untrailingslashit( $category_base . $tag_slug )
		);
		
		update_option( 'woocommerce_permalinks', $permalinks );	
	}

	// Register post types
	$woocommerce->init_taxonomy();

	// Add default taxonomies
	woocommerce_default_taxonomies();

	// Install folder for uploading files and prevent hotlinking
	$upload_dir =  wp_upload_dir();
	$downloads_url = $upload_dir['basedir'] . '/woocommerce_uploads';

	if ( wp_mkdir_p( $downloads_url ) && ! file_exists( $downloads_url.'/.htaccess' ) ) {
		if ( $file_handle = @fopen( $downloads_url . '/.htaccess', 'w' ) ) {
			fwrite( $file_handle, 'deny from all' );
			fclose( $file_handle );
		}
	}

	// Install folder for logs
	$logs_url = WP_PLUGIN_DIR . "/" . plugin_basename( dirname( dirname( __FILE__ ) ) ) . '/logs';

	if ( wp_mkdir_p( $logs_url ) && ! file_exists( $logs_url . '/.htaccess' ) ) {
		if ( $file_handle = @fopen( $logs_url . '/.htaccess', 'w' ) ) {
			fwrite( $file_handle, 'deny from all' );
			fclose( $file_handle );
		}
	}

	// Clear transient cache
	$woocommerce->clear_product_transients();

	// Recompile LESS styles if they are custom
	if ( get_option( 'woocommerce_frontend_css' ) == 'yes' ) {

		// Handle Colour Settings
		$colors = get_option( 'woocommerce_frontend_css_colors' );

		if  (  (
				! empty( $colors['primary'] ) &&
				! empty( $colors['secondary'] ) &&
				! empty( $colors['highlight'] ) &&
				! empty( $colors['content_bg'] ) &&
				! empty( $colors['subtext'] )
			) && (
				$colors['primary'] != '#ad74a2' ||
				$colors['secondary'] != '#f7f6f7' ||
				$colors['highlight'] != '#85ad74' ||
				$colors['content_bg'] != '#ffffff' ||
				$colors['subtext'] != '#777777'
				) ) {

			// Write less file
			woocommerce_compile_less_styles();

		}

	}

	// Update version
	update_option( 'woocommerce_db_version', $woocommerce->version );
}


/**
 * Default options
 *
 * Sets up the default options used on the settings page
 *
 * @access public
 * @return void
 */
function woocommerce_default_options() {
	global $woocommerce_settings;

	// Include settings so that we can run through defaults
	include_once( 'settings/settings-init.php' );

	foreach ( $woocommerce_settings as $section ) {

		foreach ( $section as $value ) {

	        if ( isset( $value['std'] ) && isset( $value['id'] ) ) {

	        	if ( $value['type'] == 'image_width' ) {

	        		add_option( $value['id'] . '_width', $value['std'] );
	        		add_option( $value['id'] . '_height', $value['std'] );

	        	} else {

	        		add_option( $value['id'], $value['std'] );

	        	}

	        }

        }

    }

    add_option( 'woocommerce_shop_slug', 'shop' );
}


/**
 * Create a page
 *
 * @access public
 * @param mixed $slug Slug for the new page
 * @param mixed $option Option name to store the page's ID
 * @param string $page_title (default: '') Title for the new page
 * @param string $page_content (default: '') Content for the new page
 * @param int $post_parent (default: 0) Parent for the new page
 * @return void
 */
function woocommerce_create_page( $slug, $option, $page_title = '', $page_content = '', $post_parent = 0 ) {
	global $wpdb;

	$option_value = get_option( $option );

	if ( $option_value > 0 && get_post( $option_value ) )
		return;

	$page_found = $wpdb->get_var( "SELECT ID FROM " . $wpdb->posts . " WHERE post_name = '$slug' LIMIT 1;" );
	if ( $page_found ) {
		if ( ! $option_value )
			update_option( $option, $page_found );
		return;
	}

	$page_data = array(
        'post_status' 		=> 'publish',
        'post_type' 		=> 'page',
        'post_author' 		=> 1,
        'post_name' 		=> $slug,
        'post_title' 		=> $page_title,
        'post_content' 		=> $page_content,
        'post_parent' 		=> $post_parent,
        'comment_status' 	=> 'closed'
    );
    $page_id = wp_insert_post( $page_data );

    update_option( $option, $page_id );
}


/**
 * Create pages that the plugin relies on, storing page id's in variables.
 *
 * @access public
 * @return void
 */
function woocommerce_create_pages() {

	// Shop page
    woocommerce_create_page( esc_sql( _x( 'shop', 'page_slug', 'woocommerce' ) ), 'woocommerce_shop_page_id', __( 'Shop', 'woocommerce' ), '' );

    // Cart page
    woocommerce_create_page( esc_sql( _x( 'cart', 'page_slug', 'woocommerce' ) ), 'woocommerce_cart_page_id', __( 'Cart', 'woocommerce' ), '[woocommerce_cart]' );

	// Checkout page
    woocommerce_create_page( esc_sql( _x( 'checkout', 'page_slug', 'woocommerce' ) ), 'woocommerce_checkout_page_id', __( 'Checkout', 'woocommerce' ), '[woocommerce_checkout]' );

    // Order tracking page
    woocommerce_create_page( esc_sql( _x( 'order-tracking', 'page_slug', 'woocommerce' ) ), 'woocommerce_order_tracking_page_id', __( 'Track your order', 'woocommerce' ), '[woocommerce_order_tracking]' );

	// My Account page
    woocommerce_create_page( esc_sql( _x( 'my-account', 'page_slug', 'woocommerce' ) ), 'woocommerce_myaccount_page_id', __( 'My Account', 'woocommerce' ), '[woocommerce_my_account]' );

	// Lost password page
	woocommerce_create_page( esc_sql( _x( 'lost-password', 'page_slug', 'woocommerce' ) ), 'woocommerce_lost_password_page_id', __( 'Lost Password', 'woocommerce' ), '[woocommerce_lost_password]', woocommerce_get_page_id( 'myaccount' ) );

	// Edit address page
    woocommerce_create_page( esc_sql( _x( 'edit-address', 'page_slug', 'woocommerce' ) ), 'woocommerce_edit_address_page_id', __( 'Edit My Address', 'woocommerce' ), '[woocommerce_edit_address]', woocommerce_get_page_id( 'myaccount' ) );

    // View order page
    woocommerce_create_page( esc_sql( _x( 'view-order', 'page_slug', 'woocommerce' ) ), 'woocommerce_view_order_page_id', __( 'View Order', 'woocommerce' ), '[woocommerce_view_order]', woocommerce_get_page_id( 'myaccount' ) );

    // Change password page
    woocommerce_create_page( esc_sql( _x( 'change-password', 'page_slug', 'woocommerce' ) ), 'woocommerce_change_password_page_id', __( 'Change Password', 'woocommerce' ), '[woocommerce_change_password]', woocommerce_get_page_id( 'myaccount' ) );

	// Pay page
    woocommerce_create_page( esc_sql( _x( 'pay', 'page_slug', 'woocommerce' ) ), 'woocommerce_pay_page_id', __( 'Checkout &rarr; Pay', 'woocommerce' ), '[woocommerce_pay]', woocommerce_get_page_id( 'checkout' ) );

    // Thanks page
    woocommerce_create_page( esc_sql( _x( 'order-received', 'page_slug', 'woocommerce' ) ), 'woocommerce_thanks_page_id', __( 'Order Received', 'woocommerce' ), '[woocommerce_thankyou]', woocommerce_get_page_id( 'checkout' ) );

}


/**
 * Set up the database tables which the plugin needs to function.
 *
 * @access public
 * @return void
 */
function woocommerce_tables_install() {
	global $wpdb, $woocommerce;

	$wpdb->hide_errors();

	$collate = '';
    if ( $wpdb->has_cap( 'collation' ) ) {
		if( ! empty($wpdb->charset ) ) 
			$collate .= "DEFAULT CHARACTER SET $wpdb->charset";
		if( ! empty($wpdb->collate ) ) 
			$collate .= " COLLATE $wpdb->collate";
    }

    require_once( ABSPATH . 'wp-admin/includes/upgrade.php' );
    
    /**
     * Data Upgrades - Prepare existing tables
     **/
    if ( version_compare( get_option('woocommerce_db_version'), '1.7', '<' ) ) { 
		// remove the existing primary key so we can add the new download_id column
		$wpdb->query( "ALTER TABLE ". $wpdb->prefix . "woocommerce_downloadable_product_permissions DROP PRIMARY KEY" );
	}

    // Table for storing attribute taxonomies - these are user defined
    $sql = "
CREATE TABLE ". $wpdb->prefix . "woocommerce_attribute_taxonomies (
  attribute_id bigint(20) NOT NULL auto_increment,
  attribute_name varchar(200) NOT NULL,
  attribute_label longtext NULL,
  attribute_type varchar(200) NOT NULL,
  attribute_orderby varchar(200) NOT NULL,
  PRIMARY KEY  (attribute_id)
) $collate;
";
    dbDelta($sql);

    // Term meta table - sadly WordPress does not have termmeta so we need our own
    $sql = "
CREATE TABLE ". $wpdb->prefix . "woocommerce_termmeta (
  meta_id bigint(20) NOT NULL AUTO_INCREMENT,
  woocommerce_term_id bigint(20) NOT NULL,
  meta_key varchar(255) NULL,
  meta_value longtext NULL,
  PRIMARY KEY  (meta_id)
) $collate;
";
    dbDelta($sql);
    
    // Table for storing user and guest download permissions
    // KEY(order_id, product_id, download_id) used for organizing downloads on the My Account page
    $sql = "
CREATE TABLE ". $wpdb->prefix . "woocommerce_downloadable_product_permissions (
  download_id varchar(32) NOT NULL,
  product_id bigint(20) NOT NULL,
  order_id bigint(20) NOT NULL DEFAULT 0,
  order_key varchar(200) NOT NULL,
  user_email varchar(200) NOT NULL,
  user_id bigint(20) NULL,
  downloads_remaining varchar(9) NULL,
  access_granted datetime NOT NULL default '0000-00-00 00:00:00',
  access_expires datetime NULL default null,
  download_count bigint(20) NOT NULL DEFAULT 0,
  PRIMARY KEY  (product_id,order_id,order_key,download_id),
  KEY (order_id,product_id,download_id)
) $collate;
";
    dbDelta($sql);

    /**
     * Data Upgrades
     **/
	if ( version_compare( get_option('woocommerce_db_version'), '1.7', '<' ) ) {

		// upgrade existing meta data
		$existing_file_paths = $wpdb->get_results( "SELECT * FROM ". $wpdb->postmeta . " WHERE meta_key = '_file_path'" );
		if ( $existing_file_paths ) {
			foreach( $existing_file_paths as $existing_file_path ) {
				$existing_file_path->meta_value = trim( $existing_file_path->meta_value );
				if ( $existing_file_path->meta_value ) 
					$file_paths = maybe_serialize( array( md5( $existing_file_path->meta_value ) => $existing_file_path->meta_value ) );
				else 
					$file_paths = '';
				$wpdb->query( $wpdb->prepare( "UPDATE " . $wpdb->postmeta . " SET meta_key = '_file_paths', meta_value = %s WHERE meta_id = %d", $file_paths, $existing_file_path->meta_id ) );
				$wpdb->query( $wpdb->prepare( "UPDATE " . $wpdb->prefix . "woocommerce_downloadable_product_permissions SET download_id = %s WHERE product_id = %d", md5( $existing_file_path->meta_value ), $existing_file_path->post_id ) );
			}
		}
	}

    if ( version_compare( get_option('woocommerce_db_version'), '1.0', '>' ) && version_compare( get_option('woocommerce_db_version'), '1.4', '<' ) ) {

	    // Update woocommerce_downloadable_product_permissions table to include order ID's as well as keys
	    $results = $wpdb->get_results( "SELECT * FROM " . $wpdb->prefix . "woocommerce_downloadable_product_permissions WHERE order_id = 0;" );

		if ( $results ) foreach ( $results as $result ) {

			if ( ! $result->order_key )
				continue;

			$order_id = $wpdb->get_var( $wpdb->prepare( "SELECT post_id FROM {$wpdb->postmeta} WHERE meta_key = '_order_key' AND meta_value = '%s' LIMIT 1;", $result->order_key ) );

			if ( $order_id ) {

				$wpdb->update( $wpdb->prefix . "woocommerce_downloadable_product_permissions", array(
					'order_id' => $order_id,
				), array(
					'product_id' => $result->product_id,
					'order_key' => $result->order_key
				), array( '%s' ), array( '%s', '%s' ) );

			}

		}

		// Upgrade old meta keys for product data
		$meta = array('sku', 'downloadable', 'virtual', 'price', 'visibility', 'stock', 'stock_status', 'backorders', 'manage_stock', 'sale_price', 'regular_price', 'weight', 'length', 'width', 'height', 'tax_status', 'tax_class', 'upsell_ids', 'crosssell_ids', 'sale_price_dates_from', 'sale_price_dates_to', 'min_variation_price', 'max_variation_price', 'featured', 'product_attributes', 'file_path', 'download_limit', 'product_url', 'min_variation_price', 'max_variation_price');

		$wpdb->query("
			UPDATE {$wpdb->postmeta}
			LEFT JOIN {$wpdb->posts} ON ( {$wpdb->postmeta}.post_id = {$wpdb->posts}.ID )
			SET meta_key = CONCAT( '_', meta_key )
			WHERE meta_key IN ( '" . implode( "', '", $meta ) . "' )
			AND {$wpdb->posts}.post_type IN ('product', 'product_variation')
		");
	}
}


/**
 * Add the default terms for WC taxonomies - product types and order statuses. Modify this at your own risk.
 *
 * @access public
 * @return void
 */
function woocommerce_default_taxonomies() {

	$product_types = array(
		'simple',
		'grouped',
		'variable',
		'external'
	);

	foreach ( $product_types as $type ) {
		if ( ! get_term_by( 'slug', sanitize_title( $type ), 'product_type' ) ) {
			wp_insert_term( $type, 'product_type' );
		}
	}

	$order_status = array(
		'pending',
		'failed',
		'on-hold',
		'processing',
		'completed',
		'refunded',
		'cancelled'
	);

	foreach ( $order_status as $status ) {
		if ( ! get_term_by( 'slug', sanitize_title($status), 'shop_order_status' ) ) {
			wp_insert_term( $status, 'shop_order_status' );
		}
	}

	// Upgrade from old downloadable/virtual product types
	$downloadable_type = get_term_by( 'slug', 'downloadable', 'product_type' );
	if ( $downloadable_type ) {
		$products = get_objects_in_term( $downloadable_type->term_id, 'product_type' );
		foreach ( $products as $product ) {
			update_post_meta( $product, '_downloadable', 'yes' );
			update_post_meta( $product, '_virtual', 'yes' );
			wp_set_object_terms( $product, 'simple', 'product_type');
		}
	}

	$virtual_type = get_term_by( 'slug', 'virtual', 'product_type' );
	if ( $virtual_type ) {
		$products = get_objects_in_term( $virtual_type->term_id, 'product_type' );
		foreach ( $products as $product ) {
			update_post_meta( $product, '_downloadable', 'no' );
			update_post_meta( $product, '_virtual', 'yes' );
			wp_set_object_terms( $product, 'simple', 'product_type');
		}
	}
}