<?php
declare( strict_types = 1 );

namespace MediaWiki\Extension\Translate\Synchronization;

use MediaWiki\Extension\Translate\Services;
use MediaWiki\Extension\Translate\Utilities\BaseMaintenanceScript;
use MediaWiki\MediaWikiServices;

/**
 * Clear the contents of the group synchronization cache
 * @author Abijeet Patro
 * @license GPL-2.0-or-later
 * @since 2021.01
 */
class ClearGroupSyncCacheMaintenanceScript extends BaseMaintenanceScript {
	public function __construct() {
		parent::__construct();
		$this->addDescription( 'Clear the contents of the group synchronization cache for a single or all groups' );

		$this->addOption(
			'group',
			'(optional) Group Id being cleared',
			self::OPTIONAL,
			self::HAS_ARG
		);
		$this->addOption(
			'all',
			'(optional) Clear all groups'
		);

		$this->requireExtension( 'Translate' );
	}

	public function execute() {
		$config = MediaWikiServices::getInstance()->getMainConfig();

		if ( !$config->get( 'TranslateGroupSynchronizationCache' ) ) {
			$this->fatalError( 'GroupSynchronizationCache is not enabled' );
		}

		$this->validateParamsAndArgs();
		$groupId = $this->getOption( 'group' );
		$all = $this->hasOption( 'all' );
		$groupSyncCache = Services::getInstance()->getGroupSynchronizationCache();

		if ( $groupId ) {
			$this->clearGroupFromSync( $groupSyncCache, $groupId );
			$this->output( "Ended synchronization for group: $groupId\n" );
		} elseif ( $all ) {
			// Remove all groups
			$groupsInSync = $groupSyncCache->getGroupsInSync();
			$this->output( 'Found ' . count( $groupsInSync ) . " groups in sync.\n" );
			foreach ( $groupsInSync as $groupId ) {
				$this->clearGroupFromSync( $groupSyncCache, $groupId );
				$this->output( "Ended synchronization for group: $groupId\n" );
			}
		}
	}

	public function validateParamsAndArgs() {
		parent::validateParamsAndArgs();

		$group = $this->getOption( 'group' );
		$all = $this->hasOption( 'all' );

		if ( $all && $group !== null ) {
			$this->fatalError( 'The "all" and "group" options cannot be used together.' );
		}

		if ( !$all && $group === null ) {
			$this->fatalError( 'One of "all" OR "group" options must be specified.' );
		}
	}

	private function clearGroupFromSync( GroupSynchronizationCache $groupSyncCache, string $groupId ): void {
		if ( !$groupSyncCache->isGroupBeingProcessed( $groupId ) ) {
			$this->fatalError( "$groupId is currently not being processed" );
		}

		$groupSyncCache->forceEndSync( $groupId );
	}
}