<?php

namespace WPML\ST\TranslationFile;

use wpdb;
use WPML\Collect\Support\Collection;
use WPML\FP\Fns;
use WPML\FP\Just;
use WPML\FP\Maybe;
use WPML\FP\Nothing;
use WPML\FP\Relation;
use WPML\LIB\WP\Cache;
use WPML\ST\Package\Domains as PackageDomains;
use WPML_Admin_Texts;
use WPML_ST_Blog_Name_And_Description_Hooks;
use WPML_ST_Translations_File_Dictionary;
use WPML\ST\Shortcode;
use WPML\ST\TranslationFile\StringCollation;

class Domains {
	use StringCollation;

	const MO_DOMAINS_CACHE_GROUP = 'WPML_ST_CACHE';
	const MO_DOMAINS_CACHE_KEY   = 'wpml_string_translation_has_mo_domains';

	/** @var wpdb $wpdb */
	private $wpdb;

	/** @var PackageDomains $package_domains */
	private $package_domains;

	/** @var WPML_ST_Translations_File_Dictionary $file_dictionary */
	private $file_dictionary;

	/** @var null|Collection $jed_domains */
	private static $jed_domains;

	/**
	 * Domains constructor.
	 *
	 * @param wpdb                                 $wpdb            The WordPress database instance.
	 * @param PackageDomains                       $package_domains The package domains instance.
	 * @param WPML_ST_Translations_File_Dictionary $file_dictionary The translations file dictionary.
	 */
	public function __construct(
		wpdb $wpdb,
		PackageDomains $package_domains,
		WPML_ST_Translations_File_Dictionary $file_dictionary
	) {
		$this->wpdb            = $wpdb;
		$this->package_domains = $package_domains;
		$this->file_dictionary = $file_dictionary;
	}


	/**
	 * @return Collection
	 */
	public function getMODomains() {
		$transient_key = self::MO_DOMAINS_CACHE_KEY;

		$cacheItem = Cache::get( self::MO_DOMAINS_CACHE_GROUP, self::MO_DOMAINS_CACHE_KEY );
		if ( $cacheItem instanceof Just ) {
			return $cacheItem->get();
		}

		// If not in object cache, try transient.
		$transient = get_transient( $transient_key );
		if ( false !== $transient ) {
			return $transient;
		}

		// If not in any cache, fetch from database.
		$excluded_domains = self::getReservedDomains()->merge( $this->getJEDDomains() );

		$sql = "
			SELECT DISTINCT context {$this->getCollateForContextColumn( $this->wpdb )}
			FROM {$this->wpdb->prefix}icl_strings
		";

		$mo_domains = wpml_collect( $this->wpdb->get_col( $sql ) )
			->diff( $excluded_domains )
			->values();

		// If we don't get any data from DB, we set cache expire time to be 15 minutes so that cache refreshes in lesser time.
		$cacheLifeTime = $mo_domains->count() <= 0 ? 15 * MINUTE_IN_SECONDS : HOUR_IN_SECONDS;

		// Try to store in object cache first.
		Cache::set(
			self::MO_DOMAINS_CACHE_GROUP,
			self::MO_DOMAINS_CACHE_KEY,
			$cacheLifeTime,
			$mo_domains
		);

		// Use transient as fallback only if object cache is not being used.
		if ( ! wp_using_ext_object_cache() ) {
			set_transient( $transient_key, $mo_domains, $cacheLifeTime );
		}

		return $mo_domains;
	}

	public static function invalidateMODomainCache() {
		static $invalidationScheduled = false;

		delete_transient( self::MO_DOMAINS_CACHE_KEY );

		if ( ! $invalidationScheduled ) {
			$invalidationScheduled = true;
			add_action(
				'shutdown',
				function () {
					Cache::flushGroup( self::MO_DOMAINS_CACHE_GROUP );
				}
			);
		}
	}


	/**
	 * Returns a collection of MO domains that
	 * WPML needs to automatically load.
	 *
	 * @return Collection
	 */
	public function getCustomMODomains() {
		$all_mo_domains    = $this->getMODomains();
		$native_mo_domains = $this->file_dictionary->get_domains( 'mo', get_locale() );

		return $all_mo_domains->reject(
			function ( $domain ) use ( $native_mo_domains ) {
				/**
				 * Admin texts, packages, string shortcodes are handled separately,
				 * so they are loaded on-demand.
				 */
				return null === $domain
					   || 0 === strpos( $domain, WPML_Admin_Texts::DOMAIN_NAME_PREFIX )
					   || $this->package_domains->isPackage( $domain )
					   || Shortcode::STRING_DOMAIN === $domain
					   || in_array( $domain, $native_mo_domains, true );
			}
		)->values();
	}

	/**
	 * @return Collection
	 */
	public function getJEDDomains() {
		if ( ! self::$jed_domains instanceof Collection ) {
			self::$jed_domains = wpml_collect( $this->file_dictionary->get_domains( 'json' ) );
		}

		return self::$jed_domains;
	}

	/**
	 * @param string $domain
	 *
	 * @return bool
	 */
	public function hasNoNativeTranslationFile( $domain ) {
		return ! wpml_collect( $this->file_dictionary->get_domains() )
			->first( Relation::equals( $domain ) );
	}

	public static function resetCache() {
		self::invalidateMODomainCache();
		self::$jed_domains = null;
	}

	/**
	 * Domains that are not handled with MO files,
	 * but have direct DB queries.
	 *
	 * @return Collection
	 */
	public static function getReservedDomains() {
		return wpml_collect(
			[
				WPML_ST_Blog_Name_And_Description_Hooks::STRING_DOMAIN,
			]
		);
	}

	/**
	 * @return Collection
	 */
	private function getPackageDomains() {
		return $this->package_domains->getDomains();
	}
}