POC showing how statfile can call the conn wshclient to get file info (#242)

cmd/server/main-server.go

@@ -26,6 +26,8 @@ import (
 	"github.com/wavetermdev/thenextwave/pkg/wconfig"
 	"github.com/wavetermdev/thenextwave/pkg/web"
 	"github.com/wavetermdev/thenextwave/pkg/wps"
+	"github.com/wavetermdev/thenextwave/pkg/wshrpc"
+	"github.com/wavetermdev/thenextwave/pkg/wshrpc/wshremote"
 	"github.com/wavetermdev/thenextwave/pkg/wshrpc/wshserver"
 	"github.com/wavetermdev/thenextwave/pkg/wshutil"
 	"github.com/wavetermdev/thenextwave/pkg/wstore"
@@ -152,6 +154,8 @@ func createMainWshClient() {
 	rpc := wshserver.GetMainRpcClient()
 	wshutil.DefaultRouter.RegisterRoute(wshutil.DefaultRoute, rpc)
 	wps.Broker.SetClient(wshutil.DefaultRouter)
+	localConnWsh := wshutil.MakeWshRpc(nil, nil, wshrpc.RpcContext{}, &wshremote.ServerImpl{})
+	wshutil.DefaultRouter.RegisterRoute(wshutil.MakeConnectionRouteId(wshrpc.LocalConnName), localConnWsh)
 }
 
 func main() {

frontend/app/store/services.ts

@@ -74,7 +74,9 @@ class FileServiceType {
     SaveFile(arg1: string, arg2: string): Promise<void> {
         return WOS.callBackendService("file", "SaveFile", Array.from(arguments))
     }
-    StatFile(arg1: string): Promise<FileInfo> {
+
+    // get file info
+    StatFile(connection: string, path: string): Promise<FileInfo> {
         return WOS.callBackendService("file", "StatFile", Array.from(arguments))
     }
 }

frontend/app/view/preview/preview.tsx

@@ -44,6 +44,7 @@ export class PreviewModel implements ViewModel {
     isCeView: jotai.PrimitiveAtom<boolean>;
 
     fileName: jotai.WritableAtom<string, [string], void>;
+    connection: jotai.Atom<string>;
     statFile: jotai.Atom<Promise<FileInfo>>;
     fullFile: jotai.Atom<Promise<FullFile>>;
     fileMimeType: jotai.Atom<Promise<string>>;
@@ -187,7 +188,6 @@ export class PreviewModel implements ViewModel {
             }
             return null;
         });
-
         this.fileName = jotai.atom<string, [string], void>(
             (get) => {
                 return get(this.blockAtom)?.meta?.file;
@@ -196,14 +196,18 @@ export class PreviewModel implements ViewModel {
                 services.ObjectService.UpdateObjectMeta(`block:${blockId}`, { file: update });
             }
         );
+        this.connection = jotai.atom<string>((get) => {
+            return get(this.blockAtom)?.meta?.connection;
+        });
         this.statFile = jotai.atom<Promise<FileInfo>>(async (get) => {
             const fileName = get(this.fileName);
             if (fileName == null) {
                 return null;
             }
+            const conn = get(this.connection) ?? "";
             // const statFile = await FileService.StatFile(fileName);
-            console.log("PreviewModel calling StatFile", fileName);
-            const statFile = await services.FileService.StatFile(fileName);
+            console.log("PreviewModel calling StatFile", conn, fileName);
+            const statFile = await services.FileService.StatFile(conn, fileName);
             return statFile;
         });
         this.fullFile = jotai.atom<Promise<FullFile>>(async (get) => {

pkg/service/fileservice/fileservice.go

@@ -11,10 +11,14 @@ import (
 	"time"
 
 	"github.com/wavetermdev/thenextwave/pkg/filestore"
+	"github.com/wavetermdev/thenextwave/pkg/tsgen/tsgenmeta"
 	"github.com/wavetermdev/thenextwave/pkg/util/utilfn"
 	"github.com/wavetermdev/thenextwave/pkg/wavebase"
 	"github.com/wavetermdev/thenextwave/pkg/wconfig"
 	"github.com/wavetermdev/thenextwave/pkg/wshrpc"
+	"github.com/wavetermdev/thenextwave/pkg/wshrpc/wshclient"
+	"github.com/wavetermdev/thenextwave/pkg/wshrpc/wshserver"
+	"github.com/wavetermdev/thenextwave/pkg/wshutil"
 )
 
 const MaxFileSize = 10 * 1024 * 1024 // 10M
@@ -40,30 +44,24 @@ func (fs *FileService) SaveFile(path string, data64 string) error {
 	return nil
 }
 
-func (fs *FileService) StatFile(path string) (*wshrpc.FileInfo, error) {
-	cleanedPath := filepath.Clean(wavebase.ExpandHomeDir(path))
-	finfo, err := os.Stat(cleanedPath)
-	if os.IsNotExist(err) {
-		return &wshrpc.FileInfo{Path: wavebase.ReplaceHomeDir(path), NotFound: true}, nil
+func (fs *FileService) StatFile_Meta() tsgenmeta.MethodMeta {
+	return tsgenmeta.MethodMeta{
+		Desc:     "get file info",
+		ArgNames: []string{"connection", "path"},
 	}
-	if err != nil {
-		return nil, fmt.Errorf("cannot stat file %q: %w", path, err)
+}
+
+func (fs *FileService) StatFile(connection string, path string) (*wshrpc.FileInfo, error) {
+	if connection == "" {
+		connection = wshrpc.LocalConnName
 	}
-	mimeType := utilfn.DetectMimeType(cleanedPath)
-	return &wshrpc.FileInfo{
-		Path:     cleanedPath,
-		Name:     finfo.Name(),
-		Size:     finfo.Size(),
-		Mode:     finfo.Mode(),
-		ModeStr:  finfo.Mode().String(),
-		ModTime:  finfo.ModTime().UnixMilli(),
-		IsDir:    finfo.IsDir(),
-		MimeType: mimeType,
-	}, nil
+	connRoute := wshutil.MakeConnectionRouteId(connection)
+	client := wshserver.GetMainRpcClient()
+	return wshclient.RemoteFileInfoCommand(client, path, &wshrpc.RpcOpts{Route: connRoute})
 }
 
 func (fs *FileService) ReadFile(path string) (*FullFile, error) {
-	finfo, err := fs.StatFile(path)
+	finfo, err := fs.StatFile("", path)
 	if err != nil {
 		return nil, fmt.Errorf("cannot stat file %q: %w", path, err)
 	}
@@ -83,7 +81,7 @@ func (fs *FileService) ReadFile(path string) (*FullFile, error) {
 		}
 		var innerFilesInfo []wshrpc.FileInfo
 		parent := filepath.Dir(finfo.Path)
-		parentFileInfo, err := fs.StatFile(parent)
+		parentFileInfo, err := fs.StatFile("", parent)
 		if err == nil && parent != finfo.Path {
 			log.Printf("adding parent")
 			parentFileInfo.Name = ".."

pkg/wshrpc/wshrpctypes.go

@@ -16,6 +16,8 @@ import (
 	"github.com/wavetermdev/thenextwave/pkg/wstore"
 )
 
+const LocalConnName = "local"
+
 const (
 	RpcType_Call             = "call"             // single response (regular rpc)
 	RpcType_ResponseStream   = "responsestream"   // stream of responses (streaming rpc)

pkg/wshutil/wshrouter.go

@@ -38,6 +38,10 @@ type WshRouter struct {
 	InputCh         chan msgAndRoute
 }
 
+func MakeConnectionRouteId(connId string) string {
+	return "conn:" + connId
+}
+
 var DefaultRouter = NewWshRouter()
 
 func NewWshRouter() *WshRouter {

pkg/wshutil/wshrpc.go

@@ -180,6 +180,12 @@ func validateServerImpl(serverImpl ServerImpl) {
 
 // closes outputCh when inputCh is closed/done
 func MakeWshRpc(inputCh chan []byte, outputCh chan []byte, rpcCtx wshrpc.RpcContext, serverImpl ServerImpl) *WshRpc {
+	if inputCh == nil {
+		inputCh = make(chan []byte, DefaultInputChSize)
+	}
+	if outputCh == nil {
+		outputCh = make(chan []byte, DefaultOutputChSize)
+	}
 	validateServerImpl(serverImpl)
 	rtn := &WshRpc{
 		Lock:               &sync.Mutex{},