PartnerExpo-Core/includes/class-partnerexpo-core-user-fields.php

<?php

/**
 * Used to register custom user fields.
*
* @link       https://github.com/Duskell
* @since      1.1.0
* @package    Partnerexpo_Core
* @subpackage Partnerexpo_Core/includes
* @author     Juhász Levente <juhasz.levente@rendszerepito.hu>
*/
class Partnerexpo_Core_User_Fields extends Partnerexpo_Core {
    protected $loader;

    public function __construct( $loader ) {
        $this->loader = $loader;
        $this->activate_fields();
    }

    private function activate_fields() {
        $this->loader->add_action( 'show_user_profile', $this, 'add_custom_user_fields' );
        $this->loader->add_action( 'edit_user_profile', $this, 'add_custom_user_fields' );
        $this->loader->add_action( 'user_new_form', $this, 'add_user_new_fields' );
        $this->loader->add_action( 'personal_options_update', $this, 'save_custom_user_fields' );
        $this->loader->add_action( 'edit_user_profile_update', $this, 'save_custom_user_fields' );
        $this->loader->add_action( 'user_register', $this, 'add_value_after_reg' );
        $this->loader->add_action( 'register_form', $this, 'registration_form' );
        $this->loader->add_filter( 'registration_errors', $this, 'registration_errors', 10, 3 );
        $this->loader->add_filter( 'comment_post', $this, 'validate_comment' );

        $this->loader->add_action( 'init', $this, 'remove_filters' );

    }

    public function remove_filters() {
        remove_all_filters('comment_form_defaults');
        remove_all_filters('comment_form_top');
        remove_all_filters('comment_form_before');
        remove_all_filters('comment_form_after');
    }

    public function add_user_new_fields() {
        wp_enqueue_media();
        ?>
        <h3>Additional Information</h3>
        <table class="form-table">
            <tr>
                <th><label for="company-logo"><?php _e( 'Céges logó', 'partnerexpo-core' ); ?></label></th>
                <td>
                    <div class='pexpo-core-business-logo-preview-wrapper'>
                        <img id='pexpo-core-business-logo-preview' src='' height='100'>
                    </div>

                    <input id="pexpo-core-upload-image-button"
                        type="button"
                        class="button"
                        value="<?php _e( 'Kép feltöltése', 'partnerexpo-core' ); ?>" />

                    <input type='hidden'
                        name='image_attachment_id'
                        id='pexpo-core-business-logo-attachment-id'
                        value='' />
                </td>
            </tr>

            <tr>
                <th><label for="company"><?php _e( 'Cég neve', 'partnerexpo-core' ); ?></label></th>
                <td>
                    <input type="text"
                        name="company"
                        id="company"
                        value=""
                        class="regular-text" />
                </td>
            </tr>
        </table>
        <?php
    }


    public function add_custom_user_fields( $user ) {
        $company = get_user_meta( $user->ID, 'company', true );
        $logo_id = get_user_meta( $user->ID, 'company_logo_attachment_id', true );
        $logo_url = $logo_id ? wp_get_attachment_url( $logo_id ) : '';
        wp_enqueue_media();
        ?>
            <h3>Additional Information</h3>
            <table class="form-table">  
                <tr>
                    <th><label for="company-logo"><?php _e( 'Céges logó', 'partnerexpo-core' ) ?></label></th>
                    <td>
                        <div class='pexpo-core-business-logo-preview-wrapper'>
                            <img id='pexpo-core-business-logo-preview' src='<?php echo esc_url( $logo_url ); ?>' height='100'>
                        </div>
                        <input id="pexpo-core-upload-image-button" type="button" class="button" value="<?php _e( 'Kép feltöltése', 'partnerexpo-core' ) ?>" />
                        <input type='hidden' name='image_attachment_id' id='pexpo-core-business-logo-attachment-id' value='<?php echo esc_attr( $logo_id ); ?>'>
                    </td>
                </tr>
                <tr>
                    <th><label for="company"><?php _e( 'Cég neve', 'partnerexpo-core' ) ?></label></th>
                    <td>
                        <input type="text" name="company" id="company" value="<?php echo esc_attr( $company ) ?>" class="regular-text" />
                    </td>
                </tr>
            </table>
        <?php
    }

    public function save_custom_user_fields( $user_id ) {
        if( ! isset( $_POST[ '_wpnonce' ] ) || ! wp_verify_nonce( $_POST[ '_wpnonce' ], 'update-user_' . $user_id ) ) {
            return;
        }
        
        if( ! current_user_can( 'edit_user', $user_id ) ) {
            return;
        }
        
        if ( isset($_POST['company']) ) {
            update_user_meta(
                $user_id,
                'company',
                sanitize_text_field($_POST['company'])
            );
        }
        if ( isset($_POST['image_attachment_id']) ) {
            update_user_meta(
                $user_id,
                'company_logo_attachment_id',
                absint($_POST['image_attachment_id'])
            );
        }
    }

    public function add_value_after_reg( $user_id ) {

        if ( ! empty( $_POST['company'] ) ) {
            update_user_meta(
                $user_id,
                'company',
                sanitize_text_field( $_POST['company'] )
            );
        }

        if ( ! empty( $_POST['image_attachment_id'] ) ) {
            update_user_meta(
                $user_id,
                'company_logo_attachment_id',
                absint( $_POST['image_attachment_id'] )
            );
        }
    }

    function registration_form() {
        ?>
        <p>
            <label for="company"><?php _e( 'Cég neve', 'partnerexpo-core' ) ?><br />
            <input type="text" name="company" id="company" class="input" value="" size="25" /></label>
        </p>
        <?php
    }

    function registration_errors( $errors, $sanitized_user_login, $user_email ) {
        if ( empty( $_POST['company'] ) || ! empty( $_POST['company'] ) && trim( $_POST['company'] ) == '' ) {
            $errors->add( 'company_error', sprintf('<strong>%s</strong>: %s',__( 'Error', 'partnerexpo-core' ),__( 'Cég megadása kötelező!', 'partnerexpo-core' ) ) );  
        }

        return $errors;
    }

    function validate_comment( $id ) {
        if ( empty( $_POST['company'] ) || ! empty( $_POST['company'] ) && trim( $_POST['company'] ) == '' ) {
            wp_delete_comment( $id, true );
            wp_die( sprintf('<strong>%s</strong>: %s',__( 'Error', 'partnerexpo-core' ),__( 'Cég megadása kötelező!', 'partnerexpo-core' ) ) );
            return false;
        }
        add_comment_meta(
            $id,
            'company',
            sanitize_text_field($_POST['company'])
        );

        if ( ! empty( $_POST['name'] ) && trim( $_POST['name'] ) != '' ) {
            add_comment_meta(
                $id,
                'name',
                sanitize_text_field($_POST['name'])
            );
        }
        
        return true;
    }
}