activation.php | searchcode

/lib/activation.php

https://github.com/ndimatteo/hpv-theme
PHP | 367 lines | 298 code | 69 blank | 0 comment | 39 complexity | 632abcb479688da409db6c0dfcdd3671 MD5 | raw file

<?php

if (is_admin() && isset($_GET['activated']) && 'themes.php' == $GLOBALS['pagenow']) {
	wp_redirect(admin_url('themes.php?page=theme_activation_options'));
	exit;
}

function roots_theme_activation_options_init() {
	if (roots_get_theme_activation_options() === false) {
		add_option('roots_theme_activation_options', roots_get_default_theme_activation_options());
	}

	register_setting(
		'roots_activation_options',
		'roots_theme_activation_options',
		'roots_theme_activation_options_validate'
		);
}

add_action('admin_init', 'roots_theme_activation_options_init');

function roots_activation_options_page_capability($capability) {
	return 'edit_theme_options';
}

add_filter('option_page_capability_roots_activation_options', 'roots_activation_options_page_capability');

function roots_theme_activation_options_add_page() {
	$roots_activation_options = roots_get_theme_activation_options();
	if (!$roots_activation_options['first_run']) {
		$theme_page = add_theme_page(
			__('Theme Activation', 'roots'),
			__('Theme Activation', 'roots'),
			'edit_theme_options',
			'theme_activation_options',
			'roots_theme_activation_options_render_page'
			);
	} else {
		if (is_admin() && isset($_GET['page']) && $_GET['page'] === 'theme_activation_options') {
			global $wp_rewrite;
			$wp_rewrite->flush_rules();
			wp_redirect(admin_url('themes.php'));
			exit;
		}
	}

}

add_action('admin_menu', 'roots_theme_activation_options_add_page', 50);

function roots_get_default_theme_activation_options() {
	$default_theme_activation_options = array(
		'first_run'                       => false,
		'create_front_page'               => false,
		'change_permalink_structure'      => false,
		'change_uploads_folder'           => false,
		'create_navigation_menus'         => false,
		'add_pages_to_primary_navigation' => false,
		);

	return apply_filters('roots_default_theme_activation_options', $default_theme_activation_options);
}

function roots_get_theme_activation_options() {
	return get_option('roots_theme_activation_options', roots_get_default_theme_activation_options());
}

function roots_theme_activation_options_render_page() { ?>

<div class="wrap">
	<?php screen_icon(); ?>
	<h2><?php printf(__('%s Theme Activation', 'roots'), wp_get_theme() ); ?></h2>
	<?php settings_errors(); ?>

	<form method="post" action="options.php">

		<?php
		settings_fields('roots_activation_options');
		$roots_activation_options = roots_get_theme_activation_options();
		$roots_default_activation_options = roots_get_default_theme_activation_options();
		?>

		<input type="hidden" value="1" name="roots_theme_activation_options[first_run]" />

		<table class="form-table">

			<tr valign="top"><th scope="row"><?php _e('Create multiple page or single page site?', 'roots'); ?></th>
				<td>
					<fieldset><legend class="screen-reader-text"><span><?php _e('Create multiple page or single page site?', 'roots'); ?></span></legend>
						<select name="roots_theme_activation_options[create_front_page]" id="create_front_page">
							<option selected="selected" value="yes"><?php echo _e('Multiple Pages', 'roots'); ?></option>
							<option value="no"><?php echo _e('Single Page', 'roots'); ?></option>
						</select>
						<br />
						<small class="description"><?php printf(__('Creates the necessary pages for selected type', 'roots')); ?></small>
					</fieldset>
				</td>
			</tr>

			<tr valign="top"><th scope="row"><?php _e('Change permalink structure?', 'roots'); ?></th>
				<td>
					<fieldset><legend class="screen-reader-text"><span><?php _e('Update permalink structure?', 'roots'); ?></span></legend>
						<select name="roots_theme_activation_options[change_permalink_structure]" id="change_permalink_structure">
							<option selected="selected" value="yes"><?php echo _e('Yes', 'roots'); ?></option>
							<option value="no"><?php echo _e('No', 'roots'); ?></option>
						</select>
						<br />
						<small class="description"><?php printf(__('Change permalink structure to /&#37;postname&#37;/', 'roots')); ?></small>
					</fieldset>
				</td>
			</tr>

			<tr valign="top"><th scope="row"><?php _e('Change uploads folder?', 'roots'); ?></th>
				<td>
					<fieldset><legend class="screen-reader-text"><span><?php _e('Update uploads folder?', 'roots'); ?></span></legend>
						<select name="roots_theme_activation_options[change_uploads_folder]" id="change_uploads_folder">
							<option selected="selected" value="yes"><?php echo _e('Yes', 'roots'); ?></option>
							<option value="no"><?php echo _e('No', 'roots'); ?></option>
						</select>
						<br />
						<small class="description"><?php printf(__('Change uploads folder to /assets/ instead of /wp-content/uploads/', 'roots')); ?></small>
					</fieldset>
				</td>
			</tr>

			<tr valign="top"><th scope="row"><?php _e('Create navigation menu?', 'roots'); ?></th>
				<td>
					<fieldset><legend class="screen-reader-text"><span><?php _e('Create navigation menu?', 'roots'); ?></span></legend>
						<select name="roots_theme_activation_options[create_navigation_menus]" id="create_navigation_menus">
							<option selected="selected" value="yes"><?php echo _e('Yes', 'roots'); ?></option>
							<option value="no"><?php echo _e('No', 'roots'); ?></option>
						</select>
						<br />
						<small class="description"><?php printf(__('Create the Primary Navigation menu and set the location', 'roots')); ?></small>
					</fieldset>
				</td>
			</tr>

			<tr valign="top"><th scope="row"><?php _e('Add pages to menu?', 'roots'); ?></th>
				<td>
					<fieldset><legend class="screen-reader-text"><span><?php _e('Add pages to menu?', 'roots'); ?></span></legend>
						<select name="roots_theme_activation_options[add_pages_to_primary_navigation]" id="add_pages_to_primary_navigation">
							<option selected="selected" value="yes"><?php echo _e('Yes', 'roots'); ?></option>
							<option value="no"><?php echo _e('No', 'roots'); ?></option>
						</select>
						<br />
						<small class="description"><?php printf(__('Add all current published pages to the Primary Navigation', 'roots')); ?></small>
					</fieldset>
				</td>
			</tr>

		</table>

		<?php submit_button(); ?>
	</form>
</div>

<?php }

