edit

lib/Client.php

@@ -52,6 +52,7 @@ class Client extends Base
 		$this->initRequest();
 		$this->initAccount();
 		$this->initSource();
+		$this->initInvite();
         $this->initSystem();
 		$this->initSourceType();
 		$this->initSystemSource();
@@ -104,6 +105,12 @@ class Client extends Base
 		$this->session->add($this->prefix . 'oauth_source', $this->source);
 	}
 
+	private function initInvite()
+	{
+		$this->invite = (isset($this->request['invite']) && $this->request['invite']) ? $this->request['invite'] : $this->session->get($this->prefix . 'oauth_invite');
+		$this->session->add($this->prefix . 'oauth_invite', $this->invite);
+	}
+
 	private function initRefer()
 	{
 		$this->refer = (isset($this->request['refer']) && $this->request['refer']) ? $this->request['refer'] : $this->session->get($this->prefix . 'oauth_refer');
@@ -294,7 +301,7 @@ class Client extends Base
             $user['country'] = $userinfo['country'];
         }
 
-    	$user = $this->passport->wechat($data, $user, $this->account, $this->system, $this->source_type, $this->system_source, $this->source);
+    	$user = $this->passport->wechat($data, $user, $this->account, $this->system, $this->source_type, $this->system_source, $this->source, $this->invite);
 
         return $user;
 	}