jwebfs/fssc/internal/handler/nfhandler.go

package handler

import (
	"encoding/json"
	"fmt"
	"fssc/config"
	"net/http"
	"os"
	"path/filepath"
)

// json 结构体
type Response struct {
	Status  string `json:"status"`
	Message string `json:"data"`
}

// 文件输出的结构
type FileJson struct {
	Fname string `json:"fname"`
	Isdir bool   `json:isdir`
}

type FilesListJson struct {
	Flist []FileJson `json:"list"`
}

// 遍历监视目录，发送到json中
func NfTest(w http.ResponseWriter, r *http.Request) {
	// 监听的根目录
	realFilePath := filepath.Join(config.G.FilePath, ".")
	downloadPath := filepath.Join(filepath.Base(config.G.FilePath), r.URL.Path[1:])
	// 时间目录的情况
	fileInfo, err := os.Stat(realFilePath)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		return
	}

	data := struct {
		DeviceName   string
		Loip         string
		Rundir       string
		IsDir        bool
		FileName     string
		DownloadPath string
		// Files        []os.DirEntry
	}{
		DeviceName:   config.G.DeviceName,
		Loip:         config.G.LocalIP,
		Rundir:       config.G.FilePath,
		DownloadPath: downloadPath,
	}

	// 文件灌入数组中
	// var gmap map[string]interface{}
	// gmap = make(map[string]interface{})

	// list json
	var flist FilesListJson

	if fileInfo.IsDir() {
		data.IsDir = true
		// 遍历目录
		files, err := os.ReadDir(realFilePath)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		// data.Files = files

		for _, v := range files {
			// gmap[k]["filename"] = v.Name()
			// gmap[k]["isdir"] = v.IsDir()
			flist.Flist = append(flist.Flist, FileJson{Fname: v.Name(), Isdir: v.IsDir()})
		}

	} else {
		data.FileName = filepath.Base(realFilePath)
		flist.Flist = append(flist.Flist, FileJson{Fname: filepath.Base(realFilePath), Isdir: true})
	}

	// 绑定到jfile
	jdata, err := json.Marshal(flist)
	if err != nil {
		fmt.Printf("gmap has error:%v\n", err)
	}

	// respone file list
	response := Response{
		Status:  "success",
		Message: string(jdata),
	}
	//
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(response)
}