function roots_theme_activation_options_validate($input) {
	$output = $defaults = roots_get_default_theme_activation_options();

	if (isset($input['first_run'])) {
		if ($input['first_run'] === '1') {
			$input['first_run'] = true;
		}
		$output['first_run'] = $input['first_run'];
	}

	if (isset($input['create_front_page'])) {
		if ($input['create_front_page'] === 'yes') {
			$input['create_front_page'] = true;
		}
		if ($input['create_front_page'] === 'no') {
			$input['create_front_page'] = false;
		}
		$output['create_front_page'] = $input['create_front_page'];
	}

	if (isset($input['change_permalink_structure'])) {
		if ($input['change_permalink_structure'] === 'yes') {
			$input['change_permalink_structure'] = true;
		}
		if ($input['change_permalink_structure'] === 'no') {
			$input['change_permalink_structure'] = false;
		}
		$output['change_permalink_structure'] = $input['change_permalink_structure'];
	}

	if (isset($input['change_uploads_folder'])) {
		if ($input['change_uploads_folder'] === 'yes') {
			$input['change_uploads_folder'] = true;
		}
		if ($input['change_uploads_folder'] === 'no') {
			$input['change_uploads_folder'] = false;
		}
		$output['change_uploads_folder'] = $input['change_uploads_folder'];
	}

	if (isset($input['create_navigation_menus'])) {
		if ($input['create_navigation_menus'] === 'yes') {
			$input['create_navigation_menus'] = true;
		}
		if ($input['create_navigation_menus'] === 'no') {
			$input['create_navigation_menus'] = false;
		}
		$output['create_navigation_menus'] = $input['create_navigation_menus'];
	}

	if (isset($input['add_pages_to_primary_navigation'])) {
		if ($input['add_pages_to_primary_navigation'] === 'yes') {
			$input['add_pages_to_primary_navigation'] = true;
		}
		if ($input['add_pages_to_primary_navigation'] === 'no') {
			$input['add_pages_to_primary_navigation'] = false;
		}
		$output['add_pages_to_primary_navigation'] = $input['add_pages_to_primary_navigation'];
	}

	return apply_filters('roots_theme_activation_options_validate', $output, $input, $defaults);
}

