Commit 62ea465a authored by Eric Mann's avatar Eric Mann
Browse files

Add CLI hooks for enabling/disabling the URL rewriting engine

parent 68beb7c4

inc/class-s3-uploads-wp-cli-command.php

+20 −0
Original line number Diff line number Diff line
@@ -361,6 +361,26 @@ class S3_Uploads_WP_CLI_Command extends WP_CLI_Command { 
		WP_CLI::success( sprintf( 'Successfully deleted %s', $prefix ) );

	}



	/**

	 * Ensable the auto-rewriting of media links to S3

	 */

	public function enable( $args, $assoc_args ) {

		if ( false === add_option( 's3_uploads_enabled', 'enabled', '', 'yes' ) ) {

			update_option( 's3_uploads_enabled', 'enabled' );

		}



		WP_CLI::success( 'Media URL rewriting enabled.' );

	}



	/**

	 * Disable the auto-rewriting of media links to S3

	 */

	public function disable( $args, $assoc_args ) {

		delete_option( 's3_uploads_enabled' );



		WP_CLI::success( 'Media URL rewriting disabled.' );

	}



	private function recurse_copy($src,$dst) {

		$dir = opendir($src);

		@mkdir($dst);

s3-uploads.php

+4 −2
Original line number Diff line number Diff line
@@ -21,8 +21,10 @@ function s3_uploads_init() { 
		return;

	}



	// Make sure the plugin is enabled

	if ( ( defined( 'S3_UPLOADS_AUTOENABLE' ) && false === S3_UPLOADS_AUTOENABLE ) && 'enabled' !== get_option( 's3_uploads_enabled' ) ) {

	// Make sure the plugin is enabled when autoenable is on, or in a CLI system

	if ( ( ! defined( 'WP_CLI') || ! WP_CLI ) &&

	     ( defined( 'S3_UPLOADS_AUTOENABLE' ) && false === S3_UPLOADS_AUTOENABLE ) &&

	     'enabled' !== get_option( 's3_uploads_enabled' ) ) {

		return;

	}