From 4df543df4f41c6f682ed62b0745d103572c39e4e Mon Sep 17 00:00:00 2001
From: Marin Atanasov <tyxla@abv.bg>
Date: Mon, 30 Sep 2019 14:43:56 +0300
Subject: [PATCH 1/5] Admin: Load connect button scripts on dashboard and
 plugins

---
 _inc/lib/admin-pages/class.jetpack-admin-page.php | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/_inc/lib/admin-pages/class.jetpack-admin-page.php b/_inc/lib/admin-pages/class.jetpack-admin-page.php
index 5cef91a7e2c86..8dc8d37a20758 100644
--- a/_inc/lib/admin-pages/class.jetpack-admin-page.php
+++ b/_inc/lib/admin-pages/class.jetpack-admin-page.php
@@ -75,6 +75,16 @@ function add_actions() {
 			delete_transient( 'activated_jetpack' );
 		}
 
+		// If Jetpack not yet connected, but user is viewing one of the pages with a Jetpack connection banner.
+		if (
+			( 'index.php' === $pagenow || 'plugins.php' === $pagenow )
+			&& ! Jetpack::is_active()
+			&& current_user_can( 'jetpack_connect' )
+			&& ! Jetpack::is_development_mode()
+		) {
+			add_action( 'admin_enqueue_scripts', array( 'Jetpack_Connection_Banner', 'enqueue_connect_button_scripts' ) );
+		}
+
 		// Check if the site plan changed and deactivate modules accordingly.
 		add_action( 'current_screen', array( $this, 'check_plan_deactivate_modules' ) );
 

From 5bb7cea11a4e0e62f848e322d31474e16dc4903a Mon Sep 17 00:00:00 2001
From: Marin Atanasov <tyxla@abv.bg>
Date: Mon, 30 Sep 2019 14:44:49 +0300
Subject: [PATCH 2/5] Connection: Add class to alt connection buttons, add
 connect in place URL

---
 class.jetpack-connection-banner.php | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/class.jetpack-connection-banner.php b/class.jetpack-connection-banner.php
index 752aa7be1d14b..24db71cdcfca4 100644
--- a/class.jetpack-connection-banner.php
+++ b/class.jetpack-connection-banner.php
@@ -179,6 +179,7 @@ public static function enqueue_connect_button_scripts() {
 				'buttonTextRegistering' => __( 'Loading...', 'jetpack' ),
 				'jetpackApiDomain'      => $jetpackApiUrl['scheme'] . '://' . $jetpackApiUrl['host'],
 				'forceVariation'        => $force_variation,
+				'connectInPlaceUrl'     => Jetpack::admin_url( 'page=jetpack#/setup' ),
 				'dashboardUrl'          => Jetpack::admin_url( 'page=jetpack#/dashboard' ),
 				'plansPromptUrl'        => Jetpack::admin_url( 'page=jetpack#/plans-prompt' ),
 			)
@@ -284,7 +285,7 @@ class="jp-wpcom-connect__hide-phone-and-smaller"
 								<span class="jp-banner__tos-blurb"><?php jetpack_render_tos_blurb(); ?></span>
 								<a
 										href="<?php echo esc_url( $this->build_connect_url_for_slide( '72' ) ); ?>"
-										class="dops-button is-primary">
+										class="dops-button is-primary jp-banner__alt-connect-button">
 									<?php esc_html_e( 'Set up Jetpack', 'jetpack' ); ?>
 								</a>
 							</div>

From 1f143d35210be7a78780bf12052b04527d7a4b86 Mon Sep 17 00:00:00 2001
From: Marin Atanasov <tyxla@abv.bg>
Date: Mon, 30 Sep 2019 14:45:47 +0300
Subject: [PATCH 3/5] Connection: Redirect alt connection buttons to connect in
 place flows

---
 _inc/connect-button.js | 61 +++++++++++++++++++++++++++++++-----------
 1 file changed, 45 insertions(+), 16 deletions(-)

