Possible fix for non-working nickname referal link

[mailer.git] / ref.php

diff --git a/ref.php b/ref.php
index 95e6400a434320925e45b232f6372166ff0f4954..700491851a4bff59c4f262a928dfac4165e3463c 100644 (file)
--- a/ref.php
+++ b/ref.php
@@ -49,52 +49,47 @@ require('inc/config-global.php');
 // No refid by default
 $URL = 'index.php';
 
-// Invalid request by default
-$ref = 0;
-
-// Base URL for redirection
-switch (getConfig('refid_target')) {
-       case 'register':
-               $URL = 'modules.php?module=index&what=register&refid=';
-               break;
-
-       case 'index':
-               $URL = 'index.php?refid=';
-               break;
-} // END - switch
-
-// Get referal ID from ref or refid variable
-if (isGetRequestElementSet('ref'))        $ref = getRequestElement('ref');
- elseif (isGetRequestElementSet('refid')) $ref = getRequestElement('refid');
-
-if (!empty($ref)) {
+if (determineReferalId() > 0) {
        // Test if nickname or numeric id
        if (isExtensionActive('nickname')) {
-               // Nickname in URL, so load the ID
-               $result = SQL_QUERY_ESC("SELECT `userid` FROM `{?_MYSQL_PREFIX?}_user_data` WHERE `nickname`='%s' LIMIT 1",
-                       array($ref), __FILE__, __LINE__);
-
-               // Load userid
-               list($ref) = SQL_FETCHROW($result);
-
-               // Free result
-               SQL_FREERESULT($result);
+               // Nickname in URL, so load the id
+               fetchUserData(determineReferalId(), 'nickname');
+
+               // Do we have an entry?
+               if (isUserDataValid()) {
+                       // Load userid
+                       $GLOBALS['refid'] = getUserData('userid');
+               } else {
+                       // So do we have a userid?
+                       fetchUserData(determineReferalId());
+
+                       // Do we have an entry?
+                       if (!isUserDataValid()) {
+                               // No entry, so no referal id
+                               $GLOBALS['refid'] = getConfig('def_refid');
+                       } // END - if
+               }
        } // END - if
 
-       // Also edit this 0 !
-       if (empty($ref)) $ref = 0;
-
        // Update session
-       setSession('refid', $ref);
-
-       // We have an refid here. So we simply add it
-       $URL .= bigintval($ref);
+       setSession('refid', determineReferalId());
 
        // Is the refid valid?
-       if ($ref > 0) {
+       if (determineReferalId() > 0) {
                // Update ref counter
                SQL_QUERY_ESC("UPDATE `{?_MYSQL_PREFIX?}_user_data` SET `ref_clicks`=`ref_clicks`+1 WHERE `userid`=%s LIMIT 1",
-                       array(bigintval($ref)), __FILE__, __LINE__);
+                       array(determineReferalId()), __FILE__, __LINE__);
+
+               // Base URL for redirection
+               switch (getConfig('refid_target')) {
+                       case 'register':
+                               $URL = 'modules.php?module=index&what=register';
+                               break;
+
+                       case 'index':
+                               $URL = 'index.php';
+                               break;
+               } // END - switch
        } // END - if
 } // END - if