Feature: added post lock feature based on posts and user

mahbubme · mahbubme · commit 3834b77bc3cc · 2018-12-19T12:55:06.000+06:00
diff --git a/admin/class-admin-settings.php b/admin/class-admin-settings.php
@@ -401,6 +401,12 @@ public function enqueue_styles() {
 
         wp_enqueue_style( 'wpuf-admin', WPUF_ASSET_URI . '/css/admin.css', false, WPUF_VERSION );
         wp_enqueue_script( 'wpuf-admin-script', WPUF_ASSET_URI . '/js/wpuf-admin.js', array( 'jquery' ), false, WPUF_VERSION );
+
+        wp_localize_script( 'wpuf-admin-script', 'wpuf_admin_script', array(
+            'ajaxurl'               => admin_url( 'admin-ajax.php' ),
+            'nonce'                 => wp_create_nonce( 'wpuf_nonce' ),
+            'cleared_schedule_lock' => __( 'Schedule lock has been cleared', 'wp-user-frontend' ),
+        ) );
     }
 
 }
diff --git a/admin/html/form-settings-post-edit.php b/admin/html/form-settings-post-edit.php
@@ -6,6 +6,7 @@
 $url                  = isset( $form_settings['edit_url'] ) ? $form_settings['edit_url'] : '';
 $update_text          = isset( $form_settings['update_text'] ) ? $form_settings['update_text'] : __( 'Update', 'wp-user-frontend' );
 $subscription         = isset( $form_settings['subscription'] ) ? $form_settings['subscription'] : null;
+$lock_edit_post       = isset( $form_settings['lock_edit_post'] ) ? $form_settings['lock_edit_post'] : '';
 ?>
 <table class="form-table">
 
@@ -93,4 +94,15 @@
             <input type="text" name="wpuf_settings[update_text]" value="<?php echo esc_attr( $update_text ); ?>">
         </td>
     </tr>
+
+    <tr class="lock-edit-post">
+        <th><?php _e( 'Lock User From Editing After', 'wp-user-frontend' ); ?></th>
+        <td>
+            <input type="number" name="wpuf_settings[lock_edit_post]" id="lock_edit_post" value="<?php echo $lock_edit_post; ?>">
+            <span><?php _e( 'hours', 'wp-user-frontend' ); ?></span>
+            <p class="description">
+                <?php _e( 'After how many hours user will be locked from editing the submitted post.', 'wp-user-frontend' ) ?>
+            </p>
+        </td>
+    </tr>
 </table>
diff --git a/admin/posting.php b/admin/posting.php
@@ -16,11 +16,12 @@ function __construct() {
         // meta boxes
         add_action( 'add_meta_boxes', array( $this, 'add_meta_boxes') );
         add_action( 'add_meta_boxes', array( $this, 'add_meta_box_form_select') );
-
+        add_action( 'add_meta_boxes', array( $this, 'add_meta_box_post_lock') );
         add_action( 'admin_enqueue_scripts', array( $this, 'enqueue_script') );
-
         add_action( 'save_post', array( $this, 'save_meta'), 1, 2 ); // save the custom fields
         add_action( 'save_post', array( $this, 'form_selection_metabox_save' ), 1, 2 ); // save edit form id
+        add_action( 'save_post', array( $this, 'post_lock_metabox_save' ), 1, 2 ); // save post lock option
+        add_action( 'wp_ajax_wpuf_clear_schedule_lock', array($this, 'clear_schedule_lock') );
     }
 
     public static function init() {
@@ -101,7 +102,6 @@ function add_meta_box_form_select() {
         }
     }
 