function roots_theme_activation_action() {
	$roots_theme_activation_options = roots_get_theme_activation_options();

	if ($roots_theme_activation_options['create_front_page'] == true) {
		$roots_theme_activation_options['create_front_page'] = false;
		
		$default_pages = array('Home', 'About', 'Menu', 'Gallery', 'News', 'Contact');
		$existing_pages = get_pages();
		$temp = array();

		foreach ($existing_pages as $page) {
			$temp[] = $page->post_title;
		}

		$pages_to_create = array_diff($default_pages, $temp);
		
		$i=0;
		foreach ($pages_to_create as $new_page_title) {
			$i++;
			$add_default_pages = array(
				'post_title' => $new_page_title,
				'post_content' => 'Enter page content here...',
				'post_status' => 'publish',
				'post_type' => 'page',
				'menu_order' => $i
				);

			$result = wp_insert_post($add_default_pages);
		}
		
		$sample = get_page_by_title('Sample Page');
		wp_delete_post( $sample->ID, true );

		$home = get_page_by_title('Home');
		update_option('show_on_front', 'page');
		update_option('page_on_front', $home->ID);

		$home_menu_order = array(
			'ID' => $home->ID,
			'menu_order' => -1
			);
		wp_update_post($home_menu_order);
		
	} elseif ($roots_theme_activation_options['create_front_page'] == false) {
		
		$default_pages = array('Home');
		$existing_pages = get_pages();
		$temp = array();

		foreach ($existing_pages as $page) {
			$temp[] = $page->post_title;
		}

		$pages_to_create = array_diff($default_pages, $temp);

		foreach ($pages_to_create as $new_page_title) {
			$add_default_pages = array(
				'post_title' => $new_page_title,
				'post_content' => 'Enter page content here...',
				'post_status' => 'publish',
				'post_type' => 'page'
				);

			$result = wp_insert_post($add_default_pages);
		}
		
		$sample = get_page_by_title('Sample Page');
		wp_delete_post( $sample->ID, true );

		$home = get_page_by_title('Home');
		update_option('show_on_front', 'page');
		update_option('page_on_front', $home->ID);

		$home_menu_order = array(
			'ID' => $home->ID,
			'menu_order' => -1
			);
		wp_update_post($home_menu_order);
		
	}

	if ($roots_theme_activation_options['change_permalink_structure']) {
		$roots_theme_activation_options['change_permalink_structure'] = false;

		if (get_option('permalink_structure') !== '/%postname%/') {
			update_option('permalink_structure', '/%postname%/');
		}

		global $wp_rewrite;
		$wp_rewrite->init();
		$wp_rewrite->flush_rules();
	}

	if ($roots_theme_activation_options['change_uploads_folder']) {
		$roots_theme_activation_options['change_uploads_folder'] = false;

		update_option('uploads_use_yearmonth_folders', 0);
		update_option('upload_path', 'assets');
	}

	if ($roots_theme_activation_options['create_navigation_menus']) {
		$roots_theme_activation_options['create_navigation_menus'] = false;

		$roots_nav_theme_mod = false;

		if (!has_nav_menu('primary_navigation')) {
			$primary_nav_id = wp_create_nav_menu('Primary Navigation', array('slug' => 'primary_navigation'));
			$roots_nav_theme_mod['primary_navigation'] = $primary_nav_id;
		}

		if ($roots_nav_theme_mod) {
			set_theme_mod('nav_menu_locations', $roots_nav_theme_mod);
		}
	}

	if ($roots_theme_activation_options['add_pages_to_primary_navigation']) {
		$roots_theme_activation_options['add_pages_to_primary_navigation'] = false;

		$primary_nav = wp_get_nav_menu_object('Primary Navigation');
		$primary_nav_term_id = (int) $primary_nav->term_id;
		$menu_items= wp_get_nav_menu_items($primary_nav_term_id);
		if (!$menu_items || empty($menu_items)) {
			$pages = get_pages();
			foreach($pages as $page) {
				$item = array(
					'menu-item-object-id' => $page->ID,
					'menu-item-object' => 'page',
					'menu-item-type' => 'post_type',
					'menu-item-status' => 'publish'
					);
				wp_update_nav_menu_item($primary_nav_term_id, 0, $item);
			}
		}
	}

	update_option('roots_theme_activation_options', $roots_theme_activation_options);
}

add_action('admin_init','roots_theme_activation_action');

function roots_deactivation_action() {
	update_option('roots_theme_activation_options', roots_get_default_theme_activation_options());
}

add_action('switch_theme', 'roots_deactivation_action');