Subscriptions.php 1.87 KB
<?php

namespace ACA\WC;

use AC;
use ACA\WC\Column;
use ACA\WC\ListScreen\Product;
use ACA\WC\ListScreenFactory\ShopSubscriptionFactory;
use ACA\WC\ListScreenFactory\WcSubscriptions;
use WC_Subscriptions;

/**
 * @since 3.4
 */
final class Subscriptions implements AC\Registerable {

	/**
	 * @return bool
	 */
	private function is_wc_subscriptions_active() {
		if ( ! class_exists( 'WC_Subscriptions', false ) ) {
			return false;
		}

		return version_compare( WC_Subscriptions::$version, '2.6', '>=' );
	}

	public function register() {
		if ( ! $this->is_wc_subscriptions_active() ) {
			return;
		}

		add_action( 'ac/column_groups', [ $this, 'register_column_groups' ] );
		add_action( 'ac/column_types', [ $this, 'add_product_columns' ] );
		add_action( 'ac/column_types', [ $this, 'add_user_columns' ] );

		AC\ListScreenFactory::add( new ShopSubscriptionFactory() );
	}

	public function register_column_groups( AC\Groups $groups ): void {
		$groups->add( 'woocommerce_subscriptions', __( 'WooCommerce Subscriptions', 'codepress-admin-columns' ), 15 );
	}

	public function add_product_columns( AC\ListScreen $list_screen ) {
		if ( $list_screen instanceof Product ) {
			$columns = [
				Column\ProductSubscription\Expires::class,
				Column\ProductSubscription\FreeTrial::class,
				Column\ProductSubscription\LimitSubscription::class,
				Column\ProductSubscription\Period::class,
			];

			foreach ( $columns as $column ) {
				$list_screen->register_column_type( new $column );
			}
		}
	}

	public function add_user_columns( AC\ListScreen $list_screen ) {
		if ( $list_screen instanceof AC\ListScreen\User ) {
			$columns = [
				Column\UserSubscription\ActiveSubscriber::class,
				Column\UserSubscription\InactiveSubscriber::class,
				Column\UserSubscription\Subscriptions::class,
			];

			foreach ( $columns as $column ) {
				$list_screen->register_column_type( new $column );
			}
		}

	}

}