Ticket #1109: 1109.1.diff

wordcamp.org/public_html/wp-content/plugins/wordcamp-site-cloner/includes/source-site-id-setting.php

@@ -114 +114 @@
          * @return null
          */
         protected function update( $source_site_id ) {
+                // Import css from source site to current site.
+                $this->import_css( $source_site_id );
+
+                // Import menus from source site to current site.
+                $this->import_menus( $source_site_id );
+        }
+
+        /**
+         * Import CSS from source site to current site.
+         *
+         * @param int $source_site_id ID of source site.
+         */
+        protected function import_css( $source_site_id ) {
                 switch_to_blog( $source_site_id );
 
                 $source_site_theme_mods = get_theme_mod( 'jetpack_custom_css' );
@@ -160 +173 @@
                 wp_update_custom_css_post( $source_site_css );
                 set_theme_mod( 'jetpack_custom_css', $source_site_theme_mods );
         }
+
+        /**
+         * Import menus from source site to current site.
+         *
+         * @param int $source_site_id ID of source site.
+         */
+        protected function import_menus( $source_site_id ) {
+                switch_to_blog( $source_site_id );
+
+                // Get all menus from source site.
+                $menus = wp_get_nav_menus();
+
+                if ( empty( $menus ) || is_wp_error( $menus ) ) {
+                        restore_current_blog();
+                        return;
+                }
+
+                // Get menu locations from source site.
+                $menu_locations = get_nav_menu_locations();
+
+                restore_current_blog();
+
+                $this->create_menus( $menus, $menu_locations );
+        }
+
+        /**
+         * Create menus and add default menu item.
+         * Also set menu locations as per source site.
+         *
+         * @param array $menus            All menus of source site.
+         * @param array $menu_locations   Menu locations of source site.
+         */
+        protected function create_menus( $menus, $menu_locations ) {
+                $ref = array();
+
+                // Create menu and add one menu item to that menu.
+                foreach ( $menus as $menu ) {
+                        $new_menu_id = wp_create_nav_menu( $menu->name );
+
+                        if ( is_wp_error( $new_menu_id ) ) {
+                                continue;
+                        }
+
+                        // save new menu id reference for old menu id. Use to set menu location.
+                        $ref[ $menu->term_id ] = $new_menu_id;
+
+                        $args = array(
+                                'menu-item-position'            => 1,
+                                'menu-item-type'                        => 'custom',
+                                'menu-item-title'                       => __( 'Home', 'wordcamporg' ),
+                                'menu-item-attr-title'      => __( 'Home', 'wordcamporg' ),
+                                'menu-item-url'                         => home_url( '/' ),
+                                'menu-item-status'                      => 'publish',
+                        );
+
+                        wp_update_nav_menu_item( $new_menu_id, 0, $args );
+                }
+
+                // Set menu to same menu location as source site.
+                if ( ! empty( $menu_locations ) ) {
+                        foreach ( $menu_locations as $menu_location => $menu_id ) {
+                                $menu_locations[ $menu_location ] = $ref[ $menu_id ];
+                        }
+                        set_theme_mod( 'nav_menu_locations', array_map( 'absint', $menu_locations ) );
+                }
+        }
 }