disk.go RawHistoryFolder

1package system
2
3import (
4	"encoding/json"
5	"fmt"
6)
7
8// Disk represents a storage device
9type Disk struct {
10	Name   string `json:"name"`
11	Size   string `json:"size"`
12	Type   string `json:"type"`
13	Model  string `json:"model"`
14	Serial string `json:"serial"`
15	Vendor string `json:"vendor"`
16	Path   string `json:"path"`
17}
18
19// lsblkOutput represents the JSON output from lsblk
20type lsblkOutput struct {
21	BlockDevices []struct {
22		Name   string `json:"name"`
23		Size   string `json:"size"`
24		Type   string `json:"type"`
25		Model  string `json:"model"`
26		Serial string `json:"serial"`
27		Vendor string `json:"vendor"`
28	} `json:"blockdevices"`
29}
30
31// ListDisks returns a list of available disks (excluding loop devices)
32func ListDisks() ([]Disk, error) {
33	result := RunSimple("lsblk", "-J", "-o", "NAME,SIZE,TYPE,MODEL,SERIAL,VENDOR", "-d", "-e", "7")
34	if result.Error != nil {
35		return nil, fmt.Errorf("failed to list disks: %w", result.Error)
36	}
37
38	var output lsblkOutput
39	if err := json.Unmarshal([]byte(result.Output), &output); err != nil {
40		return nil, fmt.Errorf("failed to parse lsblk output: %w", err)
41	}
42
43	var disks []Disk
44	for _, dev := range output.BlockDevices {
45		if dev.Type == "disk" {
46			disks = append(disks, Disk{
47				Name:   dev.Name,
48				Size:   dev.Size,
49				Type:   dev.Type,
50				Model:  dev.Model,
51				Serial: dev.Serial,
52				Vendor: dev.Vendor,
53				Path:   "/dev/" + dev.Name,
54			})
55		}
56	}
57
58	return disks, nil
59}