Refresh features on save

assets/js/features/provider.js

@@ -38,14 +38,15 @@ export const FeatureSettingsProvider = ({
 	children,
 	defaultSettings,
 	epioLogoUrl,
-	features,
+	features: initialFeatures,
 	indexMeta,
 	syncedSettings,
 }) => {
 	const [isBusy, setIsBusy] = useState(false);
 	const [isSyncing, setIsSyncing] = useState(!!indexMeta);
 	const [settings, setSettings] = useState({ ...defaultSettings });
 	const [savedSettings, setSavedSettings] = useState({ ...syncedSettings });
+	const [features, setFeatures] = useState(initialFeatures);
 
 	/**
 	 * Get a feature's data by its slug.
@@ -168,6 +169,10 @@ export const FeatureSettingsProvider = ({
 			});
 
 			setSavedSettings(response.data);
+
+			if (response.features) {
+				setFeatures(response.features);
+			}
 		} finally {
 			setIsBusy(false);
 		}

includes/classes/REST/Features.php

@@ -220,9 +220,14 @@ public function update_settings( \WP_REST_Request $request ) {
 			FeaturesStore::factory()->update_feature( $slug, $feature, true, 'draft' );
 		}
 
+		$features_objects = FeaturesStore::factory()->registered_features;
+		$features_data    = array_map( fn( $feature ) => $feature->get_json(), $features_objects );
+		$features_data    = array_values( $features_data );
+
 		return [
-			'data'    => $current_settings,
-			'success' => true,
+			'data'     => $current_settings,
+			'success'  => true,
+			'features' => $features_data,
 		];
 	}
 }