← All changes | app/Draft/Controllers/ImageController.php +11 -0 3.2.0 → 3.2.2 View file →

		@@ -44,9 +44,20 @@
44	44	require_once ABSPATH . 'wp-admin/includes/file.php';
45	45	require_once ABSPATH . 'wp-admin/includes/image.php';
46	46	}
47	47
	48	+ // media_sideload_image() takes no file name; it names the file from the URL.
	49	+ $markAiFilename = function ($file) {
	50	+ $info = pathinfo($file['name']);
	51	+ $file['name'] = "{$info['filename']}-ai-generated.{$info['extension']}";
	52	+ return $file;
	53	+ };
	54	+ if ($request->get_param('ai_generated')) {
	55	+ add_filter('wp_handle_sideload_prefilter', $markAiFilename);
	56	+ }
	57	+
48	58	$imageId = \media_sideload_image($request->get_param('source'), 0, null, 'id');
	59	+ remove_filter('wp_handle_sideload_prefilter', $markAiFilename);
49	60
50	61	// Without this the caller can't tell a failed upload from a declined one.
51	62	if (\is_wp_error($imageId)) {
52	63	return new \WP_REST_Response(['message' => $imageId->get_error_message()], 500);