bokoFs/main.go

package main

import (
	"encoding/json"
	"fmt"
	"os"
	"strings"

	smart "imuslab.com/bokofs/bokofsd/mod/middleware/SMART"
)

var dir string

func main() {

	//Get all the devices under /dev that is either sd or nvme
	deviceFiles, err := os.ReadDir("/dev")
	if err != nil {
		panic(err)
	}

	for _, deviceFile := range deviceFiles {
		if deviceFile.IsDir() {
			continue
		}
		if strings.HasPrefix(deviceFile.Name(), "sd") || strings.HasPrefix(deviceFile.Name(), "nvme") {
			if strings.HasPrefix(deviceFile.Name(), "sd") && len(deviceFile.Name()) > 3 {
				continue
			}
			if strings.HasPrefix(deviceFile.Name(), "nvme") && strings.Contains(deviceFile.Name()[4:], "p") {
				continue
			}
			fullPath := "/dev/" + deviceFile.Name()
			fmt.Println(fullPath)

			if !smart.IsNVMeDevice(fullPath) && !smart.IsSATADevice(fullPath) {
				fmt.Println("Unsupported disk type")
				continue
			}
			//Get the SMART data printout in json
			smartdata, err := smart.GetSMARTData(fullPath)
			if err != nil {
				fmt.Println(err)
			}

			js, _ := json.MarshalIndent(smartdata, "", " ")
			fmt.Println(string(js))
		}
	}

	/*
		dirFlag := flag.String("d", "./", "Directory to serve from. Default is CWD")
		httpPort := flag.Int("p", 80, "Port to serve on (Plain HTTP)")
		httpsPort := flag.Int("ps", 443, "Port to serve TLS on")
		serveSecure := flag.Bool("s", false, "Serve HTTPS. Default false")

		flag.Parse()

		dir = *dirFlag

		srv := &webdav.Handler{
			FileSystem: webdav.Dir(dir),
			LockSystem: webdav.NewMemLS(),
			Logger: func(r *http.Request, err error) {
				if err != nil {
					log.Printf("WEBDAV [%s]: %s, ERROR: %s\n", r.Method, r.URL, err)
				} else {
					log.Printf("WEBDAV [%s]: %s \n", r.Method, r.URL)
				}
			},
		}
		http.Handle("/", srv)
		if *serveSecure == true {
			if _, err := os.Stat("./cert.pem"); err != nil {
				fmt.Println("[x] No cert.pem in current directory. Please provide a valid cert")
				return
			}
			if _, er := os.Stat("./key.pem"); er != nil {
				fmt.Println("[x] No key.pem in current directory. Please provide a valid cert")
				return
			}

			go http.ListenAndServeTLS(fmt.Sprintf(":%d", *httpsPort), "cert.pem", "key.pem", nil)
		}
		if err := http.ListenAndServe(fmt.Sprintf(":%d", *httpPort), nil); err != nil {
			log.Fatalf("Error with WebDAV server: %v", err)
		}
	*/
}