-
     /**
      * Form selection meta box in post types
      *
@@ -127,7 +127,7 @@ function form_selection_metabox() {
             <?php } ?>
         </select>
         <div>
-            <p><a href="https://wedevs.com/docs/wp-user-frontend-pro/tutorials/purpose-of-the-wpuf-form-metabox/" target="_blank"><?php _e( 'Purpose of this metabox', 'wp-user-frontend' ); ?></a></p>
+            <p><a href="https://wedevs.com/docs/wp-user-frontend-pro/tutorials/purpose-of-the-wpuf-form-metabox/" target="_blank"><?php _e( 'Learn more', 'wp-user-frontend' ); ?></a></p>
         </div>
         <?php
     }
@@ -158,6 +158,89 @@ function form_selection_metabox_save( $post_id, $post ) {
         update_post_meta( $post->ID, '_wpuf_form_id', $_POST['wpuf_form_select'] );
     }
 
+    /**
+     * Meta box for post lock
+     *
+     * Registers a meta box in public post types to select the desired WPUF
+     * form select box to assign a form id.
+     *
+     * @since 3.0.2
+     *
+     * @return void
+     */
+    function add_meta_box_post_lock() {
+        $post_types = get_post_types( array('public' => true) );
+
+        foreach ($post_types as $post_type) {
+            add_meta_box( 'wpuf-post-lock', __('WPUF Lock User', 'wp-user-frontend'), array($this, 'post_lock_metabox'), $post_type, 'side', 'high' );
+        }
+    }
+
+    /**
+     * Post lock meta box in post types
+     *
+     * Registered via $this->add_meta_box_post_lock()
+     *
+     * @since 3.0.2
+     *
+     * @global object $post
+     */
+    function post_lock_metabox() {
+        global $post;
+
+        $msg                 = '';
+        $edit_post_lock      = get_post_meta( $post->ID, '_wpuf_lock_editing_post', true );
+        $edit_post_lock_time = get_post_meta( $post->ID, '_wpuf_lock_user_editing_post_time', true );
+
+        if( !empty( $edit_post_lock_time ) && $edit_post_lock_time > time() ) {
+            $time         = date( 'Y-m-d H:i:s', $edit_post_lock_time );
+            $local_time   = get_date_from_gmt( $time, get_option('date_format') . ' ' . get_option('time_format') );
+            $msg          = sprintf( __( 'Frontend edit access for this post will be automatically locked after %s, <a id="wpuf_clear_schedule_lock" data="%s" href="#">Clear Schedule Lock</a> Or,', 'wp-user-frontend' ), $local_time, $post->ID );
+        }
+
+        ?>
+
+        <input type="hidden" name="wpuf_lock_editing_post_nonce" value="<?php echo wp_create_nonce( plugin_basename( __FILE__ ) ); ?>" />
+
+        <p><?php echo $msg; ?></p>
+
+        <label>
+            <input type="hidden" name="wpuf_lock_post" value="no">
+            <input type="checkbox" name="wpuf_lock_post" value="yes" <?php checked($edit_post_lock, 'yes'); ?>>
+            <?php _e( 'Lock Post', 'wp-user-frontend' ); ?>
+        </label>
+        <p style="margin-top: 10px"><?php _e( 'Lock user from editing this post from the frontend dashboard', 'wp-user-frontend' ); ?></p>
+        <?php
+    }
+
+    /**
+     * Save the lock post option
+     *
+     * @since 3.0.2
+     *
+     * @param int $post_id
+     * @param object $post
+     * @return int|void
+     */
+    function post_lock_metabox_save( $post_id, $post ) {
+        $edit_post_lock_time = isset( $_POST['_wpuf_lock_user_editing_post_time'] ) ? $_POST['_wpuf_lock_user_editing_post_time'] : '';
+
+        if ( !isset($_POST['wpuf_lock_post'])) {
+            return $post->ID;
+        }
+
+        if ( !wp_verify_nonce( $_POST['wpuf_lock_editing_post_nonce'], plugin_basename( __FILE__ ) ) ) {
+            return $post->ID;
+        }
+
+        // Is the user allowed to edit the post or page?
+        if ( !current_user_can( 'edit_post', $post->ID ) ) {
+            return $post->ID;
+        }
+
+        update_post_meta( $post->ID, '_wpuf_lock_editing_post', $_POST['wpuf_lock_post'] );
+    }
+
     /**
      * Meta box to show WPUF Custom Fields
      *
@@ -430,4 +513,23 @@ function save_meta( $post_id, $post = null ) {
         WPUF_Frontend_Form_Post::update_post_meta( $meta_vars, $post_id );
     }
 
+    /**
+     * Clear Schedule lock
+     *
+     * @since 3.0.2
+     */
+    public function clear_schedule_lock() {
+        check_ajax_referer( 'wpuf_nonce', 'nonce' );
+
+        $post_id = isset( $_POST['post_id'] ) ? $_POST['post_id'] : '';
+        if ( !empty( $post_id ) ) {
+            $edit_post_lock_time = get_post_meta( $post_id, '_wpuf_lock_user_editing_post_time', true );
+
+            if ( !empty( $edit_post_lock_time ) ) {
+                update_post_meta( $post_id, '_wpuf_lock_user_editing_post_time', '' );
+            }
+        }
+        exit;
+    }
+
 }
