feat(geolite): implement GeoLite2 database download from cloud

Author: 0xJacky

.claude/settings.local.json

@@ -17,7 +17,9 @@
       "Bash(cp:*)",
       "mcp__eslint__lint-files",
       "Bash(go generate:*)",
-      "Bash(pnpm eslint:*)"
+      "Bash(pnpm eslint:*)",
+      "Read(//workspaces/cosy/settings/**)",
+      "Bash(go doc:*)"
     ],
     "deny": []
   }

.gitignore

@@ -25,3 +25,4 @@ internal/**/*.gen.go
 log-index/
 *.prof
 *.test
+GeoLite2-City.mmdb

api/geolite/download.go

@@ -0,0 +1,114 @@
+package geolite
+
+import (
+	"net/http"
+
+	"github.com/0xJacky/Nginx-UI/internal/geolite"
+	"github.com/gin-gonic/gin"
+	"github.com/gorilla/websocket"
+	"github.com/uozi-tech/cosy/logger"
+)
+
+const (
+	StatusInfo     = "info"
+	StatusError    = "error"
+	StatusProgress = "progress"
+)
+
+type DownloadProgressResp struct {
+	Status   string  `json:"status"`
+	Progress float64 `json:"progress"`
+	Message  string  `json:"message"`
+}
+
+func DownloadGeoLiteDB(c *gin.Context) {
+	var upgrader = websocket.Upgrader{
+		CheckOrigin: func(r *http.Request) bool {
+			return true
+		},
+	}
+
+	// Upgrade HTTP to WebSocket
+	ws, err := upgrader.Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		logger.Error(err)
+		return
+	}
+	defer ws.Close()
+
+	sendMessage := func(status, message string, progress float64) {
+		if err := ws.WriteJSON(DownloadProgressResp{
+			Status:   status,
+			Progress: progress,
+			Message:  message,
+		}); err != nil {
+			logger.Error("Failed to send WebSocket message:", err)
+		}
+	}
+
+	// Check if database already exists
+	if geolite.DBExists() {
+		sendMessage(StatusInfo, "Database already exists, removing old version...", 0)
+		// Optionally remove old database here if you want to force re-download
+	}
+
+	sendMessage(StatusInfo, "Starting download...", 0)
+
+	// Download progress channel
+	downloadProgressChan := make(chan float64, 100)
+	downloadDone := make(chan error, 1)
+
+	// Start download in goroutine
+	go func() {
+		downloadDone <- geolite.DownloadGeoLiteDB(downloadProgressChan)
+	}()
+
+	// Track download progress (0-50%)
+	downloadComplete := false
+	for !downloadComplete {
+		select {
+		case progress := <-downloadProgressChan:
+			// Scale download progress to 0-50%
+			scaledProgress := progress * 0.5
+			sendMessage(StatusProgress, "Downloading GeoLite2 database...", scaledProgress)
+		case err := <-downloadDone:
+			if err != nil {
+				sendMessage(StatusError, "Download failed: "+err.Error(), 0)
+				return
+			}
+			downloadComplete = true
+			sendMessage(StatusInfo, "Download complete", 50)
+		}
+	}
+
+	sendMessage(StatusInfo, "Decompressing database...", 50)
+
+	// Decompress progress channel
+	decompressProgressChan := make(chan float64, 100)
+	decompressDone := make(chan error, 1)
+
+	// Start decompression in goroutine
+	go func() {
+		decompressDone <- geolite.DecompressGeoLiteDB(decompressProgressChan)
+	}()
+
+	// Track decompression progress (50-100%)
+	decompressComplete := false
+	for !decompressComplete {
+		select {
+		case progress := <-decompressProgressChan:
+			// Scale decompress progress to 50-100%
+			scaledProgress := 50 + (progress * 0.5)
+			sendMessage(StatusProgress, "Decompressing database...", scaledProgress)
+		case err := <-decompressDone:
+			if err != nil {
+				sendMessage(StatusError, "Decompression failed: "+err.Error(), 50)
+				return
+			}
+			decompressComplete = true
+			sendMessage(StatusInfo, "Database ready", 100)
+		}
+	}
+
+	sendMessage(StatusInfo, "GeoLite2 database downloaded and installed successfully", 100)
+}

