<?php

/**
 * Class WPML_Beaver_Builder_Register_Strings
 */
class WPML_Beaver_Builder_Register_Strings extends WPML_Page_Builders_Register_Strings {

	/**
	 * @param array $data_array
	 * @param array $package
	 */
	protected function register_strings_for_modules( array $data_array, array $package ) {
		foreach ( $data_array as $data ) {
			if ( is_array( $data ) ) {
				$data = $this->sort_modules_before_string_registration( $data );
				$this->register_strings_for_modules( $data, $package );
			} elseif ( is_object( $data ) ) {
				if ( isset( $data->type ) && 'module' === $data->type ) {
					$this->register_strings_for_node( $data->node, $data->settings, $package );
				}
			}
		}
	}

	/**
	 * The modules are not in the order they appear on the page,
	 * so we need to sort it before to register the strings.
	 *
	 * @param array $modules
	 *
	 * @return array
	 */
	private function sort_modules_before_string_registration( array $modules ) {
		uasort( $modules, array( $this, 'sort_modules_by_position_only' ) );
		return $this->sort_modules_by_parent_and_child( $modules );
	}

	/**
	 * We receive all modules as a flat tree and we need to reorder from:
	 * - child A
	 * - child A
	 * - parent A
	 * - child B
	 * - parent B
	 * - child B
	 * - child C
	 *
	 * To:
	 * - parent A
	 * - child A
	 * - child B
	 * - parent B
	 * - child A
	 * - child B
	 * - child C
	 *
	 * The relative positions are already sorted by `sort_modules_by_position_only`
	 *
	 * @param array        $all_modules
	 * @param string|null  $parent_hash
	 * @param array        $sorted_modules
	 *
	 * @return array
	 */
	private function sort_modules_by_parent_and_child( array $all_modules, $parent_hash = null, array $sorted_modules = array() ){
		foreach ( $all_modules as $hash => $module ) {

			if ( $module->parent === $parent_hash ) {
				$sorted_modules[ $hash ] = $module;
				unset( $all_modules[ $hash ] );
				$sorted_modules = $this->sort_modules_by_parent_and_child( $all_modules, $module->node, $sorted_modules );
			}
		}

		return $sorted_modules;
	}

	/**
	 * @param stdClass $a
	 * @param stdClass $b
	 *
	 * @return int
	 */
	private function sort_modules_by_position_only( stdClass $a, stdClass $b ) {
		return ( (int) $a->position < (int) $b->position ) ? -1 : 1;
	}
}