LICENS-45 Add auth logic

src/Uplink/API/Validation_Response.php

@@ -313,6 +313,7 @@ public function handle_api_errors() : stdClass {
 			'id',
 			'slug',
 			'version',
+			'auth_required',
 			'homepage',
 			'download_url',
 			'upgrade_notice',
@@ -492,8 +493,22 @@ public function to_wp_format() {
 			}
 		}
 
-		//Other fields need to be renamed and/or transformed.
-		$info->download_link = isset( $this->response->download_url ) ? $this->response->download_url . '&pu_get_download=1' : '';
+		if ( empty( $this->response->auth_required ) || $this->resource->has_valid_auth_token( (array) $this->response->origin ) ) {
+			$info->download_link = isset($this->response->download_url) ? $this->response->download_url . '&pu_get_download=1' : '';
+		} else {
+			$url 		  = $this->response->origin->url;
+			$query_params = [
+				'callback_uri' => urlencode( sprintf( '%s/stellarwp/connect', get_site_url() ) ),
+				'refer'		   => urlencode( wp_get_referer() ),
+			];
+			$url 		  = sprintf( '%s/stellarwp/oauth_connect/login?%s', $url, http_build_query( $query_params ) );
+
+			$info->api_upgrade   = sprintf(
+				esc_html__( 'Please connect plugin on Setting page in order to receive updates. You can use plugin settings page or follow this <a href="%s">link</a>', '%TEXTDOMAIN%' ),
+				$url
+			);
+			$info->download_link = '';
+		}
 
 		if ( ! empty( $this->author_homepage ) && ! empty( $this->response->author ) ) {
 			$info->author = sprintf( '<a href="%s">%s</a>', esc_url( $this->author_homepage ), $this->response->author );

src/Uplink/Admin/Actions.php

@@ -0,0 +1,95 @@
+<?php declare(strict_types=1);
+
+namespace StellarWP\Uplink\Admin;
+
+use StellarWP\Uplink\API;
+use StellarWP\Uplink\Config;
+use StellarWP\Uplink\Resources\Collection;
+use StellarWP\Uplink\Site\Data;
+
+class Actions {
+
+	const QUERY_VAR = 'stellarwp_action';
+
+	/**
+	 * @return void
+	 * @action init
+	 */
+	public function register_route() {
+		add_rewrite_endpoint( 'stellarwp', EP_ROOT, self::QUERY_VAR );
+	}
+
+	/**
+	 * @param \WP $wp
+	 *
+	 * @return void
+	 * @action parse_request
+	 */
+	public function handle_auth_request( $wp ) {
+		if ( empty( $wp->query_vars[ self::QUERY_VAR ] ) ) {
+			return;
+		}
+
+		$args = explode( '/', $wp->query_vars[ self::QUERY_VAR ] );
+
+		if (  ! empty( $args['disconnect'] ) ) { // @phpstan-ignore-line
+			$this->handle_disconnect();
+		}
+
+		$this->handle_connect( $args );
+	}
+
+	/**
+	 * Remove auth token§
+	 */
+	public function handle_disconnect() {
+		$license = $this->get_license_object();
+
+		delete_option( sprintf( 'stellarwp_origin_%s_auth_token', $license->origin->slug ?? '' ) );
+
+		wp_safe_redirect( wp_get_referer() );
+		exit();
+	}
+
+	public function handle_connect( $args ) {
+		if ( empty( $args['token'] ) ) {
+			$url = $this->get_origin_url();
+			if ( empty( $url ) ) {
+				return;
+			}
+
+			$query_params = [
+				'callback_uri' => urlencode( sprintf( '%s/stellarwp/connect', get_site_url() ) ),
+				'refer'		   => urlencode( wp_get_referer() ),
+			];
+			$url = sprintf( '%s/stellarwp/oauth_connect/login?%s', $url, http_build_query( $query_params ) );
+
+			wp_safe_redirect( $url );
+			exit();
+		}
+
+		Config::get_container()->get( Data::class )->save_auth_token( $args['token'] );
+
+		wp_safe_redirect( $args['refer'] );
+		exit();
+	}
+
+	protected function get_origin_url() {
+		$license = $this->get_license_object();
+		$api     = Config::get_container()->get( API\Client::class );
+		$origin  = $api->post('/origin', [ 'slug' => $license->get_slug() ] );
+
+		if ( ! empty( $origin ) ) {
+			return $origin->url . '/stellarwp_connect';
+		}
+
+		return '';
+	}
+
+	protected function get_license_object() {
+		$collection = Config::get_container()->get( Collection::class );
+		$plugin     = $collection->current();
+
+		return $plugin->get_license_object();
+	}
+}

src/Uplink/Admin/Auth.php

@@ -0,0 +1,34 @@
+<?php declare(strict_types=1);
+
+namespace StellarWP\Uplink\Admin;
+
+use StellarWP\Uplink\Config;
+use StellarWP\Uplink\Resources\Collection;
+
+class Auth {
+
+	public function do_auth_html() {
+		$collection = Config::get_container()->get( Collection::class );
+		$plugin     = $collection->current();
+		$license    = $plugin->get_license_object();
+
+		$token   = get_option( sprintf( 'stellarwp_origin_%s_auth_token', $license->origin->slug ?? '' ), '' );
+		$message = esc_html__( 'Connect to origin', '%TEXTDOMAIN%' );
+		$classes = [ 'button', 'button-primary' ];
+		$url     = '/stellarwp/connect';
+
+		if ( ! empty( $token ) ) {
+			$message = esc_html__( 'Disconnect from origin', '%TEXTDOMAIN%' );
+			$classes = [ 'button', 'button-secondary'];
+			$url     = '/stellarwp/disconnect';
+		}
+
+		return sprintf(
+			'<a href="%s" class="%s">%s</a>',
+			$url,
+			implode( ' ', $classes ),
+			$message
+		);
+	}
+
+}

src/Uplink/Admin/Provider.php

@@ -45,6 +45,7 @@ public function register_hooks() {
 		add_action( 'admin_enqueue_scripts', [ $this, 'store_admin_notices' ], 10, 1 );
 		add_action( 'admin_notices', [ $this, 'admin_notices' ], 10, 0 );
 		add_action( 'load-plugins.php', [ $this, 'remove_default_update_message' ], 50, 0 );
+		add_action( 'parse_request', [ $this, 'auth_request' ], 10, 1 );
 	}
 
 	/**
@@ -62,6 +63,18 @@ public function filter_plugins_api( $result, $action, $args ) {
 		return $this->container->get( Plugins_Page::class )->inject_info( $result, $action, $args );
 	}
 
+	/**
+	 * Handle auth and disconnect requests to origin
+	 *
+	 * @param \WP $wp
+	 */
+	public function auth_request( $wp ) {
+		if ( ! is_user_logged_in() ) {
+			return;
+		}
+		$this->container->get( Actions::class )->handle_auth_request( $wp );
+	}
+
 	/**
 	 * Filter the plugins transient.
 	 *
@@ -94,6 +107,7 @@ public function ajax_validate_license() {
 	 * @return void
 	 */
 	public function admin_init() {
+		$this->container->get( Actions::class )->register_route();
 		$this->container->get( License_Field::class )->register_settings();
 	}
 

src/Uplink/Resources/Resource.php

@@ -499,4 +499,20 @@ public function validate_license( $key = null, $do_network_validate = false ) {
 
 		return $results;
 	}
+
+	public function has_valid_auth_token( array $origin ) {
+		$token = get_option( sprintf( 'stellarwp_origin_%s_auth_token', $origin['slug'] ?? '' ), '' );
+
+		if ( empty( $token ) ) {
+			return false;
+		}
+
+		$token = json_decode( $token, true );
+
+		if ( empty( $token ) ) {
+			return false;
+		}
+
+		return $token['expiration'] > time();
+	}
 }

