edit-view.php

<?php defined( 'ABSPATH' ) || exit;

class TDB_Edit_View extends TDB_View {

  protected $object_id = 0;
  protected $object = null;
  protected $editing = false;
  protected $message = false;
  protected $columns = 2;

  public function __construct( $name, $args ) {

      parent::__construct( $name, $args );

      $this->columns = isset( $args['columns'] ) ? $args['columns'] : 2;

  }

  public function init() {

      global $tdb_registered_tables, $tdb_table;

      if( ! isset( $tdb_registered_tables[$this->name] ) ) {
          wp_die( __( 'Invalid item type.' ) );
      }

      // Setup global $tdb_table
      $tdb_table = $tdb_registered_tables[$this->name];

      // If not TDB object, die
      if ( ! $tdb_table )
          wp_die( __( 'Invalid item type.' ) );

      // If not TDB object allow ui, die
      if ( ! $tdb_table->show_ui ) {
        wp_die( __( 'Sorry, you are not allowed to edit items of this type.' ) );
      }

      if( isset( $_POST['tdb-save'] ) ) {
          // Saving
          $this->save();
      }

      $primary_key = $tdb_table->db->primary_key;

      if( isset( $_GET[$primary_key] ) ) {
          // Editing object
          $this->object_id = (int) $_GET[$primary_key];
          $this->object = $tdb_table->db->get( $this->object_id );
          $this->editing = true;

          // If not id, return to list
          if ( empty( $this->object_id ) ) {
              wp_redirect( tdb_get_list_link( $tdb_table->name ) );
              exit();
          }

          // If not object, die
          if ( ! $this->object ) {
              wp_die( __( 'You attempted to edit an item that doesn&#8217;t exist. Perhaps it was deleted?' ) );
          }

          // If not current user can edit, die
          if ( ! current_user_can( $tdb_table->cap->edit_item, $this->object_id ) ) {
              wp_die( __( 'Sorry, you are not allowed to edit this item.' ) );
          }

      } else {

          // See filter "tdb_{$tdb_table->name}_default_data"
          $this->object_id = tdb_insert_object( array() );
          // If not id, return to list
          if ( empty( $this->object_id ) ) {
              wp_redirect( tdb_get_list_link( $tdb_table->name ) );
              exit();
          }

          $this->object = tdb_get_object( $this->object_id );

          // If not object, die
          if ( ! $this->object )
              wp_die( __( 'Unable to create the draft item.' ) );

          // If not current user can create, die
          if ( ! current_user_can( $tdb_table->cap->create_items, $this->object_id )
              && ! current_user_can( $tdb_table->cap->edit_item, $this->object_id ) ) {
              wp_die( __( 'Sorry, you are not allowed to create items of this type.' ) );
          }

          // Redirect to edit screen to prevent add a draft item multiples times
          wp_redirect( tdb_get_edit_link( $tdb_table->name, $this->object_id ) );
      }

  }

  /**
   * Screen settings text displayed in the Screen Options tab.
   *
   * @param string    $screen_settings    Screen settings.
   * @param WP_Screen $screen             WP_Screen object.
   */
  public function screen_settings( $screen_settings, $screen ) {

      $this->render_meta_boxes_preferences();
      $this->render_screen_layout();

  }

  /**
   * Render the meta boxes preferences.
   *
   * @since 1.0.0
   *
   * @global array $wp_meta_boxes
   */
  public function render_meta_boxes_preferences() {

      global $wp_meta_boxes, $tdb_table;

      // TODO: Forced to place it here until fix issue with load-{pagehook}

      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( 'add_meta_boxes', $tdb_table->name, $this->object );

      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( "add_meta_boxes_{$tdb_table->name}", $this->object );

      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( 'do_meta_boxes', $tdb_table->name, 'normal', $this->object );
      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( 'do_meta_boxes', $tdb_table->name, 'advanced', $this->object );
      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( 'do_meta_boxes', $tdb_table->name, 'side', $this->object );

      if ( ! isset( $wp_meta_boxes[ $tdb_table->name ] ) ) {
          return;
      }
      ?>

      <fieldset class="metabox-prefs">
          <legend><?php _e( 'Boxes' ); ?></legend>
          <?php meta_box_prefs( $tdb_table->name ); ?>
      </fieldset>

      <?php
  }

