Discuz3.3注册程序修改添加记录推荐人账号

Discuz3.3注册入口地址为：member.php?mod=register

一.member.php：

打开之后，代码非常简单。

其中有一句：

 $mod = !in_array($discuz->var['mod'], $modarray) && (!preg_match('/^\w+$/', $discuz->var['mod']) || !file_exists(DISCUZ_ROOT.'./source/module/member/member_'.$discuz->var['mod'].'.php')) ? 'register' : $discuz->var['mod'];

这一句里面的 $discuz->var['mod']来源，有专门的技术贴分析：http://www.phpfensi.com/cms/20150930/10204.html

 require libfile('function/member');//载入menber 函数
 require libfile('class/member');//载入menber 类库
 runhooks();  //运行hook点

 require DISCUZ_ROOT.'./source/module/member/member_'.$mod.'.php';//载入member_register.php页面。

从这里面来看，主要注册程序页为：source/module/member/member_register.php页面。

二.source/module/member/member_register.php:

if(!defined('IN_DISCUZ')) {
    exit('Access Denied');
}//来源检测

define('NOROBOT', TRUE);

$ctl_obj = new register_ctl();//实例化register_ctl类
$ctl_obj->setting = $_G['setting'];//获取全局变量中setting值
$ctl_obj->template = 'member/register';//载入模板template\default\member\register.htm
$ctl_obj->on_register();//执行on_register函数

注册执行的主函数为：register_ctl类中的on_register()，register_ctl类的位置是：source/class/class_member.php文件中。这个文件在第一部分已经通过载入类库方式载入。

三.source/class/class_member.php:

function register_ctl(){}
function on_register(){}

这个文件只有这两个函数，register_ctl(){}初始化，载入相关类库。on_register()注册程序。

注册程序很长，选几个关键点来看：

大约700行左右（这个程序页面我稍微修改过，跟官方原版略有区别）：

            if(!$activation) {
                $ffromuser=getuserbyuid($fromuid);//这个系我添加，上面程序已经获取了$fromuid推荐人ID，这里我获取了一下推荐人的数据（注：这是个人员数据数组对象， $ffromuser['username']是用户名）。
                $uid = uc_user_register(addslashes($username), $password, $email, $questionid, $answer, $_G['clientip'], $ffromuser['username']);//ucenter中心添加用户。

这个程序uc_user_register()是添加用户的具体实现方法。

uc_user_register()这个程序的位置是：uc_client\client.php页面里面。

四.uc_client\client.php：

function uc_user_register($username, $password, $email, $questionid = '', $answer = '', $regip = '',$fromuser = '') {
    return call_user_func(UC_API_FUNC, 'user', 'register', array('username'=>$username, 'password'=>$password, 'email'=>$email, 'questionid'=>$questionid, 'answer'=>$answer, 'regip'=>$regip, 'fromuser'=>$fromuser));
}

代码很简单，调用了call_user_func()方法实现了多接口复用。其中参数：

UC_API_FUNC参数在该页面头部定义了：

define('UC_API_FUNC', UC_CONNECT == 'mysql' ? 'uc_api_mysql' : 'uc_api_post');//在我这边连接方式是mysql，根据配置，也可以使用api post方式传送用户注册信息。

uc_api_mysql()函数，在本页面上部定义了：

function uc_api_mysql($model, $action, $args=array()) {
    global $uc_controls;
    if(empty($uc_controls[$model])) {
        if(function_exists("mysql_connect")) {
            include_once UC_ROOT.'./lib/db.class.php';
        } else {
            include_once UC_ROOT.'./lib/dbi.class.php';
        }
        include_once UC_ROOT.'./model/base.php';
        include_once UC_ROOT."./control/$model.php";
        eval("\$uc_controls['$model'] = new {$model}control();");
    }
    if($action{0} != '_') {
        $args = uc_addslashes($args, 1, TRUE);
        $action = 'on'.$action;
        $uc_controls[$model]->input = $args;
        return $uc_controls[$model]->$action($args);
    } else {
        return '';
    }
}

include_once UC_ROOT."./control/$model.php";表示载入/control/user.php模块。

$action = 'on'.$action;表示调用函数onregister();

五.uc_client\control\user.php:

    function onregister() {
        $this->init_input();
        $username = $this->input('username');
        $password =  $this->input('password');
        $email = $this->input('email');
        $questionid = $this->input('questionid');
        $answer = $this->input('answer');
        $regip = $this->input('regip');
        $fromuser = $this->input('fromuser');//这一行是我添加的推荐人信息。

        if(($status = $this->_check_username($username)) < 0) {
            return $status;
        }
        if(($status = $this->_check_email($email)) < 0) {
            return $status;
        }
        $uid = $_ENV['user']->add_user($username, $password, $email, 0, $questionid, $answer, $regip, $fromuser);//这一行是我修改的推荐人账号传参。
        return $uid;
    }

$_ENV是PHP中的一个包含服务器端环境变量的数组。它是PHP中一个超级全局变量，我们可以在PHP程序的任何地方直接访问它。

在 uc_client/model/base.php 中，使用了load()方法定义了 $_ENV[$model] 并赋值。

function load($model, $base = NULL) {
    $base = $base ? $base : $this;
    if(empty($_ENV[$model])) {
        require_once UC_ROOT."./model/$model.php";
        eval('$_ENV[$model] = new '.$model.'model($base);');
    }
    return $_ENV[$model];
}

这里的作用，我们不深究，知道这里调用的是UC_ROOT."./model/user.php中的add_user()方法。

六.uc_client/model/user.php：

大约L129：

    function add_user($username, $password, $email, $uid = 0, $questionid = '', $answer = '', $regip = '', $fromuser = '') {
        $regip = empty($regip) ? $this->base->onlineip : $regip;
        $salt = substr(uniqid(rand()), -6);
        $password = md5(md5($password).$salt);
        $sqladd = $uid ? "uid='".intval($uid)."'," : '';
        $sqladd .= $questionid > 0 ? " secques='".$this->quescrypt($questionid, $answer)."'," : " secques='',";
        $this->db->query("INSERT INTO ".UC_DBTABLEPRE."members SET $sqladd username='$username', password='$password', email='$email', regip='$regip', regdate='".$this->base->time."', salt='$salt'");
        $uid = $this->db->insert_id();
        $this->db->query("INSERT INTO ".UC_DBTABLEPRE."memberfields SET uid='$uid'");
        return $uid;
    }

到此，参数就传递到写入数据库的部分了，具体如何放置就不属于本文讨论范围，相信大家会有很多很好的解决方案。

所有需要修改的部分就完成了。总共涉及6个页面。