diff --git a/assets/js/wpuf-admin.js b/assets/js/wpuf-admin.js
@@ -26,6 +26,33 @@ jQuery(function($) {
         $(this.children[0]).attr("checked", "checked");
         $(".wpuf-form-layouts li").removeClass('active');
         $(this).toggleClass('active');
-    })
+    });
+
+    // Clear schedule lock
+    $('#wpuf_clear_schedule_lock').on('click', function(e) {
+        e.preventDefault();
+        var post_id = $(this).attr('data');
 
+        $.ajax({
+            url: wpuf_admin_script.ajaxurl,
+            type: 'POST',
+            data: {
+                'action'    : 'wpuf_clear_schedule_lock',
+                'nonce'     : wpuf_admin_script.nonce,
+                'post_id'   : post_id
+            },
+            success:function(data) {
+                swal({
+                    type: 'success',
+                    title: wpuf_admin_script.cleared_schedule_lock,
+                    showConfirmButton: false,
+                    timer: 1500
+                });
+            },
+            error: function(errorThrown){
+                console.log(errorThrown);
+            }
+        });
+        $(this).closest("p").hide();
+    });
 });
diff --git a/class/frontend-form-post.php b/class/frontend-form-post.php
@@ -179,6 +179,24 @@ function edit_post_shortcode( $atts ) {
             return '<div class="wpuf-info">' . __( 'Invalid post', 'wp-user-frontend' ) . '</div>';
         }
 
