Merge branch 'refs/heads/feat-file'

dao/db.go

@@ -82,6 +82,18 @@ func Init() error {
 		return err
 	}
 
+	err = db.AutoMigrate(&ConfigFile{})
+	if err != nil {
+		fmt.Println("config file table:", err)
+		return err
+	}
+
+	err = db.AutoMigrate(&FileAuth{})
+	if err != nil {
+		fmt.Println("file auth table:", err)
+		return err
+	}
+
 	err = db.AutoMigrate(&Friend{})
 	if err != nil {
 		fmt.Println("friend table:", err)

dao/file.go

@@ -18,6 +18,23 @@ type File struct {
 	Md5           string `gorm:"column:md5;type:varchar(255);uniqueIndex:idx_file_name"`
 }
 
+type FileAuth struct {
+	gorm.Model
+	AuthID       int    `gorm:"column:auth_id"`
+	FileID       int    `gorm:"column:file_id"`
+	UserFileName string `gorm:"column:user_file_name;type:varchar(255);uniqueIndex:idx_file_name"` //对于同一个文件，不同用户有不同的文件名
+	UploadType   string `gorm:"column:upload_type"`                                                // 上传类型: im,avatar,file,config,config为系统文件
+	IsPrivate    int    `gorm:"column:is_private"`                                                 // 是否私有,私有文件只能自己下载或者通过分享链接下载,1为私有，0为公开
+	ShareCode    string `gorm:"column:share_code"`                                                 // 分享码，用于分享时的验证，构建分享链接
+}
+
+type ConfigFile struct {
+	gorm.Model
+	AuthID   int    `gorm:"column:auth_id"`
+	FileName string `gorm:"column:file_name"`
+	FilePath string `gorm:"column:file_path"`
+}
+
 func CreateFile(fileStoreName, fileName, fileType, filePath, md5Str string, fileSize, authID int, NeedAuth bool) File {
 	file := File{FileStoreName: fileStoreName, FileName: fileName, FileType: fileType, FilePath: filePath, FileSize: fileSize, AuthID: authID, NeedAuth: NeedAuth, Md5: md5Str}
 	result := DB.Create(&file)
@@ -96,3 +113,67 @@ func FindFileByMd5(md5 string) File {
 	}
 	return file
 }
+
+func CreateFileAuth(authID, fileID int, userFileName, uploadType string, isPrivate int, shareCode string) FileAuth {
+	fileAuth := FileAuth{AuthID: authID, FileID: fileID, UserFileName: userFileName, UploadType: uploadType, IsPrivate: isPrivate, ShareCode: shareCode}
+	var result *gorm.DB
+	if proto.Config.SERVER_SQL_LOG {
+		result = DB.Debug().Create(&fileAuth)
+	} else {
+		result = DB.Create(&fileAuth)
+	}
+	if result.Error != nil {
+		return FileAuth{}
+	}
+	return fileAuth
+}
+
+func CreateConfigFile(file ConfigFile) (id uint, err error) {
+	var result *gorm.DB
+	if proto.Config.SERVER_SQL_LOG {
+		result = DB.Debug().Create(&file)
+	} else {
+		result = DB.Create(&file)
+	}
+	return file.ID, result.Error
+}
+
+func FindConfigFileByID(id int, user_id int) ConfigFile {
+	var file ConfigFile
+	if proto.Config.SERVER_SQL_LOG {
+		DB.Debug().Where("id = ? and auth_id = ?", id, user_id).First(&file)
+	} else {
+		DB.Where("id = ? and auth_id = ?", id, user_id).First(&file)
+	}
+	return file
+}
+
+func DeleteConfigFileByID(id int) error {
+	var res *gorm.DB
+	if proto.Config.SERVER_SQL_LOG {
+		res = DB.Debug().Delete(&ConfigFile{}, id)
+	} else {
+		res = DB.Delete(&ConfigFile{}, id)
+	}
+	return res.Error
+}
+
+func UpdateConfigFileByID(id int, file ConfigFile) error {
+	var res *gorm.DB
+	if proto.Config.SERVER_SQL_LOG {
+		res = DB.Debug().Model(&ConfigFile{}).Where("id = ?", id).Updates(&file)
+	} else {
+		res = DB.Model(&ConfigFile{}).Where("id = ?", id).Updates(&file)
+	}
+	return res.Error
+}
+
+func FindConfigFileByAuthID(auth_id int) []ConfigFile {
+	var files []ConfigFile
+	if proto.Config.SERVER_SQL_LOG {
+		DB.Debug().Where("auth_id = ?", auth_id).Find(&files)
+	} else {
+		DB.Where("auth_id = ?", auth_id).Find(&files)
+	}
+	return files
+}

handler/file.go

@@ -0,0 +1,194 @@
+package handler
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"videoplayer/dao"
+	"videoplayer/proto"
+	"videoplayer/service"
+)
+
+func SetUpFileGroup(router *gin.Engine) {
+	fileGroup := router.Group("/file")
+	fileGroup.POST("/config/add", AddConfigFile)
+	fileGroup.POST("/config/delete", DeleteConfigFile)
+	fileGroup.POST("/config/update", UpdateConfigFile)
+	fileGroup.POST("/config/search", SearchConfigFile)
+	fileGroup.POST("/upload", UploadFileV2)
+	fileGroup.GET("/:filename", GetFile)
+
+}
+
+func AddConfigFile(c *gin.Context) {
+	id, _ := c.Get("id")
+	user_id := int(id.(float64))
+	var req proto.ConfigFileReq
+	if err := c.ShouldBind(&req); err == nil {
+		err2 := service.CreateConfigFile(&req, user_id)
+		if err2 != nil {
+			c.JSON(http.StatusOK, gin.H{"error": "add config file failed:" + err2.Error(), "code": proto.AddConfigFileFailed, "message": "failed"})
+			return
+		} else {
+			c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success"})
+			return
+		}
+
+	} else {
+		c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
+		return
+	}
+
+}
+
+func DeleteConfigFile(c *gin.Context) {
+	id, _ := c.Get("id")
+	user_id := int(id.(float64))
+	var req proto.ConfigFileReq
+	if err := c.ShouldBind(&req); err == nil {
+		err2 := service.DeleteConfigFile(&req, user_id)
+		if err2 != nil {
+			c.JSON(http.StatusOK, gin.H{"error": "delete config file failed:" + err2.Error(), "code": proto.DeleteConfigFailed, "message": "failed"})
+			return
+		} else {
+			c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success"})
+			return
+		}
+
+	} else {
+		c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
+		return
+	}
+}
+
+func UpdateConfigFile(c *gin.Context) {
+	id, _ := c.Get("id")
+	user_id := int(id.(float64))
+	var req proto.ConfigFileReq
+	if err := c.ShouldBind(&req); err == nil {
+		var configFileService service.ConfigFileService
+		err2 := configFileService.UpdateConfigFile(&req, user_id)
+		if err2 != nil {
+			c.JSON(http.StatusOK, gin.H{"error": "update config file failed:" + err2.Error(), "code": proto.UpdateConfigFailed, "message": "failed"})
+			return
+		} else {
+			c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success"})
+			return
+		}
+
+	} else {
+		c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
+		return
+	}
+}
+
+func SearchConfigFile(c *gin.Context) {
+	id, _ := c.Get("id")
+	user_id := int(id.(float64))
+	var req proto.ConfigFileReq
+	if err := c.ShouldBind(&req); err == nil {
+		var configFileService service.ConfigFileService
+		if req.Type == "one" {
+			//有文件内容
+			configFile, err2 := configFileService.SearchOneConfigFile(&req, user_id)
+			if err2 == nil {
+				c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "msg": "success", "data": configFile})
+			} else {
+				c.JSON(http.StatusOK, gin.H{"code": proto.SearchConfigFileFailed, "msg": "info:" + err2.Error(), "data": configFile})
+			}
+		} else if req.Type == "all" {
+			configFileList, err3 := configFileService.SearchAllConfigFile(user_id)
+			if err3 == nil {
+				c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "msg": "success", "data": configFileList})
+			} else {
+				c.JSON(http.StatusOK, gin.H{"code": proto.SearchConfigFileFailed, "msg": "info:" + err3.Error(), "data": configFileList})
+			}
+		} else {
+			c.JSON(http.StatusOK, gin.H{"error": "search config file type error", "code": proto.ParameterError, "message": "failed"})
+		}
+	} else {
+		c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
+		return
+	}
+}
+
+func UploadFileV2(c *gin.Context) {
+	//先查看是否有权限
+	id, _ := c.Get("id")
+	id1 := int(id.(float64))
+
+	var req proto.FileUploadReq
+	//获取post form参数
+	if err := c.ShouldBind(&req); err == nil {
+		//检查参数
+		if err2 := service.CheckUploadRequestParameters(&req); err2 != nil {
+			c.JSON(http.StatusOK, gin.H{"error": "upload form parameter check error:" + err2.Error(), "code": proto.ParameterError, "message": "failed"})
+			return
+		}
+		//
+
+	} else {
+		c.JSON(http.StatusOK, gin.H{"error": "upload form parameter decode error:" + err.Error(), "code": proto.ParameterError, "message": "failed"})
+		return
+	}
+
+	//从请求头获取upload_type
+	uploadType := c.PostForm("upload_type")
+	authType := c.PostForm("auth_type")
+	md5_ := c.PostForm("md5")
+	if uploadType == "" {
+		c.JSON(http.StatusOK, gin.H{"error": "upload_type is empty", "code": proto.ParameterError, "message": "failed"})
+		return
+	}
+
+	user := dao.FindUserByUserID(id1)
+	if user.Upload == false {
+		c.JSON(http.StatusOK, gin.H{"error": "no upload Permissions", "code": proto.NoUploadPermissions, "message": "failed"})
+		return
+	}
+	//上传文件
+	file, err := c.FormFile("file")
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{"error": "upload file failed", "code": proto.UploadFileFailed, "message": "failed"})
+		return
+	}
+	//计算文件md5值
+	if md5_ == "" {
+		file_, _ := file.Open()
+		md5_ = service.CalculateFileMd5(file_)
+		if md5_ == "" {
+			c.JSON(http.StatusOK, gin.H{"error": "计算文件MD5值失败", "code": proto.UploadFileFailed, "message": "failed"})
+			return
+		}
+	}
+	//查询文件是否已存在
+	fileExist := dao.FindFileByMd5(md5_)
+	if fileExist.ID != 0 {
+		fileExist.FilePath = ""
+		c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": fileExist})
+		return
+	}
+
+	//保存文件
+	filePath, fileStoreName, err := service.SaveFile(c, file, uploadType)
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{"error": "save file failed", "code": proto.SaveFileFailed, "message": "failed"})
+		return
+	}
+	//保存文件信息
+	fileSize := int(file.Size)
+	fileName := file.Filename
+	fileType := file.Header.Get("file_type")
+	var auth_type_ bool
+	if authType == "public" || authType == "" {
+		auth_type_ = false
+	} else if authType == "private" {
+		auth_type_ = true
+	}
+	file_record := dao.CreateFile(fileStoreName, fileName, fileType, filePath, md5_, fileSize, id1, auth_type_)
+	if file_record.ID == 0 {
+		c.JSON(http.StatusOK, gin.H{"error": "save file info failed", "code": proto.SaveFileInfoFailed, "message": "failed"})
+		return
+	}
+	file_record.FilePath = ""
+	c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": file_record})
+}

