<?php
/* phpcs:disable WordPress.Files.FileName */
/**
 * Restart Workflow on GV Approval and Subscription Payment
 *
 * GOAL:
 * - Adds Start Step settings to (a) delay initial start until GravityView approval and (b) restart a
 *   completed workflow on each subscription payment. Optionally update a date field on restart and
 *   include manual restart buttons in the entry details sidebar.
 *
 */

	// to add checkboxes
add_filter(
	'gravityflow_step_settings_fields',
	static function ( $settings, $current_step_id ) {
		if ( ! function_exists( 'gravity_flow' ) || ! class_exists( 'GFCommon' ) ) {
			return $settings;
		}
		$step = gravity_flow()->get_step( $current_step_id );
		if ( $step && method_exists( $step, 'get_type' ) ) {
			$step_type = $step->get_type();
			if ( 'workflow_start' === $step_type ) {
				$settings[0]['fields'][] = [
					'label'   => 'Delay Workflow',
					'type'    => 'checkbox',
					'name'    => 'delay_workflow_checkbox',
					'choices' => [
						[
							'label' => 'Delay the workflow until Gravity View approval',
							'name'  => 'delay_workflow_checkbox',
						],
					],
				];
				$settings[0]['fields'][] = [
					'label'   => 'Restart Workflow',
					'type'    => 'checkbox',
					'name'    => 'restart_workflow_checkbox',
					'choices' => [
						[
							'label' => 'Restart the workflow on subscription payment',
							'name'  => 'restart_workflow_checkbox',
						],
					],
				];
				$form                    = gravity_flow()->get_current_form();
				$api                     = class_exists( 'Gravity_Flow_API' ) ? new Gravity_Flow_API( $form['id'] ) : null;
				$all_steps               = $api ? $api->get_steps() : [];
				if ( is_array( $all_steps ) && ! empty( $all_steps ) ) {
					array_shift( $all_steps );
				}
				if ( ! empty( $all_steps ) ) {
					$step_choices = [];
					foreach ( $all_steps as $workflow_step ) {
						$step_choices[] = [
							'label' => $workflow_step->get_name(),
							'value' => $workflow_step->get_id(),
						];
					}
					$settings[0]['fields'][] = [
						'name'    => 'restart_target_step',
						'label'   => 'Send to Step',
						'type'    => 'select',
						'choices' => $step_choices,
						'tooltip' => 'Select which step to send the entry to when workflow restarts. Defaults to first step.',
					];
				}
				$form        = gravity_flow()->get_current_form();
				$date_fields = GFCommon::get_fields_by_type( $form, [ 'date' ] );
				if ( ! empty( $date_fields ) ) {
					$date_field_choices = [];
					foreach ( $date_fields as $date_field ) {
						$date_field_choices[] = [
							'label' => $date_field['label'],
							'value' => $date_field['id'],
						];
					}
					array_unshift(
                        $date_field_choices,
                        [
							'label' => 'Select a date field',
							'value' => '',
						]
                        );
					$settings[0]['fields'][] = [
						'name'    => 'update_date_field',
						'label'   => 'Update Date Field',
						'type'    => 'select',
						'choices' => $date_field_choices,
						'fields'  => [
							[
								'name'    => 'also_for_gv',
								'label'   => 'Also update the date field on Gravity View Approval',
								'type'    => 'checkbox',
								'choices' => [
									[
										'label' => '',
										'name'  => 'also_for_gv',
									],
								],
							],
						],
					];
				}
			}
		}
		return $settings;
	},
	10,
	2
);


add_action(
	'gravityflow_step_start',
	static function ( $step_id, $entry_id, $form_id, $step_status, $step ) {
		$delay_view_approval = $step->__get( 'delay_workflow_checkbox' );
		if ( $delay_view_approval ) {
			$entry = GFAPI::get_entry( (int) $entry_id );
			if ( is_wp_error( $entry ) ) {
				return;
			}
			if ( ! class_exists( 'Gravity_Flow_API' ) || ! function_exists( 'gravity_flow' ) ) {
				return;
			}
			$api    = new Gravity_Flow_API( (int) $form_id );
			$status = $api->cancel_workflow( $entry );
			if ( $status ) {
				gravity_flow()->add_timeline_note( (int) $entry_id, 'Initial workflow cancelled. Set to start on View approval.' );
			}
		}
	},
	10,
	5
);

