From 1d970aa6ba1c7da0ba67285339f96fb44dd4e565 Mon Sep 17 00:00:00 2001 From: Daniel Legt Date: Mon, 22 Jan 2024 00:28:07 +0200 Subject: [PATCH] Rootless Run + Debug Tools * ReadME * Optimized config, now it's cached * ReWrote and optimized logic.go * rewrote and optimized + Styling improvements --- .vscode/launch.json | 15 ++++ README.md | 48 ++++++++-- go.mod | 15 ++-- go.sum | 17 ++-- lib/config/config.go | 11 ++- lib/hardware/logic.go | 198 +++++++++++++++++++++++++++++++---------- lib/hardware/models.go | 65 ++++++++++---- static/style.css | 47 +++++++++- templates/index.html | 5 +- 9 files changed, 333 insertions(+), 88 deletions(-) create mode 100644 .vscode/launch.json diff --git a/.vscode/launch.json b/.vscode/launch.json new file mode 100644 index 0000000..0f8103e --- /dev/null +++ b/.vscode/launch.json @@ -0,0 +1,15 @@ +{ + // Use IntelliSense to learn about possible attributes. + // Hover to view descriptions of existing attributes. + // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387 + "version": "0.2.0", + "configurations": [ + { + "name": "Launch Package", + "type": "go", + "request": "launch", + "mode": "auto", + "program": "${workspaceFolder}" + } + ] +} \ No newline at end of file diff --git a/README.md b/README.md index 2e661ed..03bb46e 100644 --- a/README.md +++ b/README.md @@ -1,11 +1,47 @@ -## About +## 📖About -## Disclaimer +Drive Health is a program written in golang to help with tracking and monitoring of your hardware's temperature. -This software is only meant to run on linux, it's something I threw together all at once in one sitting without really thinking about structure or even caring about anything but making it work. +This tool has been created by [me](https://danlegt.com) with the purpose of installing it in different servers I own with different configurations to help keep track of the temperature of different hard-disks, ssds, nvme drives, etc... The testing has been very limited to only 4 different computers and not on laptops so expect some mishaps. -It's actually meant to run on my own server just so I can track the temperature of my harddrives in a live environment, I just want this service to run, record the status of the harddrives, and be able via my VPN to access it in a private manner, it's not meant to be dockerized, it's not meant to be ran in a VM, it's meant to be ran either as a `service` or in a [screen](https://git.savannah.gnu.org/cgit/screen.git) in the background. +## ❗Disclaimer -It's not really meant to run all the time or anything like that, I just wanted to write this disclaimer, feel free to use the code any way you want to :) +I'm not exactly a linux hardware wizard, so I honestly have no clue about a lot of things and I myself can tell there's a lot to improve upon and that there's a lot of other things missing that are a little bit more obscure, I personally don't currently own any m.2 sata drives to test the code on, or many of the other drive types, I have only tested on HDD, SSD and NVMe drives, any issues opened would help me so much! -## How to use +## Requirements +1. A linux machine, this will NOT work on macOS or on Windows, it's meant to be ran on servers as a service with which administrators can privately connect to for temperature logging. + +2. Please make sure you have the [**drivetemp kernel drive**](https://docs.kernel.org/hwmon/drivetemp.html) you can check this by running `sudo modprobe drivetemp`. +The program depends on this to be able to log the temperature of your devices. + + +## 📖How to use + +The program is straight forward to use really, edit the [.env](./.env) file and make the changes you would like applied. + +### Docker ( Recommended/Hassle free ) + +### SystemD +```toml +[Unit] +Description=Drive Health Service +After=network.target + +[Service] +Type=simple +User=daniel # Your user here +WorkingDirectory=/home/daniel/services/drive-health # The path to the service's directory +ExecStart=/home/daniel/services/drive-health/drive-health # The path to the binary +Restart=on-failure + +[Install] +WantedBy=multi-user.target +``` + +## ❔FAQ + +### How does it work? +Currently the program does not depend on any hardware library as I couldn't find anything that would not require root access while giving me the possibility to interrogate the temperature of the drives, I chose not to depend on `lsblk` either, so how does the program work? Well it looks in `/sys/block` and simply + +### Why not just run as root? +I really, really, **really** want to avoid asking people to run **ANY** program I write as root and even try and prevent that from happening since that's how things can go bad, especially because I am runnig actions over hardware items. I think you can see how easy it is for a mistake or a **malicious attack** to easily deal damage \ No newline at end of file diff --git a/go.mod b/go.mod index a36143c..3f50cf4 100644 --- a/go.mod +++ b/go.mod @@ -3,13 +3,20 @@ module tea.chunkbyte.com/kato/drive-health go 1.21.6 require ( - github.com/anatol/smart.go v0.0.0-20230705044831-c3b27137baa3 // indirect + github.com/gin-gonic/gin v1.9.1 + github.com/joho/godotenv v1.5.1 + github.com/wcharczuk/go-chart/v2 v2.1.1 + gorm.io/driver/sqlite v1.5.4 + gorm.io/gorm v1.25.5 +) + +require ( + github.com/blend/go-sdk v1.20220411.3 // indirect github.com/bytedance/sonic v1.10.2 // indirect github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect github.com/chenzhuoyu/iasm v0.9.1 // indirect github.com/gabriel-vasile/mimetype v1.4.3 // indirect github.com/gin-contrib/sse v0.1.0 // indirect - github.com/gin-gonic/gin v1.9.1 // indirect github.com/go-playground/locales v0.14.1 // indirect github.com/go-playground/universal-translator v0.18.1 // indirect github.com/go-playground/validator/v10 v10.17.0 // indirect @@ -17,7 +24,6 @@ require ( github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect github.com/jinzhu/inflection v1.0.0 // indirect github.com/jinzhu/now v1.1.5 // indirect - github.com/joho/godotenv v1.5.1 // indirect github.com/json-iterator/go v1.1.12 // indirect github.com/klauspost/cpuid/v2 v2.2.6 // indirect github.com/leodido/go-urn v1.2.4 // indirect @@ -28,7 +34,6 @@ require ( github.com/pelletier/go-toml/v2 v2.1.1 // indirect github.com/twitchyliquid64/golang-asm v0.15.1 // indirect github.com/ugorji/go/codec v1.2.12 // indirect - github.com/wcharczuk/go-chart/v2 v2.1.1 // indirect golang.org/x/arch v0.7.0 // indirect golang.org/x/crypto v0.18.0 // indirect golang.org/x/image v0.11.0 // indirect @@ -37,6 +42,4 @@ require ( golang.org/x/text v0.14.0 // indirect google.golang.org/protobuf v1.32.0 // indirect gopkg.in/yaml.v3 v3.0.1 // indirect - gorm.io/driver/sqlite v1.5.4 // indirect - gorm.io/gorm v1.25.5 // indirect ) diff --git a/go.sum b/go.sum index 929f381..da6a1ef 100644 --- a/go.sum +++ b/go.sum @@ -1,5 +1,5 @@ -github.com/anatol/smart.go v0.0.0-20230705044831-c3b27137baa3 h1:kAF2MWFD8tyDqD74OQizymjj2cnZAURwSzBrEslCDnI= -github.com/anatol/smart.go v0.0.0-20230705044831-c3b27137baa3/go.mod h1:llkexGSe52bW0OjNva0kvIqGZxfSnVfpKHrnKBI2+pU= +github.com/blend/go-sdk v1.20220411.3 h1:GFV4/FQX5UzXLPwWV03gP811pj7B8J2sbuq+GJQofXc= +github.com/blend/go-sdk v1.20220411.3/go.mod h1:7lnH8fTi6U4i1fArEXRyOIY2E1X4MALg09qsQqY1+ak= github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM= github.com/bytedance/sonic v1.10.0-rc/go.mod h1:ElCzW+ufi8qKqNW0FY314xriJhyJhuoJ3gFZdAHF7NM= github.com/bytedance/sonic v1.10.2 h1:GQebETVBxYB7JGWJtLBi07OVzWwt+8dWA00gEVW2ZFE= @@ -12,6 +12,7 @@ github.com/chenzhuoyu/iasm v0.9.0/go.mod h1:Xjy2NpN3h7aUqeqM+woSuuvxmIe6+DDsiNLI github.com/chenzhuoyu/iasm v0.9.1 h1:tUHQJXo3NhBqw6s33wkGn9SP3bvrWLdlVIJ3hQBL7P0= github.com/chenzhuoyu/iasm v0.9.1/go.mod h1:Xjy2NpN3h7aUqeqM+woSuuvxmIe6+DDsiNLIrkAmYog= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0= github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk= @@ -19,6 +20,8 @@ github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg= github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU= +github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= +github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= @@ -29,7 +32,7 @@ github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g= github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k= -github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= @@ -57,6 +60,7 @@ github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9G github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= github.com/pelletier/go-toml/v2 v2.1.1 h1:LWAJwfNvjQZCFIDKWYQaM62NcYeYViCmWIwmOStowAI= github.com/pelletier/go-toml/v2 v2.1.1/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdUvyaeZurnPPDc= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= @@ -67,6 +71,7 @@ github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/ github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk= github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= @@ -102,8 +107,6 @@ golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBc golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= -golang.org/x/sys v0.9.0 h1:KS/R3tvhPqvJvwcKfnBHJwwthS11LRhmM5D59eEXa0s= -golang.org/x/sys v0.9.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU= golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= @@ -121,9 +124,11 @@ golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtn golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= -golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= google.golang.org/protobuf v1.32.0 h1:pPC6BG5ex8PDFnkbrGU3EixyhKcQ2aDuBS36lqK/C7I= google.golang.org/protobuf v1.32.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= diff --git a/lib/config/config.go b/lib/config/config.go index fa049a7..3dcf0f9 100644 --- a/lib/config/config.go +++ b/lib/config/config.go @@ -20,8 +20,15 @@ type DHConfig struct { DebugMode bool `json:"debugMode"` } -func GetConfiguration() DHConfig { - config := DHConfig{ +var config *DHConfig = nil + +func GetConfiguration() *DHConfig { + + if config != nil { + return config + } + + config = &DHConfig{ DiskFetchFrequency: 5, CleanupServiceFrequency: 3600, MaxHistoryAge: 2592000, diff --git a/lib/hardware/logic.go b/lib/hardware/logic.go index f400987..ad5a12c 100644 --- a/lib/hardware/logic.go +++ b/lib/hardware/logic.go @@ -1,70 +1,78 @@ package hardware import ( - "bufio" - "bytes" "fmt" - "os/exec" + "os" + "path/filepath" + "strconv" "strings" "time" "gorm.io/gorm" + "tea.chunkbyte.com/kato/drive-health/lib/config" ) func GetSystemHardDrives(db *gorm.DB, olderThan *time.Time, newerThan *time.Time) ([]*HardDrive, error) { - - // Execute the lsblk command to get detailed block device information - cmd := exec.Command("lsblk", "-d", "-o", "NAME,TRAN,SIZE,MODEL,SERIAL,TYPE") - var out bytes.Buffer - cmd.Stdout = &out - err := cmd.Run() - if err != nil { - fmt.Println("Failed to execute command:", err) - return nil, err - } - var systemHardDrives []*HardDrive - // Scan the output line by line - scanner := bufio.NewScanner(&out) - for scanner.Scan() { - line := scanner.Text() - - // Skip the header line - if strings.Contains(line, "NAME") { - continue - } - - // Split the line into columns - cols := strings.Fields(line) - if len(cols) < 6 { - continue - } - - // Filter out nvme drives (M.2) - if cols[1] != "usb" { - hd := &HardDrive{ - Name: cols[0], - Transport: cols[1], - Size: cols[2], - Model: cols[3], - Serial: cols[4], - Type: cols[5], - } - systemHardDrives = append(systemHardDrives, hd) - } + // List all block devices + devices, err := os.ReadDir("/sys/block/") + if err != nil { + return nil, fmt.Errorf("failed to list block devices: %w", err) } - // Handle error - if err := scanner.Err(); err != nil { - return nil, err + for _, device := range devices { + deviceName := device.Name() + + // Skip non-physical devices (like loop and ram devices) + // TODO: Read more about this, there might be some other devices we should or should not skip + if strings.HasPrefix(deviceName, "loop") || strings.HasPrefix(deviceName, "ram") { + continue + } + + // Read device details + model, _ := os.ReadFile(fmt.Sprintf("/sys/block/%s/device/model", deviceName)) + serial, _ := os.ReadFile(fmt.Sprintf("/sys/block/%s/device/serial", deviceName)) + sizeBytes, _ := os.ReadFile(fmt.Sprintf("/sys/block/%s/size", deviceName)) + + size := convertSizeToString(sizeBytes) + transport := getTransportType(deviceName) + + // TODO: Maybe find a better way? + if size == "0 Bytes" { + // This looks like an invalid device, skip it. + if config.GetConfiguration().DebugMode { + fmt.Printf("[🟨] Igoring device:[/dev/%s], reported size of 0\n", deviceName) + } + continue + } + + hwid, err := getHardwareID(deviceName) + if err != nil { + if config.GetConfiguration().DebugMode { + fmt.Printf("[🟨] No unique identifier found for device:[/dev/%s] unique identifier\n", deviceName) + } + continue + } + + hd := &HardDrive{ + Name: deviceName, + Transport: transport, + Model: strings.TrimSpace(string(model)), + Serial: strings.TrimSpace(string(serial)), + Size: size, + Type: getDriveType(deviceName), + HWID: hwid, + } + + systemHardDrives = append(systemHardDrives, hd) } var updatedHardDrives []*HardDrive for _, sysHDD := range systemHardDrives { var existingHD HardDrive - q := db.Where("serial = ? AND model = ? AND type = ?", sysHDD.Serial, sysHDD.Model, sysHDD.Type) + q := db.Where("hw_id = ?", sysHDD.HWID) if newerThan != nil && olderThan != nil { q = q.Preload("Temperatures", "time_stamp < ? AND time_stamp > ?", newerThan, olderThan) @@ -90,3 +98,101 @@ func GetSystemHardDrives(db *gorm.DB, olderThan *time.Time, newerThan *time.Time return updatedHardDrives, nil } + +func getTransportType(deviceName string) string { + transportLink, err := filepath.EvalSymlinks(fmt.Sprintf("/sys/block/%s/device", deviceName)) + if err != nil { + return "Unknown" + } + + if strings.Contains(transportLink, "/usb/") { + return "USB" + } else if strings.Contains(transportLink, "/ata") { + return "SATA" + } else if strings.Contains(transportLink, "/nvme/") { + return "NVMe" + } + + return "Other" +} + +func convertSizeToString(sizeBytes []byte) string { + // Convert the size from a byte slice to a string, then to an integer + sizeStr := strings.TrimSpace(string(sizeBytes)) + sizeSectors, err := strconv.ParseInt(sizeStr, 10, 64) + if err != nil { + return "Unknown" + } + + // Convert from 512-byte sectors to bytes + sizeInBytes := sizeSectors * 512 + + // Define size units + const ( + _ = iota // ignore first value by assigning to blank identifier + KB float64 = 1 << (10 * iota) + MB + GB + TB + ) + + var size float64 = float64(sizeInBytes) + var unit string + + // Determine the unit to use + switch { + case size >= TB: + size /= TB + unit = "TB" + case size >= GB: + size /= GB + unit = "GB" + case size >= MB: + size /= MB + unit = "MB" + case size >= KB: + size /= KB + unit = "KB" + default: + unit = "Bytes" + } + + // Return the formatted size + return fmt.Sprintf("%.2f %s", size, unit) +} + +// Look throug /sys/block/device/ and try and find the unique identifier of the device. +func getHardwareID(deviceName string) (string, error) { + // Define potential ID file paths + idFilePaths := []string{ + "/sys/block/" + deviceName + "/device/wwid", + "/sys/block/" + deviceName + "/device/wwn", + "/sys/block/" + deviceName + "/device/serial", + } + + // Try to read each file and return the first successful read + for _, path := range idFilePaths { + if idBytes, err := os.ReadFile(path); err == nil { + return strings.TrimSpace(string(idBytes)), nil + } + } + + // Return an empty string if no ID is found + return "", fmt.Errorf("could not find unique identifier for %s", deviceName) +} + +// Figure out what kind of device this is by reading if it's rotational or not +func getDriveType(deviceName string) string { + // Check if the drive is rotational (HDD) + if isRotational, _ := os.ReadFile(fmt.Sprintf("/sys/block/%s/queue/rotational", deviceName)); string(isRotational) == "1\n" { + return "HDD" + } + + // Check if the drive is NVMe + if strings.HasPrefix(deviceName, "nvme") { + return "NVMe" + } + + // Default to SSD for non-rotational and non-NVMe drives + return "SSD" +} diff --git a/lib/hardware/models.go b/lib/hardware/models.go index 9ea4355..47f091f 100644 --- a/lib/hardware/models.go +++ b/lib/hardware/models.go @@ -2,9 +2,12 @@ package hardware import ( "fmt" + "os" + "path/filepath" + "strconv" + "strings" "time" - "github.com/anatol/smart.go" "gorm.io/gorm" ) @@ -19,6 +22,7 @@ type HardDrive struct { Model string Serial string Type string + HWID string Temperatures []HardDriveTemperature `gorm:"foreignKey:HardDriveID"` } @@ -39,26 +43,49 @@ type Snapshots struct { List []*HardwareSnapshot } -// Fetch the temperature of the device, optinally update the reference object func (h *HardDrive) GetTemperature() int { - // Fetch the device by name - disk, err := smart.Open("/dev/" + h.Name) - if err != nil { - fmt.Printf("Failed to open device %s: %s\n", h.Name, err) - return -1 - } - defer disk.Close() - - // Fetch SMART data - smartInfo, err := disk.ReadGenericAttributes() - if err != nil { - fmt.Printf("Failed to get SMART data for %s: %s\n", h.Name, err) - return -1 + // Try HDD/SSD path + temp, found := h.getTemperatureFromPath("/sys/block/" + h.Name + "/device/hwmon/") + if found { + return temp } - // Parse the temperature - temperature := int(smartInfo.Temperature) + // Try NVMe path + temp, found = h.getTemperatureFromPath("/sys/block/" + h.Name + "/device/") + if found { + return temp + } - // Return the found value - return temperature + fmt.Printf("[🛑] Failed to get temperature for %s\n", h.Name) + return -1 +} + +func (h *HardDrive) getTemperatureFromPath(basePath string) (int, bool) { + hwmonDirs, err := os.ReadDir(basePath) + if err != nil { + return 0, false + } + + for _, dir := range hwmonDirs { + if strings.HasPrefix(dir.Name(), "hwmon") { + tempPath := filepath.Join(basePath, dir.Name(), "temp1_input") + if _, err := os.Stat(tempPath); err == nil { + tempBytes, err := os.ReadFile(tempPath) + if err != nil { + continue + } + + tempStr := strings.TrimSpace(string(tempBytes)) + temperature, err := strconv.Atoi(tempStr) + if err != nil { + continue + } + + // Convert millidegree Celsius to degree Celsius + return temperature / 1000, true + } + } + } + + return 0, false } diff --git a/static/style.css b/static/style.css index 8d7ff02..afae1d9 100644 --- a/static/style.css +++ b/static/style.css @@ -10,7 +10,17 @@ --fg0: #bbc0ca; --fg1: #434c56; - --acc: #bbc0ca; + --acc1: #2aa3f4; + --acc1BG0: #2aa3f450; + --acc1BG1: #2aa3f430; + + --acc2: #2af488; + --acc2BG0: #2af48850; + --acc2BG1: #2af48830; + + --acc3: #f4e02a; + --acc3BG0: #f4e02a50; + --acc3BG1: #f4e02a30; } :root { @@ -55,6 +65,41 @@ html, body { margin: 0; } +.badge { + font-size: 12px; + font-weight: bold; + padding: .1rem .4rem; + + display: inline-block; + border-radius: 3px; + + color: var(--acc1); + background-color: var(--acc1BG0); + border: 1px solid var(--acc1BG1); +} + +.badge[type="HDD"] { + color: var(--acc2); + background-color: var(--acc2BG0); + border: 1px solid var(--acc2BG1); +} + +.badge[type="NVMe"] { + color: var(--acc3); + background-color: var(--acc3BG0); + border: 1px solid var(--acc3BG1); +} + +.grooved { + background-color: var(--bg0); + border: 1px solid var(--bg1); + + font-size: 12px; + padding: .1rem .3rem; + + border-radius: 4px; +} + /* Table */ diff --git a/templates/index.html b/templates/index.html index 20e335c..bd9707a 100644 --- a/templates/index.html +++ b/templates/index.html @@ -99,8 +99,9 @@ {{ if len .drives }} {{ range .drives }} -
-

{{.Name}}:{{.Serial}} [{{.Size}}]

+
+
/dev/{{.Name}}
+

{{.Model}}:{{.HWID}} {{.Size}}

{{ .Model }} Image