cloud_disk.go 28 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001
  1. package controllers
  2. import (
  3. "archive/zip"
  4. "encoding/json"
  5. "eta/eta_api/models"
  6. "eta/eta_api/models/company"
  7. "eta/eta_api/services"
  8. "eta/eta_api/services/alarm_msg"
  9. "eta/eta_api/utils"
  10. "fmt"
  11. "github.com/rdlucklib/rdluck_tools/http"
  12. "os"
  13. "path"
  14. "strconv"
  15. "strings"
  16. "time"
  17. )
  18. // CloudDiskController 共享云盘
  19. type CloudDiskController struct {
  20. BaseAuthController
  21. }
  22. // MenuCreate 创建文件夹
  23. // @Title 创建文件夹
  24. // @Description 创建文件夹
  25. // @Param request body models.CloudDiskMenuCreateReq true "type json string"
  26. // @Success 200 Ret=200 操作成功
  27. // @router /menu/create [post]
  28. func (this *CloudDiskController) MenuCreate() {
  29. br := new(models.BaseResponse).Init()
  30. defer func() {
  31. this.Data["json"] = br
  32. this.ServeJSON()
  33. }()
  34. var req models.CloudDiskMenuCreateReq
  35. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  36. br.Msg = "参数解析异常!"
  37. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  38. return
  39. }
  40. req.MenuName = strings.TrimSpace(req.MenuName)
  41. if req.MenuName == "" {
  42. br.Msg = "文件夹名称不可为空"
  43. return
  44. }
  45. // 权限校验
  46. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  47. if e != nil {
  48. br.Msg = "操作失败"
  49. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  50. return
  51. }
  52. if !ok {
  53. br.Msg = "无权操作"
  54. return
  55. }
  56. // 重名校验
  57. existItem := new(models.CloudDiskMenu)
  58. existCond := ` AND menu_name = ? AND parent_id = ? `
  59. existPars := make([]interface{}, 0)
  60. existPars = append(existPars, req.MenuName, req.ParentId)
  61. if e := existItem.GetItemByCondition(existCond, existPars); e != nil && e.Error() != utils.ErrNoRow() {
  62. br.Msg = "操作失败"
  63. br.ErrMsg = "获取重名目录失败, Err: " + e.Error()
  64. return
  65. }
  66. if existItem != nil && existItem.MenuId > 0 {
  67. br.Msg = "该文件夹已存在"
  68. return
  69. }
  70. nowTime := time.Now().Local()
  71. newMenu := &models.CloudDiskMenu{
  72. MenuName: req.MenuName,
  73. ParentId: req.ParentId,
  74. AdminId: this.SysUser.AdminId,
  75. AdminName: this.SysUser.RealName,
  76. CreateTime: nowTime,
  77. ModifyTime: nowTime,
  78. }
  79. if e := newMenu.Create(); e != nil {
  80. br.Msg = "操作失败"
  81. br.ErrMsg = "新增目录失败, Err: " + e.Error()
  82. return
  83. }
  84. br.Ret = 200
  85. br.Success = true
  86. br.Msg = "操作成功"
  87. }
  88. // MenuRename 重命名文件夹
  89. // @Title 重命名文件夹
  90. // @Description 重命名文件夹
  91. // @Param request body models.CloudDiskMenuRenameReq true "type json string"
  92. // @Success 200 Ret=200 操作成功
  93. // @router /menu/rename [post]
  94. func (this *CloudDiskController) MenuRename() {
  95. br := new(models.BaseResponse).Init()
  96. defer func() {
  97. this.Data["json"] = br
  98. this.ServeJSON()
  99. }()
  100. var req models.CloudDiskMenuRenameReq
  101. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  102. br.Msg = "参数解析异常!"
  103. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  104. return
  105. }
  106. if req.MenuId <= 0 {
  107. br.Msg = "参数有误"
  108. return
  109. }
  110. req.MenuName = strings.TrimSpace(req.MenuName)
  111. if req.MenuName == "" {
  112. br.Msg = "文件夹名称不可为空"
  113. return
  114. }
  115. // 权限校验
  116. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  117. if e != nil {
  118. br.Msg = "操作失败"
  119. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  120. return
  121. }
  122. if !ok {
  123. br.Msg = "无权操作"
  124. return
  125. }
  126. // 重名校验
  127. menuItem := new(models.CloudDiskMenu)
  128. if e := menuItem.GetItemById(req.MenuId); e != nil {
  129. br.Msg = "操作失败"
  130. br.ErrMsg = "获取目录信息失败, Err: " + e.Error()
  131. return
  132. }
  133. existItem := new(models.CloudDiskMenu)
  134. existCond := ` AND menu_name = ? AND parent_id = ? AND menu_id <> ? `
  135. existPars := make([]interface{}, 0)
  136. existPars = append(existPars, req.MenuName, menuItem.ParentId, menuItem.MenuId)
  137. if e := existItem.GetItemByCondition(existCond, existPars); e != nil && e.Error() != utils.ErrNoRow() {
  138. br.Msg = "操作失败"
  139. br.ErrMsg = "获取重名目录失败, Err: " + e.Error()
  140. return
  141. }
  142. if existItem != nil && existItem.MenuId > 0 {
  143. br.Msg = "该文件夹已存在"
  144. return
  145. }
  146. menuItem.MenuName = req.MenuName
  147. menuItem.ModifyTime = time.Now().Local()
  148. if e := menuItem.Update([]string{"MenuName", "ModifyTime"}); e != nil {
  149. br.Msg = "操作失败"
  150. br.ErrMsg = "目录重命名失败, Err: " + e.Error()
  151. return
  152. }
  153. br.Ret = 200
  154. br.Success = true
  155. br.Msg = "操作成功"
  156. }
  157. // MenuDelete 删除文件夹
  158. // @Title 删除文件夹
  159. // @Description 删除文件夹
  160. // @Param request body models.CloudDiskMenuDeleteReq true "type json string"
  161. // @Success 200 Ret=200 操作成功
  162. // @router /menu/delete [post]
  163. func (this *CloudDiskController) MenuDelete() {
  164. br := new(models.BaseResponse).Init()
  165. defer func() {
  166. this.Data["json"] = br
  167. this.ServeJSON()
  168. }()
  169. var req models.CloudDiskMenuDeleteReq
  170. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  171. br.Msg = "参数解析异常!"
  172. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  173. return
  174. }
  175. if req.MenuId <= 0 {
  176. br.Msg = "参数有误"
  177. return
  178. }
  179. // 权限校验
  180. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  181. if e != nil {
  182. br.Msg = "操作失败"
  183. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  184. return
  185. }
  186. if !ok {
  187. br.Msg = "无权操作"
  188. return
  189. }
  190. // 获取文件夹子目录IDs
  191. menuIds := make([]int, 0)
  192. menuIds = append(menuIds, req.MenuId)
  193. childIds, e := services.GetCloudDiskMenuIdsByParentId(req.MenuId)
  194. if e != nil {
  195. br.Msg = "操作失败"
  196. br.ErrMsg = "获取子目录IDs失败, ErrMsg: " + e.Error()
  197. return
  198. }
  199. menuIds = append(menuIds, childIds...)
  200. // 删除目录及目录下所有文件
  201. if e = models.DeleteCloudDiskMenuAndResourcesByMenuIds(menuIds); e != nil {
  202. br.Msg = "操作失败"
  203. br.ErrMsg = "删除目录及文件失败, Err: " + e.Error()
  204. return
  205. }
  206. br.Ret = 200
  207. br.Success = true
  208. br.Msg = "操作成功"
  209. }
  210. // MenuTree 目录分类树
  211. // @Title 目录分类树
  212. // @Description 目录分类树
  213. // @Success 200 Ret=200 获取成功
  214. // @router /menu/tree [get]
  215. func (this *CloudDiskController) MenuTree() {
  216. br := new(models.BaseResponse).Init()
  217. defer func() {
  218. this.Data["json"] = br
  219. this.ServeJSON()
  220. }()
  221. treeData, e := services.GetCloudDiskMenuTree()
  222. if e != nil {
  223. br.Msg = "获取失败"
  224. br.ErrMsg = "获取目录分类树失败, Err: " + e.Error()
  225. return
  226. }
  227. br.Ret = 200
  228. br.Success = true
  229. br.Msg = "获取成功"
  230. br.Data = treeData
  231. }
  232. // MenuResources 目录文件
  233. // @Title 目录文件
  234. // @Description 获取目录下所有文件
  235. // @Param MenuId query int true "目录ID"
  236. // @Success 200 Ret=200 获取成功
  237. // @router /menu/resources [get]
  238. func (this *CloudDiskController) MenuResources() {
  239. br := new(models.BaseResponse).Init()
  240. defer func() {
  241. this.Data["json"] = br
  242. this.ServeJSON()
  243. }()
  244. menuId, _ := this.GetInt("MenuId", 0)
  245. if menuId <= 0 {
  246. br.Msg = "参数有误"
  247. return
  248. }
  249. menuItem := new(models.CloudDiskMenu)
  250. if e := menuItem.GetItemById(menuId); e != nil {
  251. br.Msg = "目录已被删除, 请刷新页面"
  252. br.ErrMsg = "获取目录信息失败, Err: " + e.Error()
  253. return
  254. }
  255. // 获取该目录子目录IDs
  256. menuIds := make([]int, 0)
  257. menuIds = append(menuIds, menuId)
  258. childIds, e := services.GetCloudDiskMenuIdsByParentId(menuId)
  259. if e != nil {
  260. br.Msg = "操作失败"
  261. br.ErrMsg = "获取子目录IDs失败, ErrMsg: " + e.Error()
  262. return
  263. }
  264. menuIds = append(menuIds, childIds...)
  265. // 获取目录下所有文件
  266. resourceOB := new(models.CloudDiskResource)
  267. resourceCond := ` AND menu_id IN (` + utils.GetOrmInReplace(len(menuIds)) + `) `
  268. resourcePars := make([]interface{}, 0)
  269. resourcePars = append(resourcePars, menuIds)
  270. resourceFieldArr := []string{"resource_id", "resource_name", "resource_suffix", "resource_url", "menu_id"}
  271. resourceList, e := resourceOB.GetItemsByCondition(resourceCond, resourcePars, resourceFieldArr, "create_time ASC")
  272. if e != nil {
  273. br.Msg = "操作失败"
  274. br.ErrMsg = "获取目录下所有文件失败, Err: " + e.Error()
  275. return
  276. }
  277. // 获取目录列表, 取出parentMap
  278. listCond := ``
  279. listPars := make([]interface{}, 0)
  280. menuList, e := menuItem.GetItemsByCondition(listCond, listPars, []string{}, "")
  281. if e != nil {
  282. br.Msg = "操作失败"
  283. br.ErrMsg = "获取目录列表失败, Err: " + e.Error()
  284. return
  285. }
  286. menuMap := make(map[int]*models.CloudDiskMenu)
  287. for i := range menuList {
  288. menuMap[menuList[i].MenuId] = menuList[i]
  289. }
  290. type MenuResource struct {
  291. ResourceId int
  292. ResourceName string
  293. ResourceSuffix string
  294. ResourceUrl string
  295. OriginName string
  296. }
  297. respList := make([]*MenuResource, 0)
  298. for i := range resourceList {
  299. n := resourceList[i].ResourceName + resourceList[i].ResourceSuffix
  300. p := services.FillMenuPath2File(resourceList[i].MenuId, resourceList[i].MenuId, n, menuMap) // 完整的文件路径
  301. respList = append(respList, &MenuResource{
  302. ResourceId: resourceList[i].ResourceId,
  303. ResourceName: p,
  304. ResourceSuffix: resourceList[i].ResourceSuffix,
  305. ResourceUrl: resourceList[i].ResourceUrl,
  306. OriginName: n,
  307. })
  308. }
  309. br.Ret = 200
  310. br.Success = true
  311. br.Msg = "获取成功"
  312. br.Data = respList
  313. }
  314. // MenuDownload 目录文件打包下载
  315. // @Title 目录文件打包下载
  316. // @Description 目录文件打包下载
  317. // @Param MenuId query int true "目录ID"
  318. // @Success 200 Ret=200 操作成功
  319. // @router /menu/download [get]
  320. func (this *CloudDiskController) MenuDownload() {
  321. br := new(models.BaseResponse).Init()
  322. defer func() {
  323. this.Data["json"] = br
  324. this.ServeJSON()
  325. }()
  326. menuId, _ := this.GetInt("MenuId", 0)
  327. if menuId <= 0 {
  328. br.Msg = "参数有误"
  329. return
  330. }
  331. menuItem := new(models.CloudDiskMenu)
  332. if e := menuItem.GetItemById(menuId); e != nil {
  333. br.Msg = "目录已被删除, 请刷新页面"
  334. br.ErrMsg = "获取目录信息失败, Err: " + e.Error()
  335. return
  336. }
  337. // 获取该目录子目录IDs
  338. menuIds := make([]int, 0)
  339. menuIds = append(menuIds, menuId)
  340. childIds, e := services.GetCloudDiskMenuIdsByParentId(menuId)
  341. if e != nil {
  342. br.Msg = "操作失败"
  343. br.ErrMsg = "获取子目录IDs失败, ErrMsg: " + e.Error()
  344. return
  345. }
  346. menuIds = append(menuIds, childIds...)
  347. // 获取目录下所有文件
  348. resourceOB := new(models.CloudDiskResource)
  349. resourceCond := ` AND menu_id IN (` + utils.GetOrmInReplace(len(menuIds)) + `) `
  350. resourcePars := make([]interface{}, 0)
  351. resourcePars = append(resourcePars, menuIds)
  352. resourceFieldArr := []string{"resource_id", "resource_name", "resource_suffix", "resource_url"}
  353. resourceList, e := resourceOB.GetItemsByCondition(resourceCond, resourcePars, resourceFieldArr, "create_time ASC")
  354. if e != nil {
  355. br.Msg = "操作失败"
  356. br.ErrMsg = "获取目录下所有文件失败, Err: " + e.Error()
  357. return
  358. }
  359. if len(resourceList) == 0 {
  360. br.Msg = "无文件可下载"
  361. return
  362. }
  363. // 创建zip
  364. zipName := time.Now().Format(utils.FormatDateTimeUnSpace) + utils.GetRandString(5) + ".zip"
  365. savePath := zipName
  366. zipFile, err := os.Create(zipName)
  367. if err != nil {
  368. br.Msg = "操作失败"
  369. br.ErrMsg = "生成压缩文件失败, Err: " + e.Error()
  370. return
  371. }
  372. zipWriter := zip.NewWriter(zipFile)
  373. // 生成zip过程中报错关闭
  374. defer func() {
  375. if err != nil {
  376. _ = zipWriter.Close()
  377. _ = zipFile.Close()
  378. }
  379. _ = os.Remove(savePath)
  380. }()
  381. // 获取资源, 写入zip
  382. for i := range resourceList {
  383. if resourceList[i].ResourceName == "" || resourceList[i].ResourceSuffix == "" || resourceList[i].ResourceUrl == "" {
  384. continue
  385. }
  386. fmt.Printf("开始压缩第%d个文件\n", i+1)
  387. ioWriter, err := zipWriter.Create(fmt.Sprintf("%s.%s", resourceList[i].ResourceName, resourceList[i].ResourceSuffix))
  388. if err != nil {
  389. if os.IsPermission(err) {
  390. br.Msg = "操作失败"
  391. br.ErrMsg = "打包权限不足, Err: " + err.Error()
  392. return
  393. }
  394. br.Msg = "操作失败"
  395. br.ErrMsg = "压缩出错, Err: " + err.Error()
  396. return
  397. }
  398. var content []byte
  399. content, err = http.Get(resourceList[i].ResourceUrl)
  400. if err != nil {
  401. br.Msg = "操作失败"
  402. br.ErrMsg = "资源获取失败, Err: " + err.Error()
  403. return
  404. }
  405. _, err = ioWriter.Write(content)
  406. if err != nil {
  407. br.Msg = "操作失败"
  408. br.ErrMsg = "压缩文件写入失败, Err: " + err.Error()
  409. return
  410. }
  411. fmt.Printf("第%d个文件写入成功\n", i+1)
  412. }
  413. // 生成zip后关闭,否则下载文件会损坏
  414. _ = zipWriter.Close()
  415. _ = zipFile.Close()
  416. this.Ctx.Output.Download(savePath, fmt.Sprintf("%s.zip", menuItem.MenuName))
  417. br.Ret = 200
  418. br.Msg = "下载成功"
  419. br.Success = true
  420. }
  421. // CheckResourceRepeat 校验目录下文件重名
  422. // @Title 校验目录下文件重名
  423. // @Description 校验目录下文件重名
  424. // @Param request body models.CloudDiskCheckResourceRepeatReq true "type json string"
  425. // @Success 200 Ret=200 操作成功
  426. // @router /resource/check_repeat [post]
  427. func (this *CloudDiskController) CheckResourceRepeat() {
  428. br := new(models.BaseResponse).Init()
  429. defer func() {
  430. this.Data["json"] = br
  431. this.ServeJSON()
  432. }()
  433. var req models.CloudDiskCheckResourceRepeatReq
  434. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  435. br.Msg = "参数解析异常!"
  436. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  437. return
  438. }
  439. // 检测接口允许menuId为0, 上传不允许
  440. if len(req.FileNames) == 0 {
  441. br.Msg = "请选择文件"
  442. return
  443. }
  444. // 获取该级目录下的文件
  445. resourceOB := new(models.CloudDiskResource)
  446. resourceCond := ` AND menu_id = ? `
  447. resourcePars := make([]interface{}, 0)
  448. resourcePars = append(resourcePars, req.MenuId)
  449. resourceFieldArr := []string{"resource_id", "resource_name", "resource_suffix"}
  450. resourceList, e := resourceOB.GetItemsByCondition(resourceCond, resourcePars, resourceFieldArr, "create_time ASC")
  451. if e != nil {
  452. br.Msg = "操作失败"
  453. br.ErrMsg = "获取目录下文件失败, Err: " + e.Error()
  454. return
  455. }
  456. resourceLen := len(resourceList)
  457. respList := make([]*models.CloudDiskCheckResourceRepeatResp, 0)
  458. for i := range req.FileNames {
  459. v := new(models.CloudDiskCheckResourceRepeatResp)
  460. v.OriginName = req.FileNames[i]
  461. r := false
  462. a := v.OriginName
  463. if resourceLen > 0 {
  464. r, a = services.CheckFileNameFormFiles(v.OriginName, "", resourceList, 0)
  465. }
  466. v.IsRepeat = r
  467. v.AvailableName = a
  468. respList = append(respList, v)
  469. }
  470. br.Ret = 200
  471. br.Success = true
  472. br.Msg = "操作成功"
  473. br.Data = respList
  474. }
  475. // ResourceUpload 上传文件
  476. // @Title 上传文件
  477. // @Description 上传文件
  478. // @Param MenuId query int true "目录ID"
  479. // @Param File query file true "文件"
  480. // @Success 200 Ret=200 操作成功
  481. // @router /resource/upload [post]
  482. func (this *CloudDiskController) ResourceUpload() {
  483. br := new(models.BaseResponse).Init()
  484. defer func() {
  485. if br.ErrMsg == "" {
  486. br.IsSendEmail = false
  487. }
  488. this.Data["json"] = br
  489. this.ServeJSON()
  490. }()
  491. sysUser := this.SysUser
  492. if sysUser == nil {
  493. br.Msg = "请登录"
  494. br.ErrMsg = "请登录,SysUser Is Empty"
  495. br.Ret = 408
  496. return
  497. }
  498. menuId, _ := this.GetInt("MenuId", 0)
  499. if menuId <= 0 {
  500. br.Msg = "不允许上传文件至根目录"
  501. return
  502. }
  503. originName := this.GetString("OriginName", "")
  504. if originName == "" {
  505. br.Msg = "请上传文件名"
  506. return
  507. }
  508. if !strings.Contains(originName, ".") {
  509. br.Msg = "文件名有误"
  510. return
  511. }
  512. f, h, e := this.GetFile("file")
  513. if e != nil {
  514. br.Msg = "获取资源信息失败"
  515. br.ErrMsg = "获取资源信息失败, Err:" + e.Error()
  516. return
  517. }
  518. defer func() {
  519. _ = f.Close()
  520. }()
  521. extIndex := strings.LastIndex(originName, ".")
  522. fileName := originName[:extIndex]
  523. // 限制文件类型
  524. confVal, e := company.GetConfigDetailByCode(company.ConfCloudDiskAllowExt)
  525. if e != nil && e.Error() != utils.ErrNoRow() {
  526. br.Msg = "操作失败"
  527. br.ErrMsg = "文件上传失败, Err: " + e.Error()
  528. return
  529. }
  530. allowExt := make([]string, 0)
  531. if confVal.ConfigValue != "" {
  532. allowExt = strings.Split(confVal.ConfigValue, ",")
  533. }
  534. ext := path.Ext(h.Filename)
  535. if !utils.InArrayByStr(allowExt, ext) {
  536. br.Msg = "不允许上传该格式文件"
  537. return
  538. }
  539. // 重名校验
  540. existItem := new(models.CloudDiskResource)
  541. existCond := ` AND resource_name = ? AND menu_id = ? `
  542. existPars := make([]interface{}, 0)
  543. existPars = append(existPars, fileName, menuId)
  544. if e := existItem.GetItemByCondition(existCond, existPars); e != nil && e.Error() != utils.ErrNoRow() {
  545. br.Msg = "操作失败"
  546. br.ErrMsg = "获取重名文件失败, Err: " + e.Error()
  547. return
  548. }
  549. if existItem != nil && existItem.MenuId > 0 {
  550. br.Msg = fmt.Sprintf("文件名%s已存在", fileName)
  551. return
  552. }
  553. uploadDir := utils.STATIC_DIR + "hongze/" + time.Now().Format("20060102")
  554. if e = os.MkdirAll(uploadDir, utils.DIR_MOD); e != nil {
  555. br.Msg = "存储目录创建失败"
  556. br.ErrMsg = "存储目录创建失败, Err:" + e.Error()
  557. return
  558. }
  559. ossFileName := utils.GetRandStringNoSpecialChar(28) + ext
  560. filePath := uploadDir + "/" + ossFileName
  561. if e = this.SaveToFile("file", filePath); e != nil {
  562. br.Msg = "文件保存失败"
  563. br.ErrMsg = "文件保存失败, Err:" + e.Error()
  564. return
  565. }
  566. defer func() {
  567. _ = os.Remove(filePath)
  568. }()
  569. // 上传到阿里云
  570. ossDir := utils.RESOURCE_DIR + "cloud_disk/"
  571. resourceUrl := ``
  572. //上传到阿里云 和 minio
  573. //if utils.ObjectStorageClient == "minio" {
  574. // resourceUrl, e = services.UploadMinIoToDir(ossFileName, filePath, ossDir, "")
  575. // if e != nil {
  576. // br.Msg = "文件上传失败"
  577. // br.ErrMsg = "文件上传失败, Err:" + e.Error()
  578. // return
  579. // }
  580. //} else {
  581. // resourceUrl, e = services.UploadAliyunToDir(ossFileName, filePath, ossDir, "")
  582. // if e != nil {
  583. // br.Msg = "文件上传失败"
  584. // br.ErrMsg = "文件上传失败, Err:" + e.Error()
  585. // return
  586. // }
  587. //}
  588. savePath := ossDir + time.Now().Format("200601/20060102/") + ossFileName
  589. ossClient := services.NewOssClient()
  590. if ossClient == nil {
  591. br.Msg = "上传失败"
  592. br.ErrMsg = "初始化OSS服务失败"
  593. return
  594. }
  595. resourceUrl, e = ossClient.UploadFile(ossFileName, filePath, savePath)
  596. if e != nil {
  597. br.Msg = "文件上传失败"
  598. br.ErrMsg = "文件上传失败,Err:" + e.Error()
  599. return
  600. }
  601. // 新增云盘资源
  602. extMap := services.GetCloudDiskResourceFileTypeExtMap()
  603. resourceIcon := extMap[ext]
  604. if resourceIcon == "" {
  605. resourceIcon = models.CloudDiskResourceFileTypeIconUnknown
  606. }
  607. resourceItem := &models.CloudDiskResource{
  608. MenuId: menuId,
  609. ResourceName: fileName,
  610. ResourceSuffix: ext,
  611. ResourceUrl: resourceUrl,
  612. ResourceIcon: resourceIcon,
  613. AdminId: this.SysUser.AdminId,
  614. AdminName: this.SysUser.RealName,
  615. Size: h.Size,
  616. CreateTime: time.Now().Local(),
  617. ModifyTime: time.Now().Local(),
  618. }
  619. if e = resourceItem.Create(); e != nil {
  620. br.Msg = "上传失败"
  621. br.ErrMsg = "新增云盘资源失败, Err: " + e.Error()
  622. return
  623. }
  624. // 更新目录大小
  625. go func() {
  626. if e = services.UpdateCloudDiskMenuSize(menuId); e != nil {
  627. alarm_msg.SendAlarmMsg("更新云盘目录大小失败, Err: "+e.Error(), 3)
  628. }
  629. }()
  630. br.Msg = "上传成功"
  631. br.Ret = 200
  632. br.Success = true
  633. br.Data = resourceItem
  634. }
  635. // ResourceRename 重命名文件
  636. // @Title 重命名文件
  637. // @Description 重命名文件
  638. // @Param request body models.CloudDiskResourceRenameReq true "type json string"
  639. // @Success 200 Ret=200 操作成功
  640. // @router /resource/rename [post]
  641. func (this *CloudDiskController) ResourceRename() {
  642. br := new(models.BaseResponse).Init()
  643. defer func() {
  644. this.Data["json"] = br
  645. this.ServeJSON()
  646. }()
  647. var req models.CloudDiskResourceRenameReq
  648. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  649. br.Msg = "参数解析异常!"
  650. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  651. return
  652. }
  653. if req.ResourceId <= 0 {
  654. br.Msg = "参数有误"
  655. return
  656. }
  657. req.ResourceName = strings.TrimSpace(req.ResourceName)
  658. if req.ResourceName == "" {
  659. br.Msg = "文件名称不可为空"
  660. return
  661. }
  662. // 权限校验-仅上传人或管理员可操作
  663. resourceItem := new(models.CloudDiskResource)
  664. if e := resourceItem.GetItemById(req.ResourceId); e != nil {
  665. br.Msg = "操作失败"
  666. br.ErrMsg = "获取文件信息失败, Err: " + e.Error()
  667. return
  668. }
  669. if resourceItem.AdminId != this.SysUser.AdminId {
  670. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  671. if e != nil {
  672. br.Msg = "操作失败"
  673. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  674. return
  675. }
  676. if !ok {
  677. br.Msg = "无权操作"
  678. return
  679. }
  680. }
  681. // 重名校验
  682. existItem := new(models.CloudDiskResource)
  683. existCond := ` AND resource_name = ? AND menu_id = ? AND resource_id <> ? `
  684. existPars := make([]interface{}, 0)
  685. existPars = append(existPars, req.ResourceName, resourceItem.MenuId, resourceItem.ResourceId)
  686. if e := existItem.GetItemByCondition(existCond, existPars); e != nil && e.Error() != utils.ErrNoRow() {
  687. br.Msg = "操作失败"
  688. br.ErrMsg = "获取重名文件失败, Err: " + e.Error()
  689. return
  690. }
  691. if existItem != nil && existItem.MenuId > 0 {
  692. br.Msg = "该文件名已存在"
  693. return
  694. }
  695. resourceItem.ResourceName = req.ResourceName
  696. resourceItem.ModifyTime = time.Now().Local()
  697. if e := resourceItem.Update([]string{"ResourceName", "ModifyTime"}); e != nil {
  698. br.Msg = "操作失败"
  699. br.ErrMsg = "文件重命名失败, Err: " + e.Error()
  700. return
  701. }
  702. br.Ret = 200
  703. br.Success = true
  704. br.Msg = "操作成功"
  705. }
  706. // ResourceDelete 删除文件
  707. // @Title 删除文件
  708. // @Description 删除文件
  709. // @Param request body models.CloudDiskResourceDeleteReq true "type json string"
  710. // @Success 200 Ret=200 操作成功
  711. // @router /resource/delete [post]
  712. func (this *CloudDiskController) ResourceDelete() {
  713. br := new(models.BaseResponse).Init()
  714. defer func() {
  715. this.Data["json"] = br
  716. this.ServeJSON()
  717. }()
  718. var req models.CloudDiskResourceDeleteReq
  719. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  720. br.Msg = "参数解析异常!"
  721. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  722. return
  723. }
  724. if req.ResourceId <= 0 {
  725. br.Msg = "参数有误"
  726. return
  727. }
  728. // 权限校验-仅上传人或管理员可操作
  729. resourceItem := new(models.CloudDiskResource)
  730. if e := resourceItem.GetItemById(req.ResourceId); e != nil {
  731. br.Msg = "操作失败"
  732. br.ErrMsg = "获取文件信息失败, Err: " + e.Error()
  733. return
  734. }
  735. if resourceItem.AdminId != this.SysUser.AdminId {
  736. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  737. if e != nil {
  738. br.Msg = "操作失败"
  739. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  740. return
  741. }
  742. if !ok {
  743. br.Msg = "无权操作"
  744. return
  745. }
  746. }
  747. if e := resourceItem.Delete(); e != nil {
  748. br.Msg = "操作失败"
  749. br.ErrMsg = "删除文件失败, Err: " + e.Error()
  750. return
  751. }
  752. // 更新目录大小
  753. go func() {
  754. if e := services.UpdateCloudDiskMenuSize(resourceItem.MenuId); e != nil {
  755. alarm_msg.SendAlarmMsg("更新云盘目录大小失败, Err: "+e.Error(), 3)
  756. }
  757. }()
  758. br.Ret = 200
  759. br.Success = true
  760. br.Msg = "操作成功"
  761. }
  762. // List 云盘列表
  763. // @Title 云盘列表
  764. // @Description 云盘列表
  765. // @Param MenuId query int false "目录ID"
  766. // @Param Keyword query string false "关键词"
  767. // @Param SortType query int false "排序方式: 0-默认; 1-创建时间倒序; 2-创建时间正序"
  768. // @Success 200 Ret=200 获取成功
  769. // @router /list [get]
  770. func (this *CloudDiskController) List() {
  771. br := new(models.BaseResponse).Init()
  772. defer func() {
  773. this.Data["json"] = br
  774. this.ServeJSON()
  775. }()
  776. menuId, _ := this.GetInt("MenuId", 0)
  777. keyword := this.GetString("Keyword", "")
  778. sortType, _ := this.GetInt("SortType", 0)
  779. menuCond := ``
  780. menuPars := make([]interface{}, 0)
  781. resourceCond := ``
  782. resourcePars := make([]interface{}, 0)
  783. if keyword != "" {
  784. // 有关键词时全局搜索
  785. menuCond += ` AND menu_name LIKE ? `
  786. menuPars = append(menuPars, utils.GetLikeKeyword(keyword))
  787. resourceCond += ` AND resource_name LIKE ? `
  788. resourcePars = append(resourcePars, utils.GetLikeKeyword(keyword))
  789. } else {
  790. menuCond += ` AND parent_id = ? `
  791. menuPars = append(menuPars, menuId)
  792. resourceCond += ` AND menu_id = ? `
  793. resourcePars = append(resourcePars, menuId)
  794. }
  795. orderRule := `create_time DESC`
  796. if sortType == 2 {
  797. orderRule = `create_time ASC`
  798. }
  799. // 父级目录
  800. parentItem := new(models.CloudDiskMenu)
  801. parentCond := ``
  802. parentPars := make([]interface{}, 0)
  803. parentList, e := parentItem.GetItemsByCondition(parentCond, parentPars, []string{}, "")
  804. if e != nil {
  805. br.Msg = "获取失败"
  806. br.ErrMsg = "获取父级目录列表失败, Err: " + e.Error()
  807. return
  808. }
  809. parentNameMap := make(map[int]string)
  810. menuMap := make(map[int]*models.CloudDiskMenu)
  811. for i := range parentList {
  812. parentNameMap[parentList[i].MenuId] = parentList[i].MenuName
  813. menuMap[parentList[i].MenuId] = parentList[i]
  814. }
  815. list, e := models.GetCloudDiskMenuAndResourceList(menuCond, resourceCond, menuPars, resourcePars, orderRule)
  816. if e != nil {
  817. br.Msg = "获取失败"
  818. br.ErrMsg = "获取云盘列表数据失败, Err: " + e.Error()
  819. return
  820. }
  821. // 管理员权限
  822. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  823. if e != nil {
  824. br.Msg = "操作失败"
  825. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  826. return
  827. }
  828. listResp := new(models.CloudDiskListResp)
  829. listResp.UploadResourceAuth = true
  830. if ok {
  831. listResp.CreateMenuAuth = true
  832. listResp.BatchDelAuth = true
  833. }
  834. menuCount := 0
  835. for i := range list {
  836. list[i].ParentName = parentNameMap[list[i].ParentId]
  837. if list[i].ItemType == 1 {
  838. menuCount += 1
  839. list[i].ItemIcon = models.CloudDiskResourceFileTypeIconMenu
  840. }
  841. list[i].SizeName = utils.FormatFileSize(list[i].Size)
  842. // 按钮权限
  843. au := new(models.CloudDiskListButton)
  844. if ok || (list[i].AdminId == this.SysUser.AdminId) {
  845. au.RenameAuth = true
  846. au.DelAuth = true
  847. }
  848. list[i].ButtonAuth = au
  849. }
  850. // 根目录无子目录时, 关闭上传文件按钮
  851. if menuId == 0 && menuCount == 0 {
  852. listResp.UploadResourceAuth = false
  853. }
  854. // 面包屑
  855. pathArr := make([]*models.CloudDiskMenuResourcePath, 0)
  856. pathArr = services.GetCloudDiskMenuOrResourcePath(menuId, 0, menuMap, menuMap[menuId], pathArr)
  857. listResp.ListPath = pathArr
  858. listResp.List = list
  859. br.Ret = 200
  860. br.Success = true
  861. br.Msg = "获取成功"
  862. br.Data = listResp
  863. }
  864. // BatchDelete 批量删除
  865. // @Title 批量删除
  866. // @Description 批量删除
  867. // @Param request body models.CloudDiskBatchDeleteReq true "type json string"
  868. // @Success 200 Ret=200 操作成功
  869. // @router /batch_delete [post]
  870. func (this *CloudDiskController) BatchDelete() {
  871. br := new(models.BaseResponse).Init()
  872. defer func() {
  873. this.Data["json"] = br
  874. this.ServeJSON()
  875. }()
  876. var req models.CloudDiskBatchDeleteReq
  877. if e := json.Unmarshal(this.Ctx.Input.RequestBody, &req); e != nil {
  878. br.Msg = "参数解析异常!"
  879. br.ErrMsg = "参数解析失败,Err:" + e.Error()
  880. return
  881. }
  882. if req.MenuIds == "" && req.ResourceIds == "" {
  883. br.Msg = "请选择目录或文件"
  884. return
  885. }
  886. menuIds := make([]int, 0)
  887. resourceIds := make([]int, 0)
  888. if req.MenuIds != "" {
  889. menuIdArr := strings.Split(req.MenuIds, ",")
  890. for i := range menuIdArr {
  891. n, e := strconv.Atoi(menuIdArr[i])
  892. if e != nil {
  893. br.Msg = "参数有误"
  894. br.ErrMsg = "参数有误, Err:" + e.Error()
  895. return
  896. }
  897. menuIds = append(menuIds, n)
  898. }
  899. }
  900. if req.ResourceIds != "" {
  901. resourceIdArr := strings.Split(req.ResourceIds, ",")
  902. for i := range resourceIdArr {
  903. n, e := strconv.Atoi(resourceIdArr[i])
  904. if e != nil {
  905. br.Msg = "参数有误"
  906. br.ErrMsg = "参数有误, Err:" + e.Error()
  907. return
  908. }
  909. resourceIds = append(resourceIds, n)
  910. }
  911. }
  912. // 权限校验-仅管理员可操作
  913. ok, e := services.CheckCloudDiskManagerAuth(this.SysUser.RoleTypeCode)
  914. if e != nil {
  915. br.Msg = "操作失败"
  916. br.ErrMsg = "操作权限校验失败, ErrMsg: " + e.Error()
  917. return
  918. }
  919. if !ok {
  920. br.Msg = "无权操作"
  921. return
  922. }
  923. if e = models.BatchDeleteCloudDiskMenuAndResource(menuIds, resourceIds); e != nil {
  924. br.Msg = "操作失败"
  925. br.ErrMsg = "批量删除目录及文件失败, Err:" + e.Error()
  926. return
  927. }
  928. // 更新目录大小
  929. if req.MenuId > 0 {
  930. go func() {
  931. if e := services.UpdateCloudDiskMenuSize(req.MenuId); e != nil {
  932. alarm_msg.SendAlarmMsg("更新云盘目录大小失败, Err: "+e.Error(), 3)
  933. }
  934. }()
  935. }
  936. br.Ret = 200
  937. br.Success = true
  938. br.Msg = "操作成功"
  939. }