src/Uplink/Site/Data.php

@@ -518,4 +518,10 @@ public function is_public(): bool {
 
 		return (bool) $is_public;
 	}
+
+	public function save_auth_token( string $data ) {
+		$data = json_decode( base64_decode( $data ), true );
+		update_option(  sprintf( 'stellarwp_origin_%s_auth_token', $data['origin'] ?? '' ), json_encode( $data ) );
+	}
+
 }

src/admin-views/fields/settings.php

@@ -6,6 +6,7 @@
  * @var bool $show_title Whether to show the title. Default true.
  */
 
+use StellarWP\Uplink\Admin\Auth;
 use StellarWP\Uplink\Admin\License_Field;
 use StellarWP\Uplink\Config;
 
@@ -14,6 +15,7 @@
 }
 
 $field = Config::get_container()->get( License_Field::class );
+$auth  = Config::get_container()->get( Auth::class );
 $group = $field->get_group_name( sanitize_title( $plugin->get_slug() ) );
 
 ?>
@@ -23,6 +25,7 @@
 
 <div class="stellarwp-uplink" data-js="stellarwp-uplink">
 	<div class="stellarwp-uplink__settings">
+		<?php echo $auth->do_auth_html(); ?>
 		<?php do_action( 'stellarwp/uplink/' . Config::get_hook_prefix(). '/license_field_before_form', $plugin->get_slug() ) ?>
 		<form method="post" action="options.php">
 			<?php settings_fields( $group ); ?>

