// Package handler — cloud-disk (网盘) endpoints: directory browsing, QR-code // login, media import and 302 playback redirects. package handler import ( "net/http" "strings" "github.com/gin-gonic/gin" "github.com/ShukeBta/MediaStationGo/internal/model" "github.com/ShukeBta/MediaStationGo/internal/service" "github.com/ShukeBta/MediaStationGo/internal/service/cloud" ) // cloudListHandler browses a configured cloud disk directory. func cloudListHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { typ := c.Param("type") if !service.IsAdminCloudConfigurable(typ) { c.JSON(http.StatusBadRequest, gin.H{"error": "unsupported cloud provider", "items": []any{}}) return } dir := c.Query("dir") entries, err := svc.StorageCfg.CloudList(c.Request.Context(), typ, dir) if err != nil { c.JSON(http.StatusOK, gin.H{"error": err.Error(), "items": []any{}}) return } c.JSON(http.StatusOK, gin.H{"items": entries}) } } func cloudMkdirHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { typ := c.Param("type") if !service.IsAdminCloudConfigurable(typ) { c.JSON(http.StatusBadRequest, gin.H{"error": "unsupported cloud provider"}) return } var in struct { Dir string `json:"dir"` Name string `json:"name" binding:"required"` } if err := c.ShouldBindJSON(&in); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } entry, err := svc.StorageCfg.CloudMkdir(c.Request.Context(), typ, in.Dir, in.Name) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"entry": entry}) } } func cloudRenameHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { typ := c.Param("type") if !service.IsAdminCloudConfigurable(typ) { c.JSON(http.StatusBadRequest, gin.H{"error": "unsupported cloud provider"}) return } var in struct { Ref string `json:"ref" binding:"required"` Name string `json:"name" binding:"required"` } if err := c.ShouldBindJSON(&in); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } entry, err := svc.StorageCfg.CloudRename(c.Request.Context(), typ, in.Ref, in.Name) if err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, gin.H{"entry": entry}) } } // cloudImportHandler turns a cloud file into a playable 302-backed media item. func cloudImportHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { typ := c.Param("type") if !service.IsAdminCloudConfigurable(typ) { c.JSON(http.StatusBadRequest, gin.H{"error": "unsupported cloud provider"}) return } var in struct { Ref string `json:"ref" binding:"required"` Name string `json:"name"` Size int64 `json:"size"` } if err := c.ShouldBindJSON(&in); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } m, err := svc.StorageCfg.CloudImport(c.Request.Context(), typ, in.Ref, in.Name, in.Size) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, m) } } // cloudMountHandler creates or reuses a cloud:// media library for a cloud // directory, then queues a recursive import scan. The scan runs outside the // request so large 115/OpenList folders do not make the UI report a timeout. func cloudMountHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { typ := c.Param("type") if !service.IsAdminCloudConfigurable(typ) { c.JSON(http.StatusBadRequest, gin.H{"error": "unsupported cloud provider"}) return } var in struct { Dir string `json:"dir"` DirPath string `json:"dir_path"` Name string `json:"name"` MediaType string `json:"media_type"` } _ = c.ShouldBindJSON(&in) if !cloud.IsCloudType(typ) { c.JSON(http.StatusBadRequest, gin.H{"error": "unsupported cloud provider"}) return } if _, err := svc.StorageCfg.CloudProvider(c.Request.Context(), typ); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } path := service.BuildCloudLibraryPath(typ, in.Dir, in.DirPath) name := strings.TrimSpace(in.Name) if name == "" { name = cloudMountLibraryName(typ, strings.TrimSpace(in.Dir), strings.TrimSpace(in.DirPath)) } mediaType := strings.TrimSpace(in.MediaType) if mediaType == "" || strings.EqualFold(mediaType, "auto") { displayDir := strings.TrimSpace(in.DirPath) if displayDir == "" { displayDir = strings.TrimSpace(in.Dir) } mediaType = service.InferCloudMountMediaType(displayDir, name) } libs, err := svc.Repo.Library.List(c.Request.Context()) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } var lib *model.Library alreadyMounted := false if conflict := service.FindCloudMountConflict(libs, typ, in.Dir, in.DirPath); conflict != nil { lib = &conflict.Library alreadyMounted = conflict.Exact if conflict.Nested { c.JSON(http.StatusOK, gin.H{ "library": lib, "skipped": true, "reason": "cloud mount overlaps an existing mounted parent/child directory", "conflict_library": conflict.Library, }) return } } if lib == nil { lib = &model.Library{Name: name, Path: path, Type: mediaType, Enabled: true} if err := svc.Repo.Library.Create(c.Request.Context(), lib); err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } } else if alreadyMounted { updates := map[string]any{} if path != "" && path != lib.Path { updates["path"] = path lib.Path = path } if mediaType != "" && mediaType != lib.Type { updates["type"] = mediaType lib.Type = mediaType } currentDisplayName, _ := service.CloudLibraryDisplayName(*lib) if name != "" && name != lib.Name && (currentDisplayName == "" || currentDisplayName != name || strings.Contains(lib.Name, " · ")) { updates["name"] = name lib.Name = name } if len(updates) > 0 { if err := svc.Repo.DB.WithContext(c.Request.Context()).Model(&model.Library{}).Where("id = ?", lib.ID).Updates(updates).Error; err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } } } if svc.Scan != nil { libID := lib.ID if svc.WSHub != nil { svc.WSHub.Publish("scan", gin.H{ "library_id": libID, "cloud": true, "queued": true, "stage": "queued", "message": "云盘扫描已加入后台队列,会递归扫描并自动加入媒体库", "estimate_message": "小目录通常几十秒;几万文件的大目录可能需要数分钟到数小时,取决于网盘接口速度", }) } _, _, _ = svc.Scan.StartCloudLibraryScan(libID, false) } c.JSON(http.StatusAccepted, gin.H{ "library": lib, "already_mounted": alreadyMounted, "scan_queued": svc.Scan != nil, "message": "挂载后会后台递归扫描,发现的媒体会自动加入当前媒体库", "estimate_message": "小目录通常几十秒;几万文件的大目录可能需要数分钟到数小时,取决于网盘接口速度", }) } } func cloudScanAllHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { if svc.Scan == nil { c.JSON(http.StatusServiceUnavailable, gin.H{"error": "scanner unavailable"}) return } statuses, err := svc.Scan.StartAllCloudLibraryScans() if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } c.JSON(http.StatusAccepted, gin.H{ "items": statuses, "scan_queued": true, "message": "已开始扫描所有启用的网盘媒体库", "resume_message": "中断后再次点击扫描会重新遍历,但已入库媒体会去重更新,只补齐缺失项。", "estimate_message": "小目录通常几十秒;几万文件的大目录可能需要数分钟到数小时,取决于网盘接口速度", }) } } func cloudScanCancelHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { if svc.Scan == nil { c.JSON(http.StatusServiceUnavailable, gin.H{"error": "scanner unavailable"}) return } libraryID := strings.TrimSpace(c.Query("library_id")) provider := strings.TrimSpace(c.Query("provider")) cancelled := 0 if libraryID != "" { if svc.Scan.CancelCloudScan(libraryID) { cancelled = 1 } } else if provider != "" { cancelled = svc.Scan.CancelCloudScansForProvider(provider) } else { cancelled = svc.Scan.CancelAllCloudScans() } c.JSON(http.StatusOK, gin.H{ "cancelled": cancelled, "message": "已发送中断信号;正在等待当前网盘请求返回后停止", }) } } func cloudScanStatusHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { if svc.Scan == nil { c.JSON(http.StatusOK, gin.H{"items": []service.CloudScanStatus{}}) return } c.JSON(http.StatusOK, gin.H{"items": svc.Scan.CloudScanStatuses()}) } } func cloudMountLibraryName(typ, dir, displayDir string) string { base := service.CloudMountProviderLabel(typ) displayDir = strings.Trim(strings.TrimSpace(strings.ReplaceAll(displayDir, "\\", "/")), "/") if displayDir != "" { parts := strings.Split(displayDir, "/") for i := len(parts) - 1; i >= 0; i-- { if part := strings.TrimSpace(parts[i]); part != "" { return part } } } if dir == "" || dir == "0" { return base } dir = strings.Trim(strings.TrimSpace(strings.ReplaceAll(dir, "\\", "/")), "/") if dir == "" { return base } parts := strings.Split(dir, "/") for i := len(parts) - 1; i >= 0; i-- { if part := strings.TrimSpace(parts[i]); part != "" { return part } } return base } // cloud115QRStartHandler begins a 115 QR-code login and returns the session + // QR image URL for the frontend to render. func cloud115QRStartHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { if c.Param("type") != cloud.Type115 { c.JSON(http.StatusBadRequest, gin.H{"error": "qr login is only supported for 115"}) return } sess, err := cloud.QRStart(c.Request.Context(), nil) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, sess) } } // cloud115QRPollHandler polls a 115 QR session; on confirmation it returns the // session cookie so the frontend can save it as the storage credential. func cloud115QRPollHandler(svc *service.Container) gin.HandlerFunc { return func(c *gin.Context) { if c.Param("type") != cloud.Type115 { c.JSON(http.StatusBadRequest, gin.H{"error": "qr login is only supported for 115"}) return } var sess cloud.QRSession if err := c.ShouldBindJSON(&sess); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } st, err := cloud.QRPoll(c.Request.Context(), nil, &sess) if err != nil { c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) return } c.JSON(http.StatusOK, st) } }