wp-headless-theme/functions.php at master · paramander/wp-headless-theme · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
<?php

add_action( 'rest_api_init', function( $server ) {
    $ns = 'headless/v1';

    register_rest_route( $ns, '/frontpage', array(
        'methods'  => WP_REST_Server::READABLE,
        'callback' => function() {
            $page_id = get_option('page_on_front');
            if ( $page_id > 0 ) {
                wp_redirect( get_rest_url() . 'wp/v2/pages/' . $page_id );
                exit;
            }

            return new WP_Error(
                'headless_no_frontpage',
                __( 'No static frontpage set', 'headless' ),
                array(
                    'status' => 404
                )
            );
        }
    ) );

    register_rest_route( $ns, '/menu-locations', array(
        'methods'  => WP_REST_Server::READABLE,
        'callback' => function() use ($ns) {
            $response       = array();
            $menu_locations = get_registered_nav_menus();
            $locations      = get_nav_menu_locations();

            if ( !empty( $menu_locations ) ) {
                foreach( $menu_locations as $slug => $description ) {
                    $data = array(
                        'slug'        => $slug,
                        'description' => $description,
                    );

                    if ( array_key_exists( $slug, $locations ) ) {
                        $menu_id     = $locations[ $slug ];
                        $menu_object = wp_get_nav_menu_object( $menu_id );
                        $menu_items  = array_map( function( $item ) {
                            return array(
                                'id'         => $item->ID,
                                'date'       => $item->post_date,
                                'date_gmt'   => $item->post_date_gmt,
                                'author'     => (int) $item->post_author,
                                'link'       => $item->url,
                                'title'      => $item->title,
                                'parent'     => (int) $item->menu_item_parent,
                                'menu_order' => $item->menu_order
                            );
                        }, wp_get_nav_menu_items( $menu_id ) );

                        $items = array();
                        $children = array();
                        while ( $item = array_pop($menu_items) ) {
                            if ( array_key_exists( $item['id'], $children ) ) {
                                $item['children'] = array_reverse( $children[ $item['id'] ] );
                            }
                            if ( $item['parent'] == '0' ) {
                                $items[] = $item;
                            } else {
                                $children[ $item['parent'] ][] = $item;
                            }
                        }
                        $items = array_reverse( $items );

                        $data['menu'] = array(
                            'term_id' => $menu_object->term_id,
                            'name'    => $menu_object->name,
                            'slug'    => $menu_object->slug,
                            'count'   => $menu_object->count,
                            'items'   => $items
                        );
                    }

                    $data['_links'] = array(
                        'collection' => array(
                            array(
                                'href' => get_rest_url() . $ns . '/menu-locations'
                            )
                        )
                    );

                    $response[] = $data;
                }
            }

            return $response;
        }
    ) );

    add_filter( 'acf/format_value/type=relationship', 'headless_theme_format_post_object_value', 20, 3 );
    add_filter( 'acf/format_value/type=post_object', 'headless_theme_format_post_object_value', 20, 3 );
} );

function headless_theme_format_post_object_value( $value, $post_id, $field ) {
    if ( $field['return_format'] !== 'object' ) {
        return $value;
    }
    remove_filter( 'acf/format_value/type=relationship', 'headless_theme_format_post_object_value', 20 );
    remove_filter( 'acf/format_value/type=post_object', 'headless_theme_format_post_object_value', 20 );

    if ( is_array( $value ) ) {
        foreach( $value as $post ) {
            $formatted[] = headless_theme_rest_format_post( $post );
        }
    } else {
        $formatted = headless_theme_rest_format_post( $value );
    }

    add_filter( 'acf/format_value/type=relationship', 'headless_theme_format_post_object_value', 20, 3 );
    add_filter( 'acf/format_value/type=post_object', 'headless_theme_format_post_object_value', 20, 3 );
    return $formatted;
};

function headless_theme_rest_format_post( $post ) {
    $server = rest_get_server();
    $controller = new WP_REST_Posts_Controller( $post->post_type );
    $post_type = get_post_type_object( $post->post_type );
    $request = WP_REST_Request::from_url( rest_url( sprintf( 'wp/v2/%s/%d', $post_type->rest_base, $post->ID ) ) );
    $request['context'] = 'embed';

    $response = $server->dispatch( $request );
	$response = apply_filters( 'rest_post_dispatch', rest_ensure_response( $response ), $server, $request );

	return $server->response_to_data( $response, isset( $_GET['_embed'] ) );
}