// to start workflow on approval
add_action(
	'gravityview/approve_entries/approved',
	static function ( $entry_id ) {
		if ( ! function_exists( 'gravity_flow' ) || ! class_exists( 'GFAPI' ) || ! class_exists( 'Gravity_Flow_API' ) ) {
			return;
		}
		$entry = GFAPI::get_entry( (int) $entry_id );
		if ( is_wp_error( $entry ) || empty( $entry ) ) {
			return;
		}
		$form_id    = isset( $entry['form_id'] ) ? (int) $entry['form_id'] : 0;
		$gwf        = gravity_flow();
		$start_step = $gwf->get_workflow_start_step( $form_id, $entry );
		if ( ! $start_step ) {
			return;
		}
		$is_checked = $start_step->__get( 'delay_workflow_checkbox' );
		if ( ! $is_checked ) {
			return;
		}
		$field_id      = $start_step->__get( 'update_date_field' );
		$original_date = null;
		if ( $start_step->__get( 'also_for_gv' ) && $field_id && isset( $entry[ $field_id ] ) ) {
			$original_date = $entry[ $field_id ];
			$today         = function_exists( 'wp_date' ) ? wp_date( 'Y-m-d', current_time( 'timestamp' ) ) : date( 'Y-m-d' ); // phpcs:ignore WordPress.DateTime.RestrictedFunctions.date_date, WordPress.DateTime.CurrentTimeTimestamp.Requested
			GFAPI::update_entry_field( $entry['id'], $field_id, $today );
		}
		$api            = new Gravity_Flow_API( $form_id );
		$target_step_id = $start_step->__get( 'restart_target_step' );
		$target_step    = null;
		if ( $target_step_id ) {
			$target_step = $gwf->get_step( $target_step_id );
		}
		if ( ! $target_step ) {
			$steps = $api->get_steps();
			if ( is_array( $steps ) && ! empty( $steps ) ) {
				array_shift( $steps );
			}
			$target_step = $steps[0] ?? null;
		}
		if ( $target_step ) {
			$gwf->add_timeline_note( $entry['id'], 'Workflow started because of Gravity View Approval.' );
			if ( null !== $original_date ) {
				add_action(
					'gravityflow_post_process_workflow',
					static function () use ( $entry, $field_id, $original_date ) {
						GFAPI::update_entry_field( $entry['id'], $field_id, $original_date );
					}
				);
			}
			$api->send_to_step( $entry, $target_step->get_id() );
            GFAPI::send_notifications( GFAPI::get_form( $form_id ), $entry, 'bld_restart_workflow' );
		} else {
            GFAPI::add_note( $entry_id, 0, 'bld-restart-workflow', 'Step not found, unable to restart workflow.' );
        }
	},
	10,
	1
);

// to restart workflow on subscription payment
add_action(
	'gform_post_add_subscription_payment',
	static function ( $entry, $action ) {
		if ( ! is_array( $action ) || ! isset( $action['type'] ) || 'add_subscription_payment' !== $action['type'] ) {
			return;
		}
		if ( ! function_exists( 'gravity_flow' ) || ! class_exists( 'GFAPI' ) || ! class_exists( 'Gravity_Flow_API' ) ) {
			return;
		}
		$entry   = is_array( $entry ) ? $entry : [];
		$form_id = isset( $entry['form_id'] ) ? (int) $entry['form_id'] : 0;
		$gwf     = gravity_flow();
		$start   = $gwf->get_workflow_start_step( $form_id, $entry );
		if ( ! $start ) {
			return;
		}
		$is_checked = $start->__get( 'restart_workflow_checkbox' );
		if ( ! $is_checked ) {
			return;
		}
		$field_id      = $start->__get( 'update_date_field' );
		$original_date = null;
		if ( $field_id && isset( $entry[ $field_id ] ) ) {
			$original_date = $entry[ $field_id ];
			$today         = function_exists( 'wp_date' ) ? wp_date( 'Y-m-d', current_time( 'timestamp' ) ) : date( 'Y-m-d' ); // phpcs:ignore WordPress.DateTime.RestrictedFunctions.date_date, WordPress.DateTime.CurrentTimeTimestamp.Requested
			GFAPI::update_entry_field( $entry['id'], $field_id, $today );
		}
		$api            = new Gravity_Flow_API( $form_id );
		$target_step_id = $start->__get( 'restart_target_step' );
		$target_step    = null;
		if ( $target_step_id ) {
			$target_step = $gwf->get_step( $target_step_id );
		}
		if ( ! $target_step ) {
			$steps = $api->get_steps();
			if ( is_array( $steps ) && ! empty( $steps ) ) {
				array_shift( $steps );
			}
			$target_step = $steps[0] ?? null;
		}
		if ( $target_step ) {
			$gwf->add_timeline_note( $entry['id'], 'Workflow started because of subscription payment.' );
			if ( null !== $original_date ) {
				add_action(
					'gravityflow_post_process_workflow',
					static function () use ( $entry, $field_id, $original_date ) {
						GFAPI::update_entry_field( $entry['id'], $field_id, $original_date );
					}
				);
			}
			$api->send_to_step( $entry, $target_step->get_id() );
            GFAPI::send_notifications( GFAPI::get_form( $form_id ), $entry, 'bld_restart_workflow' );
		} else {
            GFAPI::add_note( $entry['id'], 0, 'bld-restart-workflow', 'Step not found, unable to restart workflow.' );
        }
	},
	10,
	2
);


