CasaOS/route/v1/persion.go

package v1

import (
	"encoding/json"
	"net/http"
	"time"

	"github.com/IceWhaleTech/CasaOS/model"
	"github.com/IceWhaleTech/CasaOS/pkg/config"
	oasis_err2 "github.com/IceWhaleTech/CasaOS/pkg/utils/oasis_err"
	"github.com/IceWhaleTech/CasaOS/service"
	model2 "github.com/IceWhaleTech/CasaOS/service/model"
	"github.com/IceWhaleTech/CasaOS/types"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	uuid "github.com/satori/go.uuid"
)

func PersonTest(c *gin.Context) {

	//service.MyService.Person().GetPersionInfo("fb2333a1-72b2-4cb4-9e31-61ccaffa55b9")

	m := model.ConnectState{}
	m.CreatedAt = time.Now()
	m.From = config.ServerInfo.Token
	m.To = "fb2333a1-72b2-4cb4-9e31-61ccaffa55b9"
	m.Type = ""
	m.UUId = uuid.NewV4().String()

	//service.MyService.Person().Handshake(m)
	msg := model.MessageModel{}
	msg.Type = "connection"
	msg.Data = "fb2333a1-72b2-4cb4-9e31-61ccaffa55b9"
	msg.From = config.ServerInfo.Token
	msg.UUId = "1234567890"
	b, _ := json.Marshal(msg)
	err := service.WebSocketConn.WriteMessage(websocket.TextMessage, b)
	if err == nil {
		return
	}
}

//get other persion file
func GetPersionFile(c *gin.Context) {
	path := c.Query("path")
	persion := c.Query("persion")
	if len(path) == 0 && len(persion) == 0 {
		c.JSON(http.StatusOK, model.Result{Success: oasis_err2.INVALID_PARAMS, Message: oasis_err2.GetMsg(oasis_err2.INVALID_PARAMS)})
		return
	}
	//任务标识
	uuid := uuid.NewV4().String()

	//1.通知对方需要下载
	service.MyService.Person().GetFileDetail(uuid, path, persion)

	//2.添加数据库

	task := model2.PersionDownloadDBModel{}
	task.UUID = uuid
	task.Name = ""
	task.Length = 0
	task.Size = 0
	task.State = types.DOWNLOADAWAIT
	task.TempPath = ""
	task.Type = 0
	service.MyService.Person().AddDownloadTask(task)

	c.JSON(http.StatusOK, model.Result{Success: oasis_err2.SUCCESS, Message: oasis_err2.GetMsg(oasis_err2.SUCCESS)})
}
func GetPersionDownloadList(c *gin.Context) {
	path := c.Query("path")
	persion := c.Query("persion")
	if len(path) == 0 && len(persion) == 0 {
		c.JSON(http.StatusOK, model.Result{Success: oasis_err2.INVALID_PARAMS, Message: oasis_err2.GetMsg(oasis_err2.INVALID_PARAMS)})
		return
	}
	//任务标识
	uuid := uuid.NewV4().String()

	//1.通知对方需要下载
	service.MyService.Person().GetFileDetail(uuid, path, persion)

	c.JSON(http.StatusOK, model.Result{Success: oasis_err2.SUCCESS, Message: oasis_err2.GetMsg(oasis_err2.SUCCESS)})
}