edit

index.php

@@ -2,7 +2,7 @@
 
 if (!defined('DEVER_APP_NAME')) {
 	# 这样定义就可以将组件重复使用
-	define('DEVER_APP_NAME', 'invite');
+	define('DEVER_APP_NAME', 'Invite');
 }
 define('DEVER_APP_LANG', '邀请组件');
 define('DEVER_APP_PATH', dirname(__FILE__) . DIRECTORY_SEPARATOR);

lib/Code.php

@@ -1,41 +0,0 @@
-<?php
-namespace Invite\Lib;
-use Dever;
-use Dever\Helper\Str;
-class Code 
-{
-    # 获取邀请码
-    public function get($uid)
-    {   
-        $info = Dever::db('code', 'invite')->find(['uid' => $uid]);
-        if ($info) {
-            return $info['value'];
-        } else {
-            //$code = Dever::uid($uid);
-            $code = Str::rand(5, 0);
-            return $this->createCode($uid, $code);
-        }
-    }
-
-    private function createCode($uid, $code)
-    {
-        $info = Dever::db('code', 'invite')->find(['value' => $code]);
-        if ($info) {
-            $code = Str::rand(5, 0);
-            return $this->createCode($uid, $code);
-        } else {
-            Dever::db('code', 'invite')->insert(['value' => $code, 'uid' => $uid]);
-            return $code;
-        }
-    }
-
-    # 根据邀请码获取邀请人uid
-    public function getUid($code)
-    {   
-        $info = Dever::db('code', 'invite')->find(['value' => $code]);
-        if ($info) {
-            return $info['uid'];
-        }
-        return false;
-    }
-}

lib/Relation.php

@@ -1,146 +0,0 @@
-<?php
-# 关系类
-namespace Invite\Lib;
-use Dever;
-
-class Relation
-{
-    private $table = 'invite/relation';
-    public function __construct($invite = '')
-    {
-        if ($invite) {
-            $this->table = $invite;
-        }
-    }
-
-    # 只记录几级关系
-    private $total = 10;
-
-    # 通用的邀请方法：
-    # uid 当前用户的上级，需要通过code邀请码来得到
-    # to_uid 被邀请人，当前登录用户，注册后得到
-    public function set($uid, $to_uid)
-    {
-        $this->setParent($uid, $to_uid);
-        $this->add($uid, $to_uid, 1);
-
-        return true;
-    }
-
-    public function setParent($uid, $to_uid, $level = 1)
-    {
-        $parent = $this->getParent($uid);
-        if ($parent) {
-            $level = $level + 1;
-            if ($level > $this->total) {
-                return;
-            }
-            $this->add($parent['uid'], $to_uid, $level);
-            $this->setParent($parent['uid'], $to_uid, $level);
-        }
-    }
-
-    # 更换上级
-    public function replaceParent($uid, $old_parent, $new_parent, $call = true)
-    {
-        $info = Dever::db($this->table)->find(['to_uid' => $uid, 'uid' => $old_parent]);
-        if ($info) {
-            $state = Dever::db($this->table)->update($info['id'], ['uid' => $new_parent]);
-            if ($state && $call) {
-                $child = $this->getChild($uid, false, false);
-                if ($child) {
-                    foreach ($child as $k => $v) {
-                        $this->replaceParent($v['to_uid'], $old_parent, $new_parent, false);
-                    }
-                }
-            }
-        }
-        return true;
-    }
-
-    # 重置上级
-    public function resetParent($uid, $parent)
-    {
-        Dever::db($this->table)->delete(['to_uid' => $uid]);
-        $this->set($parent, $uid);
-
-        $child = $this->getChild($uid, 1, false);
-        if ($child) {
-            foreach ($child as $k => $v) {
-                $this->resetParent($v['to_uid'], $uid);
-            }
-        }
-        return true;
-    }
-
-    # 清理邀请关系
-    public function dropParent($uid, $parent)
-    {
-        return Dever::db($this->table)->delete(['uid' => $parent, 'to_uid' => $uid]);
-    }
-
-    # 获取某个用户的上级数据
-    public function getParent($uid, $level = 1)
-    {
-        return Dever::db($this->table)->find(['to_uid' => $uid, 'level' => $level]);
-    }
-
-    # 获取某个用户的所有上级数据
-    public function getParentAll($uid, $level = false)
-    {
-        $where['to_uid'] = $uid;
-        if ($level) {
-            $where['level'] = $level;
-        }
-        return Dever::db($this->table)->select($where);
-    }
-
-    # 获取某个用户的下级数据
-    public function getChild($uid, $level = 1, $page = 10)
-    {
-        $where['uid'] = $uid;
-        if ($level) {
-            $where['level'] = $level;
-        }
-        $set = [];
-        if ($page) {
-            $set['num'] = $page;
-        }
-        return Dever::db($this->table)->select($where, $set);
-    }
-
-    # 获取某个用户在x小时之内的下级数据
-    public function getChildNum($uid, $level = 1, $start = false, $end = false, $method = 'count')
-    {
-        $where['uid'] = $uid;
-        if ($level) {
-            $where['level'] = $level;
-        }
-        if ($start) {
-            $where['cdate#'] = ['>=', strtotime($start)];
-        }
-        if ($end) {
-            $where['cdate##'] = ['<=', strtotime($end)];
-        }
-        if ($method == 'count') {
-            $method = 'count';
-        } else {
-            $method = 'select';
-        }
-        return Dever::db($this->table)->$method($where);
-    }
-
-    # 插入数据
-    public function add($uid, $to_uid, $level = 1)
-    {   
-        $data['uid'] = $uid;
-        $data['to_uid'] = $to_uid;
-        $data['level'] = $level;
-        $info = Dever::db($this->table)->find($data);
-        if (!$info) {
-            return Dever::db($this->table)->insert($data);
-        }
-        
-        return false;
-    }
-}