add_action(
	'gform_entry_detail_sidebar_middle',
	static function ( $form, $entry ) {
		if ( ! function_exists( 'gravity_flow' ) || ! class_exists( 'GFAPI' ) ) {
			return;
		}
		// Capability guard: only show to users who can view workflow details.
		if ( class_exists( 'GFCommon' ) && ! GFCommon::current_user_can_any( 'gravityflow_workflow_detail' ) ) {
			return;
		}
		$start_step = gravity_flow()->get_workflow_start_step( isset( $form['id'] ) ? (int) $form['id'] : 0, $entry );
		if ( ! $start_step ) {
			return;
		}
		$delay_view_approval       = $start_step->__get( 'delay_workflow_checkbox' );
		$restart_workflow_checkbox = $start_step->__get( 'restart_workflow_checkbox' );
		if ( ! $delay_view_approval && ! $restart_workflow_checkbox ) {
			return;
		}

		// Handle POSTed actions.
        if ( isset( $_POST['restart_workflow_nonce'] ) && isset( $_POST['restart-workflow-button'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Missing
			$nonce_ok = wp_verify_nonce( sanitize_text_field( wp_unslash( $_POST['restart_workflow_nonce'] ) ), 'restart_workflow_nonce' ); // phpcs:ignore WordPress.Security.NonceVerification.Missing
			$btn      = sanitize_text_field( wp_unslash( $_POST['restart-workflow-button'] ) ); // phpcs:ignore WordPress.Security.NonceVerification.Missing
			if ( $nonce_ok && ( 'restart_workflow_checkbox' === $btn || 'delay_workflow_checkbox' === $btn ) ) {
				// Emulate send-to-first-step like other triggers above.
				$form_id        = isset( $entry['form_id'] ) ? (int) $entry['form_id'] : 0;
				$gwf            = gravity_flow();
				$api            = class_exists( 'Gravity_Flow_API' ) ? new Gravity_Flow_API( $form_id ) : null;
				$target_step_id = $start_step->__get( 'restart_target_step' );
				$target_step    = null;
				if ( $target_step_id ) {
					$target_step = $gwf->get_step( $target_step_id );
				}
				if ( ! $target_step ) {
					$steps = $api ? $api->get_steps() : [];
					if ( is_array( $steps ) && ! empty( $steps ) ) {
						array_shift( $steps );
					}
					$target_step = $steps[0] ?? null;
				}
				if ( $target_step ) {
					$reason = ( 'restart_workflow_checkbox' === $btn ) ? 'Subscription Payment (manual)' : 'Gravity View Approval (manual)';
					$gwf->add_timeline_note( $entry['id'], 'Workflow started manually: ' . $reason );
					$api->send_to_step( $entry, $target_step->get_id() );
                    GFAPI::send_notifications( GFAPI::get_form( $form_id ), $entry, 'bld_restart_workflow' );
                } else {
                    GFAPI::add_note( $entry['id'], 0, 'bld-restart-workflow', 'Step not found, unable to restart workflow.' );
                }
			}
		}

		$nonce = wp_create_nonce( 'restart_workflow_nonce' );
		ob_start();
		?>
		<div class="postbox">
			<div class="postbox-header">
				<h2 class="hndle ui-sortable-handle">Restart Workflow</h2>
			</div>
			<div class="inside">
				<input type="hidden" name="restart_workflow_nonce" value="<?php echo esc_attr( $nonce ); ?>" />
				<?php if ( $delay_view_approval ) : ?>
					<button class="button button-large" name="restart-workflow-button" value="delay_workflow_checkbox">GV Approval</button><br /><br />
				<?php endif; ?>
				<?php if ( $restart_workflow_checkbox ) : ?>
					<button class="button button-large" name="restart-workflow-button" value="restart_workflow_checkbox">Subscription Payment</button>
				<?php endif; ?>
			</div>
		</div>
		<?php
		echo ob_get_clean(); // phpcs:ignore WordPress.Security.EscapeOutput.OutputNotEscaped
	},
	10,
	2
);

add_filter(
    'gform_notification_events',
    static function ( $notification_events ) {
		$notification_events['bld_restart_workflow'] = 'BLD Snippet - Restart Workflow on GV Approval or Subscription Payment';
		return $notification_events;
	}
);