tests/wpunit/Resources/ResourceTest.php

@@ -0,0 +1,49 @@
+<?php declare( strict_types=1 );
+
+namespace StellarWP\Uplink\Tests\Resources;
+
+use StellarWP\Uplink\API\Validation_Response;
+use StellarWP\Uplink\Register;
+use StellarWP\Uplink\Uplink;
+
+class ResourceTest extends \StellarWP\Uplink\Tests\UplinkTestCase {
+
+	public $resource;
+
+	public function setUp() {
+		parent::setUp();
+
+		$root 		    = dirname( __DIR__, 3 );
+		$this->resource = Register::plugin(
+			'sample',
+			'Lib Sample',
+			$root . '/plugin.php',
+			Uplink::class,
+			'1.0.10',
+			Uplink::class
+		);
+	}
+
+	public function it_should_check_auth_token_valid() {
+		$result = $this->resource->has_valid_auth_token( [ 'slug' => 'sample' ] );
+		$this->assertFalse( $result );
+
+		update_option( sprintf( 'stellarwp_origin_%s_auth_token', 'sample' ), [
+			'token'      => '11111',
+			'expiration' => strtotime( '-1 day'),
+			'origin'     => '',
+		] );
+		$result = $this->resource->has_valid_auth_token( [ 'slug' => 'sample' ] );
+		$this->assertFalse( $result );
+
+		update_option( sprintf( 'stellarwp_origin_%s_auth_token', 'sample' ), [
+			'token'      => '11111',
+			'expiration' => strtotime( '+1 day'),
+			'origin'     => '',
+		] );
+
+		$result = $this->resource->has_valid_auth_token( [ 'slug' => 'sample' ] );
+		$this->assertTrue( $result );
+	}
+
+}

tests/wpunit/Site/DataTest.php

@@ -55,4 +55,18 @@ public function it_should_collect_full_stats() {
 
 		remove_filter( 'stellarwp/uplink/test/use_full_stats', '__return_true' );
 	}
+
+	public function it_should_save_auth_token() {
+		$token = json_encode( [
+			'token'      => '11111',
+			'expiration' => strtotime( '+1 day'),
+			'origin'     => 'sample',
+		] );
+
+		$data = $this->container->make( Uplink\Site\Data::class );
+		$data->save_auth_token( base64_encode( $token ) );
+
+		$options_value = get_option( sprintf( 'stellarwp_origin_%s_auth_token', 'sample' ) );
+		$this->assertSame( $token, $options_value );
+	}
 }