From 74b88185dcdfc35e1bb4b951e6a342c844a3ee57 Mon Sep 17 00:00:00 2001
From: sawka <mike.sawka@gmail.com>
Date: Mon, 8 Aug 2022 09:52:50 -0700
Subject: [PATCH] add cache for ensuresessiondir

---
 pkg/base/base.go     | 31 ++++++++++++++++++++++++++++++-
 pkg/packet/packet.go |  7 ++++---
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/pkg/base/base.go b/pkg/base/base.go
index 4933273c..1f848e53 100644
--- a/pkg/base/base.go
+++ b/pkg/base/base.go
@@ -16,6 +16,7 @@ import (
 	"path"
 	"path/filepath"
 	"strings"
+	"sync"
 
 	"github.com/google/uuid"
 )
@@ -30,6 +31,9 @@ const SessionsDirBaseName = "sessions"
 const MShellVersion = "0.1.0"
 const RemoteIdFile = "remoteid"
 
+var sessionDirCache = make(map[string]string)
+var baseLock = &sync.Mutex{}
+
 type CommandFileNames struct {
 	PtyOutFile    string
 	StdinFifo     string
@@ -114,6 +118,22 @@ func GetMShellHomeDir() string {
 	return ExpandHomeDir(DefaultMShellHome)
 }
 
+func GetPtyOutFile(ck CommandKey, seqNum int) (string, error) {
+	if err := ck.Validate("ck"); err != nil {
+		return "", fmt.Errorf("cannot get command files: %w", err)
+	}
+	if seqNum < 0 {
+		return "", fmt.Errorf("invalid seqnum, cannot be negative")
+	}
+	sessionId, cmdId := ck.Split()
+	sdir, err := EnsureSessionDir(sessionId)
+	if err != nil {
+		return "", err
+	}
+	base := path.Join(sdir, cmdId)
+	return fmt.Sprintf("%s.%d.ptyout", base, seqNum), nil
+}
+
 func GetCommandFileNames(ck CommandKey) (*CommandFileNames, error) {
 	if err := ck.Validate("ck"); err != nil {
 		return nil, fmt.Errorf("cannot get command files: %w", err)
@@ -166,8 +186,14 @@ func EnsureSessionDir(sessionId string) (string, error) {
 	if sessionId == "" {
 		return "", fmt.Errorf("Bad sessionid, cannot be empty")
 	}
+	baseLock.Lock()
+	sdir, ok := sessionDirCache[sessionId]
+	baseLock.Unlock()
+	if ok {
+		return sdir, nil
+	}
 	mhome := GetMShellHomeDir()
-	sdir := path.Join(mhome, SessionsDirBaseName, sessionId)
+	sdir = path.Join(mhome, SessionsDirBaseName, sessionId)
 	info, err := os.Stat(sdir)
 	if errors.Is(err, fs.ErrNotExist) {
 		err = os.MkdirAll(sdir, 0777)
@@ -182,6 +208,9 @@ func EnsureSessionDir(sessionId string) (string, error) {
 	if !info.IsDir() {
 		return "", fmt.Errorf("session dir '%s' must be a directory", sdir)
 	}
+	baseLock.Lock()
+	sessionDirCache[sessionId] = sdir
+	baseLock.Unlock()
 	return sdir, nil
 }
 
diff --git a/pkg/packet/packet.go b/pkg/packet/packet.go
index 0961bfaa..5fef24a7 100644
--- a/pkg/packet/packet.go
+++ b/pkg/packet/packet.go
@@ -461,9 +461,10 @@ func MakeCmdStartPacket(reqId string) *CmdStartPacketType {
 }
 
 type TermOpts struct {
-	Rows int    `json:"rows"`
-	Cols int    `json:"cols"`
-	Term string `json:"term"`
+	Rows    int    `json:"rows"`
+	Cols    int    `json:"cols"`
+	Term    string `json:"term"`
+	CmdSize int64  `json:"cmdsize,omitempty"`
 }
 
 type RemoteFd struct {