22

✏️ Editing: Export_Code.php

<?php

namespace Code_Snippets\Migration\Export;

use Code_Snippets\Model\Snippet;

/**
 * Handles exporting the code from snippets.
 *
 * @package Code_Snippets
 */
class Export_Code extends Export {

/**
	 * The type of snippet being exported.
	 *
	 * @var string
	 */
	protected string $export_type;

/**
	 * Constructor.
	 *
	 * @param array<int>  $ids         The IDs of the snippets to export.
	 * @param bool|null   $network     Whether to export network-wide snippets.
	 * @param string|null $export_type The type of snippet to export (e.g., 'php', 'html', 'css', 'js', 'cond').
	 */
	public function __construct( array $ids, ?bool $network = null, ?string $export_type = null ) {
		parent::__construct( $ids, $network );

if ( $export_type && in_array( $export_type, Snippet::get_types(), true ) ) {
			$this->export_type = $export_type;
		} else {
			// If no export type is specified, default to the type of the first snippet.
			$snippets_list = $this->get_snippets_list();
			$this->export_type = $snippets_list[0]->type;
		}
	}

/**
	 * Get the type of snippets being exported.
	 *
	 * @return string
	 */
	public function get_export_type(): string {
		return $this->export_type;
	}

/**
	 * Get the file extension for the export format.
	 *
	 * @return string
	 */
	public function get_file_extension(): string {
		return 'cond' === $this->export_type
			? 'json'
			: $this->export_type;
	}

/**
	 * Bundle a snippets into a PHP file.
	 */
	protected function export_snippets_php(): string {
		$result = "<?php\n";

foreach ( $this->get_snippets_list() as $snippet ) {
			$code = trim( $snippet->code );

if ( ( 'php' !== $snippet->type && 'html' !== $snippet->type ) || ! $code ) {
				continue;
			}

$result .= "\n/**\n * $snippet->display_name\n";

if ( ! empty( $snippet->desc ) ) {
				// Convert description to PhpDoc.
				$desc = wp_strip_all_tags( str_replace( "\n", "\n * ", $snippet->desc ) );
				$result .= " *\n * $desc\n";
			}

$result .= " */\n";

if ( 'content' === $snippet->scope ) {
				$shortcode_tag = apply_filters( 'code_snippets_export_shortcode_tag', "code_snippets_export_$snippet->id", $snippet );

$code = sprintf(
					"add_shortcode( '%s', function () {\n\tob_start();\n\t?>\n\n\t%s\n\n\t<?php\n\treturn ob_get_clean();\n} );",
					$shortcode_tag,
					str_replace( "\n", "\n\t", $code )
				);
			}

$result .= "$code\n";
		}

return $result;
	}

/**
	 * Export conditions in JSON format.
	 *
	 * @return string
	 */
	protected function export_conditions_json(): string {
		$conditions_data = [];
		$fields_to_copy = [ 'name', 'desc', 'tags' ];

foreach ( $this->get_snippets_list() as $snippet ) {
			$condition_data = [];

if ( ! $snippet->code || 'cond' !== $snippet->type ) {
				continue;
			}

$rules = json_decode( $snippet->code, false );

if ( json_last_error() !== JSON_ERROR_NONE ) {
				continue;
			}

foreach ( $fields_to_copy as $field ) {
				if ( ! empty( $snippet->$field ) ) {
					$condition_data[ $field ] = $snippet->$field;
				}
			}

$condition_data['rules'] = $rules;
			$conditions_data[] = $condition_data;
		}

return wp_json_encode( 1 === count( $conditions_data ) ? $conditions_data[0] : $conditions_data, JSON_PRETTY_PRINT );
	}

/**
	 * Export snippets in a generic format, compatible with CSS or JavaScript.
	 */
	protected function export_snippets_css_js(): string {
		$result = '';

foreach ( $this->get_snippets_list() as $snippet ) {
			$snippet = new Snippet( $snippet );

if ( $this->get_export_type() !== $snippet->type ) {
				continue;
			}

$result .= "\n/*\n";

if ( $snippet->name ) {
				$result .= wp_strip_all_tags( $snippet->name ) . "\n\n";
			}

if ( ! empty( $snippet->desc ) ) {
				$result .= wp_strip_all_tags( $snippet->desc ) . "\n";
			}

$result .= "*/\n\n$snippet->code\n\n";
		}

return $result;
	}

/**
	 * Generate a downloadable code file from the exported snippets.
	 */
	public function generate_export(): string {
		switch ( $this->get_export_type() ) {
			case 'php':
			case 'html':
				return $this->export_snippets_php();

case 'cond':
				return $this->export_conditions_json();

default:
				return $this->export_snippets_css_js();
		}
	}
}

MR SKB