diff --git a/_inc/connect-button.js b/_inc/connect-button.js
index 20c51951c564d..99a8cd582211b 100644
--- a/_inc/connect-button.js
+++ b/_inc/connect-button.js
@@ -1,30 +1,44 @@
 /* global jpConnect */
 
 jQuery( document ).ready( function( $ ) {
-	var connectButton = $( '.jp-connect-button' );
+	var connectButton = $( '.jp-connect-button, .jp-banner__alt-connect-button' ).eq( 0 );
 	var tosText = $( '.jp-connect-full__tos-blurb' );
-	connectButton.click( function( event ) {
+	var jetpackConnectIframe = $( '<iframe class="jp-jetpack-connect__iframe" />' );
+	var connectionCards = $( '#jetpack-connection-cards' );
+
+	connectButton.on( 'click', function( event ) {
 		event.preventDefault();
-		$( '#jetpack-connection-cards' ).fadeOut( 600 );
-		if ( ! jetpackConnectButton.isRegistering ) {
-			if ( 'original' === jpConnect.forceVariation ) {
-				// Forcing original connection flow, `JETPACK_SHOULD_USE_CONNECTION_IFRAME = false`
-				// or we're dealing with Safari which has issues with handling 3rd party cookies.
-				jetpackConnectButton.handleOriginalFlow();
-			} else if ( 'in_place' === jpConnect.forceVariation ) {
-				// Forcing new connection flow, `JETPACK_SHOULD_USE_CONNECTION_IFRAME = true`.
-				jetpackConnectButton.handleConnectInPlaceFlow();
-			} else {
-				// Forcing A/B test driven connection flow variation, `JETPACK_SHOULD_USE_CONNECTION_IFRAME` not defined.
-				jetpackConnectButton.startConnectionFlow();
-			}
+
+		if ( connectionCards.length ) {
+			connectionCards.fadeOut( 600 );
 		}
+
+		jetpackConnectButton.selectAndStartConnectionFlow();
 	} );
-	var jetpackConnectIframe = $( '<iframe class="jp-jetpack-connect__iframe" />' );
 
 	var jetpackConnectButton = {
 		isRegistering: false,
 		isPaidPlan: false,
+		selectAndStartConnectionFlow: function() {
+			var connectionCards = $( '#jetpack-connection-cards' );
+			if ( connectionCards.length ) {
+				connectionCards.fadeOut( 600 );
+			}
+
+			if ( ! jetpackConnectButton.isRegistering ) {
+				if ( 'original' === jpConnect.forceVariation ) {
+					// Forcing original connection flow, `JETPACK_SHOULD_USE_CONNECTION_IFRAME = false`
+					// or we're dealing with Safari which has issues with handling 3rd party cookies.
+					jetpackConnectButton.handleOriginalFlow();
+				} else if ( 'in_place' === jpConnect.forceVariation ) {
+					// Forcing new connection flow, `JETPACK_SHOULD_USE_CONNECTION_IFRAME = true`.
+					jetpackConnectButton.handleConnectInPlaceFlow();
+				} else {
+					// Forcing A/B test driven connection flow variation, `JETPACK_SHOULD_USE_CONNECTION_IFRAME` not defined.
+					jetpackConnectButton.startConnectionFlow();
+				}
+			}
+		},
 		startConnectionFlow: function() {
 			var abTestName = 'jetpack_connect_in_place_v2';
 			$.ajax( {
@@ -48,6 +62,12 @@ jQuery( document ).ready( function( $ ) {
 			window.location = connectButton.attr( 'href' );
 		},
 		handleConnectInPlaceFlow: function() {
+			// Alternative connection buttons should redirect to the main one for the "connect in place" flow.
+			if ( connectButton.hasClass( 'jp-banner__alt-connect-button' ) ) {
+				window.location = jpConnect.connectInPlaceUrl;
+				return;
+			}
+
 			jetpackConnectButton.isRegistering = true;
 			tosText.hide();
 			connectButton.hide();
@@ -136,4 +156,13 @@ jQuery( document ).ready( function( $ ) {
 			jetpackConnectButton.handleOriginalFlow();
 		},
 	};
+
+	var hash = location.hash.replace( /#\//, '' );
+	if ( 'setup' === hash ) {
+		if ( connectionCards.length ) {
+			connectionCards.hide();
+		}
+
+		jetpackConnectButton.selectAndStartConnectionFlow();
+	}
 } );

From 20ef8f24180d085471ae895e2daf068996e8d9fe Mon Sep 17 00:00:00 2001
From: Marin Atanasov <tyxla@abv.bg>
Date: Mon, 30 Sep 2019 15:56:48 +0300
Subject: [PATCH 4/5] Add a comment

---
 _inc/connect-button.js | 1 +
 1 file changed, 1 insertion(+)

diff --git a/_inc/connect-button.js b/_inc/connect-button.js
index 99a8cd582211b..45320c8089e4f 100644
--- a/_inc/connect-button.js
+++ b/_inc/connect-button.js
@@ -157,6 +157,7 @@ jQuery( document ).ready( function( $ ) {
 		},
 	};
 
+	// When we visit /wp-admin/admin.php?page=jetpack#/setup, immediately start the connection flow.
 	var hash = location.hash.replace( /#\//, '' );
 	if ( 'setup' === hash ) {
 		if ( connectionCards.length ) {

From 22063d8e6eae6dc0ee8caa165528dcabb53d3fc5 Mon Sep 17 00:00:00 2001
From: Marin Atanasov <tyxla@abv.bg>
Date: Mon, 30 Sep 2019 16:07:59 +0300
Subject: [PATCH 5/5] Also hide the cancel button

---
 _inc/connect-button.js | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/_inc/connect-button.js b/_inc/connect-button.js
index 45320c8089e4f..c30775d19278f 100644
--- a/_inc/connect-button.js
+++ b/_inc/connect-button.js
@@ -4,13 +4,15 @@ jQuery( document ).ready( function( $ ) {
 	var connectButton = $( '.jp-connect-button, .jp-banner__alt-connect-button' ).eq( 0 );
 	var tosText = $( '.jp-connect-full__tos-blurb' );
 	var jetpackConnectIframe = $( '<iframe class="jp-jetpack-connect__iframe" />' );
-	var connectionCards = $( '#jetpack-connection-cards' );
+	var connectionHelpSections = $(
+		'#jetpack-connection-cards, .jp-connect-full__dismiss-paragraph'
+	);
 
 	connectButton.on( 'click', function( event ) {
 		event.preventDefault();
 
-		if ( connectionCards.length ) {
-			connectionCards.fadeOut( 600 );
+		if ( connectionHelpSections.length ) {
+			connectionHelpSections.fadeOut( 600 );
 		}
 
 		jetpackConnectButton.selectAndStartConnectionFlow();
@@ -20,9 +22,9 @@ jQuery( document ).ready( function( $ ) {
 		isRegistering: false,
 		isPaidPlan: false,
 		selectAndStartConnectionFlow: function() {
-			var connectionCards = $( '#jetpack-connection-cards' );
-			if ( connectionCards.length ) {
-				connectionCards.fadeOut( 600 );
+			var connectionHelpSections = $( '#jetpack-connection-cards' );
+			if ( connectionHelpSections.length ) {
+				connectionHelpSections.fadeOut( 600 );
 			}
 
 			if ( ! jetpackConnectButton.isRegistering ) {
@@ -160,8 +162,8 @@ jQuery( document ).ready( function( $ ) {
 	// When we visit /wp-admin/admin.php?page=jetpack#/setup, immediately start the connection flow.
 	var hash = location.hash.replace( /#\//, '' );
 	if ( 'setup' === hash ) {
-		if ( connectionCards.length ) {
-			connectionCards.hide();
+		if ( connectionHelpSections.length ) {
+			connectionHelpSections.hide();
 		}
 
 		jetpackConnectButton.selectAndStartConnectionFlow();