  /**
   * Render the option for number of columns on the page
   *
   * @since 1.0.0
   */
  public function render_screen_layout() {

      if ( $this->columns <= 1 ) {
          return;
      }

      $screen_layout_columns = get_current_screen()->get_columns();

      if( ! $screen_layout_columns ) {
          $screen_layout_columns = $this->columns;
      }

      $num = $this->columns;

      ?>
      <fieldset class='columns-prefs'>
          <legend class="screen-layout"><?php _e( 'Layout' ); ?></legend><?php
          for ( $i = 1; $i <= $num; ++$i ):
              ?>
              <label class="columns-prefs-<?php echo $i; ?>">
                  <input type='radio' name='screen_columns' value='<?php echo esc_attr( $i ); ?>'
                      <?php checked( $screen_layout_columns, $i ); ?> />
                  <?php printf( _n( '%s column', '%s columns', $i ), number_format_i18n( $i ) ); ?>
              </label>
              <?php
          endfor; ?>
      </fieldset>
      <?php
  }

  public function submit_meta_box( $object ) {

      global $tdb_table;

      /**
       * Fires at top of the submit meta box.
       *
       * @since 1.0.0
       *
       * @param object        $object         Object.
       * @param TDB_Table      $tdb_table       TDB Table object.
       * @param bool          $editing        True if edit screen, false if is adding a new one.
       * @param TDB_Edit_View  $view           Edit view object.
       */
      do_action( "tdb_{$tdb_table->name}_edit_screen_submit_meta_box_top", $object, $tdb_table, $this->editing, $this );

      $submit_label = __( 'Add' );

      if( $this->editing ) {
          $submit_label = __( 'Update' );
      }

      /**
       * Filter to override the submit button label.
       *
       * @since 1.0.0
       *
       * @param string        $submit_label   The submit label.
       * @param object        $object         Object.
       * @param TDB_Table      $tdb_table       TDB Table object.
       * @param bool          $editing        True if edit screen, false if is adding a new one.
       * @param TDB_Edit_View  $view           Edit view object.
       */
      $submit_label = apply_filters( "tdb_{$tdb_table->name}_edit_screen_submit_label", $submit_label, $object, $tdb_table, $this->editing, $this );

      $primary_key = $tdb_table->db->primary_key;
      $object_id = $object->$primary_key;

      ?>

      <div class="submitbox" id="submitpost">

          <?php
          /**
           * Fires at top of submit meta box submit post element.
           *
           * @since 1.0.0
           *
           * @param object        $object         Object.
           * @param TDB_Table      $tdb_table       TDB Table object.
           * @param bool          $editing        True if edit screen, false if is adding a new one.
           * @param TDB_Edit_View  $view           Edit view object.
           */
          do_action( "tdb_{$tdb_table->name}_edit_screen_submit_meta_box_submit_post_top", $object, $tdb_table, $this->editing, $this ); ?>

          <div id="minor-publishing">

              <?php ob_start();
              /**
               * Fires inside minor publishing actions from submit meta box.
               *
               * @since 1.0.0
               *
               * @param object        $object         Object.
               * @param TDB_Table      $tdb_table       TDB Table object.
               * @param bool          $editing        True if edit screen, false if is adding a new one.
               * @param TDB_Edit_View  $view           Edit view object.
               */
              do_action( "tdb_{$tdb_table->name}_edit_screen_submit_meta_box_minor_publishing_actions", $object, $tdb_table, $this->editing, $this );
              $minor_publishing_actions = ob_get_clean(); ?>

              <?php // Since minor-publishing-actions has a margin, check if minor publishing actions has any content to render it or not
              if( ! empty( $minor_publishing_actions ) ) : ?>
                  <div id="minor-publishing-actions"><?php echo $minor_publishing_actions; ?></div>
              <?php endif; ?>

              <?php ob_start();
              /**
               * Fires inside misc publishing actions from submit meta box.
               *
               * @since 1.0.0
               *
               * @param object        $object         Object.
               * @param TDB_Table      $tdb_table       TDB Table object.
               * @param bool          $editing        True if edit screen, false if is adding a new one.
               * @param TDB_Edit_View  $view           Edit view object.
               */
              do_action( "tdb_{$tdb_table->name}_edit_screen_submit_meta_box_misc_publishing_actions", $object, $tdb_table, $this->editing, $this );
              $misc_publishing_actions = ob_get_clean(); ?>

              <?php // Since misc-publishing-actions has a margin, check if misc publishing actions has any content to render it or not
              if( ! empty( $misc_publishing_actions ) ) : ?>
                  <div id="misc-publishing-actions"><?php echo $misc_publishing_actions; ?></div>
              <?php endif; ?>

              <div class="clear"></div>

          </div>

          <div id="major-publishing-actions">

              <?php
              if ( current_user_can( $tdb_table->cap->delete_item, $object_id ) ) {

                  printf(
                      '<a href="%s" class="submitdelete deletion" onclick="%s" aria-label="%s">%s</a>',
                      tdb_get_delete_link( $tdb_table->name, $object_id ),
                      "return confirm('" .
                      esc_attr( __( "Are you sure you want to delete this item?\\n\\nClick \\'Cancel\\' to go back, \\'OK\\' to confirm the delete." ) ) .
                      "');",
                      esc_attr( __( 'Delete permanently' ) ),
                      __( 'Delete Permanently' )
                  );

              } ?>

              <div id="publishing-action">
                  <span class="spinner"></span>
                  <?php submit_button( $submit_label, 'primary large', 'tdb-save', false ); ?>
              </div>

              <div class="clear"></div>

          </div>

          <?php
          /**
           * Fires at bottom of submit meta box submit post element.
           *
           * @since 1.0.0
           *
           * @param object        $object         Object.
           * @param TDB_Table      $tdb_table       TDB Table object.
           * @param bool          $editing        True if edit screen, false if is adding a new one.
           * @param TDB_Edit_View  $view           Edit view object.
           */
          do_action( "tdb_{$tdb_table->name}_edit_screen_submit_meta_box_submit_post_bottom", $object, $tdb_table, $this->editing, $this ); ?>

      </div>

      <?php

      /**
       * Fires at bottom the submit meta box.
       *
       * @since 1.0.0
       *
       * @param object        $object         Object.
       * @param TDB_Table      $tdb_table       TDB Table object.
       * @param bool          $editing        True if edit screen, false if is adding a new one.
       * @param TDB_Edit_View  $view           Edit view object.
       */
      do_action( "tdb_{$tdb_table->name}_edit_screen_submit_meta_box_bottom", $object, $tdb_table, $this->editing, $this );
  }