proto/conf.go

@@ -11,7 +11,7 @@ var Config ConfigStruct
 var SigningKey = []byte{}
 var Url_map = map[string]bool{"/login": true, "/register": true, "/uuid": true, "/gqr": true, "/cid/callback": true, "/tool/monitor": true, "/user/sync": true, "/tool/file/": true} // 不需要token验证的url
 var Per_menu_map = map[string]int{"/video/": 1, "/device/": 2, "/cid/": 3}
-
+var File_Type = map[string]int{"im": 1, "avatar": 2, "file": 3, "config": 4} // 文件类型
 const (
 	MYSQL_USER     = "video_t2"
 	MYSQL_DB       = "video_t2"

proto/file_req.go

@@ -0,0 +1,28 @@
+package proto
+
+import "time"
+
+type FileUploadReq struct {
+	UploadType string `json:"upload_type" form:"upload_type" binding:"required"`
+	AuthType   string `json:"auth_type" form:"auth_type"`
+	Md5        string `json:"md5" form:"md5"`   //文件md5值
+	Type       string `json:"type" form:"type"` //类型，im,avatar,file,config,config为系统文件
+}
+
+type ConfigFileReq struct {
+	ID       int    `json:"id" form:"id" binding:"required"`
+	Type     string `json:"type" form:"type" binding:"required"` //查询类型，one,all
+	DelFile  bool   `json:"del_file" form:"del_file"`            //删除文件
+	FileName string `json:"file_name" form:"file_name" binding:"required"`
+	FilePath string `json:"file_path" form:"file_path" binding:"required"`
+	Content  string `json:"content" form:"content" binding:"required"`
+}
+
+type SearchOneConfigFileResp struct {
+	ID        uint      `json:"id" form:"id"`
+	CreatedAt time.Time `json:"created_at" form:"created_at"`
+	UpdatedAt time.Time `json:"updated_at" form:"updated_at"`
+	FileName  string    `json:"file_name" form:"file_name"`
+	FilePath  string    `json:"file_path" form:"file_path"`
+	Content   string    `json:"content" form:"content"`
+}

proto/status.go

@@ -63,4 +63,9 @@ const (
 	DeleteFileInfoFailed = 78 // 删除文件信息失败
 
 	DataFormatError = 80 // 数据格式错误
+
+	AddConfigFileFailed    = 90 // 添加配置文件失败
+	UpdateConfigFailed     = 91 // 更新配置失败
+	DeleteConfigFailed     = 92 // 删除配置失败
+	SearchConfigFileFailed = 93 // 获取配置失败
 )

service/fileService.go

@@ -10,6 +10,7 @@ import (
 	"os"
 	"path"
 	"time"
+	"videoplayer/dao"
 	"videoplayer/proto"
 	"videoplayer/worker"
 )
@@ -55,3 +56,152 @@ func CalculateFileMd5(file io.Reader) string {
 	}
 	return fmt.Sprintf("%x", hash.Sum(nil))
 }