api/geolite/status.go

@@ -0,0 +1,38 @@
+package geolite
+
+import (
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/0xJacky/Nginx-UI/internal/geolite"
+	"github.com/gin-gonic/gin"
+	"github.com/uozi-tech/cosy"
+)
+
+type StatusResp struct {
+	Exists       bool   `json:"exists"`
+	Path         string `json:"path"`
+	Size         int64  `json:"size"`
+	LastModified string `json:"last_modified"`
+}
+
+func GetStatus(c *gin.Context) {
+	dbPath := geolite.GetDBPath()
+	resp := StatusResp{
+		Exists: geolite.DBExists(),
+		Path:   dbPath,
+	}
+
+	if resp.Exists {
+		fileInfo, err := os.Stat(dbPath)
+		if err != nil {
+			cosy.ErrHandler(c, err)
+			return
+		}
+		resp.Size = fileInfo.Size()
+		resp.LastModified = fileInfo.ModTime().Format(time.RFC3339)
+	}
+
+	c.JSON(http.StatusOK, resp)
+}

app/components.d.ts

@@ -11,6 +11,7 @@ declare module 'vue' {
     AAlert: typeof import('ant-design-vue/es')['Alert']
     AApp: typeof import('ant-design-vue/es')['App']
     AAutoComplete: typeof import('ant-design-vue/es')['AutoComplete']
+    AAvatar: typeof import('ant-design-vue/es')['Avatar']
     ABadge: typeof import('ant-design-vue/es')['Badge']
     ABreadcrumb: typeof import('ant-design-vue/es')['Breadcrumb']
     ABreadcrumbItem: typeof import('ant-design-vue/es')['BreadcrumbItem']
@@ -69,6 +70,7 @@ declare module 'vue' {
     ATag: typeof import('ant-design-vue/es')['Tag']
     ATextarea: typeof import('ant-design-vue/es')['Textarea']
     ATooltip: typeof import('ant-design-vue/es')['Tooltip']
+    ATypographyParagraph: typeof import('ant-design-vue/es')['TypographyParagraph']
     ATypographyText: typeof import('ant-design-vue/es')['TypographyText']
     ATypographyTitle: typeof import('ant-design-vue/es')['TypographyTitle']
     AutoCertFormAutoCertForm: typeof import('./src/components/AutoCertForm/AutoCertForm.vue')['default']
@@ -84,6 +86,7 @@ declare module 'vue' {
     ConfigHistoryDiffViewer: typeof import('./src/components/ConfigHistory/DiffViewer.vue')['default']
     DevDebugPanelDevDebugPanel: typeof import('./src/components/DevDebugPanel/DevDebugPanel.vue')['default']
     FooterToolbarFooterToolBar: typeof import('./src/components/FooterToolbar/FooterToolBar.vue')['default']
+    GeoLiteDownloadGeoLiteDownload: typeof import('./src/components/GeoLiteDownload/GeoLiteDownload.vue')['default']
     ICPICP: typeof import('./src/components/ICP/ICP.vue')['default']
     InspectConfigInspectConfig: typeof import('./src/components/InspectConfig/InspectConfig.vue')['default']
     LLMChatMessage: typeof import('./src/components/LLM/ChatMessage.vue')['default']

app/src/api/geolite.ts

@@ -0,0 +1,16 @@
+import { http } from '@uozi-admin/request'
+
+export interface GeoLiteStatus {
+  exists: boolean
+  path: string
+  size: number
+  last_modified: string
+}
+
+const geolite = {
+  async getStatus() {
+    return http.get<GeoLiteStatus>('geolite/status')
+  },
+}
+
+export default geolite