  public function save() {

      global $tdb_registered_tables, $tdb_table;

      // If not TDB object, die
      if ( ! $tdb_table )
          wp_die( __( 'Invalid item type.' ) );

      // If not TDB object allow ui, die
      if ( ! $tdb_table->show_ui ) {
          wp_die( __( 'Sorry, you are not allowed to edit items of this type.' ) );
      }

      $primary_key = $tdb_table->db->primary_key;

      if( ! isset( $_POST[$primary_key] ) ) {
          wp_die( __( 'Invalid item type.' ) );
      }

      $object_id = $_POST[$primary_key];

      // Nonce check
      if ( ! isset( $_REQUEST['_wpnonce'] ) ) {
          wp_die( __( 'Sorry, you are not allowed to edit this item.' ) );
      }

      if ( ! wp_verify_nonce( $_REQUEST['_wpnonce'], 'tdb_edit_' . $object_id ) ) {
          wp_die( __( 'Sorry, you are not allowed to edit this item.' ) );
      }

      $object_data = &$_POST;

      unset( $object_data['tdb-save'] );

      $success = tdb_update_object( $object_data );

      $location = add_query_arg( array( $primary_key => $object_id ), $this->get_link() );

      if( $success ) {
          $location = add_query_arg( array( 'message' => 1 ), $location );
      } else {
          $location = add_query_arg( array( 'message' => 0 ), $location );
      }

      wp_redirect( $location );
      exit;

  }

