Merge pull request #51 from xwp/feature/add-toolbar-link_issue_48

Added a toolbar link to edit post screen view.
xwp · Jul 5, 2016 · cb45595 · cb45595
2 parents 3653198 + f692f04
commit cb45595
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 9 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -16,10 +16,6 @@ env:
     - WP_VERSION=trunk WP_MULTISITE=1
     - WP_VERSION=latest WP_MULTISITE=1
 
-branches:
-    only:
-        - master
-
 install:
     - export DEV_LIB_PATH=dev-lib
     - if [ ! -e "$DEV_LIB_PATH" ] && [ -L .travis.yml ]; then export DEV_LIB_PATH=$( dirname $( readlink .travis.yml ) ); fi

diff --git a/dev-lib b/dev-lib
diff --git a/php/class-customize-snapshot-manager.php b/php/class-customize-snapshot-manager.php
@@ -765,11 +765,44 @@ public function get_snapshot_uuid() {
 	}
 
 	/**
-	 * Replaces the "Customize" link in the Toolbar.
+	 * Toolbar modifications for Customize Snapshot
 	 *
 	 * @param \WP_Admin_Bar $wp_admin_bar WP_Admin_Bar instance.
 	 */
 	public function customize_menu( $wp_admin_bar ) {
+
+		$this->replace_customize_link( $wp_admin_bar );
+		$this->add_post_edit_screen_link( $wp_admin_bar );
+
+		add_action( 'wp_before_admin_bar_render', 'wp_customize_support_script' );
+	}
+
+	/**
+	 * Adds a "Snapshot in Dashboard" link to the Toolbar when in Snapshot mode.
+	 *
+	 * @param \WP_Admin_Bar $wp_admin_bar WP_Admin_Bar instance.
+	 */
+	public function add_post_edit_screen_link( $wp_admin_bar ) {
+		$snapshot_post = $this->snapshot->post();
+		if ( ! $snapshot_post ) {
+			return;
+		}
+
+		$wp_admin_bar->add_node( array(
+			'parent' => 'customize',
+			'id' => 'snapshot-view-link',
+			'title' => __( 'View Snapshot', 'customize-snapshots' ),
+			'href' => get_edit_post_link( $snapshot_post->ID ),
+		) );
+	}
+
+	/**
+	 * Replaces the "Customize" link in the Toolbar.
+	 *
+	 * @param \WP_Admin_Bar $wp_admin_bar WP_Admin_Bar instance.
+	 */
+	public function replace_customize_link( $wp_admin_bar ) {
+
 		// Don't show for users who can't access the customizer or when in the admin.
 		if ( ! current_user_can( 'customize' ) || is_admin() ) {
 			return;
@@ -795,7 +828,6 @@ public function customize_menu( $wp_admin_bar ) {
 				),
 			)
 		);
-		add_action( 'wp_before_admin_bar_render', 'wp_customize_support_script' );
 	}
 
 	/**

diff --git a/tests/php/test-class-customize-snapshot-manager.php b/tests/php/test-class-customize-snapshot-manager.php
@@ -158,14 +158,16 @@ function test_clean_current_url() {
 	}
 
 	/**
+	 * Test Customize_Snapshot_Manager::create_post_type().
+	 *
 	 * @see Customize_Snapshot_Manager::create_post_type()
 	 */
 	function test_create_post_type() {
 		$pobj = get_post_type_object( Customize_Snapshot_Manager::POST_TYPE );
-		$this->assertInstanceOf( 'stdClass', $pobj );
+		$this->assertNotNull( $pobj );
 		$this->assertEquals( Customize_Snapshot_Manager::POST_TYPE, $pobj->name );
 
-		// Test some defaults
+		// Test some defaults.
 		$this->assertFalse( is_post_type_hierarchical( Customize_Snapshot_Manager::POST_TYPE ) );
 		$this->assertEquals( array(), get_object_taxonomies( Customize_Snapshot_Manager::POST_TYPE ) );
 	}