+
+func CheckUploadRequestParameters(req *proto.FileUploadReq) error {
+	var err error
+	if req.AuthType == "" {
+		err = fmt.Errorf("auth_type is empty")
+	}
+	if req.UploadType == "" {
+		req.UploadType = "1"
+	}
+	if req.UploadType != "1" {
+		req.UploadType = "2"
+	}
+	if proto.File_Type[req.Type] == 0 {
+		err = fmt.Errorf("file type is invalid")
+	}
+	return err
+}
+
+func CreateConfigFile(req *proto.ConfigFileReq, userId int) error {
+	var err error
+	user := GetUserByIDWithCache(userId)
+	if user.ID == 0 || user.Role != "admin" {
+		err = fmt.Errorf("user not found or no permission")
+		return err
+	}
+	//查看系统中是否存在文件,不存在则创建
+	file := req.FilePath + "/" + req.FileName
+	_, err = os.Stat(file)
+	if err != nil {
+		//创建文件
+		f, err2 := os.Create(file)
+		if err2 != nil {
+			err = err2
+			return err
+		}
+		defer func(f *os.File) {
+			err := f.Close()
+			if err != nil {
+				fmt.Println("Error closing file")
+			}
+		}(f)
+	}
+	//创建
+	configFile := dao.ConfigFile{FilePath: req.FilePath, FileName: req.FileName, AuthID: userId}
+	_, err3 := dao.CreateConfigFile(configFile)
+	if err3 != nil {
+		err = err3
+		return err
+	}
+	return err
+}
+
+func DeleteConfigFile(req *proto.ConfigFileReq, userId int) error {
+	var err error
+	user := GetUserByIDWithCache(userId)
+	if user.ID == 0 || user.Role != "admin" {
+		err = fmt.Errorf("user not found or no permission")
+		return err
+	}
+	//删除文件
+	config_file := dao.FindConfigFileByID(req.ID, userId)
+	if config_file.ID == 0 {
+		err = fmt.Errorf("config file not found")
+		return err
+	}
+	err = dao.DeleteConfigFileByID(req.ID)
+	if req.DelFile {
+		file := config_file.FilePath + "/" + config_file.FileName
+		err = os.Remove(file)
+		if err != nil {
+			return err
+		}
+	}
+	//删除数据库记录
+	return err
+}
+
+type ConfigFileService struct {
+}
+
+func (c *ConfigFileService) UpdateConfigFile(req *proto.ConfigFileReq, userId int) error {
+	var err error
+	user := GetUserByIDWithCache(userId)
+	if user.ID == 0 || user.Role != "admin" {
+		err = fmt.Errorf("user not found or no permission")
+		return err
+	}
+	config_file := dao.FindConfigFileByID(req.ID, userId)
+	if config_file.ID == 0 {
+		err = fmt.Errorf("config file not found")
+		return err
+	}
+	//修改文件名
+	if req.FileName != "" {
+		file := config_file.FilePath + "/" + config_file.FileName
+		new_file := config_file.FilePath + "/" + req.FileName
+		err = os.Rename(file, new_file)
+		if err != nil {
+			return err
+		}
+		err = dao.UpdateConfigFileByID(req.ID, dao.ConfigFile{FileName: req.FileName})
+	}
+	if req.Content != "" {
+		file := config_file.FilePath + "/" + config_file.FileName
+		f, err2 := os.OpenFile(file, os.O_WRONLY|os.O_TRUNC, 0644) //打开文件，清空文件内容，写入新内容，不存在则创建
+		if err2 != nil {
+			err = err2
+			return err
+		}
+		defer func(f *os.File) {
+			err3 := f.Close()
+			if err3 != nil {
+				fmt.Println("Error closing file")
+			}
+		}(f)
+		_, err = f.WriteString(req.Content)
+		if err != nil {
+			return err
+		}
+	}
+	return err
+}
+
+func (c *ConfigFileService) SearchOneConfigFile(req *proto.ConfigFileReq, userId int) ([]proto.SearchOneConfigFileResp, error) {
+	user := GetUserByIDWithCache(userId)
+	if user.ID == 0 || user.Role != "admin" {
+		return []proto.SearchOneConfigFileResp{}, fmt.Errorf("user not found or no permission")
+	}
+	config_file := dao.FindConfigFileByID(req.ID, userId)
+	if config_file.ID == 0 {
+		return []proto.SearchOneConfigFileResp{}, fmt.Errorf("config file not found")
+	}
+	file := config_file.FilePath + "/" + config_file.FileName
+	content, err2 := os.ReadFile(file)
+	if err2 != nil {
+		return []proto.SearchOneConfigFileResp{}, err2
+	}
+	resp := []proto.SearchOneConfigFileResp{{ID: config_file.ID, FileName: config_file.FileName, Content: string(content), CreatedAt: config_file.CreatedAt, UpdatedAt: config_file.UpdatedAt}}
+	return resp, nil
+}
+
+func (c *ConfigFileService) SearchAllConfigFile(userId int) ([]dao.ConfigFile, error) {
+	user := GetUserByIDWithCache(userId)
+	if user.ID == 0 || user.Role != "admin" {
+		return []dao.ConfigFile{}, fmt.Errorf("user not found or no permission")
+	}
+	config_files := dao.FindConfigFileByAuthID(userId)
+	return config_files, nil
+}

