create-block-theme/admin/create-theme/theme-utils.php

<?php

class Theme_Utils {
	public static function is_absolute_url( $url ) {
		return ! empty( $url ) && isset( parse_url( $url )['host'] );
	}

	public static function get_theme_slug( $new_theme_name ) {
		return sanitize_title( $new_theme_name );
	}

	public static function get_file_extension_from_url( $url ) {
		$extension = pathinfo( $url, PATHINFO_EXTENSION );
		return $extension;
	}

	public static function replace_namespace( $content, $old_slug, $new_slug, $old_name, $new_name ) {
		$new_slug_underscore = str_replace( '-', '_', $new_slug ) . '_';
		$old_slug_underscore = str_replace( '-', '_', $old_slug ) . '_';

		// Generate placeholders
		$placeholder_slug            = md5( $old_slug );
		$placeholder_slug_underscore = md5( $old_slug_underscore );
		$placeholder_name            = md5( $old_name );

		// Replace old values with placeholders
		$content = str_replace( $old_slug_underscore, $placeholder_slug_underscore, $content );
		$content = str_replace( $old_slug, $placeholder_slug, $content );
		$content = str_replace( $old_name, $placeholder_name, $content );

		// Replace placeholders with new values
		$content = str_replace( $placeholder_slug_underscore, $new_slug_underscore, $content );
		$content = str_replace( $placeholder_slug, $new_slug, $content );
		$content = str_replace( $placeholder_name, $new_name, $content );

		return $content;
	}

	public static function clone_theme_to_folder( $location, $new_slug, $new_name ) {

		$theme    = wp_get_theme();
		$old_slug = $theme->get( 'TextDomain' );
		$old_name = $theme->get( 'Name' );

		// Get real path for our folder
		$theme_path = get_stylesheet_directory();

		// Create recursive directory iterator
		/** @var SplFileInfo[] $files */
		$files = new \RecursiveIteratorIterator(
			new \RecursiveDirectoryIterator( $theme_path, \RecursiveDirectoryIterator::SKIP_DOTS ),
			\RecursiveIteratorIterator::SELF_FIRST
		);

		// Add all the files (except for templates)
		foreach ( $files as $name => $file ) {

			// Get real and relative path for current file
			$file_path     = wp_normalize_path( $file );
			$relative_path = substr( $file_path, strlen( $theme_path ) + 1 );

			// Create Directories
			if ( $file->isDir() ) {
				wp_mkdir_p( $location . DIRECTORY_SEPARATOR . $files->getSubPathname() );
			}

			// If the path is for templates/parts ignore it
			if (
				strpos( $file_path, 'block-template-parts/' ) ||
				strpos( $file_path, 'block-templates/' ) ||
				strpos( $file_path, 'templates/' ) ||
				strpos( $file_path, 'parts/' )
			) {
				continue;
			}

			// Replace only text files, skip png's and other stuff.
			$contents               = file_get_contents( $file_path );
			$valid_extensions       = array( 'php', 'css', 'scss', 'js', 'txt', 'html' );
			$valid_extensions_regex = implode( '|', $valid_extensions );

			if ( preg_match( "/\.({$valid_extensions_regex})$/", $relative_path ) ) {
				// Replace namespace values if provided
				if ( $new_slug ) {
					$contents = self::replace_namespace( $contents, $old_slug, $new_slug, $old_name, $new_name );
				}
			}

			// Add current file to target
			file_put_contents( $location . DIRECTORY_SEPARATOR . $relative_path, $contents );
		}
	}

