/*
 * @Author: LinkLeong link@icewhale.com
 * @Date: 2022-07-11 16:02:29
 * @LastEditors: LinkLeong
 * @LastEditTime: 2022-08-17 19:14:50
 * @FilePath: /CasaOS/route/v1/storage.go
 * @Description:
 * @Website: https://www.casaos.io
 * Copyright (c) 2022 by icewhale, All Rights Reserved.
 */
package v1

import (
	"path/filepath"
	"reflect"

	"github.com/IceWhaleTech/CasaOS/model"
	"github.com/IceWhaleTech/CasaOS/pkg/utils/common_err"
	"github.com/IceWhaleTech/CasaOS/service"
	"github.com/gin-gonic/gin"
)

func GetStorageList(c *gin.Context) {
	system := c.Query("system")
	storages := []model.Storages{}
	disks := service.MyService.Disk().LSBLK(false)
	diskNumber := 1
	children := 1
	findSystem := 0
	for _, d := range disks {
		if d.Tran != "usb" {
			tempSystemDisk := false
			children = 1
			tempDisk := model.Storages{
				DiskName: d.Model,
				Path:     d.Path,
				Size:     d.Size,
			}

			storageArr := []model.Storage{}
			temp := service.MyService.Disk().SmartCTL(d.Path)
			if reflect.DeepEqual(temp, model.SmartctlA{}) {
				temp.SmartStatus.Passed = true
			}
			for _, v := range d.Children {
				if v.MountPoint != "" {
					if findSystem == 0 {
						if v.MountPoint == "/" {
							tempDisk.DiskName = "System"
							findSystem = 1
							tempSystemDisk = true
						}
						if len(v.Children) > 0 {
							for _, c := range v.Children {
								if c.MountPoint == "/" {
									tempDisk.DiskName = "System"
									findSystem = 1
									tempSystemDisk = true
									break
								}
							}
						}
					}

					stor := model.Storage{}
					stor.MountPoint = v.MountPoint
					stor.Size = v.FSSize
					stor.Avail = v.FSAvail
					stor.Path = v.Path
					stor.Type = v.FsType
					stor.DriveName = v.Name
					if len(v.Label) == 0 {
						if stor.MountPoint == "/" {
							stor.Label = "System"
						} else {
							stor.Label = filepath.Base(stor.MountPoint)
						}

						children += 1
					} else {
						stor.Label = v.Label
					}
					storageArr = append(storageArr, stor)
				}
			}

			if len(storageArr) > 0 {
				if tempSystemDisk && len(system) > 0 {
					tempStorageArr := []model.Storage{}
					for i := 0; i < len(storageArr); i++ {
						if storageArr[i].MountPoint != "/boot/efi" && storageArr[i].Type != "swap" {
							tempStorageArr = append(tempStorageArr, storageArr[i])
						}
					}
					tempDisk.Children = tempStorageArr
					storages = append(storages, tempDisk)
					diskNumber += 1
				} else if !tempSystemDisk {
					tempDisk.Children = storageArr
					storages = append(storages, tempDisk)
					diskNumber += 1
				}

			}
		}
	}

	c.JSON(common_err.SUCCESS, model.Result{Success: common_err.SUCCESS, Message: common_err.GetMsg(common_err.SUCCESS), Data: storages})
}