You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

183 lines
5.3 KiB

// MIT License
//
// Copyright (c) 2024 godoos.com
// Email: xpbb@qq.com
// GitHub: github.com/phpk/godoos
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
package localchat
import (
"bytes"
"encoding/json"
"fmt"
"godo/libs"
"io"
"log"
"net/http"
"os"
"path/filepath"
"time"
)
func downloadFiles(msg UdpMessage) error {
postUrl := fmt.Sprintf("http://%s:56780/localchat/getfiles", msg.IP)
postData, err := json.Marshal(msg.Message)
if err != nil {
return fmt.Errorf("failed to marshal post data: %v", err)
}
log.Printf("Sending POST request to %s with data: %s", postUrl, string(postData))
resp, err := http.Post(postUrl, "application/json", bytes.NewBuffer(postData))
if err != nil {
return fmt.Errorf("failed to make POST request: %v", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
body, _ := io.ReadAll(resp.Body)
return fmt.Errorf("server returned status code: %v, body: %s", resp.StatusCode, body)
}
// 接收文件的目录
baseDir, err := libs.GetOsDir()
if err != nil {
log.Printf("Failed to get OS directory: %v", err)
return fmt.Errorf("failed to get OS directory")
}
resPath := filepath.Join("C", "Users", "Reciv", time.Now().Format("2006-01-02"))
receiveDir := filepath.Join(baseDir, resPath)
if !libs.PathExists(receiveDir) {
err := os.MkdirAll(receiveDir, 0755)
if err != nil {
log.Printf("Failed to create receive directory: %v", err)
return fmt.Errorf("failed to create receive directory")
}
}
// 处理响应中的文件
if err := handleResponse(resp.Body, receiveDir); err != nil {
log.Fatalf("Failed to handle response: %v", err)
}
fmt.Println("Files downloaded successfully")
return nil
}
func handleResponse(reader io.Reader, saveDir string) error {
body, err := io.ReadAll(reader)
if err != nil {
return fmt.Errorf("failed to read response body: %v", err)
}
log.Printf("Received file list: %v", string(body))
var fileList FileList
if err := json.Unmarshal(body, &fileList); err != nil {
return fmt.Errorf("failed to unmarshal file list: %v", err)
}
log.Printf("Received file list: %v", fileList.Files)
for _, filePath := range fileList.Files {
err := saveFileOrFolder(filePath, saveDir)
if err != nil {
return fmt.Errorf("failed to save file or folder: %v", err)
}
}
return nil
}
func saveFileOrFolder(filePath string, saveDir string) error {
relativePath := filePath
localFilePath := filepath.Join(saveDir, relativePath)
// 检查是否为文件夹
if _, err := os.Stat(localFilePath); os.IsNotExist(err) {
// 如果不存在,则创建文件夹
err := os.MkdirAll(localFilePath, 0755)
if err != nil {
return fmt.Errorf("failed to create directory: %v", err)
}
} else if fileInfo, err := os.Stat(localFilePath); err == nil && fileInfo.IsDir() {
// 如果是文件夹,则递归处理
err := saveFolder(relativePath, saveDir)
if err != nil {
return fmt.Errorf("failed to save folder: %v", err)
}
} else {
// 如果是文件,则直接保存
err := saveFile(localFilePath)
if err != nil {
return fmt.Errorf("failed to save file: %v", err)
}
}
return nil
}
func saveFolder(folderName string, saveDir string) error {
localFolderPath := filepath.Join(saveDir, folderName)
err := os.MkdirAll(localFolderPath, 0755)
if err != nil {
return fmt.Errorf("failed to create directory: %v", err)
}
// 获取文件夹内容
folderPath := filepath.Join(saveDir, folderName)
files, err := os.ReadDir(folderPath)
if err != nil {
return fmt.Errorf("failed to read directory: %v", err)
}
for _, file := range files {
subPath := filepath.Join(folderName, file.Name())
if file.IsDir() {
err := saveFolder(subPath, localFolderPath)
if err != nil {
return fmt.Errorf("failed to save folder: %v", err)
}
} else {
err := saveFile(filepath.Join(localFolderPath, file.Name()))
if err != nil {
return fmt.Errorf("failed to save file: %v", err)
}
}
}
return nil
}
func saveFile(filePath string) error {
file, err := os.Open(filePath)
if err != nil {
return fmt.Errorf("failed to open file: %v", err)
}
defer file.Close()
// 创建本地文件
localFile, err := os.Create(filePath)
if err != nil {
return fmt.Errorf("failed to create file: %v", err)
}
defer localFile.Close()
_, err = io.Copy(localFile, file)
if err != nil {
return fmt.Errorf("failed to copy file: %v", err)
}
return nil
}