<?php

namespace MediaWiki\Extension\CheckUser\Tests\Integration\Services;

use Firebase\JWT\JWT;
use MediaWiki\Config\ConfigException;
use MediaWiki\Extension\CheckUser\Services\TokenManager;
use MediaWiki\Request\FauxRequest;
use MediaWiki\Session\SessionManager;
use MediaWikiIntegrationTestCase;
use Wikimedia\Timestamp\ConvertibleTimestamp;

/**
 * Test class for TokenManager class
 *
 * @group CheckUser
 * @group Database
 *
 * @covers \MediaWiki\Extension\CheckUser\Services\TokenManager
 */
class TokenManagerTest extends MediaWikiIntegrationTestCase {

	public function setUp(): void {
		parent::setUp();
		ConvertibleTimestamp::setFakeTime( 0 );
		JWT::$timestamp = 60;
	}

	public function tearDown(): void {
		parent::tearDown();
		ConvertibleTimestamp::setFakeTime( null );
		JWT::$timestamp = null;
	}

	public function testEncodeDecode() {
		$tokenManager = new TokenManager( 'abcdef' );
		$targets = [ 'Example', '10.0.0.0/8' ];
		$request = new FauxRequest( [], false, [
			'CheckUserTokenKey' => base64_encode( 'test' ),
		] );

		$encoded = $tokenManager->encode( $request->getSession(), [
			'targets' => $targets,
		] );

		$decoded = $tokenManager->decode( $request->getSession(), $encoded );
		$this->assertIsArray( $decoded );
		$this->assertCount( 1, $decoded );
		$this->assertArrayHasKey( 'targets', $decoded );
		$this->assertSame( $targets, $decoded['targets'] );
	}

	public function testDecodeSecretFailure() {
		$this->expectExceptionMessage( 'Signature verification failed' );

		$tokenManager = new TokenManager( 'abcdef' );
		$session = SessionManager::singleton()->getEmptySession();
		$encoded = $tokenManager->encode( $session, [] );

		$tokenManager = new TokenManager( 'abcdef2' );
		$tokenManager->decode( $session, $encoded );
	}

	public function testDecodeSessionFailure() {
		$this->expectExceptionMessage( 'Signature verification failed' );

		$tokenManager = new TokenManager( 'abcdef' );
		$encoded = $tokenManager->encode( SessionManager::singleton()->getEmptySession(), [] );
		$tokenManager->decode( SessionManager::singleton()->getEmptySession(), $encoded );
	}

	public function testDecodeExpiredToken() {
		$this->expectExceptionMessage( 'Expired token' );

		$tokenManager = new TokenManager( 'abcdef' );
		$targets = [ 'Example', '10.0.0.0/8' ];
		$request = new FauxRequest( [], false, [ 'CheckUserTokenKey' => base64_encode( 'test' ) ] );

		$encoded = $tokenManager->encode( $request->getSession(), [ 'targets' => $targets ] );

		// Simulate that the token has expired by setting the $timestamp
		JWT::$timestamp = 96401;
		$tokenManager->decode( $request->getSession(), $encoded );
	}

	public function testConstructorWithNoSecret() {
		$this->expectException( ConfigException::class );
		new TokenManager( '' );
	}
}
