/var/www/html/wp-content/plugins/elementor/modules/site-navigation/data/endpoints/add-new-post.php


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
<?php

namespace Elementor\Modules\SiteNavigation\Data\Endpoints;

use 
Elementor\Data\V2\Base\Endpoint;
use 
Elementor\Plugin;
use 
Elementor\User;

if ( ! 
defined'ABSPATH' ) ) {
    exit; 
// Exit if accessed directly.
}

class 
Add_New_Post extends Endpoint {

    protected function 
register() {
        
$args = [
            
'post_type' => [
                
'description' => 'Post type to create',
                
'type' => 'string',
                
'required' => false,
                
'default' => 'post',
                
'sanitize_callback' => function ( $value ) {
                    return 
sanitize_text_field$value );
                },
                
'validate_callback' => 'rest_validate_request_arg',
            ],
        ];

        
$this->register_items_route\WP_REST_Server::CREATABLE$args );
    }

    public function 
get_name() {
        return 
'add-new-post';
    }

    public function 
get_format() {
        return 
'site-navigation/add-new-post';
    }

    public function 
create_items$request ) {
        
$post_type $request->get_param'post_type' );

        if ( ! 
$this->validate_post_type$post_type ) ) {
            return new 
\WP_Error400sprintf'Post type %s does not exist.'$post_type ), [ 'status' => 400 ] );
        }

        if ( ! 
User::is_current_user_can_edit_post_type$post_type ) ) {
            return new 
\WP_Error401sprintf'User dont have capability to create page of type - %s.'$post_type ), [ 'status' => 401 ] );
        }

        
// Temporary solution for the fact that documents creation not using the actual registered post types.
        
$post_type $this->map_post_type$post_type );

        
$document Plugin::$instance->documents->create$post_type );

        if ( 
is_wp_error$document ) ) {
            return new 
\WP_Error500sprintf'Error while creating %s.'$post_type ) );
        }

        return [
            
'id' => $document->get_main_id(),
            
'edit_url' => $document->get_edit_url(),
        ];
    }

    private function 
validate_post_type$post_type ): bool {
        
$post_types get_post_types();

        return 
in_array$post_type$post_types );
    }

    
/**
     * Map post type to Elementor document type.
     *
     * @param $post_type
     *
     * @return string
     */
    
private function map_post_type$post_type ): string {
        
$post_type_map = [
            
'page' => 'wp-page',
            
'post' => 'wp-post',
        ];

        return 
$post_type_map$post_type ] ?? $post_type;
    }
}