1: <?php
  2: 
  3: /**
  4:  * User property that handle relationship between
  5:  * users and posts.
  6:  */
  7: class Papi_Property_User extends Papi_Property_Dropdown {
  8: 
  9:     /**
 10:      * The convert type.
 11:      *
 12:      * @var string
 13:      */
 14:     public $convert_type = 'int';
 15: 
 16:     /**
 17:      * Format the value of the property before it's returned
 18:      * to WordPress admin or the site.
 19:      *
 20:      * @param  mixed  $value
 21:      * @param  string $slug
 22:      * @param  int    $post_id
 23:      *
 24:      * @return array
 25:      */
 26:     public function format_value( $value, $slug, $post_id ) {
 27:         if ( is_object( $value ) && isset( $value->ID ) ) {
 28:             $value = $value->ID;
 29:         }
 30: 
 31:         if ( is_numeric( $value ) ) {
 32:             return new WP_User( $value );
 33:         }
 34: 
 35:         return $value;
 36:     }
 37: 
 38:     /**
 39:      * Get default settings.
 40:      *
 41:      * @return array
 42:      */
 43:     public function get_default_settings() {
 44:         return [
 45:             'capabilities' => [],
 46:             'placeholder'  => '',
 47:             'select2'      => true
 48:         ];
 49:     }
 50: 
 51:     /**
 52:      * Get value from database.
 53:      *
 54:      * @return int
 55:      */
 56:     public function get_value() {
 57:         $user = parent::get_value();
 58: 
 59:         if ( is_object( $user ) && isset( $user->ID ) ) {
 60:             return $user->ID;
 61:         }
 62: 
 63:         return 0;
 64:     }
 65: 
 66:     /**
 67:      * Get users as dropdown items.
 68:      *
 69:      * @return array
 70:      */
 71:     public function get_items() {
 72:         $capabilities = papi_to_array( $this->get_setting( 'capabilities' ) );
 73:         $users        = get_users();
 74:         $items        = [];
 75: 
 76:         foreach ( $users as $user ) {
 77:             $allcaps = $user->allcaps;
 78: 
 79:             if ( count( array_diff( $capabilities, array_keys( $allcaps ) ) ) === 0 ) {
 80:                 $items[$user->display_name] = $user->ID;
 81:             }
 82:         }
 83: 
 84:         return $items;
 85:     }
 86: 
 87:     /**
 88:      * Import value to the property.
 89:      *
 90:      * @param  mixed  $value
 91:      * @param  string $slug
 92:      * @param  int    $post_id
 93:      *
 94:      * @return int|null
 95:      */
 96:     public function import_value( $value, $slug, $post_id ) {
 97:         if ( $value instanceof WP_User ) {
 98:             $value = $value->ID;
 99:         }
100: 
101:         if ( is_numeric( $value ) ) {
102:             return (int) $value;
103:         }
104: 
105:         return $this->default_value;
106:     }
107: 
108:     /**
109:      * Change value after it's loaded from the database.
110:      *
111:      * @param  mixed  $value
112:      * @param  string $slug
113:      * @param  int    $post_id
114:      *
115:      * @return int
116:      */
117:     public function load_value( $value, $slug, $post_id ) {
118:         return (int) $value;
119:     }
120: 
121:     /**
122:      * Update value before it's saved to the database.
123:      *
124:      * @param mixed  $value
125:      * @param string $slug
126:      * @param int    $post_id
127:      *
128:      * @return int
129:      */
130:     public function update_value( $value, $slug, $post_id ) {
131:         if ( $value instanceof WP_User ) {
132:             $value = $value->ID;
133:         }
134: 
135:         return (int) $value;
136:     }
137: }
138: