Issue 313: Create new connector for Theme Editor

connectors/editor.php

@@ -0,0 +1,182 @@
+<?php
+
+class WP_Stream_Connector_Editor extends WP_Stream_Connector {
+
+	/**
+	 * Context name
+	 *
+	 * @var string
+	 */
+	public static $name = 'editor';
+
+	/**
+	 * Actions registered for this context
+	 *
+	 * @var array
+	 */
+	public static $actions = array();
+
+	/**
+	 * Actions registered for this context
+	 *
+	 * @var array
+	 */
+	private static $edited_file = array();
+
+	/**
+	 * Register all context hooks
+	 *
+	 * @return void
+	 */
+	public static function register() {
+		parent::register();
+		add_action( 'load-theme-editor.php', array( __CLASS__, 'get_edition_data' ) );
+		add_filter( 'wp_redirect', array( __CLASS__, 'log_changes' ) );
+	}
+
+	/**
+	 * Return translated context label
+	 *
+	 * @return string Translated context label
+	 */
+	public static function get_label() {
+		return __( 'Theme Editor', 'stream' );
+	}
+
+	/**
+	 * Return translated action labels
+	 *
+	 * @return array Action label translations
+	 */
+	public static function get_action_labels() {
+		return array(
+			'updated' => __( 'Updated', 'stream' ),
+		);
+	}
+
+	/**
+	 * Return translated context labels
+	 *
+	 * @return array Context label translations
+	 */
+	public static function get_context_labels() {
+		$themes = wp_get_themes();
+
+		$themes_slugs = array_map(
+			function( $theme ) {
+				return $theme->get_template();
+			},
+			$themes
+		);
+
+		$themes_names = array_map(
+			function( $theme ) {
+				return (string) $theme;
+			},
+			$themes
+		);
+
+		return array_combine( $themes_slugs, $themes_names );
+	}
+
+	/**
+	 * Add action links to Stream drop row in admin list screen
+	 *
+	 * @filter wp_stream_action_links_{connector}
+	 * @param  array $links      Previous links registered
+	 * @param  int   $record     Stream record
+	 * @return array             Action links
+	 */
+	public static function action_links( $links, $record ) {
+		if ( current_user_can( 'edit_theme_options' ) ) {
+			$file_name  = get_stream_meta( $record->ID, 'file', true );
+			$theme_name = get_stream_meta( $record->ID, 'theme', true );
+
+			$links[ __( 'Edit File', 'stream' ) ] = admin_url(
+				sprintf(
+					'theme-editor.php?theme=%s&file=%s',
+					$theme_name,
+					$file_name
+				)
+			);
+
+			$links[ __( 'Edit Theme', 'stream' ) ] = admin_url(
+				sprintf(
+					'themes.php?theme=%s',
+					$theme_name
+				)
+			);
+		}
+
+		return $links;
+	}
+
+	/**
+	 * @action load-theme-editor.php
+	 */
+	public static function get_edition_data() {
+		if ( 'POST' !== $_SERVER['REQUEST_METHOD'] ) {
+			return;
+		}
+
+		if ( ! isset( $_POST['action'] ) || 'update' !== $_POST['action'] ) {
+			return;
+		}
+
+		$theme_name = ( isset( $_POST['theme'] ) && $_POST['theme'] ? $_POST['theme'] : get_stylesheet() );
+		$theme = wp_get_theme( $theme_name );
+
+		if ( ! $theme->exists() || ( $theme->errors() && 'theme_no_stylesheet' === $theme->errors()->get_error_code() ) ) {
+			return;
+		}
+
+		$allowed_files = $theme->get_files( 'php', 1 );
+		$style_files = $theme->get_files( 'css' );
+		$allowed_files['style.css'] = $style_files['style.css'];
+
+		if ( empty( $_POST['file'] ) ) {
+			$file_name = 'style.css';
+			$file_path = $allowed_files['style.css'];
+		} else {
+			$file_name = $_POST['file'];
+			$file_path = $theme->get_stylesheet_directory() . '/' . $file_name;
+		}
+
+		$file_contents_before = file_get_contents( $file_path );
+
+		self::$edited_file = compact(
+			'file_name',
+			'file_path',
+			'file_contents_before',
+			'theme'
+		);
+	}
+
+	/**
+	 * @filter wp_redirect
+	 */
+	public static function log_changes( $location ) {
+		if ( ! empty( self::$edited_file ) ) {
+			$file_contents_after = file_get_contents( self::$edited_file['file_path'] );
+
+			if ( $file_contents_after !== self::$edited_file['file_contents_before'] ) {
+				$theme_slug = self::$edited_file['theme']->get_template();
+				$properties = array(
+					'file'       => self::$edited_file['file_name'],
+					'theme_name' => (string) self::$edited_file['theme'],
+					'theme'      => $theme_slug,
+				);
+
+				self::log(
+					__( '"%1$s" was updated in "%2$s"', 'stream' ),
+					$properties,
+					null,
+					array( $theme_slug => 'updated' )
+				);
+			}
+		}
+
+		return $location;
+	}
+
+}

connectors/installer.php

@@ -312,7 +312,7 @@ public static function callback_pre_set_site_transient_update_plugins( $value )
 	}
 
 	public static function callback_wp_redirect( $location ) {
-		if ( ! preg_match( '#(plugin|theme)-editor.php#', $location, $match ) ) {
+		if ( ! preg_match( '#(plugin)-editor.php#', $location, $match ) ) {
 			return $location;
 		}