ultimatemember/includes/core/class-setup.php

<?php
namespace um\core;

// Exit if accessed directly
if ( ! defined( 'ABSPATH' ) ) exit;

if ( ! class_exists( 'Setup' ) ) {
    class Setup {

        var $setup_shortcode = array();

        function __construct() {

            //add_action('init',  array(&$this, 'install_basics'), 9);


        }


        function run_setup() {
            $this->install_basics();
            $this->install_default_forms();
            $this->set_default_settings();
            $this->set_default_role_meta();
        }


        /***
         ***	@Basics
         ***/
        function install_basics() {
            if ( ! get_option( '__ultimatemember_sitekey' ) )
                update_option( '__ultimatemember_sitekey', str_replace( array( 'http://', 'https://' ), '', sanitize_user( get_bloginfo('url') ) ) . '-' . wp_generate_password( 20, false ) );
        }


        /***
         ***	@Default Forms
         ***/
        function install_default_forms() {

            $options = get_option( 'um_options' );
            $options = empty( $options ) ? array() : $options;

            if ( current_user_can( 'manage_options' ) && ! get_option( 'um_is_installed' ) ) {

                update_option( 'um_is_installed', 1 );

                //Install default options
                foreach ( UM()->config()->settings_defaults as $key => $value ) {
                    $options[$key] = $value;
                }

                // Install Core Forms
                foreach ( UM()->config()->core_forms as $id ) {

                    /**
                    If page does not exist
                    Create it
                     **/
                    $page_exists = UM()->query()->find_post_id( 'um_form', '_um_core', $id );
                    if ( ! $page_exists ) {

                        if ( $id == 'register' ) {
                            $title = 'Default Registration';
                        } else if ( $id == 'login' ) {
                            $title = 'Default Login';
                        } else {
                            $title = 'Default Profile';
                        }

                        $form = array(
                            'post_type' 	  	=> 'um_form',
                            'post_title'		=> $title,
                            'post_status'		=> 'publish',
                            'post_author'   	=> get_current_user_id(),
                        );

                        $form_id = wp_insert_post( $form );

                        foreach( UM()->config()->core_form_meta[$id] as $key => $value ) {
                            if ( $key == '_um_custom_fields' ) {
                                $array = unserialize( $value );
                                update_post_meta( $form_id, $key, $array );
                            } else {
                                update_post_meta( $form_id, $key, $value );
                            }
                        }

                        $this->setup_shortcode[$id] = '[ultimatemember form_id='.$form_id.']';

                        $core_forms[ $form_id ] = $form_id;

                    }
                    /** DONE **/

                }

                if ( isset( $core_forms ) )
                    update_option( 'um_core_forms', $core_forms );

                // Install Core Directories
                foreach ( UM()->config()->core_directories as $id ) {

                    /**
                    If page does not exist
                    Create it
                     **/
                    $page_exists = UM()->query()->find_post_id( 'um_directory', '_um_core', $id );
                    if ( ! $page_exists ) {

                        $title = 'Members';

                        $form = array(
                            'post_type' 	  	=> 'um_directory',
                            'post_title'		=> $title,
                            'post_status'		=> 'publish',
                            'post_author'   	=> get_current_user_id(),
                        );

                        $form_id = wp_insert_post( $form );

                        foreach ( UM()->config()->core_directory_meta[$id] as $key => $value ) {
                            if ( $key == '_um_custom_fields' ) {
                                $array = unserialize( $value );
                                update_post_meta( $form_id, $key, $array );
                            } else {
                                update_post_meta($form_id, $key, $value);
                            }
                        }

                        $this->setup_shortcode[$id] = '[ultimatemember form_id='.$form_id.']';

                        $core_directories[ $form_id ] = $form_id;

                    }
                    /** DONE **/

                }

                if ( isset( $core_directories ) ) update_option( 'um_core_directories', $core_directories );


                // Install Core Pages
                $core_pages = array();
                foreach ( UM()->config()->core_pages as $slug => $array ) {

                    /**
                    If page does not exist
                    Create it
                     **/
                    $page_exists = UM()->query()->find_post_id( 'page', '_um_core', $slug );
                    if ( ! $page_exists ) {

                        if ( $slug == 'logout' ) {
                            $content = '';
                        } else if ( $slug == 'account' ) {
                            $content = '[ultimatemember_account]';
                        } else if ( $slug == 'password-reset' ) {
                            $content = '[ultimatemember_password]';
                        } else if ( $slug == 'user' ){
                            $content = $this->setup_shortcode['profile'];
                        } else {
                            $content = $this->setup_shortcode[$slug];
                        }

                        $user_page = array(
                            'post_title'		=> $array['title'],
                            'post_content'		=> $content,
                            'post_name'			=> $slug,
                            'post_type' 	  	=> 'post',
                            'post_status'		=> 'publish',
                            'post_author'   	=> get_current_user_id(),
                            'comment_status'    => 'closed'
                        );

                        $post_id = wp_insert_post( $user_page );
                        wp_update_post( array( 'ID' => $post_id, 'post_type' => 'page' ) );

                        update_post_meta( $post_id, '_um_core', $slug );

                        $core_pages[ $slug ] = $post_id;

                    } else {
                        $core_pages[ $slug ] = $page_exists;
                    }
                    /** DONE **/
                }

                foreach ( $core_pages as $slug => $page_id ) {
	                $key = UM()->options()->get_core_page_id( $slug );
                    $options[ $key ] = $page_id;
                }
            }

            update_option( 'um_options', $options );
        }


        /**
         * Set default UM settings
         */
        function set_default_settings() {
            $options = get_option( 'um_options' );
            $options = empty( $options ) ? array() : $options;

            foreach ( UM()->config()->settings_defaults as $key => $value ) {
                //set new options to default
                if ( ! isset( $options[$key] ) )
                    $options[$key] = $value;
            }

            update_option( 'um_options', $options );
        }


        /**
         * Set UM roles meta to Default WP roles
         */
        function set_default_role_meta() {

			//for set accounts without account status approved status
            UM()->query()->count_users_by_status( 'unassigned' );

            foreach ( UM()->config()->default_roles_metadata as $role => $meta ) {
                update_option( "um_role_{$role}_meta", $meta );
            }
        }
    }
}