如何集成注册邀请码功能到WordPress主题中

有的客户需要邀请码注册，这就将实现代码贡献出来，在实现邀请码的功能时有参考过一个插件。

思路：

1. 需要新建一个数据表来保存邀请码。

2. 后台需要两个页面：邀请码列表、添加邀请码。

3. 添加邀请码的时候需要能设置前缀，一次生成多个邀请码，邀请码长度可自定义，每个邀请码使用次数可设置。

数据表

code:邀请码、max:邀请码使用次数、users:使用这个验证码的所有用户、status:验证码是否可用。

数据库操作

数据库操作部分代码包括：1. 建立数据库。2.对数据的获取、增加、删除、更改等操作。

1. <?php
2. //第一次启用主题时执行
3. function ashuwp_load_theme() {
4.   global $pagenow;
5.   if ( is_admin() && ‘themes.php’ == $pagenow && isset( $_GET[‘activated’] ) ){
6.     ashuwp_invitation_code_install();
7.   }
8. }
9. add_action( ‘load-themes.php’, ‘ashuwp_load_theme’ );
11. //建立数据表
12. function ashuwp_invitation_code_install(){
13.   global $wpdb;
15.   $table_name = $wpdb->prefix . ‘invitation_code’;
16.   if( $wpdb->get_var(“SHOW TABLES LIKE ‘$table_name’”) != $table_name ) :
17.     $sql = ” CREATE TABLE `”.$wpdb->prefix.“invitation_code` (
18.       `id` BIGINT NOT NULL AUTO_INCREMENT PRIMARY KEY ,
19.       `code` varchar(40),
20.       `max` INT NOT NULL,
21.       `users` varchar(20),
22.       `status` tinyint
23.       ) ENGINE = MYISAM DEFAULT CHARSET=utf8;”;
25.     require_once(ABSPATH . ‘wp-admin/includes/upgrade.php’);
26.     dbDelta($sql);
27.   endif;
28. }
30. //插入数据
31. function ashuwp_insert_invitation_code( $code, $max = 1, $users, $status){
32.   global $wpdb;
34.   if($code==”){
35.     return false;
36.   }
37.   $code = trim($code);
39.   $code_exists = ashuwp_check_invitation_code($code);
41.   if(!$code_exists){
42.     $insert = “insert into “.$wpdb->prefix.“invitation_code (code,max,users,status) values( ‘$code’, ‘$max’, ”,’1′)”;
43.     $wpdb->query($insert);
44.     return true;
45.   }else{
46.     return false;
47.   }
48. }
50. //检查邀请码是否已存在
51. function ashuwp_check_invitation_code( $code ){
52.   global $wpdb;
54.   $sql = “select * from “.$wpdb->prefix.“invitation_code where code=’$code’”;
56.   $result = $wpdb->get_results($sql);
58.   if(!empty($result)){
59.     return true;
60.   }else{
61.     return false;
62.   }
63. }
65. //获取邀请码
66. function ashuwp_get_invitation_code($per_page=50, $page=1){
67.   global $wpdb;
69.   $page = (int)$page;
70.   $per_page = (int)$per_page;
71.   if(!$page){
72.     $page = 1;
73.   }
74.   if(!$per_page){
75.     $per_page = 50;
76.   }
78.   $begin = $per_page*($page-1);
79.   $end = $per_page*$page;
81.   $sql = “select * from “.$wpdb->prefix.“invitation_code limit $begin,$end”;
83.   $results = $wpdb->get_results($sql,’ARRAY_A’);
85.   return $results;
86. }
88. //邀请码的删除、启用、禁用。
89. function ashuwp_operation_invitation_code( $id, $action ){
90.   global $wpdb;
92.   $id = (int)$id;
93.   if(!$id){
94.     return false;
95.   }
97.   if(!in_array($action,array(‘delete‘,’deactive’,’active’))){
98.     return false;
99.   }
101.   if($action ==’delete‘){
102.     $sql = “delete from “.$wpdb->prefix .“invitation_code where id=’$id’”;
103.   }
105.   if($action ==’deactive’){
106.     $sql = “update “.$wpdb->prefix .“invitation_code set status=0 where id=’$id’”;
107.   }
109.   if($action ==’active’){
110.     $sql = “update “.$wpdb->prefix .“invitation_code set status=1 where id=’$id’”;
111.   }
113.   $result = $wpdb->query($sql);
115.   if($result){
116.     return true;
117.   }else{
118.     return false;
119.   }
120. }

网站后台
网站后台部分代码包括：1. 邀请码列表页面。2. 邀请码增加页面。

