class.jetpack-sync-module-menus.php 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778
  1. <?php
  2. class Jetpack_Sync_Module_Menus extends Jetpack_Sync_Module {
  3. private $nav_items_just_added = array();
  4. function name() {
  5. return 'menus';
  6. }
  7. public function init_listeners( $callable ) {
  8. add_action( 'wp_create_nav_menu', $callable, 10, 2 );
  9. add_action( 'wp_update_nav_menu', array( $this, 'update_nav_menu' ), 10, 2 );
  10. add_action( 'wp_add_nav_menu_item', array( $this, 'update_nav_menu_add_item' ), 10, 3 );
  11. add_action( 'wp_update_nav_menu_item', array( $this, 'update_nav_menu_update_item' ), 10, 3 );
  12. add_action( 'post_updated', array( $this, 'remove_just_added_menu_item' ), 10, 2 );
  13. add_action( 'jetpack_sync_updated_nav_menu', $callable, 10, 2 );
  14. add_action( 'jetpack_sync_updated_nav_menu_add_item', $callable, 10, 4 );
  15. add_action( 'jetpack_sync_updated_nav_menu_update_item', $callable, 10, 4 );
  16. add_action( 'delete_nav_menu', $callable, 10, 3 );
  17. }
  18. public function update_nav_menu( $menu_id, $menu_data = array() ) {
  19. if ( empty( $menu_data ) ) {
  20. return;
  21. }
  22. /**
  23. * Helps sync log that a nav menu was updated.
  24. *
  25. * @since 5.0.0
  26. *
  27. * @param int $menu_id, the id of the menu
  28. * @param object $menu_data
  29. */
  30. do_action( 'jetpack_sync_updated_nav_menu', $menu_id, $menu_data );
  31. }
  32. public function update_nav_menu_add_item( $menu_id, $nav_item_id, $nav_item_args ) {
  33. $menu_data = wp_get_nav_menu_object( $menu_id );
  34. $this->nav_items_just_added[] = $nav_item_id;
  35. /**
  36. * Helps sync log that a new menu item was added.
  37. *
  38. * @since 5.0.0
  39. *
  40. * @param int $menu_id, the id of the menu
  41. * @param object $menu_data
  42. * @param int $nav_item_id
  43. * @param int $nav_item_args
  44. */
  45. do_action( 'jetpack_sync_updated_nav_menu_add_item', $menu_id, $menu_data, $nav_item_id, $nav_item_args );
  46. }
  47. public function update_nav_menu_update_item( $menu_id, $nav_item_id, $nav_item_args ) {
  48. if ( in_array( $nav_item_id, $this->nav_items_just_added ) ) {
  49. return;
  50. }
  51. $menu_data = wp_get_nav_menu_object( $menu_id );
  52. /**
  53. * Helps sync log that an update to the menu item happened.
  54. *
  55. * @since 5.0.0
  56. *
  57. * @param int $menu_id, the id of the menu
  58. * @param object $menu_data
  59. * @param int $nav_item_id
  60. * @param int $nav_item_args
  61. */
  62. do_action( 'jetpack_sync_updated_nav_menu_update_item', $menu_id, $menu_data, $nav_item_id, $nav_item_args );
  63. }
  64. public function remove_just_added_menu_item( $nav_item_id, $post_after ) {
  65. if ( 'nav_menu_item' !== $post_after->post_type ) {
  66. return;
  67. }
  68. $this->nav_items_just_added = array_diff( $this->nav_items_just_added, array( $nav_item_id ) );
  69. }
  70. }