main.go

@@ -77,14 +77,11 @@ func Main() {
 
 	initUsbGadget()
 
-	if err := setInitialVirtualMediaState(); err != nil {
+	err = setInitialVirtualMediaState()
+	if err != nil {
 		logger.Warn().Err(err).Msg("failed to set initial virtual media state")
 	}
 
-	if err := initImagesFolder(); err != nil {
-		logger.Warn().Err(err).Msg("failed to init images folder")
-	}
-
 	go func() {
 		time.Sleep(15 * time.Minute)
 		for {

resource/jetkvm_native.sha256

@@ -1 +1 @@
-6dabd0e657dd099280d9173069687786a4a8c9c25cf7f9e7ce2f940cab67c521
+4b925c7aa73d2e35a227833e806658cb17e1d25900611f93ed70b11ac9f1716d

usb_mass_storage.go

@@ -94,20 +94,9 @@ var nbdDevice *NBDDevice
 
 const imagesFolder = "/userdata/jetkvm/images"
 
-func initImagesFolder() error {
-	err := os.MkdirAll(imagesFolder, 0755)
-	if err != nil {
-		return fmt.Errorf("failed to create images folder: %w", err)
-	}
-	return nil
-}
-
 func rpcMountBuiltInImage(filename string) error {
 	logger.Info().Str("filename", filename).Msg("Mount Built-In Image")
-	if err := initImagesFolder(); err != nil {
-		return err
-	}
-
+	_ = os.MkdirAll(imagesFolder, 0755)
 	imagePath := filepath.Join(imagesFolder, filename)
 
 	// Check if the file exists in the imagesFolder
@@ -235,14 +224,13 @@ func getInitialVirtualMediaState() (*VirtualMediaState, error) {
 	}
 
 	// TODO: check if it's WebRTC or HTTP
-	switch diskPath {
-	case "":
+	if diskPath == "" {
 		return nil, nil
-	case "/dev/nbd0":
+	} else if diskPath == "/dev/nbd0" {
 		initialState.Source = HTTP
 		initialState.URL = "/"
 		initialState.Size = 1
-	default:
+	} else {
 		initialState.Filename = filepath.Base(diskPath)
 		// get size from file
 		logger.Info().Str("diskPath", diskPath).Msg("getting file size")