  public function pre_render() {

      global $tdb_registered_tables, $tdb_table;

      $messages = array(
          0 => __( '%s could not be updated.' ),
          1 => __( '%s updated successfully.' ),
      );

      /**
       * Filters the table updated messages (string keys allowed!).
       *
       * @since 1.0.0
       *
       * @param array $messages Post updated messages. For defaults @see $messages declarations above.
       */
      $messages = apply_filters( 'tdb_table_updated_messages', $messages );

      // Setup screen message
      if ( isset($_GET['message']) ) {

          if ( isset($messages[$_GET['message']]) )
              $this->message = sprintf( $messages[$_GET['message']], $tdb_table->labels->singular_name );

      }

      wp_enqueue_script( 'post' );

      if ( wp_is_mobile() ) {
          wp_enqueue_script( 'jquery-touch-punch' );
      }

      // Register submitdiv metabox
      add_meta_box( 'submitdiv', __( 'Save Changes' ), array( $this, 'submit_meta_box' ), $tdb_table->name, 'side', 'core' );

      /**
       * Fires after all built-in meta boxes have been added.
       *
       * @since 1.0.0
       *
       * @param string  $post_type Post type.
       * @param WP_Post $post      Post object.
       */
      do_action( 'add_meta_boxes', $tdb_table->name, $this->object );

      /**
       * Fires after all built-in meta boxes have been added, contextually for the given post type.
       *
       * The dynamic portion of the hook, `$post_type`, refers to the post type of the post.
       *
       * @since 1.0.0
       *
       * @param WP_Post $post Post object.
       */
      do_action( "add_meta_boxes_{$tdb_table->name}", $this->object );

      /**
       * Fires after meta boxes have been added.
       *
       * Fires once for each of the default meta box contexts: normal, advanced, and side.
       *
       * @since 1.0.0
       *
       * @param string  $post_type Type of the object.
       * @param string  $context   string  Meta box context.
       * @param WP_Post $object    The object.
       */
      do_action( 'do_meta_boxes', $tdb_table->name, 'normal', $this->object );
      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( 'do_meta_boxes', $tdb_table->name, 'advanced', $this->object );
      /** This action is documented in wp-admin/edit-form-advanced.php */
      do_action( 'do_meta_boxes', $tdb_table->name, 'side', $this->object );

      // TODO: Need to add it manually through screen_settings() function
      //add_screen_option( 'layout_columns', array( 'max' => $this->columns, 'default' => $this->columns ) );

  }

  public function render() {

    global $tdb_registered_tables, $tdb_table;

    $this->pre_render();

    if( $this->editing ) {
      $title = $tdb_table->labels->edit_item;
      $new_url = ( $tdb_table->views->add ? $tdb_table->views->add->get_link() : false );
    } else {
      $title = $tdb_table->labels->add_new_item;
    }

    ?>
    <div class="wrap">

      <h1 class="wp-heading-inline"><?php echo $title; ?></h1>

      <?php if ( isset( $new_url ) && $new_url && current_user_can( $tdb_table->cap->create_items ) ) :
          echo ' <a href="' . esc_url( $new_url ) . '" class="page-title-action">' . esc_html( $tdb_table->labels->add_new_item ) . '</a>';
      endif; ?>

      <hr class="wp-header-end">

      <?php if ( $this->message ) : ?>
          <div id="message" class="updated notice notice-success is-dismissible"><p><?php echo $this->message; ?></p></div>
      <?php endif; ?>

      <form name="tdb_edit_form" action="" method="post" id="tdb_edit_form">

        <input type="hidden" id="object_id" name="<?php echo $tdb_table->db->primary_key; ?>" value="<?php echo $this->object_id; ?>">
        <?php wp_nonce_field( 'tdb_edit_' . $this->object_id ); ?>

        <?php
        /**
         * Fires at the beginning of the edit form.
         *
         * At this point, the required hidden fields and nonces have already been output.
         *
         * @since 1.0.0
         *
         * @param stdClass $object Object.
         */
        do_action( 'tdb_edit_form_top', $this->object ); ?>

        <div id="poststuff">

          <div id="post-body" class="metabox-holder columns-<?php echo get_current_screen()->get_columns() === 1 || $this->columns === 1 ? '1' : '2'; ?>">

            <div id="postbox-container-1" class="postbox-container">
              <?php do_meta_boxes( $tdb_table->name, 'side', $this->object ); ?>
            </div>

            <div id="postbox-container-2" class="postbox-container">
              <?php do_meta_boxes( $tdb_table->name, 'normal', $this->object ); ?>
              <?php do_meta_boxes( $tdb_table->name, 'advanced', $this->object ); ?>
            </div>
          </div><!-- /post-body -->
          <br class="clear" />
        </div><!-- /poststuff -->
        <?php
        /**
         * Fires at the end of the edit form.
         *
         * @since 1.0.0
         *
         * @param stdClass $object Object.
         */
        do_action( 'tdb_edit_form_bottom', $this->object );

        ?>
      </form>
    </div>
    <?php
  }
}