From 6fec22a6e5f3a4023cf3d308571cca40373d50b6 Mon Sep 17 00:00:00 2001
From: Zack Katz <zack@katz.co>
Date: Wed, 4 Nov 2020 19:46:19 -0500
Subject: [PATCH] Add some extension/layout support

For DataTables and Ratings & Reviews

See #2
---
 class-gravityview-admin-view.php | 58 ++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/class-gravityview-admin-view.php b/class-gravityview-admin-view.php
index 391bcb4..60d33fd 100644
--- a/class-gravityview-admin-view.php
+++ b/class-gravityview-admin-view.php
@@ -48,6 +48,27 @@ public function add_hooks() {
 
 		add_action( 'gravityview_before', array( $this, 'maybe_output_notices' ) );
 
+		// Ratings & Reviews relies on priority 15
+		add_action( 'gravityview_after', array( $this, 'set_post_global' ), -100 );
+		add_action( 'gravityview_after', array( $this, 'unset_post_global' ), 10000 );
+
+		if ( class_exists( 'GV_Extension_DataTables' ) ) {
+			// Support DataTables
+			add_action( 'gravityview/template/after', array( $this, 'set_post_global' ), -100 );
+			add_action( 'admin_enqueue_scripts', array( $this, 'set_post_global' ), -100 );
+			add_action( 'gravityview/template/after', array( $this, 'unset_post_global' ), 10000 );
+			add_action( 'admin_enqueue_scripts', array( $this, 'unset_post_global' ), 1000 );
+
+			$GV_Extension_DataTables = new GV_Extension_DataTables;
+
+			// load DataTables core logic
+			$GV_Extension_DataTables->core_actions();
+
+			$DT = new GV_Extension_DataTables_Data();
+			add_action( 'admin_enqueue_scripts', array( $DT, 'add_scripts_and_styles' ) );
+		}
+
+	}
 
 	/**
 	 * Modify the links shown in the Connected Form links in the Data Source box
@@ -75,6 +96,43 @@ private function is_admin_view_request() {
 
 		return true;
 	}
+
+	/**
+	 * Set the $post global in the admin screen
+	 *
+	 * @global \WP_Post|null $post
+	 *
+	 * @param int|\GV\View $view View being rendered
+	 */
+	public function set_post_global( $view = 0 ) {
+
+		if ( ! $this->is_admin_view_request() ) {
+			return;
+		}
+
+		global $post;
+
+		if ( $post ) {
+			return;
+		}
+
+		$backup_view_id = $view instanceof \GV\View ? $view->ID : $view;
+
+		$post = get_post( \GV\Utils::_GET( 'gvid', $backup_view_id ) );
+	}
+
+	/**
+	 * @param int|\GV\View $view View being rendered
+	 */
+	public function unset_post_global( $view ) {
+
+		if ( ! $this->is_admin_view_request() ) {
+			return;
+		}
+
+		global $post;
+
+		$post = null;
 	}
 
 	/**