Working SMB file share

cmd/run.go

@@ -4,9 +4,11 @@ import (
 	"context"
 	"fmt"
 	"log/slog"
+	"net"
 	"os"
 
 	"github.com/AlexSSD7/linsk/vm"
+	"github.com/sethvargo/go-password/password"
 	"github.com/spf13/cobra"
 )
 
@@ -19,6 +21,12 @@ var runCmd = &cobra.Command{
 		vmMountDevName := args[1]
 		fsType := args[2]
 
+		networkSharePort, err := getClosestAvailPort(9000)
+		if err != nil {
+			slog.Error("Failed to get closest available host port for network file share", "error", err)
+			os.Exit(1)
+		}
+
 		// TODO: `slog` library prints entire stack traces for errors which makes reading errors challenging.
 
 		os.Exit(runVM(args[0], func(ctx context.Context, i *vm.Instance, fm *vm.FileManager) int {
@@ -31,10 +39,31 @@ var runCmd = &cobra.Command{
 				return 1
 			}
 
-			fmt.Println("Mounted! Now sleeping")
+			sharePWD, err := password.Generate(16, 10, 0, false, false)
+			if err != nil {
+				slog.Error("Failed to generate ephemeral password for network file share", "error", err)
+				return 1
+			}
+
+			shareURI := "smb://linsk:" + sharePWD + "@127.0.0.1:" + fmt.Sprint(networkSharePort)
+
+			fmt.Fprintf(os.Stderr, "================\n[Network File Share Config]\nThe network file share was started. Please use the credentials below to connect to the file server.\n\nType: SMB\nServer Address: smb://127.0.0.1:%v\nUsername: linsk\nPassword: %v\n\nShare URI: %v\n================\n", networkSharePort, sharePWD, shareURI)
+
+			err = fm.StartSMB([]byte(sharePWD))
+			if err != nil {
+				slog.Error("Failed to start SMB server", "error", err)
+				return 1
+			}
+
+			slog.Info("Started the network share successfully", "type", "smb")
+
 			<-ctx.Done()
 			return 0
-		}))
+		}, []vm.PortForwardingConfig{{
+			HostIP:   net.ParseIP("127.0.0.1"), // TODO: Make this changeable.
+			HostPort: networkSharePort,
+			VMPort:   445,
+		}}))
 
 		return nil
 	},