v1.0

wp-security-pack/includes/class-wpsp-updater.php

@@ -0,0 +1,178 @@
+<?php
+/**
+ * Plugin updater for Codeberg releases.
+ *
+ * @package WP_Security_Pack
+ */
+
+// Prevent direct access.
+if ( ! defined( 'ABSPATH' ) ) {
+	exit;
+}
+
+/**
+ * Handles plugin updates from Codeberg.
+ */
+class WPSP_Updater {
+
+	/**
+	 * Codeberg raw URL for update info.
+	 */
+	const UPDATE_URL = 'https://codeberg.org/ArkHost/WP-Security-Pack/raw/branch/main/update-info.json';
+
+	/**
+	 * Plugin slug.
+	 */
+	const SLUG = 'wp-security-pack';
+
+	/**
+	 * Cache key.
+	 */
+	const CACHE_KEY = 'wpsp_update_info';
+
+	/**
+	 * Initialize updater.
+	 */
+	public function __construct() {
+		add_filter( 'pre_set_site_transient_update_plugins', array( $this, 'check_update' ) );
+		add_filter( 'plugins_api', array( $this, 'plugin_info' ), 10, 3 );
+		add_filter( 'upgrader_source_selection', array( $this, 'fix_folder_name' ), 10, 4 );
+	}
+
+	/**
+	 * Check for updates.
+	 *
+	 * @param object $transient Transient data.
+	 * @return object
+	 */
+	public function check_update( $transient ) {
+		if ( empty( $transient->checked ) ) {
+			return $transient;
+		}
+
+		$remote = $this->get_remote_info();
+		if ( ! $remote ) {
+			return $transient;
+		}
+
+		$current_version = WPSP_VERSION;
+		if ( version_compare( $current_version, $remote->version, '<' ) ) {
+			$transient->response[ WPSP_PLUGIN_BASENAME ] = (object) array(
+				'slug'        => self::SLUG,
+				'plugin'      => WPSP_PLUGIN_BASENAME,
+				'new_version' => $remote->version,
+				'url'         => $remote->homepage ?? 'https://codeberg.org/ArkHost/WP-Security-Pack',
+				'package'     => $remote->download_url,
+				'icons'       => array(),
+				'banners'     => array(),
+			);
+		}
+
+		return $transient;
+	}
+
+	/**
+	 * Plugin info for the update screen.
+	 *
+	 * @param false|object|array $result Result.
+	 * @param string             $action API action.
+	 * @param object             $args   Arguments.
+	 * @return false|object
+	 */
+	public function plugin_info( $result, $action, $args ) {
+		if ( 'plugin_information' !== $action || self::SLUG !== ( $args->slug ?? '' ) ) {
+			return $result;
+		}
+
+		$remote = $this->get_remote_info();
+		if ( ! $remote ) {
+			return $result;
+		}
+
+		return (object) array(
+			'name'          => $remote->name ?? 'WP Security Pack',
+			'slug'          => self::SLUG,
+			'version'       => $remote->version,
+			'author'        => '<a href="https://arkhost.com">ArkHost</a>',
+			'homepage'      => $remote->homepage ?? 'https://codeberg.org/ArkHost/WP-Security-Pack',
+			'download_link' => $remote->download_url,
+			'sections'      => array(
+				'description' => $remote->description ?? 'WordPress security without the nonsense.',
+				'changelog'   => $remote->changelog ?? '',
+			),
+			'requires'      => $remote->requires ?? '5.0',
+			'tested'        => $remote->tested ?? '6.9',
+			'requires_php'  => $remote->requires_php ?? '7.4',
+		);
+	}
+
+	/**
+	 * Fix folder name after download.
+	 * Codeberg archives use "WP-Security-Pack" but we need "wp-security-pack".
+	 *
+	 * @param string      $source        Source path.
+	 * @param string      $remote_source Remote source path.
+	 * @param WP_Upgrader $upgrader      Upgrader instance.
+	 * @param array       $args          Extra arguments.
+	 * @return string|WP_Error
+	 */
+	public function fix_folder_name( $source, $remote_source, $upgrader, $args ) {
+		global $wp_filesystem;
+
+		// Only process our plugin.
+		if ( ! isset( $args['plugin'] ) || WPSP_PLUGIN_BASENAME !== $args['plugin'] ) {
+			return $source;
+		}
+
+		$correct_name = self::SLUG;
+		$new_source   = trailingslashit( $remote_source ) . $correct_name . '/';
+
+		// If source already correct, return as is.
+		if ( trailingslashit( $source ) === $new_source ) {
+			return $source;
+		}
+
+		// Rename folder.
+		if ( $wp_filesystem->move( $source, $new_source ) ) {
+			return $new_source;
+		}
+
+		return new WP_Error( 'rename_failed', 'Could not rename plugin folder.' );
+	}
+
+	/**
+	 * Get remote update info.
+	 *
+	 * @return object|false
+	 */
+	private function get_remote_info() {
+		$cached = get_transient( self::CACHE_KEY );
+		if ( false !== $cached ) {
+			return $cached;
+		}
+
+		$response = wp_remote_get( self::UPDATE_URL, array(
+			'timeout' => 10,
+			'headers' => array( 'Accept' => 'application/json' ),
+		) );
+
+		if ( is_wp_error( $response ) || 200 !== wp_remote_retrieve_response_code( $response ) ) {
+			// Cache failure for 1 hour to avoid hammering.
+			set_transient( self::CACHE_KEY, false, HOUR_IN_SECONDS );
+			return false;
+		}
+
+		$body = wp_remote_retrieve_body( $response );
+		$data = json_decode( $body );
+
+		if ( ! $data || empty( $data->version ) ) {
+			set_transient( self::CACHE_KEY, false, HOUR_IN_SECONDS );
+			return false;
+		}
+
+		// Cache for 12 hours.
+		set_transient( self::CACHE_KEY, $data, 12 * HOUR_IN_SECONDS );
+
+		return $data;
+	}
+}