image-optimization / modules / optimization / components / upload-optimization.php

upload-optimization.php in Image Optimizer – Compress Images and Convert to WebP or AVIF 1.7.4, at modules/optimization/components/upload-optimization.php

120 lines 3.2 KB

Raw Download Zip

1	<?php
2
3	namespace ImageOptimization\Modules\Optimization\Components;
4
5	use ImageOptimization\Classes\Async_Operation\{
6	Async_Operation,
7	Async_Operation_Hook,
8	Async_Operation_Queue,
9	};
10
11	use ImageOptimization\Classes\Image\{
12	Exceptions\Invalid_Image_Exception,
13	Image_Meta,
14	Image_Optimization_Error_Type,
15	Image_Status,
16	};
17
18	use ImageOptimization\Modules\Optimization\Classes\{
19	Exceptions\Image_File_Already_Exists_Error,
20	Exceptions\Image_Validation_Error,
21	Optimize_Image,
22	Validate_Image,
23	};
24
25	use ImageOptimization\Classes\Logger;
26	use ImageOptimization\Classes\Exceptions\Quota_Exceeded_Error;
27	use ImageOptimization\Modules\Connect\Classes\Exceptions\Connection_Error;
28	use ImageOptimization\Modules\Settings\Classes\Settings;
29
30	use Throwable;
31	use ImageOptimization\Plugin;
32
33	if ( ! defined( 'ABSPATH' ) ) {
34	exit; // Exit if accessed directly.
35	}
36
37	class Upload_Optimization {
38	public function handle_upload( array $metadata, int $attachment_id, string $context ) {
39	if ( 'create' !== $context ) {
40	return $metadata;
41	}
42
43	if ( ! Settings::get( Settings::OPTIMIZE_ON_UPLOAD_OPTION_NAME ) ) {
44	return $metadata;
45	}
46
47	// @var ImageOptimizer/Modules/ConnectManager/Module
48	$module = Plugin::instance()->modules_manager->get_modules( 'connect-manager' );
49
50	if ( ! $module->connect_instance->is_connected() \|\| ! $module->connect_instance->is_activated() ) {
51	return $metadata;
52	}
53
54	try {
55	Validate_Image::is_valid( $attachment_id );
56	} catch ( Invalid_Image_Exception \|Image_Validation_Error $iie ) {
57	return $metadata;
58	}
59
60	$meta = new Image_Meta( $attachment_id );
61
62	try {
63	$meta
64	->set_status( Image_Status::OPTIMIZATION_IN_PROGRESS )
65	->save();
66
67	Async_Operation::create(
68	Async_Operation_Hook::OPTIMIZE_ON_UPLOAD,
69	[ 'attachment_id' => $attachment_id ],
70	Async_Operation_Queue::OPTIMIZE
71	);
72	} catch ( Throwable $t ) {
73	$meta
74	->set_status( Image_Status::OPTIMIZATION_FAILED )
75	->save();
76	}
77
78	return $metadata;
79	}
80
81	/** @async */
82	public function optimize_image_on_upload( int $image_id ) {
83	try {
84	$oi = new Optimize_Image(
85	$image_id,
86	'upload',
87	);
88
89	$oi->optimize();
90	} catch ( Quota_Exceeded_Error $qe ) {
91	( new Image_Meta( $image_id ) )
92	->set_status( Image_Status::OPTIMIZATION_FAILED )
93	->set_error_type( Image_Optimization_Error_Type::QUOTA_EXCEEDED )
94	->save();
95	} catch ( Connection_Error $ce ) {
96	( new Image_Meta( $image_id ) )
97	->set_status( Image_Status::OPTIMIZATION_FAILED )
98	->set_error_type( Image_Optimization_Error_Type::CONNECTION_ERROR )
99	->save();
100	} catch ( Image_File_Already_Exists_Error $fe ) {
101	( new Image_Meta( $image_id ) )
102	->set_status( Image_Status::OPTIMIZATION_FAILED )
103	->set_error_type( Image_Optimization_Error_Type::FILE_ALREADY_EXISTS )
104	->save();
105	} catch ( Throwable $t ) {
106	Logger::error( 'Upload optimization error. Reason: ' . $t->getMessage() );
107
108	( new Image_Meta( $image_id ) )
109	->set_status( Image_Status::OPTIMIZATION_FAILED )
110	->set_error_type( Image_Optimization_Error_Type::GENERIC )
111	->save();
112	}
113	}
114
115	public function __construct() {
116	add_action( 'wp_generate_attachment_metadata', [ $this, 'handle_upload' ], 10, 3 );
117	add_action( Async_Operation_Hook::OPTIMIZE_ON_UPLOAD, [ $this, 'optimize_image_on_upload' ] );
118	}
119	}
120