1. <?php
2. class ashuwp_invitation_code_admin {
3.   static public $instance;
5.   public function __construct(){
6.     add_action( ‘admin_menu’, array(&$this, ‘ashuwp_invitation_code_menu’) );
8.   }
9.   function ashuwp_invitation_code_menu(){
10.     add_menu_page( ‘邀请码’, ‘邀请码’, ‘manage_options’, ‘invitation_code’, array(&$this, ‘invitation_code_list’),”,27);
11.     add_submenu_page(‘invitation_code’, ‘添加邀请码’, ‘添加邀请码’, ‘manage_options’, ‘invitation_code_add’, array(&$this, ‘invitation_code_add’));
12.   }
14.   function invitation_code_list(){
15.     if( isset($_GET[‘code_action’]) && in_array($_GET[‘code_action’],array(‘delete‘,’deactive’,’active’)) && isset($_GET[‘code_id’]) ){
16.       $code_id = (int)$_GET[‘code_id’];
17.       if(!$code_id){
18.         return;
19.       }
21.       $result = ashuwp_operation_invitation_code( $code_id, $_GET[‘code_action’] );
23.     }
24.     $code_lists = ashuwp_get_invitation_code(999,1);
25.   ?>
26.     <div class=“wrap”>
27.       <h2 class=“wp-heading-inline”>邀请码</h2>
28.       <a href=“<?php echo admin_url( ‘admin.php?page=invitation_code_add’ ); ?>” class=“page-title-action”>添加</a>
29.       <hr class=“wp-header-end”>
30.       <?php
31.       if(isset($result)){
32.         if($result){
33.       ?>
34.       <div id=“message” class=“notice notice-success”>操作成功。</div>
35.       <?php
36.         }else{
37.         ?>
38.         <div id=“message” class=“notice notice-error”>操作失败。</div>
39.         <?php
40.         }
41.       }
42.       ?>
43.       <ul class=“subsubsub”><li class=“all”>全部<span class=“count”>（<?php echo count($code_lists); ?>）</span></ul>
45.       <table class=“wp-list-table widefat fixed”>
46.         <thead>
47.           <tr>
48.             <th scope=“col”>ID</th>
49.             <th scope=“col”>邀请码</th>
50.             <th scope=“col”>统计(最大使用数/已使用)</th>
51.             <th scope=“col”>用户</th>
52.             <th scope=“col”>操作</th>
53.           </tr>
54.         </thead>
55.         <tbody>
56.         <?php
57.         if($code_lists){
58.           foreach($code_lists as $code){
59.             $users = array();
60.             if(!empty($code[‘users’])){
61.               $users = explode( ‘,’, $code[‘users’] );
62.             }
63.             $used = count($users);
64.           ?>
65.           <tr>
66.             <td><?php echo $code[‘id’]; ?></td>
67.             <td>
68.               <?php echo $code[‘code’]; ?>
69.               <?php
70.               if(empty($code[‘status’]) || !$code[‘status’]){
71.                 echo ‘-已禁用’;
72.               }
73.               ?>
74.             </td>
75.             <td>
76.               <?php echo $code[‘max’].’/’.$used; ?>
77.             </td>
78.             <td>
79.               <?php
80.               foreach( $users as $user_id ){
81.                 $user = get_user_by(‘id’, $user_id);
82.                 if(!empty($user)){
83.                 ?>
84.                 <a href=“<?php echo admin_url( ‘user-edit.php?user_id=’.$user->ID ); ?>”><?php echo $user->user_login; ?></a>,
85.                 <?php
86.                 }
88.               }
89.               ?>
90.             </td>
91.             <td>
92.               <a href=“<?php echo admin_url( ‘admin.php?page=invitation_code&code_action=delete&code_id=’.$code[‘id’] ); ?>”>删除</a>
93.               <?php
94.               if(empty($code[‘status’]) || !$code[‘status’]){
95.               ?>
96.               <a href=“<?php echo admin_url( ‘admin.php?page=invitation_code&code_action=active&code_id=’.$code[‘id’] ); ?>”>启用</a>
97.               <?php
98.               }else{
99.               ?>
100.               <a href=“<?php echo admin_url( ‘admin.php?page=invitation_code&code_action=deactive&code_id=’.$code[‘id’] ); ?>”>禁用</a>
101.               <?php
102.               }
103.               ?>
104.             </td>
105.           </tr>
106.           <?php
107.           }
108.         }
109.         ?>
110.         </tbody>
111.         <tfoot>
112.           <tr>
113.             <th scope=“col”>ID</th>
114.             <th scope=“col”>邀请码</th>
115.             <th scope=“col”>统计</th>
116.             <th scope=“col”>用户</th>
117.             <th scope=“col”>操作</th>
118.           </tr>
119.         </tfoot>
120.       </table>
121.       <div class=“tablenav bottom”>
122.         <div class=“tablenav-pages”>
123.           <span class=“pagination-links”>
125.           </span>
126.         </div>
127.       </div>
128.     </div>
129.   <?php
130.   }
132.   function invitation_code_add(){
133.     $data_codes = ashuwp_get_invitation_code(999,1);
134.     $code_list = array();
135.     foreach($data_codes as $code){
136.       $code_list[] = $code[‘code’];
137.     }
138.     if(isset($_REQUEST[‘submit’]) && isset($_REQUEST[‘ashuwp_invitation_code_field’]) && check_admin_referer(‘ashuwp_invitation_code_action’, ‘ashuwp_invitation_code_field’) ) {
140.       $code_prefix = ”;
141.       if(!empty($_POST[‘code_prefix’])){
142.         $code_prefix = trim($_POST[‘code_prefix’]);
143.       }
145.       $code_length = ”;
146.       if(!empty($_POST[‘code_length’])){
147.         $code_length = (int)$_POST[‘code_length’];
148.       }
149.       if(!$code_length){
150.         $code_length = 8;
151.       }
153.       $code_number = 1;
154.       if(!empty($_POST[‘code_number’])){
155.         $code_number = (int)$_POST[‘code_number’];
156.       }
157.       if(!$code_number){
158.         $code_number = 1;
159.       }
161.       $code_counter = ”;
162.       if(!empty($_POST[‘code_counter’])){
163.         $code_counter = (int)$_POST[‘code_counter’];
164.       }
165.       if(!$code_counter){
166.         $code_counter = 1;
167.       }
170.       $i=1;
171.       $code_tem = array();
172.       while ( $i <= $code_number ){
173.         $tem = strtoupper( $code_prefix . wp_generate_password( $code_length, false ) );
174.         if(!in_array($tem,$code_list)){
175.           $i++;
176.           $code_tem[] = $tem;
177.           ashuwp_insert_invitation_code( $tem, $code_counter, ”, 1);
179.         }
180.       }
181.     }
182.   ?>
183.     <div class=“wrap”>
184.       <h2 class=“wp-heading-inline”>添加邀请码</h2>
185.       <a href=“<?php echo admin_url( ‘admin.php?page=invitation_code_add’ ); ?>” class=“page-title-action”>添加</a>
186.       <hr class=“wp-header-end”>
187.       <?php
188.       if(!empty($code_tem)){
189.       ?>
190.       <div id=“message” class=“notice notice-success”>
191.         <p>邀请码添加成功：</p>
192.         <?php
193.         foreach($code_tem as $text){
194.           echo ‘<p>’.$text.'</p>’;
195.         }
196.         ?>
197.       </div>
198.       <?php
199.       }
200.       ?>
201.       <form action=“” method=“post”>
202.         <table class=“form-table”>
203.           <tbody>
204.             <tr>
205.               <td><label for=“code_prefix”>邀请码前缀</label></td>
206.               <td>
207.                 <input type=“text” id=“code_prefix” name=“code_prefix” class=“regular-text”  value=“”/>
208.                 <p class=“description”>前缀可不填。</p>
209.               </td>
210.             </tr>
211.             <tr>
212.               <td><label for=“code_length”>邀请码字符长度</label></td>
213.               <td>
214.                 <input type=“text” id=“code_length” name=“code_length” class=“regular-text”  value=“”/>
215.                 <p class=“description”>字符长度不包括前缀，默认8个字符。</p>
216.               </td>
217.             </tr>
218.             <tr>
219.               <td><label for=“code_number”>邀请码个数</label></td>
220.               <td>
221.                 <input type=“text” id=“code_number” name=“code_number” class=“regular-text” value=“”/>
222.                 <p class=“description”>本次生成多少个邀请码，默认1个。</p>
223.               </td>
224.             </tr>
225.             <tr>
226.               <td><label for=“code_counter”>允许使用的次数</label></td>
227.               <td>
228.                 <input type=“text” id=“code_counter” name=“code_counter” class=“regular-text”  value=“”/>
229.                 <p class=“description”>每个邀请码允许使用的次数，默认1次。</p>
230.               </td>
231.             </tr>
232.           </tbody>
233.         </table>
234.         <p class=“submit”>
235.           <?php wp_nonce_field( ‘ashuwp_invitation_code_action’,’ashuwp_invitation_code_field’ ); ?>
236.           <input type=“submit” name=“submit” id=“submit” class=“button button-primary” value=“生成邀请码”>
237.         </p>
238.       </form>
239.     </div>
240.   <?php
241.   }
243. }
245. $invitation_code = new ashuwp_invitation_code_admin();

效果展示

1. 增加邀请码

2. 邀请码列表

结束语

上面范例代码，仅做参考使用，可根据实际自行优化，具体如何使用邀请码也请自行斟酌。

来自：阿树工作室