1 <?php
  2 /**
  3  * Add extra profile fields for users in admin.
  4  *
  5  * @author      WooThemes
  6  * @category    Admin
  7  * @package     WooCommerce/Admin
  8  * @version     2.1.0
  9  */
 10 
 11 if ( ! defined( 'ABSPATH' ) ) exit; // Exit if accessed directly
 12 
 13 if ( ! class_exists( 'WC_Admin_Profile' ) ) :
 14 
 15 /**
 16  * WC_Admin_Profile Class
 17  */
 18 class WC_Admin_Profile {
 19 
 20     /**
 21      * Hook in tabs.
 22      */
 23     public function __construct() {
 24         add_action( 'show_user_profile', array( $this, 'add_customer_meta_fields' ) );
 25         add_action( 'edit_user_profile', array( $this, 'add_customer_meta_fields' ) );
 26 
 27         add_action( 'personal_options_update', array( $this, 'save_customer_meta_fields' ) );
 28         add_action( 'edit_user_profile_update', array( $this, 'save_customer_meta_fields' ) );
 29 
 30         add_action( 'show_user_profile', array( $this, 'add_api_key_field' ) );
 31         add_action( 'edit_user_profile', array( $this, 'add_api_key_field' ) );
 32 
 33         add_action( 'personal_options_update', array( $this, 'generate_api_key' ) );
 34         add_action( 'edit_user_profile_update', array( $this, 'generate_api_key' ) );
 35     }
 36 
 37     /**
 38      * Get Address Fields for the edit user pages.
 39      *
 40      * @return array Fields to display which are filtered through woocommerce_customer_meta_fields before being returned
 41      */
 42     public function get_customer_meta_fields() {
 43         $show_fields = apply_filters('woocommerce_customer_meta_fields', array(
 44             'billing' => array(
 45                 'title' => __( 'Customer Billing Address', 'woocommerce' ),
 46                 'fields' => array(
 47                     'billing_first_name' => array(
 48                             'label' => __( 'First name', 'woocommerce' ),
 49                             'description' => ''
 50                         ),
 51                     'billing_last_name' => array(
 52                             'label' => __( 'Last name', 'woocommerce' ),
 53                             'description' => ''
 54                         ),
 55                     'billing_company' => array(
 56                             'label' => __( 'Company', 'woocommerce' ),
 57                             'description' => ''
 58                         ),
 59                     'billing_address_1' => array(
 60                             'label' => __( 'Address 1', 'woocommerce' ),
 61                             'description' => ''
 62                         ),
 63                     'billing_address_2' => array(
 64                             'label' => __( 'Address 2', 'woocommerce' ),
 65                             'description' => ''
 66                         ),
 67                     'billing_city' => array(
 68                             'label' => __( 'City', 'woocommerce' ),
 69                             'description' => ''
 70                         ),
 71                     'billing_postcode' => array(
 72                             'label' => __( 'Postcode', 'woocommerce' ),
 73                             'description' => ''
 74                         ),
 75                     'billing_state' => array(
 76                             'label' => __( 'State/County', 'woocommerce' ),
 77                             'description' => __( 'State/County or state code', 'woocommerce' ),
 78                         ),
 79                     'billing_country' => array(
 80                             'label' => __( 'Country', 'woocommerce' ),
 81                             'description' => __( '2 letter Country code', 'woocommerce' ),
 82                         ),
 83                     'billing_phone' => array(
 84                             'label' => __( 'Telephone', 'woocommerce' ),
 85                             'description' => ''
 86                         ),
 87                     'billing_email' => array(
 88                             'label' => __( 'Email', 'woocommerce' ),
 89                             'description' => ''
 90                         )
 91                 )
 92             ),
 93             'shipping' => array(
 94                 'title' => __( 'Customer Shipping Address', 'woocommerce' ),
 95                 'fields' => array(
 96                     'shipping_first_name' => array(
 97                             'label' => __( 'First name', 'woocommerce' ),
 98                             'description' => ''
 99                         ),
100                     'shipping_last_name' => array(
101                             'label' => __( 'Last name', 'woocommerce' ),
102                             'description' => ''
103                         ),
104                     'shipping_company' => array(
105                             'label' => __( 'Company', 'woocommerce' ),
106                             'description' => ''
107                         ),
108                     'shipping_address_1' => array(
109                             'label' => __( 'Address 1', 'woocommerce' ),
110                             'description' => ''
111                         ),
112                     'shipping_address_2' => array(
113                             'label' => __( 'Address 2', 'woocommerce' ),
114                             'description' => ''
115                         ),
116                     'shipping_city' => array(
117                             'label' => __( 'City', 'woocommerce' ),
118                             'description' => ''
119                         ),
120                     'shipping_postcode' => array(
121                             'label' => __( 'Postcode', 'woocommerce' ),
122                             'description' => ''
123                         ),
124                     'shipping_state' => array(
125                             'label' => __( 'State/County', 'woocommerce' ),
126                             'description' => __( 'State/County or state code', 'woocommerce' )
127                         ),
128                     'shipping_country' => array(
129                             'label' => __( 'Country', 'woocommerce' ),
130                             'description' => __( '2 letter Country code', 'woocommerce' )
131                         )
132                 )
133             )
134         ));
135         return $show_fields;
136     }
137 
138     /**
139      * Show Address Fields on edit user pages.
140      *
141      * @param mixed $user User (object) being displayed
142      */
143     public function add_customer_meta_fields( $user ) {
144         if ( ! current_user_can( 'manage_woocommerce' ) )
145             return;
146 
147         $show_fields = $this->get_customer_meta_fields();
148 
149         foreach( $show_fields as $fieldset ) :
150             ?>
151             <h3><?php echo $fieldset['title']; ?></h3>
152             <table class="form-table">
153                 <?php
154                 foreach( $fieldset['fields'] as $key => $field ) :
155                     ?>
156                     <tr>
157                         <th><label for="<?php echo esc_attr( $key ); ?>"><?php echo esc_html( $field['label'] ); ?></label></th>
158                         <td>
159                             <input type="text" name="<?php echo esc_attr( $key ); ?>" id="<?php echo esc_attr( $key ); ?>" value="<?php echo esc_attr( get_user_meta( $user->ID, $key, true ) ); ?>" class="regular-text" /><br/>
160                             <span class="description"><?php echo wp_kses_post( $field['description'] ); ?></span>
161                         </td>
162                     </tr>
163                     <?php
164                 endforeach;
165                 ?>
166             </table>
167             <?php
168         endforeach;
169     }
170 
171     /**
172      * Save Address Fields on edit user pages
173      *
174      * @param mixed $user_id User ID of the user being saved
175      */
176     public function save_customer_meta_fields( $user_id ) {
177         $save_fields = $this->get_customer_meta_fields();
178 
179         foreach( $save_fields as $fieldset )
180             foreach( $fieldset['fields'] as $key => $field )
181                 if ( isset( $_POST[ $key ] ) )
182                     update_user_meta( $user_id, $key, wc_clean( $_POST[ $key ] ) );
183     }
184 
185     /**
186      * Display the API key info for a user
187      *
188      * @since 2.1
189      * @param WP_User $user
190      */
191     public function add_api_key_field( $user ) {
192 
193         if ( ! current_user_can( 'manage_woocommerce' ) )
194             return;
195 
196         $permissions = array(
197             'read'       => __( 'Read', 'woocommerce' ),
198             'write'      => __( 'Write', 'woocommerce' ),
199             'read_write' => __( 'Read/Write', 'woocommerce' ),
200         );
201 
202         if ( current_user_can( 'edit_user', $user->ID ) ) {
203             ?>
204                 <table class="form-table">
205                     <tbody>
206                         <tr>
207                             <th><label for="woocommerce_api_keys"><?php _e( 'WooCommerce API Keys', 'woocommerce' ); ?></label></th>
208                             <td>
209                                 <?php if ( empty( $user->woocommerce_api_consumer_key ) ) : ?>
210                                     <input name="woocommerce_generate_api_key" type="checkbox" id="woocommerce_generate_api_key" value="0" />
211                                     <span class="description"><?php _e( 'Generate API Key', 'woocommerce' ); ?></span>
212                                 <?php else : ?>
213                                     <strong><?php _e( 'Consumer Key:', 'woocommerce' ); ?>&nbsp;</strong><code id="woocommerce_api_consumer_key"><?php echo $user->woocommerce_api_consumer_key ?></code><br/>
214                                     <strong><?php _e( 'Consumer Secret:', 'woocommerce' ); ?>&nbsp;</strong><code id="woocommerce_api_consumer_secret"><?php echo $user->woocommerce_api_consumer_secret; ?></code><br/>
215                                     <strong><?php _e( 'Permissions:', 'woocommerce' ); ?>&nbsp;</strong><span id="woocommerce_api_key_permissions"><select name="woocommerce_api_key_permissions" id="woocommerce_api_key_permissions"><?php
216                                         foreach ( $permissions as $permission_key => $permission_name ) { echo '<option value="' . esc_attr( $permission_key ) . '" '.selected($permission_key, $user->woocommerce_api_key_permissions, false).'>'.esc_html( $permission_name ) . '</option>';} ?>
217                                     </select></span><br/>
218                                     <input name="woocommerce_generate_api_key" type="checkbox" id="woocommerce_generate_api_key" value="0" />
219                                     <span class="description"><?php _e( 'Revoke API Key', 'woocommerce' ); ?></span>
220                                 <?php endif; ?>
221                             </td>
222                         </tr>
223                     </tbody>
224                 </table>
225             <?php
226         }
227     }
228 
229     /**
230      * Generate and save (or delete) the API keys for a user
231      *
232      * @since 2.1
233      * @param int $user_id
234      */
235     public function generate_api_key( $user_id ) {
236 
237         if ( current_user_can( 'edit_user', $user_id ) ) {
238 
239             $user = get_userdata( $user_id );
240 
241             // creating/deleting key
242             if ( isset( $_POST['woocommerce_generate_api_key'] ) ) {
243 
244                 // consumer key
245                 if ( empty( $user->woocommerce_api_consumer_key ) ) {
246 
247                     $consumer_key = 'ck_' . hash( 'md5', $user->user_login . date( 'U' ) . mt_rand() );
248 
249                     update_user_meta( $user_id, 'woocommerce_api_consumer_key', $consumer_key );
250 
251                 } else {
252 
253                     delete_user_meta( $user_id, 'woocommerce_api_consumer_key' );
254                 }
255 
256                 // consumer secret
257                 if ( empty( $user->woocommerce_api_consumer_secret ) ) {
258 
259                     $consumer_secret = 'cs_' . hash( 'md5', $user->ID . date( 'U' ) . mt_rand() );
260 
261                     update_user_meta( $user_id, 'woocommerce_api_consumer_secret', $consumer_secret );
262 
263                 } else {
264 
265                     delete_user_meta( $user_id, 'woocommerce_api_consumer_secret' );
266                 }
267 
268                 // permissions
269                 if ( empty( $user->woocommerce_api_key_permissions ) ) {
270 
271                     if ( isset( $_POST['woocommerce_api_key_permissions'] ) ) {
272 
273                         $permissions = ( in_array( $_POST['woocommerce_api_key_permissions'], array( 'read', 'write', 'read_write' ) ) ) ? $_POST['woocommerce_api_key_permissions'] : 'read';
274 
275                     } else {
276 
277                         $permissions = 'read';
278                     }
279 
280                     update_user_meta( $user_id, 'woocommerce_api_key_permissions', $permissions );
281 
282                 } else {
283 
284                     delete_user_meta( $user_id, 'woocommerce_api_key_permissions' );
285                 }
286 
287             } else {
288 
289                 // updating permissions for key
290                 if ( ! empty( $_POST['woocommerce_api_key_permissions'] ) && $user->woocommerce_api_key_permissions !== $_POST['woocommerce_api_key_permissions'] ) {
291 
292                     $permissions = ( ! in_array( $_POST['woocommerce_api_key_permissions'], array( 'read', 'write', 'read_write' ) ) ) ? 'read' : $_POST['woocommerce_api_key_permissions'];
293 
294                     update_user_meta( $user_id, 'woocommerce_api_key_permissions', $permissions );
295                 }
296             }
297         }
298     }
299 
300 }
301 
302 endif;
303 
304 return new WC_Admin_Profile();
305 
WooCommerce API documentation generated by ApiGen 2.8.0