From 4737960d2236038677ffb994eacf584f7b0601b4 Mon Sep 17 00:00:00 2001
From: hellcat <>
Date: Fri, 28 Mar 2025 20:45:26 +0800
Subject: [PATCH] 3

---
 src/Services/TomTool/VisitorCount.php | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/src/Services/TomTool/VisitorCount.php b/src/Services/TomTool/VisitorCount.php
index d3b57f45..7bba13db 100755
--- a/src/Services/TomTool/VisitorCount.php
+++ b/src/Services/TomTool/VisitorCount.php
@@ -12,17 +12,18 @@ final class VisitorCount
     public static function From()
     {
         
+//        var_dump(self::normalizeUrl($_SERVER['HTTP_REFERER']));
         // 如果以被设置，什么都不做
         // 如果未被设置，入库并保存cookie，注册用
         if (!isset($_COOKIE['VisitorFrom'])) {
             
             if (isset($_SERVER['HTTP_REFERER'])) {
                 
-                $sFrom = $_SERVER['HTTP_REFERER'];
+                $sFrom = self::normalizeUrl($_SERVER['HTTP_REFERER']);
                 
             } else {
                 
-                $sFrom = "直接";
+                $sFrom = $_ENV['short_url'];
                 
             }
             
@@ -36,4 +37,18 @@ final class VisitorCount
         
     }
     
+    public static function normalizeUrl($url) {
+        
+        $parsedUrl = parse_url($url);
+        
+        $host = isset($parsedUrl['host']) ? $parsedUrl['host'] : '';
+        
+        if (strpos($host, 'www.') === 0) {
+            $host = substr($host, 4); // 去掉 'www.'
+        }
+        
+        return $host;
+        
+    }
+    
 }