+        $edit_post_lock      = get_post_meta( $post_id, '_wpuf_lock_editing_post', true );
+        $edit_post_lock_time = get_post_meta( $post_id, '_wpuf_lock_user_editing_post_time', true );
+
+        if ( $edit_post_lock == 'yes' ) {
+            return '<div class="wpuf-info">' . apply_filters( 'wpuf_edit_post_lock_user_notice', __( 'Your edit access for this post has been locked by an administrator.', 'wp-user-frontend' ) ) . '</div>';
+        }
+
+        if ( !empty( $edit_post_lock_time ) &&  $edit_post_lock_time < time() ) {
+            return '<div class="wpuf-info">' . apply_filters( 'wpuf_edit_post_lock_expire_notice', __( 'Your allocated time for editing this post has been expired.', 'wp-user-frontend' ) ) . '</div>';
+        }
+
+        if ( wpuf_get_user()->edit_post_locked() ) {
+            if ( !empty( wpuf_get_user()->edit_post_lock_reason() ) ) {
+                return '<div class="wpuf-info">' . wpuf_get_user()->edit_post_lock_reason() . '</div>';
+            }
+            return '<div class="wpuf-info">' . apply_filters( 'wpuf_user_edit_post_lock_notice', __( 'Your post edit access has been locked by an administrator.', 'wp-user-frontend' ) ) . '</div>';
+        }
+
         //is editing enabled?
         if ( wpuf_get_option( 'enable_post_edit', 'wpuf_dashboard', 'yes' ) != 'yes' ) {
             return '<div class="wpuf-info">' . __( 'Post Editing is disabled', 'wp-user-frontend' ) . '</div>';
@@ -491,6 +509,17 @@ function submit_post() {
 
         $post_id = wp_insert_post( $postarr );
 
+        // add _wpuf_lock_editing_post_time meta to
+        // lock user from editing the published post after a certain time
+        if ( !$is_update ) {
+            $lock_edit_post = isset( $form_settings['lock_edit_post'] ) ? floatval( $form_settings['lock_edit_post'] ) : 0;
+
+            if ( $post_id && $lock_edit_post > 0 ) {
+                $lock_edit_post_time    = time() + ( $lock_edit_post * 60 * 60 );
+                update_post_meta( $post_id, '_wpuf_lock_user_editing_post_time', $lock_edit_post_time );
+            }
+        }
+
         if ( $post_id ) {
 
             self::update_post_meta( $meta_vars, $post_id );
diff --git a/includes/class-user.php b/includes/class-user.php
@@ -61,6 +61,24 @@ public function lock_reason() {
         return get_user_meta( $this->id, 'wpuf_lock_cause', true );
     }
 
+    /**
+     * Check if a user's post edit capability is locked
+     *
+     * @return boolean
+     */
+    public function edit_post_locked() {
+        return 'yes' ==  get_user_meta( $this->id, 'wpuf_edit_postlock', true );
+    }
+
+    /**
+     * Get the edit post lock reason
+     *
+     * @return string
+     */
+    public function edit_post_lock_reason() {
+        return get_user_meta( $this->id, 'wpuf_edit_post_lock_cause', true );
+    }
+
     /**
      * Handles user subscription
      *
diff --git a/includes/free/edit-profile.php b/includes/free/edit-profile.php
@@ -241,9 +241,12 @@ function post_lock_form( $profileuser ) {
         $wpuf_subscription = $current_user->subscription();
         $post_locked       = $current_user->post_locked();
         $lock_reason       = $current_user->lock_reason();
+        $edit_post_locked  = $current_user->edit_post_locked();
+        $edit_lock_reason  = $current_user->edit_post_lock_reason();
 
         if ( is_admin() && current_user_can( 'edit_users' ) ) {
-            $select = ( $post_locked == true ) ? 'yes' : 'no';
+            $select           = ( $post_locked == true ) ? 'yes' : 'no';
+            $edit_post_select = ( $edit_post_locked == true ) ? 'yes' : 'no';
             ?>
             <div class="wpuf-user-post-lock">
                 <h3><?php _e( 'WPUF Post Lock', 'wp-user-frontend' ); ?></h3>
@@ -258,13 +261,29 @@ function post_lock_form( $profileuser ) {
                             <span class="description"><?php _e( 'Lock user from creating new post.', 'wp-user-frontend' ); ?></span></em>
                         </td>
                     </tr>
-
                     <tr>
                         <th><label for="post-lock"><?php _e( 'Lock Reason:', 'wp-user-frontend' ); ?> </label></th>
                         <td>
                             <input type="text" name="wpuf_lock_cause" id="wpuf_lock_cause" class="regular-text" value="<?php echo esc_attr( $lock_reason ); ?>" />
                         </td>
                     </tr>
+
+                    <tr>
+                        <th><label for="post-lock"><?php _e( 'Lock Edit Post:', 'wp-user-frontend' ); ?> </label></th>
+                        <td>
+                            <select name="wpuf_edit_postlock" id="edit-post-lock">
+                                <option value="no"<?php selected( $edit_post_select, 'no' ); ?>>No</option>
+                                <option value="yes"<?php selected( $edit_post_select, 'yes' ); ?>>Yes</option>
+                            </select>
+                            <span class="description"><?php _e( 'Lock user from editing post.', 'wp-user-frontend' ); ?></span></em>
+                        </td>
+                    </tr>
+                    <tr>
+                        <th><label for="post-lock"><?php _e( 'Edit Post Lock Reason:', 'wp-user-frontend' ); ?> </label></th>
+                        <td>
+                            <input type="text" name="wpuf_edit_post_lock_cause" id="wpuf_edit_post_lock_cause" class="regular-text" value="<?php echo esc_attr( $edit_lock_reason ); ?>" />
+                        </td>
+                    </tr>
                 </table>
             </div>
             <?php
@@ -280,6 +299,8 @@ function post_lock_update( $user_id ) {
         if ( is_admin() && current_user_can( 'edit_users' ) ) {
             update_user_meta( $user_id, 'wpuf_postlock', $_POST['wpuf_postlock'] );
             update_user_meta( $user_id, 'wpuf_lock_cause', $_POST['wpuf_lock_cause'] );
+            update_user_meta( $user_id, 'wpuf_edit_postlock', $_POST['wpuf_edit_postlock'] );
+            update_user_meta( $user_id, 'wpuf_edit_post_lock_cause', $_POST['wpuf_edit_post_lock_cause'] );
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -401,6 +401,12 @@ public function enqueue_styles() {`
`401`	`401`
`402`	`402`	`wp_enqueue_style( 'wpuf-admin', WPUF_ASSET_URI . '/css/admin.css', false, WPUF_VERSION );`
`403`	`403`	`wp_enqueue_script( 'wpuf-admin-script', WPUF_ASSET_URI . '/js/wpuf-admin.js', array( 'jquery' ), false, WPUF_VERSION );`
	`404`	`+`
	`405`	`+ wp_localize_script( 'wpuf-admin-script', 'wpuf_admin_script', array(`
	`406`	`+ 'ajaxurl' => admin_url( 'admin-ajax.php' ),`
	`407`	`+ 'nonce' => wp_create_nonce( 'wpuf_nonce' ),`
	`408`	`+ 'cleared_schedule_lock' => __( 'Schedule lock has been cleared', 'wp-user-frontend' ),`
	`409`	`+ ) );`
`404`	`410`	`}`
`405`	`411`
`406`	`412`	`}`