	/**
	 * Relocate the theme to a new folder and activate the newly relocated theme.
	 */
	public static function relocate_theme( $new_theme_subfolder ) {

		$current_theme_subfolder = '';
		$theme_dir               = get_stylesheet();

		$source      = get_theme_root() . DIRECTORY_SEPARATOR . $theme_dir;
		$destination = get_theme_root() . DIRECTORY_SEPARATOR . $theme_dir;

		if ( str_contains( get_stylesheet(), '/' ) ) {
			$current_theme_subfolder = substr( get_stylesheet(), 0, strrpos( get_stylesheet(), '/' ) );
			$theme_dir               = substr( get_stylesheet(), strrpos( get_stylesheet(), '/' ) + 1 );
			$source                  = get_theme_root() . DIRECTORY_SEPARATOR . $current_theme_subfolder . DIRECTORY_SEPARATOR . $theme_dir;
			$destination             = get_theme_root() . DIRECTORY_SEPARATOR . $theme_dir;
		}

		if ( $new_theme_subfolder ) {
			$destination = get_theme_root() . DIRECTORY_SEPARATOR . $new_theme_subfolder . DIRECTORY_SEPARATOR . $theme_dir;
			wp_mkdir_p( get_theme_root() . DIRECTORY_SEPARATOR . $new_theme_subfolder );
		}

		if ( $source === $destination ) {
			return;
		}

		global $wp_filesystem;
		if ( ! $wp_filesystem ) {
			require_once ABSPATH . 'wp-admin/includes/file.php';
			WP_Filesystem();
		}

		$success = move_dir( $source, $destination, false );

		if ( ! $success ) {
			return new \WP_Error( 'problem_moving', __( 'There was a problem moving the theme', 'create-block-theme' ) );
		}

		if ( $new_theme_subfolder ) {
			switch_theme( $new_theme_subfolder . '/' . $theme_dir );
		} else {
			switch_theme( $theme_dir );
		}
	}

	public static function is_valid_screenshot( $file ) {

		$allowed_screenshot_types = array(
			'png' => 'image/png',
		);
		$filetype                 = wp_check_filetype( $file['name'], $allowed_screenshot_types );
		if ( is_uploaded_file( $file['tmp_name'] ) && in_array( $filetype['type'], $allowed_screenshot_types, true ) && $file['size'] < 2097152 ) {
			return 1;
		}
		return 0;
	}

	public static function is_valid_screenshot_file( $file_path ) {
		return Theme_Utils::get_screenshot_file_extension( $file_path ) !== null;
	}

	public static function get_screenshot_file_extension( $file_path ) {
		$allowed_screenshot_types = array(
			'png'  => 'image/png',
			'gif'  => 'image/gif',
			'jpg'  => 'image/jpeg',
			'jpeg' => 'image/jpeg',
			'webp' => 'image/webp',
			'avif' => 'image/avif',
		);
		$filetype                 = wp_check_filetype( $file_path, $allowed_screenshot_types );
		if ( in_array( $filetype['type'], $allowed_screenshot_types, true ) ) {
			return $filetype['ext'];
		}
		return null;
	}

	public static function copy_screenshot( $file_path ) {

		$new_screeenshot_id = attachment_url_to_postid( $file_path );

		if ( ! $new_screeenshot_id ) {
			return new \WP_Error( 'screenshot_not_found', __( 'Screenshot not found', 'create-block-theme' ) );
		}

		$new_screenshot_metadata = wp_get_attachment_metadata( $new_screeenshot_id );
		$upload_dir              = wp_get_upload_dir();

		$new_screenshot_location = path_join( $upload_dir['basedir'], $new_screenshot_metadata['file'] );

		$new_screenshot_filetype = Theme_Utils::get_screenshot_file_extension( $file_path );
		$new_location            = path_join( get_stylesheet_directory(), 'screenshot.' . $new_screenshot_filetype );

		// copy and resize the image
		$image_editor = wp_get_image_editor( $new_screenshot_location );
		$image_editor->resize( 1200, 900, true );
		$image_editor->save( $new_location );

		return true;
	}

	public static function replace_screenshot( $new_screenshot_path ) {
		if ( ! Theme_Utils::is_valid_screenshot_file( $new_screenshot_path ) ) {
			return new \WP_Error( 'invalid_screenshot', __( 'Invalid screenshot file', 'create-block-theme' ) );
		}

		// Remove the old screenshot
		$old_screenshot = wp_get_theme()->get_screenshot( 'relative' );
		if ( $old_screenshot ) {
			unlink( path_join( get_stylesheet_directory(), $old_screenshot ) );
		}

		// Copy the new screenshot
		return Theme_Utils::copy_screenshot( $new_screenshot_path );
	}

}