service/userService.go

@@ -1,6 +1,7 @@
 package service
 
 import (
+	"encoding/json"
 	"errors"
 	"fmt"
 	"regexp"
@@ -51,6 +52,38 @@ func GetUserByID(id int) []proto.User {
 	return dao.FindUserByID(id)
 }
 
+// 获取用户信息，有redis缓存
+func GetUserByIDWithCache(id int) dao.User {
+	if id <= 0 {
+		return dao.User{}
+	}
+	var user dao.User
+	//先从redis获取
+	key := "user_info_" + strconv.Itoa(id)
+	user_str := worker.GetRedis(key)
+	if user_str != "" {
+		err := json.Unmarshal([]byte(user_str), &user)
+		if err != nil {
+			fmt.Println("get user info , json unmarshal error:", err, "\tuser_str:", user_str)
+			return dao.User{}
+		}
+	} else {
+		user = dao.FindUserByID2(id)
+		if user.ID != 0 {
+			userJson, err := json.Marshal(user)
+			if err != nil {
+				fmt.Println("get user info , json marshal error:", err)
+				return dao.User{}
+			}
+			success := worker.SetRedis(key, string(userJson))
+			if !success {
+				fmt.Println("set redis error,user json:", string(userJson))
+			}
+		}
+	}
+	return user
+}
+
 func GetUserByNameLike(name string) []proto.User {
 	return dao.FindUserByNameLike(name)
 }
@@ -63,6 +96,7 @@ func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) {
 		//添加修改用户信息到同步列表
 		if err == nil {
 			err2 := setSyncUserDataSet("update", user_id)
+			UpdateUserCache(user_id)
 			if err2 != nil {
 				fmt.Println("set sync user data set error:", err2)
 				return user_id, nil
@@ -74,6 +108,7 @@ func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) {
 		if err == nil {
 			//添加修改用户信息到同步列表
 			err2 := setSyncUserDataSet("update", req.ID)
+			UpdateUserCache(req.ID)
 			if err2 != nil {
 				fmt.Println("set sync user data set error:", err2)
 				return req.ID, nil
@@ -85,6 +120,21 @@ func UpdateUser(user_id int, req proto.UpdateUserInfoReq) (int, error) {
 	}
 }
 
+func UpdateUserCache(id int) {
+	key := "user_info_" + strconv.Itoa(id)
+	if worker.IsContainKey(key) {
+		user := GetUserByID(id)
+		userJson, err := json.Marshal(user)
+		if err != nil {
+			fmt.Println("get user info , json marshal error:", err)
+		}
+		success := worker.SetRedis(key, string(userJson))
+		if !success {
+			fmt.Println("set redis error,user json:", string(userJson))
+		}
+	}
+}
+
 func DeleteUserService(id, user_id int) int {
 	res := 0
 	if user_id == id {