Add page_type additional field

Author: frozzare

README.md

@@ -8,6 +8,8 @@ Add-on for the WordPress REST API, requires Papi 3.0.0.
 
 ## Addtional fields
 
+### Fields
+
 Post types that has page types will get addtional field attached to the object with the `fields` that contains all fields for the the current post page type. This can be removed.
 
 Example response with additional `fields` field:
@@ -23,6 +25,17 @@ Example response with additional `fields` field:
 ]
 ```
 
+### Page type field
+
+```json
+[
+  {
+    "id": 1,
+    "page_type": "example-page-type"
+  }
+]
+```
+
 ## Endpoints
 
 ### Fields

src/class-papi-rest-api-additional-fields.php

@@ -33,7 +33,11 @@ public function setup_fields() {
 
 		foreach ( $post_types as $post_type ) {
 			register_api_field( $post_type, 'fields', [
-				'get_callback'    => [$this, 'get_fields']
+				'get_callback' => [$this, 'get_fields']
+			] );
+
+			register_api_field( $post_type, 'page_type', [
+				'get_callback' => [$this, 'get_page_type']
 			] );
 		}
 	}
@@ -62,6 +66,19 @@ public function get_fields( array $data, $field_name, WP_REST_Request $request )
 
 		return $fields;
 	}
+
+	/**
+	 * Get page type.
+	 *
+	 * @param  array           $data
+	 * @param  string          $field_name
+	 * @param  WP_REST_Request $request
+	 *
+	 * @return array
+	 */
+	public function get_page_type( array $data, $field_name, WP_REST_Request $request ) {
+		return papi_get_page_type_id( $data['ID'] ) ?: '';
+	}
 }
 
 new Papi_REST_API_Additional_Fields;

tests/cases/class-papi-rest-api-additional-fields-test.php

@@ -30,4 +30,16 @@ public function test_get_fields() {
 		$fields = $this->class->get_fields( ['ID' => $post_id], 'fields', new WP_REST_Request );
 		$this->assertEquals( ['name' => 'Fredrik', 'text' => 'Hello, world!'], $fields );
 	}
+
+	public function test_get_page_type() {
+		$post_id = $this->factory->post->create();
+
+		$page_type = $this->class->get_page_type( ['ID' => $post_id], 'page_type', new WP_REST_Request );
+		$this->assertSame( $page_type, '' );
+
+		update_post_meta( $post_id, papi_get_page_type_key(), 'simple-content-page-type' );
+
+		$page_type = $this->class->get_page_type( ['ID' => $post_id], 'page_type', new WP_REST_Request );
+		$this->assertSame( $page_type, get_post_meta( $post_id, papi_get_page_type_key(), true ) );
+	}
 }