mirror of
https://github.com/JustKato/drive-health.git
synced 2026-03-04 08:29:45 +02:00
Compare commits
15 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 40e02abe87 | |||
| c8fa24f11c | |||
| 2776ad8e52 | |||
| f3905bb822 | |||
| 92baa56a1c | |||
| 4c877e7162 | |||
| cbe252fe94 | |||
| 07dec16aa4 | |||
| 545eed44cd | |||
| eb94ce4552 | |||
| c0f1ed6879 | |||
| c4aef27eda | |||
| 8f8da162e9 | |||
| 5cc58c7d53 | |||
| 21f24899a2 |
26
.env.example
26
.env.example
@@ -1,3 +1,25 @@
|
||||
#
|
||||
# All time references are in seconds
|
||||
#
|
||||
#
|
||||
############################################################
|
||||
|
||||
# The frequency at which to fetch the temperature of ALL disks and add it to the database
|
||||
DISK_FETCH_FREQUENCY=5
|
||||
MEMORY_DUMP_FREQUENCY=60
|
||||
MAX_HISTORY_AGE=2592000
|
||||
|
||||
# How ofthen should the program clean the database of old logs
|
||||
CLEANUP_SERVICE_FREQUENCY=3600
|
||||
|
||||
# The maximum age of logs in seconds
|
||||
# 1 Day = 86400
|
||||
# 1 Week = 604800
|
||||
# 1 Month ~= 2592000
|
||||
# Recommended 1 week
|
||||
MAX_HISTORY_AGE=2592000
|
||||
|
||||
# The ip:port to listen to for the application
|
||||
LISTEN=":8080"
|
||||
|
||||
# Basic Security, these are required to view the data
|
||||
IDENTITY_USERNAME=admin
|
||||
IDENTITY_PASSWORD=admin
|
||||
3
.gitignore
vendored
3
.gitignore
vendored
@@ -1,2 +1,5 @@
|
||||
snapshots.dat
|
||||
.env
|
||||
dist
|
||||
data.db
|
||||
data.sqlite
|
||||
17
build.sh
Executable file
17
build.sh
Executable file
@@ -0,0 +1,17 @@
|
||||
#!/bin/bash
|
||||
|
||||
APP_NAME="drive-health"
|
||||
DIST_DIR="dist"
|
||||
|
||||
# Create the dist directory if it doesn't exist
|
||||
mkdir -p $DIST_DIR
|
||||
|
||||
# Build the application
|
||||
echo "Building the application..."
|
||||
GOOS=linux GOARCH=amd64 go build -o $DIST_DIR/$APP_NAME
|
||||
|
||||
# Copy additional resources (like .env, static files, templates) to the dist directory
|
||||
echo "Copying additional resources..."
|
||||
cp -r .env static templates $DIST_DIR/
|
||||
|
||||
echo "Compilation and packaging completed."
|
||||
8
go.mod
8
go.mod
@@ -14,21 +14,29 @@ require (
|
||||
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||
github.com/go-playground/validator/v10 v10.17.0 // indirect
|
||||
github.com/goccy/go-json v0.10.2 // indirect
|
||||
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
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-sqlite3 v1.14.19 // indirect
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||
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
|
||||
golang.org/x/net v0.20.0 // indirect
|
||||
golang.org/x/sys v0.16.0 // indirect
|
||||
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
|
||||
)
|
||||
|
||||
46
go.sum
46
go.sum
@@ -27,9 +27,15 @@ github.com/go-playground/validator/v10 v10.17.0 h1:SmVVlfAOtlZncTxRuinDPomC2DkXJ
|
||||
github.com/go-playground/validator/v10 v10.17.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
|
||||
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/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=
|
||||
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
|
||||
github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
|
||||
github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
|
||||
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
|
||||
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
|
||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||
@@ -42,6 +48,8 @@ github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
|
||||
github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-sqlite3 v1.14.19 h1:fhGleo2h1p8tVChob4I9HpmVFIAkKGpiukdrgQbWfGI=
|
||||
github.com/mattn/go-sqlite3 v1.14.19/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
|
||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||
@@ -64,21 +72,55 @@ github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS
|
||||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||
github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
|
||||
github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
|
||||
github.com/wcharczuk/go-chart/v2 v2.1.1 h1:2u7na789qiD5WzccZsFz4MJWOJP72G+2kUuJoSNqWnE=
|
||||
github.com/wcharczuk/go-chart/v2 v2.1.1/go.mod h1:CyCAUt2oqvfhCl6Q5ZvAZwItgpQKZOkCJGb+VGv6l14=
|
||||
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
|
||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||
golang.org/x/arch v0.7.0 h1:pskyeJh/3AmoQ8CPE95vxHLqp1G1GfGNXTmcl9NEKTc=
|
||||
golang.org/x/arch v0.7.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||
golang.org/x/crypto v0.18.0 h1:PGVlW0xEltQnzFZ55hkuX5+KLyrMYhHld1YHO4AKcdc=
|
||||
golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg=
|
||||
golang.org/x/image v0.11.0 h1:ds2RoQvBvYTiJkwpSFDwCcDFNX7DqjL2WsUgTNk0Ooo=
|
||||
golang.org/x/image v0.11.0/go.mod h1:bglhjqbqVuEb9e9+eNR45Jfu7D+T4Qan+NhQk8Ck2P8=
|
||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
|
||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||
golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
||||
golang.org/x/net v0.20.0 h1:aCL9BSgETF1k+blQaYUBx9hJ9LOGP3gAVemcZlf1Kpo=
|
||||
golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
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=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||
golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
|
||||
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
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=
|
||||
google.golang.org/protobuf v1.32.0 h1:pPC6BG5ex8PDFnkbrGU3EixyhKcQ2aDuBS36lqK/C7I=
|
||||
google.golang.org/protobuf v1.32.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||
@@ -86,5 +128,9 @@ gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8
|
||||
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=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gorm.io/driver/sqlite v1.5.4 h1:IqXwXi8M/ZlPzH/947tn5uik3aYQslP9BVveoax0nV0=
|
||||
gorm.io/driver/sqlite v1.5.4/go.mod h1:qxAuCol+2r6PannQDpOP1FP6ag3mKi4esLnB/jHed+4=
|
||||
gorm.io/gorm v1.25.5 h1:zR9lOiiYf09VNh5Q1gphfyia1JpiClIWG9hQaxB/mls=
|
||||
gorm.io/gorm v1.25.5/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=
|
||||
nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
|
||||
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
||||
|
||||
@@ -1,29 +1,33 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"log"
|
||||
"os"
|
||||
"strconv"
|
||||
|
||||
"github.com/joho/godotenv"
|
||||
)
|
||||
|
||||
type DHConfig struct {
|
||||
DiskFetchFrequency int `json:"diskFetchFrequency"`
|
||||
MemoryDumpFrequency int `json:"memoryDumpFrequency"`
|
||||
MaxHistoryAge int `json:"maxHistoryAge"`
|
||||
CleanupServiceFrequency int `json:"cleanupServiceFrequency"`
|
||||
DiskFetchFrequency int `json:"diskFetchFrequency"`
|
||||
MaxHistoryAge int `json:"maxHistoryAge"`
|
||||
|
||||
DatabaseFilePath string `json:"databaseFilePath"`
|
||||
|
||||
Listen string `json:"listen"`
|
||||
|
||||
IdentityUsername string `json:"identityUsername"`
|
||||
IdentityPassword string `json:"identityPassword"`
|
||||
}
|
||||
|
||||
func GetConfiguration() DHConfig {
|
||||
// Load .env file if it exists
|
||||
if err := godotenv.Load(); err != nil {
|
||||
log.Println("No .env file found")
|
||||
}
|
||||
|
||||
config := DHConfig{
|
||||
DiskFetchFrequency: 5, // default value
|
||||
MemoryDumpFrequency: 60, // default value
|
||||
MaxHistoryAge: 2592000, // default value
|
||||
DiskFetchFrequency: 5,
|
||||
CleanupServiceFrequency: 3600,
|
||||
MaxHistoryAge: 2592000,
|
||||
DatabaseFilePath: "./data.sqlite",
|
||||
IdentityUsername: "admin",
|
||||
IdentityPassword: "admin",
|
||||
|
||||
Listen: ":8080",
|
||||
}
|
||||
|
||||
if val, exists := os.LookupEnv("DISK_FETCH_FREQUENCY"); exists {
|
||||
@@ -32,9 +36,9 @@ func GetConfiguration() DHConfig {
|
||||
}
|
||||
}
|
||||
|
||||
if val, exists := os.LookupEnv("MEMORY_DUMP_FREQUENCY"); exists {
|
||||
if val, exists := os.LookupEnv("CLEANUP_SERVICE_FREQUENCY"); exists {
|
||||
if intValue, err := strconv.Atoi(val); err == nil {
|
||||
config.MemoryDumpFrequency = intValue
|
||||
config.CleanupServiceFrequency = intValue
|
||||
}
|
||||
}
|
||||
|
||||
@@ -44,5 +48,21 @@ func GetConfiguration() DHConfig {
|
||||
}
|
||||
}
|
||||
|
||||
if val, exists := os.LookupEnv("LISTEN"); exists {
|
||||
config.Listen = val
|
||||
}
|
||||
|
||||
if val, exists := os.LookupEnv("DATABASE_FILE_PATH"); exists {
|
||||
config.DatabaseFilePath = val
|
||||
}
|
||||
|
||||
if val, exists := os.LookupEnv("IDENTITY_USERNAME"); exists {
|
||||
config.IdentityUsername = val
|
||||
}
|
||||
|
||||
if val, exists := os.LookupEnv("IDENTITY_PASSWORD"); exists {
|
||||
config.IdentityPassword = val
|
||||
}
|
||||
|
||||
return config
|
||||
}
|
||||
|
||||
@@ -6,9 +6,12 @@ import (
|
||||
"fmt"
|
||||
"os/exec"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func GetSystemHardDrives() ([]*HardDrive, error) {
|
||||
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")
|
||||
@@ -20,7 +23,7 @@ func GetSystemHardDrives() ([]*HardDrive, error) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var hardDrives []*HardDrive
|
||||
var systemHardDrives []*HardDrive
|
||||
|
||||
// Scan the output line by line
|
||||
scanner := bufio.NewScanner(&out)
|
||||
@@ -39,17 +42,16 @@ func GetSystemHardDrives() ([]*HardDrive, error) {
|
||||
}
|
||||
|
||||
// Filter out nvme drives (M.2)
|
||||
if cols[1] != "nvme" {
|
||||
if cols[1] != "usb" {
|
||||
hd := &HardDrive{
|
||||
Name: cols[0],
|
||||
Transport: cols[1],
|
||||
Size: cols[2],
|
||||
Model: cols[3],
|
||||
Serial: cols[4],
|
||||
Type: cols[5],
|
||||
Temperature: 0,
|
||||
Name: cols[0],
|
||||
Transport: cols[1],
|
||||
Size: cols[2],
|
||||
Model: cols[3],
|
||||
Serial: cols[4],
|
||||
Type: cols[5],
|
||||
}
|
||||
hardDrives = append(hardDrives, hd)
|
||||
systemHardDrives = append(systemHardDrives, hd)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -58,5 +60,33 @@ func GetSystemHardDrives() ([]*HardDrive, error) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return hardDrives, nil
|
||||
var updatedHardDrives []*HardDrive
|
||||
|
||||
for _, sysHDD := range systemHardDrives {
|
||||
var existingHD HardDrive
|
||||
q := db.Where("serial = ? AND model = ? AND type = ?", sysHDD.Serial, sysHDD.Model, sysHDD.Type)
|
||||
|
||||
if newerThan != nil && olderThan != nil {
|
||||
q = q.Preload("Temperatures", "time_stamp < ? AND time_stamp > ?", newerThan, olderThan)
|
||||
}
|
||||
|
||||
result := q.First(&existingHD)
|
||||
|
||||
if result.Error == gorm.ErrRecordNotFound {
|
||||
// Hard drive not found, create new
|
||||
db.Create(&sysHDD)
|
||||
updatedHardDrives = append(updatedHardDrives, sysHDD)
|
||||
} else {
|
||||
// Hard drive found, update existing
|
||||
existingHD.Name = sysHDD.Name
|
||||
existingHD.Transport = sysHDD.Transport
|
||||
existingHD.Size = sysHDD.Size
|
||||
existingHD.Model = sysHDD.Model
|
||||
existingHD.Type = sysHDD.Type
|
||||
db.Save(&existingHD)
|
||||
updatedHardDrives = append(updatedHardDrives, &existingHD)
|
||||
}
|
||||
}
|
||||
|
||||
return updatedHardDrives, nil
|
||||
}
|
||||
|
||||
64
lib/hardware/models.go
Normal file
64
lib/hardware/models.go
Normal file
@@ -0,0 +1,64 @@
|
||||
package hardware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/anatol/smart.go"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type HardDrive struct {
|
||||
ID uint `gorm:"primarykey"`
|
||||
CreatedAt time.Time
|
||||
UpdatedAt time.Time
|
||||
DeletedAt gorm.DeletedAt `gorm:"index"`
|
||||
Name string
|
||||
Transport string
|
||||
Size string
|
||||
Model string
|
||||
Serial string
|
||||
Type string
|
||||
Temperatures []HardDriveTemperature `gorm:"foreignKey:HardDriveID"`
|
||||
}
|
||||
|
||||
type HardDriveTemperature struct {
|
||||
gorm.Model
|
||||
HardDriveID uint
|
||||
TimeStamp time.Time
|
||||
Temperature int
|
||||
}
|
||||
|
||||
// A snapshot in time of the current state of the harddrives
|
||||
type HardwareSnapshot struct {
|
||||
TimeStamp time.Time
|
||||
HDD []*HardDrive
|
||||
}
|
||||
|
||||
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
|
||||
}
|
||||
|
||||
// Parse the temperature
|
||||
temperature := int(smartInfo.Temperature)
|
||||
|
||||
// Return the found value
|
||||
return temperature
|
||||
}
|
||||
@@ -1,46 +0,0 @@
|
||||
package hardware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/anatol/smart.go"
|
||||
)
|
||||
|
||||
type HardDrive struct {
|
||||
Name string
|
||||
Transport string
|
||||
Size string
|
||||
Model string
|
||||
Serial string
|
||||
Type string
|
||||
Temperature int
|
||||
}
|
||||
|
||||
// Fetch the temperature of the device, optinally update the reference object
|
||||
func (h *HardDrive) GetTemperature(updateTemp bool) 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
|
||||
}
|
||||
|
||||
// Parse the temperature
|
||||
temperature := int(smartInfo.Temperature)
|
||||
|
||||
// Optionally update the reference object's temperature
|
||||
if updateTemp {
|
||||
h.Temperature = temperature
|
||||
}
|
||||
|
||||
// Return the found value
|
||||
return temperature
|
||||
}
|
||||
@@ -1,128 +1,186 @@
|
||||
package svc
|
||||
|
||||
import (
|
||||
"encoding/gob"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/wcharczuk/go-chart/v2"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/config"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/hardware"
|
||||
)
|
||||
|
||||
// The path to where the snapshot database is located
|
||||
const SNAPSHOT_LIST_PATH = "./snapshots.dat"
|
||||
var db *gorm.DB
|
||||
|
||||
// A simple in-memory buffer for the history of snapshots
|
||||
var snapShotBuffer []*HardwareSnapshot
|
||||
// Initialize the database connection
|
||||
func InitDB() {
|
||||
var err error
|
||||
dbPath := config.GetConfiguration().DatabaseFilePath
|
||||
|
||||
// A snapshot in time of the current state of the harddrives
|
||||
type HardwareSnapshot struct {
|
||||
TimeStamp time.Time
|
||||
HDD []*hardware.HardDrive
|
||||
}
|
||||
|
||||
type Snapshots struct {
|
||||
List []*HardwareSnapshot
|
||||
}
|
||||
|
||||
// The function itterates through all hard disks and takes a snapshot of their state,
|
||||
// returns a struct which contains metadata as well as the harddrives themselves.
|
||||
func TakeHardwareSnapshot() (*HardwareSnapshot, error) {
|
||||
drives, err := hardware.GetSystemHardDrives()
|
||||
db, err = gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
// This should basically never happen, unless the path to the database
|
||||
// is inaccessible, doesn't exist or there's no permission to it, which
|
||||
// should and will crash the program
|
||||
panic("failed to connect database")
|
||||
}
|
||||
|
||||
snapShot := &HardwareSnapshot{
|
||||
TimeStamp: time.Now(),
|
||||
HDD: []*hardware.HardDrive{},
|
||||
// Migrate the schema
|
||||
db.AutoMigrate(&hardware.HardDrive{}, &hardware.HardDriveTemperature{})
|
||||
}
|
||||
|
||||
// Fetch the open database pointer
|
||||
func GetDatabaseRef() *gorm.DB {
|
||||
return db
|
||||
}
|
||||
|
||||
// Log the temperature of the disks
|
||||
func LogDriveTemps() error {
|
||||
drives, err := hardware.GetSystemHardDrives(db, nil, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, hdd := range drives {
|
||||
hdd.GetTemperature(true)
|
||||
snapShot.HDD = append(snapShot.HDD, hdd)
|
||||
}
|
||||
|
||||
// Append to the in-memory listing
|
||||
snapShotBuffer = append(snapShotBuffer, snapShot)
|
||||
|
||||
// Return the snapshot just in case there is any need to modify it,
|
||||
// any modification to it will also affect the current buffer from memory.
|
||||
return snapShot, nil
|
||||
}
|
||||
|
||||
// The function wil check if the `.dat` file is present, if it is then it will load it into memory
|
||||
func UpdateHardwareSnapshotsFromFile() {
|
||||
file, err := os.Open(SNAPSHOT_LIST_PATH)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return // File does not exist, no snapshots to load
|
||||
}
|
||||
panic(err) // Handle error according to your error handling policy
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
decoder := gob.NewDecoder(file)
|
||||
var snapshots Snapshots
|
||||
if err := decoder.Decode(&snapshots); err != nil {
|
||||
if err == io.EOF {
|
||||
return // End of file reached
|
||||
}
|
||||
panic(err) // Handle error according to your error handling policy
|
||||
}
|
||||
|
||||
snapShotBuffer = snapshots.List
|
||||
|
||||
fmt.Printf("Loaded %v snapshots from .dat", len(snapShotBuffer))
|
||||
}
|
||||
|
||||
// Get the list of snapshots that have been buffered in memory
|
||||
func GetHardwareSnapshot() []*HardwareSnapshot {
|
||||
return snapShotBuffer
|
||||
}
|
||||
|
||||
// Dump the current snapshot history from memory to file
|
||||
func SaveSnapshotsToFile() error {
|
||||
file, err := os.Create(SNAPSHOT_LIST_PATH)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
encoder := gob.NewEncoder(file)
|
||||
snapshots := Snapshots{List: snapShotBuffer}
|
||||
if err := encoder.Encode(snapshots); err != nil {
|
||||
return err
|
||||
temp := hdd.GetTemperature()
|
||||
db.Create(&hardware.HardDriveTemperature{
|
||||
HardDriveID: hdd.ID,
|
||||
TimeStamp: time.Now(),
|
||||
Temperature: temp,
|
||||
})
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func RunService() {
|
||||
// Run the logging service, this will periodically log the temperature of the disks with the LogDriveTemps function
|
||||
func RunLoggerService() {
|
||||
fmt.Println("[🦝] Initializing Temperature Logging Service...")
|
||||
|
||||
tickTime := time.Duration(config.GetConfiguration().DiskFetchFrequency) * time.Second
|
||||
|
||||
// Snapshot taking routine
|
||||
go func() {
|
||||
waitTime := time.Duration(config.GetConfiguration().DiskFetchFrequency) * time.Second
|
||||
for {
|
||||
time.Sleep(waitTime)
|
||||
_, err := TakeHardwareSnapshot()
|
||||
time.Sleep(tickTime)
|
||||
err := LogDriveTemps()
|
||||
if err != nil {
|
||||
fmt.Printf("Hardware Fetch Error: %s", err)
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
// Periodic saving routine
|
||||
go func() {
|
||||
for {
|
||||
waitTime := time.Duration(config.GetConfiguration().MemoryDumpFrequency) * time.Second
|
||||
time.Sleep(waitTime)
|
||||
err := SaveSnapshotsToFile()
|
||||
if err != nil {
|
||||
fmt.Printf("Memory Dump Error: %s", err)
|
||||
fmt.Printf("[🛑] Temperature logging failed: %s\n", err)
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// Generate a PNG based upon a HDD id and a date range
|
||||
func GetDiskGraphImage(hddID int, newerThan *time.Time, olderThan *time.Time) (*bytes.Buffer, error) {
|
||||
var hdd hardware.HardDrive
|
||||
// Fetch by a combination of fields
|
||||
q := db.Where("id = ?", hddID)
|
||||
|
||||
if newerThan == nil || olderThan == nil {
|
||||
q = q.Preload("Temperatures")
|
||||
} else {
|
||||
q = q.Preload("Temperatures", "time_stamp < ? AND time_stamp > ?", newerThan, olderThan)
|
||||
}
|
||||
|
||||
// Query for the instance
|
||||
result := q.First(&hdd)
|
||||
if result.Error != nil {
|
||||
return nil, result.Error
|
||||
}
|
||||
|
||||
// Prepare slices for X (time) and Y (temperature) values
|
||||
var xValues []time.Time
|
||||
var yValues []float64
|
||||
for _, temp := range hdd.Temperatures {
|
||||
xValues = append(xValues, temp.TimeStamp)
|
||||
yValues = append(yValues, float64(temp.Temperature))
|
||||
}
|
||||
|
||||
// Allocate a buffer for the graph image
|
||||
graphImageBuffer := bytes.NewBuffer([]byte{})
|
||||
|
||||
// TODO: Graph dark theme
|
||||
|
||||
// Generate the chart
|
||||
graph := chart.Chart{
|
||||
Title: fmt.Sprintf("%s:%s[%s]", hdd.Name, hdd.Serial, hdd.Size),
|
||||
TitleStyle: chart.Style{
|
||||
FontSize: 14,
|
||||
},
|
||||
|
||||
// TODO: Implement customizable sizing
|
||||
Width: 1280,
|
||||
|
||||
Background: chart.Style{
|
||||
Padding: chart.Box{
|
||||
Top: 20, Right: 20, Bottom: 20, Left: 20,
|
||||
},
|
||||
},
|
||||
|
||||
XAxis: chart.XAxis{
|
||||
Name: "Time",
|
||||
ValueFormatter: func(v interface{}) string {
|
||||
if ts, isValidTime := v.(float64); isValidTime {
|
||||
t := time.Unix(int64(ts/1e9), 0)
|
||||
|
||||
return t.Format("Jan 2 2006, 15:04")
|
||||
}
|
||||
|
||||
return ""
|
||||
},
|
||||
Style: chart.Style{},
|
||||
GridMajorStyle: chart.Style{
|
||||
StrokeColor: chart.ColorAlternateGray,
|
||||
StrokeWidth: 0.5,
|
||||
},
|
||||
GridMinorStyle: chart.Style{
|
||||
StrokeColor: chart.ColorAlternateGray.WithAlpha(64),
|
||||
StrokeWidth: 0.25,
|
||||
},
|
||||
},
|
||||
YAxis: chart.YAxis{
|
||||
Name: "Temperature (C)",
|
||||
Style: chart.Style{},
|
||||
GridMajorStyle: chart.Style{
|
||||
StrokeColor: chart.ColorAlternateGray,
|
||||
StrokeWidth: 0.5,
|
||||
},
|
||||
GridMinorStyle: chart.Style{
|
||||
StrokeColor: chart.ColorAlternateGray.WithAlpha(64),
|
||||
StrokeWidth: 0.25,
|
||||
},
|
||||
},
|
||||
Series: []chart.Series{
|
||||
chart.TimeSeries{
|
||||
Name: "Temperature",
|
||||
XValues: xValues,
|
||||
YValues: yValues,
|
||||
Style: chart.Style{
|
||||
StrokeColor: chart.ColorCyan,
|
||||
StrokeWidth: 2.0,
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
// Add a legend to the chart
|
||||
graph.Elements = []chart.Renderable{
|
||||
chart.Legend(&graph, chart.Style{
|
||||
Padding: chart.Box{
|
||||
Top: 5, Right: 5, Bottom: 5, Left: 5,
|
||||
},
|
||||
FontSize: 10,
|
||||
}),
|
||||
}
|
||||
|
||||
// Render the chart into the byte buffer
|
||||
err := graph.Render(chart.PNG, graphImageBuffer)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return graphImageBuffer, nil
|
||||
}
|
||||
|
||||
45
lib/svc/service_cleanup.go
Normal file
45
lib/svc/service_cleanup.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package svc
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/config"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/hardware"
|
||||
)
|
||||
|
||||
// Delete all thermal entries that are older than X amount of seconds
|
||||
func CleanupOldData() error {
|
||||
cfg := config.GetConfiguration()
|
||||
|
||||
beforeDate := time.Now().Add(-1 * time.Duration(cfg.MaxHistoryAge) * time.Second)
|
||||
|
||||
deleteResult := db.Where("time_stamp < ?", beforeDate).Delete(&hardware.HardDriveTemperature{})
|
||||
if deleteResult.Error != nil {
|
||||
fmt.Printf("[🛑] Error during cleanup: %s\n", deleteResult.Error)
|
||||
return db.Error
|
||||
}
|
||||
|
||||
if deleteResult.RowsAffected > 0 {
|
||||
fmt.Printf("[🛑] Cleaned up %v entries before %s\n", deleteResult.RowsAffected, beforeDate)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func RunCleanupService() {
|
||||
fmt.Println("[🦝] Initializing Log Cleanup Service...")
|
||||
|
||||
tickTime := time.Duration(config.GetConfiguration().CleanupServiceFrequency) * time.Second
|
||||
|
||||
// Snapshot taking routine
|
||||
go func() {
|
||||
for {
|
||||
time.Sleep(tickTime)
|
||||
err := CleanupOldData()
|
||||
if err != nil {
|
||||
fmt.Printf("🛑 Cleanup process failed: %s\n", err)
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
@@ -3,6 +3,8 @@ package web
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/hardware"
|
||||
@@ -10,19 +12,83 @@ import (
|
||||
)
|
||||
|
||||
func setupApi(r *gin.Engine) {
|
||||
api := r.Group("/v1/api")
|
||||
api := r.Group("/api/v1")
|
||||
|
||||
// Fetch the chart image for the disk's temperature
|
||||
api.GET("/disks/:diskid/chart", func(ctx *gin.Context) {
|
||||
diskIDString := ctx.Param("diskid")
|
||||
diskId, err := strconv.Atoi(diskIDString)
|
||||
if err != nil {
|
||||
ctx.AbortWithStatusJSON(400, gin.H{
|
||||
"error": err.Error(),
|
||||
"message": "Invalid Disk ID",
|
||||
})
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
var olderThan, newerThan *time.Time
|
||||
|
||||
if ot := ctx.Query("older"); ot != "" {
|
||||
fmt.Printf("ot = %s\n", ot)
|
||||
if otInt, err := strconv.ParseInt(ot, 10, 64); err == nil {
|
||||
otTime := time.UnixMilli(otInt)
|
||||
olderThan = &otTime
|
||||
}
|
||||
}
|
||||
|
||||
if nt := ctx.Query("newer"); nt != "" {
|
||||
fmt.Printf("nt = %s\n", nt)
|
||||
if ntInt, err := strconv.ParseInt(nt, 10, 64); err == nil {
|
||||
ntTime := time.UnixMilli(ntInt)
|
||||
newerThan = &ntTime
|
||||
}
|
||||
}
|
||||
|
||||
fmt.Printf("olderThan = %s\n", olderThan)
|
||||
fmt.Printf("newerThan = %s\n", newerThan)
|
||||
|
||||
graphData, err := svc.GetDiskGraphImage(diskId, newerThan, olderThan)
|
||||
if err != nil {
|
||||
ctx.AbortWithStatusJSON(500, gin.H{
|
||||
"error": err.Error(),
|
||||
"message": "Graph generation issue",
|
||||
})
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// Set the content type header
|
||||
ctx.Writer.Header().Set("Content-Type", "image/png")
|
||||
|
||||
// Write the image data to the response
|
||||
ctx.Writer.WriteHeader(http.StatusOK)
|
||||
_, err = graphData.WriteTo(ctx.Writer)
|
||||
if err != nil {
|
||||
ctx.AbortWithStatusJSON(500, gin.H{
|
||||
"error": err.Error(),
|
||||
"message": "Write error",
|
||||
})
|
||||
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
// Get a list of all the disks
|
||||
api.GET("/disks", func(ctx *gin.Context) {
|
||||
|
||||
olderThan := time.Now().Add(time.Minute * time.Duration(10) * -1)
|
||||
newerThan := time.Now()
|
||||
|
||||
// Fetch the disk list
|
||||
disks, err := hardware.GetSystemHardDrives()
|
||||
disks, err := hardware.GetSystemHardDrives(svc.GetDatabaseRef(), &olderThan, &newerThan)
|
||||
if err != nil {
|
||||
ctx.Error(err)
|
||||
}
|
||||
|
||||
if ctx.Request.URL.Query().Get("temp") != "" {
|
||||
for _, d := range disks {
|
||||
temp := d.GetTemperature(true)
|
||||
fmt.Printf("Disk Temp: %v", temp)
|
||||
d.GetTemperature()
|
||||
}
|
||||
}
|
||||
|
||||
@@ -32,10 +98,4 @@ func setupApi(r *gin.Engine) {
|
||||
})
|
||||
})
|
||||
|
||||
api.GET("/snapshots", func(ctx *gin.Context) {
|
||||
snapshots := svc.GetHardwareSnapshot()
|
||||
|
||||
ctx.JSON(http.StatusOK, snapshots)
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
11
lib/web/auth_middleware.go
Normal file
11
lib/web/auth_middleware.go
Normal file
@@ -0,0 +1,11 @@
|
||||
package web
|
||||
|
||||
import "github.com/gin-gonic/gin"
|
||||
|
||||
func BasicAuthMiddleware(username, password string) gin.HandlerFunc {
|
||||
authorized := gin.Accounts{
|
||||
username: password,
|
||||
}
|
||||
|
||||
return gin.BasicAuth(authorized)
|
||||
}
|
||||
@@ -1,10 +1,14 @@
|
||||
package web
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/hardware"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/svc"
|
||||
)
|
||||
|
||||
func setupFrontend(r *gin.Engine) {
|
||||
@@ -12,20 +16,51 @@ func setupFrontend(r *gin.Engine) {
|
||||
r.Static("/static", "./static")
|
||||
|
||||
// Set up a route for the root URL
|
||||
r.GET("/", func(c *gin.Context) {
|
||||
|
||||
hardDrives, err := hardware.GetSystemHardDrives()
|
||||
r.GET("/", func(ctx *gin.Context) {
|
||||
hardDrives, err := hardware.GetSystemHardDrives(svc.GetDatabaseRef(), nil, nil)
|
||||
if err != nil {
|
||||
c.AbortWithStatus(500)
|
||||
ctx.AbortWithStatus(500)
|
||||
}
|
||||
|
||||
for _, hdd := range hardDrives {
|
||||
hdd.GetTemperature(true)
|
||||
hdd.GetTemperature()
|
||||
}
|
||||
|
||||
var olderThan, newerThan *time.Time
|
||||
|
||||
if ot := ctx.Query("older"); ot != "" {
|
||||
fmt.Printf("ot = %s\n", ot)
|
||||
if otInt, err := strconv.ParseInt(ot, 10, 64); err == nil {
|
||||
otTime := time.UnixMilli(otInt)
|
||||
olderThan = &otTime
|
||||
}
|
||||
}
|
||||
|
||||
if nt := ctx.Query("newer"); nt != "" {
|
||||
fmt.Printf("nt = %s\n", nt)
|
||||
if ntInt, err := strconv.ParseInt(nt, 10, 64); err == nil {
|
||||
ntTime := time.UnixMilli(ntInt)
|
||||
newerThan = &ntTime
|
||||
}
|
||||
}
|
||||
|
||||
if olderThan == nil {
|
||||
genTime := time.Now().Add(time.Hour * -1)
|
||||
|
||||
olderThan = &genTime
|
||||
}
|
||||
|
||||
if newerThan == nil {
|
||||
genTime := time.Now()
|
||||
|
||||
newerThan = &genTime
|
||||
}
|
||||
|
||||
// Render the HTML template
|
||||
c.HTML(http.StatusOK, "index.html", gin.H{
|
||||
ctx.HTML(http.StatusOK, "index.html", gin.H{
|
||||
"drives": hardDrives,
|
||||
"older": olderThan.UnixMilli(),
|
||||
"newer": newerThan.UnixMilli(),
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
@@ -2,12 +2,16 @@ package web
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/config"
|
||||
)
|
||||
|
||||
func SetupRouter() *gin.Engine {
|
||||
// Initialize the Gin engine
|
||||
cfg := config.GetConfiguration()
|
||||
r := gin.Default()
|
||||
|
||||
r.Use(BasicAuthMiddleware(cfg.IdentityUsername, cfg.IdentityPassword))
|
||||
|
||||
// Setup Health Pings
|
||||
setupHealth(r)
|
||||
// Setup Api
|
||||
|
||||
26
main.go
26
main.go
@@ -9,30 +9,40 @@ import (
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/joho/godotenv"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/config"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/svc"
|
||||
"tea.chunkbyte.com/kato/drive-health/lib/web"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Load existing snapshots from file
|
||||
svc.UpdateHardwareSnapshotsFromFile()
|
||||
// Load .env file if it exists
|
||||
if err := godotenv.Load(); err != nil {
|
||||
log.Println("[🟨] No .env file found")
|
||||
}
|
||||
|
||||
// Init the database
|
||||
svc.InitDB()
|
||||
cfg := config.GetConfiguration()
|
||||
|
||||
router := web.SetupRouter()
|
||||
|
||||
srv := &http.Server{
|
||||
Addr: ":8080",
|
||||
Addr: cfg.Listen,
|
||||
Handler: router,
|
||||
}
|
||||
|
||||
// Run the server in a goroutine
|
||||
go func() {
|
||||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
log.Fatalf("listen: %s\n", err)
|
||||
log.Fatalf("[🛑] listening failed: %s\n", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// Run the hardware service
|
||||
svc.RunService()
|
||||
svc.RunLoggerService()
|
||||
// Run the cleanup service
|
||||
svc.RunCleanupService()
|
||||
|
||||
// Setting up signal capturing
|
||||
quit := make(chan os.Signal, 1)
|
||||
@@ -40,14 +50,14 @@ func main() {
|
||||
|
||||
// Block until a signal is received
|
||||
<-quit
|
||||
log.Println("Shutting down server...")
|
||||
log.Println("[🦝] Shutting down server...")
|
||||
|
||||
// Graceful shutdown
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
if err := srv.Shutdown(ctx); err != nil {
|
||||
log.Fatal("Server forced to shutdown:", err)
|
||||
log.Fatal("[🛑] Server forced to shutdown:", err)
|
||||
}
|
||||
|
||||
log.Println("Server exiting")
|
||||
log.Println("[🦝] Server exiting")
|
||||
}
|
||||
|
||||
181
static/main.js
181
static/main.js
@@ -1,134 +1,55 @@
|
||||
function stringToColor(str) {
|
||||
let hash = 0;
|
||||
for (let i = 0; i < str.length; i++) {
|
||||
hash = str.charCodeAt(i) + ((hash << 5) - hash);
|
||||
}
|
||||
let color = '#';
|
||||
for (let i = 0; i < 3; i++) {
|
||||
const value = (hash >> (i * 8)) & 0xFF;
|
||||
color += ('00' + value.toString(16)).substr(-2);
|
||||
}
|
||||
return color;
|
||||
/**
|
||||
* @typedef {number}
|
||||
*/
|
||||
var initialInputOlder = 0; // miliseconds Unix TimeStamp
|
||||
/**
|
||||
* @typedef {number}
|
||||
*/
|
||||
var initialInputNewer = 0; // miliseconds Unix TimeStamp
|
||||
|
||||
|
||||
/**
|
||||
* @typedef {HTMLInputElement}
|
||||
*/
|
||||
var olderThanInputElement;
|
||||
/**
|
||||
* @typedef {HTMLInputElement}
|
||||
*/
|
||||
var newerThanInputElement;
|
||||
|
||||
document.addEventListener(`DOMContentLoaded`, initializePage)
|
||||
|
||||
function initializePage() {
|
||||
|
||||
// Update the page's time filter
|
||||
initialInputOlder = Number(document.getElementById(`inp-older`).textContent.trim())
|
||||
initialInputNewer = Number(document.getElementById(`inp-newer`).textContent.trim())
|
||||
|
||||
// Bind the date elements
|
||||
olderThanInputElement = document.getElementById(`olderThan`);
|
||||
newerThanInputElement = document.getElementById(`newerThan`);
|
||||
|
||||
olderThanInputElement.value = convertTimestampToDateTimeLocal(initialInputOlder);
|
||||
newerThanInputElement.value = convertTimestampToDateTimeLocal(initialInputNewer);
|
||||
|
||||
}
|
||||
|
||||
document.addEventListener('DOMContentLoaded', function() {
|
||||
const diskTableBody = document.getElementById('disk-table-body');
|
||||
const ctx = document.getElementById('temperatureChart').getContext('2d');
|
||||
let temperatureChart = new Chart(ctx, {
|
||||
type: 'line',
|
||||
data: {
|
||||
datasets: []
|
||||
},
|
||||
options: {
|
||||
scales: {
|
||||
x: {
|
||||
type: 'time',
|
||||
time: {
|
||||
unit: 'second',
|
||||
displayFormats: {
|
||||
second: 'HH:mm:ss'
|
||||
}
|
||||
},
|
||||
title: {
|
||||
display: true,
|
||||
text: 'Time'
|
||||
}
|
||||
},
|
||||
y: {
|
||||
beginAtZero: true,
|
||||
title: {
|
||||
display: true,
|
||||
text: 'Temperature (°C)'
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
// Handle one of the date elements having their value changed.
|
||||
function applyDateInterval() {
|
||||
const olderTimeStamp = new Date(olderThanInputElement.value).getTime()
|
||||
const newerTimeStamp = new Date(newerThanInputElement.value).getTime()
|
||||
|
||||
function fetchAndUpdateDisks() {
|
||||
fetch('/v1/api/disks?temp=true')
|
||||
.then(response => response.json())
|
||||
.then(data => {
|
||||
updateDiskTable(data.disks);
|
||||
})
|
||||
.catch(error => console.error('Error fetching disk data:', error));
|
||||
}
|
||||
window.location.href = `/?older=${olderTimeStamp}&newer=${newerTimeStamp}`;
|
||||
}
|
||||
|
||||
function updateDiskTable(disks) {
|
||||
let tableHTML = '';
|
||||
disks.forEach(disk => {
|
||||
tableHTML += `
|
||||
<tr>
|
||||
<td>${disk.Name}</td>
|
||||
<td>${disk.Transport}</td>
|
||||
<td>${disk.Size}</td>
|
||||
<td>${disk.Model}</td>
|
||||
<td>${disk.Serial}</td>
|
||||
<td>${disk.Type}</td>
|
||||
<td>${disk.Temperature}</td>
|
||||
</tr>
|
||||
`;
|
||||
});
|
||||
diskTableBody.innerHTML = tableHTML;
|
||||
}
|
||||
|
||||
function fetchAndUpdateTemperatureChart() {
|
||||
fetch('/v1/api/snapshots')
|
||||
.then(response => response.json())
|
||||
.then(snapshots => {
|
||||
updateTemperatureChart(snapshots);
|
||||
})
|
||||
.catch(error => console.error('Error fetching temperature data:', error));
|
||||
}
|
||||
|
||||
function updateTemperatureChart(snapshots) {
|
||||
// Clear existing datasets
|
||||
temperatureChart.data.datasets = [];
|
||||
|
||||
snapshots.forEach(snapshot => {
|
||||
const time = new Date(snapshot.TimeStamp);
|
||||
snapshot.HDD.forEach(disk => {
|
||||
let dataset = temperatureChart.data.datasets.find(d => d.label === disk.Name);
|
||||
if (!dataset) {
|
||||
dataset = {
|
||||
label: disk.Name,
|
||||
data: [],
|
||||
fill: false,
|
||||
borderColor: stringToColor(disk.Name),
|
||||
borderWidth: 1
|
||||
};
|
||||
temperatureChart.data.datasets.push(dataset);
|
||||
}
|
||||
|
||||
dataset.data.push({
|
||||
x: time,
|
||||
y: disk.Temperature
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
temperatureChart.update();
|
||||
}
|
||||
|
||||
// Chart.js zoom and pan configuration
|
||||
temperatureChart.options.plugins.zoom = {
|
||||
zoom: {
|
||||
wheel: {
|
||||
enabled: true,
|
||||
},
|
||||
pinch: {
|
||||
enabled: true
|
||||
},
|
||||
mode: 'x',
|
||||
},
|
||||
pan: {
|
||||
enabled: true,
|
||||
mode: 'x',
|
||||
}
|
||||
};
|
||||
|
||||
fetchAndUpdateDisks();
|
||||
fetchAndUpdateTemperatureChart();
|
||||
setInterval(fetchAndUpdateDisks, 5000);
|
||||
setInterval(fetchAndUpdateTemperatureChart, 5000);
|
||||
});
|
||||
/**
|
||||
* Converts a Unix timestamp to a standard datetime string
|
||||
* @param {number} timestamp - The Unix timestamp in milliseconds.
|
||||
* @returns {string} - A normal string with Y-m-d H:i:s format
|
||||
*/
|
||||
function convertTimestampToDateTimeLocal(timestamp) {
|
||||
const date = new Date(timestamp);
|
||||
const offset = date.getTimezoneOffset() * 60000; // offset in milliseconds
|
||||
const localDate = new Date(date.getTime() - offset);
|
||||
return localDate.toISOString().slice(0, 19);
|
||||
}
|
||||
|
||||
142
static/style.css
142
static/style.css
@@ -1,45 +1,143 @@
|
||||
body {
|
||||
font-family: Arial, sans-serif;
|
||||
background-color: #333;
|
||||
color: #fff;
|
||||
margin: 0;
|
||||
@import url('https://fonts.googleapis.com/css2?family=Noto+Sans+Mono:wght@100..900&family=Roboto:wght@100;300;400&display=swap');
|
||||
|
||||
:root {
|
||||
--bg0: #202327;
|
||||
--bg1: #282d33;
|
||||
--bg2: #31373f;
|
||||
--bg3: #3e4248;
|
||||
--bg4: #1a1c1f;
|
||||
|
||||
--fg0: #bbc0ca;
|
||||
--fg1: #434c56;
|
||||
|
||||
--acc: #bbc0ca;
|
||||
}
|
||||
|
||||
:root {
|
||||
color: var(--fg0);
|
||||
font-family: "Noto Sans Mono", "Roboto", sans-serif;
|
||||
}
|
||||
|
||||
html, body {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
|
||||
width: 100vw;
|
||||
|
||||
overflow: auto;
|
||||
|
||||
background-color: var(--bg0);
|
||||
}
|
||||
|
||||
.container {
|
||||
width: 80%;
|
||||
margin: auto;
|
||||
margin: 1rem auto;
|
||||
max-width: 768px;
|
||||
|
||||
border-radius: 8px;
|
||||
border: 1px solid var(--fg1);
|
||||
|
||||
background-color: var(--bg1);
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
h1 {
|
||||
text-align: center;
|
||||
padding: 20px 0;
|
||||
.container-titlebar {
|
||||
width: 100%;
|
||||
|
||||
background-color: var(--bg2);
|
||||
}
|
||||
|
||||
.container .pad {
|
||||
padding: .5rem 1rem;
|
||||
}
|
||||
|
||||
.container-titlebar h4 {
|
||||
padding: 0;
|
||||
margin: 0;
|
||||
}
|
||||
|
||||
|
||||
/* Table */
|
||||
|
||||
table {
|
||||
width: 100%;
|
||||
margin-top: 20px;
|
||||
border-collapse: collapse;
|
||||
}
|
||||
|
||||
table, th, td {
|
||||
border: 1px solid #ddd;
|
||||
table thead tr {
|
||||
border-bottom: 1px solid var(--bg2);
|
||||
}
|
||||
|
||||
th, td {
|
||||
text-align: left;
|
||||
padding: 8px;
|
||||
.graph-image {
|
||||
max-width: 100%;
|
||||
}
|
||||
|
||||
th {
|
||||
background-color: #555;
|
||||
.disk-graph-entry {
|
||||
background-color: var(--bg3);
|
||||
|
||||
border-radius: 8px;
|
||||
|
||||
padding: .3rem .5rem;
|
||||
}
|
||||
|
||||
tr:nth-child(even) {
|
||||
background-color: #666;
|
||||
|
||||
/* Controls */
|
||||
|
||||
input {
|
||||
padding: .25rem .5rem;
|
||||
font-size: 16px;
|
||||
background-color: var(--bg3);
|
||||
color: var(--fg0);
|
||||
|
||||
border: 1px solid var(--fg1);
|
||||
border-radius: 4px;
|
||||
}
|
||||
|
||||
tr:hover {
|
||||
background-color: #555;
|
||||
.btn {
|
||||
font-size: 16px;
|
||||
|
||||
padding: .5rem 1rem;
|
||||
|
||||
border: 1px solid var(--fg1);
|
||||
border-radius: 6px;
|
||||
|
||||
background-color: var(--bg4);
|
||||
color: var(--fg0);
|
||||
|
||||
cursor: pointer;
|
||||
}
|
||||
|
||||
.btn:hover {
|
||||
background-color: var(--bg3);
|
||||
}
|
||||
|
||||
.input-grp {
|
||||
display: flex;
|
||||
flex-flow: column;
|
||||
}
|
||||
|
||||
.input-grp label {
|
||||
margin-bottom: .25rem;
|
||||
font-weight: bold;
|
||||
}
|
||||
|
||||
.graph-controls {
|
||||
display: flex;
|
||||
flex-flow: wrap;
|
||||
|
||||
width: 100%;
|
||||
}
|
||||
|
||||
.graph-controls input:nth-child(0) {
|
||||
margin-right: 1rem;
|
||||
}
|
||||
|
||||
.controls-panel {
|
||||
display: flex;
|
||||
flex-flow: column;
|
||||
|
||||
padding: 1rem 0;
|
||||
|
||||
border-bottom: 1px solid var(--fg1);
|
||||
margin-bottom: 1rem;
|
||||
|
||||
}
|
||||
@@ -6,30 +6,114 @@
|
||||
<link rel="stylesheet" href="/static/style.css">
|
||||
<title>Drive Health Dashboard</title>
|
||||
</head>
|
||||
|
||||
{{ $older := .older }}
|
||||
{{ $newer := .newer }}
|
||||
|
||||
<body>
|
||||
<div class="container">
|
||||
<h1>Drive Health Dashboard</h1>
|
||||
<table>
|
||||
<thead>
|
||||
<!-- ... table headers ... -->
|
||||
</thead>
|
||||
<tbody id="disk-table-body">
|
||||
<!-- Data will be populated here by JavaScript -->
|
||||
</tbody>
|
||||
</table>
|
||||
<div class="container bordered">
|
||||
|
||||
<hr>
|
||||
|
||||
<div class="chart-container" style="position: relative; height:40vh; width:80vw; overflow-x: scroll;">
|
||||
<canvas id="temperatureChart"></canvas>
|
||||
<div class="container-titlebar">
|
||||
<div class="pad">
|
||||
<h4>Available Disks</h4>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="container-body">
|
||||
<div class="pad">
|
||||
{{ if len .drives }}
|
||||
<table id="disks-table">
|
||||
<thead>
|
||||
<tr>
|
||||
<td>ID</td>
|
||||
<td>Name</td>
|
||||
<td>Model</td>
|
||||
<td>Serial</td>
|
||||
<td>Temperature</td>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody id="disk-table-body">
|
||||
{{ range .drives }}
|
||||
{{ $temp := .GetTemperature }}
|
||||
|
||||
<tr>
|
||||
<td>#{{ .ID }}</td>
|
||||
<td> {{ .Name }}</td>
|
||||
<td> {{ .Model }}</td>
|
||||
<td> {{ .Serial }}</td>
|
||||
|
||||
{{ if gt $temp 50 }} <!-- Temperature greater than 50°C -->
|
||||
<td style="color: red;">{{ $temp }}°C</td>
|
||||
{{ else if gt $temp 30 }} <!-- Temperature between 31°C and 50°C -->
|
||||
<td style="color: orange;">{{ $temp }}°C</td>
|
||||
{{ else }} <!-- Temperature 30°C or below -->
|
||||
<td style="color: lime;">{{ $temp }}°C</td>
|
||||
{{ end }}
|
||||
</tr>
|
||||
{{ end }}
|
||||
</tbody>
|
||||
</table>
|
||||
{{ else }}
|
||||
<p>No hard drives found.</p>
|
||||
{{ end }}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
|
||||
</div>
|
||||
|
||||
<script src="https://cdn.jsdelivr.net/npm/chart.js@latest"></script>
|
||||
<script src="https://cdn.jsdelivr.net/npm/chartjs-adapter-date-fns@latest"></script>
|
||||
<script src="https://cdn.jsdelivr.net/npm/chartjs-plugin-zoom@latest"></script>
|
||||
<div class="container bordered">
|
||||
|
||||
<div class="container-titlebar">
|
||||
<div class="pad">
|
||||
<h4>Temperature Graph</h4>
|
||||
</div>
|
||||
</div>
|
||||
<div class="container-body">
|
||||
<div class="pad">
|
||||
|
||||
<!-- Controls -->
|
||||
<div class="controls-panel">
|
||||
<div class="graph-controls">
|
||||
<span id="inp-older" style="display: none !important" hidden="true">{{ .older }}</span>
|
||||
<span id="inp-newer" style="display: none !important" hidden="true">{{ .newer }}</span>
|
||||
|
||||
<div class="input-grp" style="margin-right: 1rem;">
|
||||
<label for="olderThan">From Date</label>
|
||||
<input id="olderThan" type="datetime-local" class="date-change-inp">
|
||||
</div>
|
||||
|
||||
<div class="input-grp">
|
||||
<label for="newerThan">To Date</label>
|
||||
<input id="newerThan" type="datetime-local" class="date-change-inp">
|
||||
</div>
|
||||
|
||||
</div>
|
||||
|
||||
<div class="btn-group">
|
||||
<button type="button" class="btn" onclick="applyDateInterval()" style="margin-top: 1rem;">
|
||||
Filter
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- Drives -->
|
||||
{{ if len .drives }}
|
||||
{{ range .drives }}
|
||||
<div class="disk-graph-entry bordered" id="disk-temp-{{ .ID }}">
|
||||
<h4>{{.Name}}:{{.Serial}} [{{.Size}}]</h4>
|
||||
<a href="/api/v1/disks/{{.ID}}/chart" target="_blank">
|
||||
<img class="graph-image" src="/api/v1/disks/{{.ID}}/chart?older={{ $older }}&newer={{ $newer }}" alt="{{ .Model }} Image">
|
||||
</a>
|
||||
</div>
|
||||
<br>
|
||||
{{ end }}
|
||||
{{ else }}
|
||||
<p>No hard drives found.</p>
|
||||
{{ end }}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
</div>
|
||||
|
||||
<script src="/static/main.js"></script>
|
||||
</body>
|
||||
|
||||
Reference in New Issue
Block a user