23s2s

✏️ Editing: launch-data.js

import {
	getDesignBuildShape,
	getHomeShape,
	getImagesShape,
	getLogoShape,
	getPagesShape,
	getPluginsShape,
	getProfileShape,
	getStringsShape,
	getStyleShape,
} from '@auto-launch/fetchers/shape';
import { clearSiteImages } from '@auto-launch/functions/wp';
import { __ } from '@wordpress/i18n';
import { create } from 'zustand';
import { devtools, persist } from 'zustand/middleware';
import { overrideWithUrlParams, urlParams, urlParamsShape } from './url-params';

const shapeToKeyValue = (shape) => {
	return Object.fromEntries(
		Object.keys(shape.shape).map((key) => [key, undefined]),
	);
};

const initialState = {
	go: false,
	// translators: this is for a action log UI. Keep it short
	statusMessages: [__('Booting things up', 'extendify-local')],
	errorMessage: null,
	errorCount: 0,
	description: null,
	descriptionBackup: undefined,
	descriptionRaw: null,
	urlParams: {},
	siteProfile: {
		...shapeToKeyValue(getProfileShape),
	},
	...shapeToKeyValue(getLogoShape),
	...shapeToKeyValue(getPluginsShape),
	...shapeToKeyValue(getStyleShape),
	...shapeToKeyValue(getStringsShape),
	...shapeToKeyValue(getImagesShape),
	...shapeToKeyValue(getHomeShape),
	...shapeToKeyValue(getPagesShape),
	...shapeToKeyValue(getDesignBuildShape),
	designBuild: undefined,
	attempt: 1,
};

// Checks that a key being set is actually something we expect
const isValidKey = (key) => Object.keys(initialState).includes(key);

const keySchemas = {
	urlParams: urlParamsShape,
	siteProfile: getProfileShape,
	...Object.fromEntries(
		[
			getLogoShape,
			getPluginsShape,
			getStyleShape,
			getStringsShape,
			getImagesShape,
			getHomeShape,
			getPagesShape,
			getDesignBuildShape,
		].flatMap((s) => Object.entries(s.shape)),
	),
};

export const useLaunchDataStore = create(
	persist(devtools(state, { name: 'Extendify Launch Data' }), {
		name: `extendify-launch-data-${window.extSharedData.siteId}`,
		merge: (p, current) => {
			// Make sure the persisted state is valid and not corrupted.
			const persisted = p && typeof p === 'object' ? p : {};

// This gives us some recovery on page reload
			const validated = Object.fromEntries(
				Object.entries(persisted)
					.filter(([key]) => key in keySchemas)
					.map(([key, value]) => {
						const result = keySchemas[key].safeParse(value);
						return [key, result.success ? result.data : undefined];
					}),
			);
			// Merge in the url params
			const { title, description, go, ...urlParamsMapped } =
				overrideWithUrlParams(urlParams);

return {
				...current,
				...persisted,
				...validated,
				// If there's a URL param here it should override these values
				title: title || persisted.title,
				description: description || persisted.description,
				descriptionRaw: description || persisted.descriptionRaw,
				descriptionBackup: persisted.descriptionBackup || description || title,
				go: go || persisted.go,
				urlParams: {
					title,
					description,
					go,
					...urlParamsMapped,
				},
			};
		},
		onRehydrateStorage: () => (state) => {
			if (!state) return;
			queueMicrotask(() => {
				useLaunchDataStore.setState(state);
			});
		},
		partialize: (state) => {
			const {
				statusMessages,
				errorMessage,
				errorCount,
				pulse,
				description,
				descriptionRaw,
				title,
				...rest
			} = state;
			return Object.fromEntries(
				Object.entries(rest).filter(([, v]) =>
					Array.isArray(v) ? v.length > 0 : Boolean(v),
				),
			);
		},
	}),
	state,
);

Sid Gifari From Gifari Industries - BD Cyber Security Team