diff --git a/tssh/trzsz.go b/tssh/trzsz.go index bfdacb8..7d7fe19 100644 --- a/tssh/trzsz.go +++ b/tssh/trzsz.go @@ -156,6 +156,23 @@ func enableTrzsz(args *sshArgs, ss *sshClientSession) error { return fmt.Errorf("get terminal size failed: %v", err) } + // custom configuration + defaultUploadPath := getExOptionConfig(args, "DefaultUploadPath") + if defaultUploadPath == "" { + defaultUploadPath = userConfig.defaultUploadPath + } + defaultDownloadPath := args.DownloadPath + if defaultDownloadPath == "" { + defaultDownloadPath = getExOptionConfig(args, "DefaultDownloadPath") + if defaultDownloadPath == "" { + defaultDownloadPath = userConfig.defaultDownloadPath + } + } + dragFileUploadCommand := getExOptionConfig(args, "DragFileUploadCommand") + if dragFileUploadCommand == "" { + dragFileUploadCommand = userConfig.dragFileUploadCommand + } + // create a TrzszFilter to support trzsz ( trz / tsz ) // // os.Stdin ┌────────┐ os.Stdin ┌─────────────┐ ServerIn ┌────────┐ @@ -185,27 +202,9 @@ func enableTrzsz(args *sshArgs, ss *sshClientSession) error { }) // setup trzsz config - defaultUploadPath := getExOptionConfig(args, "DefaultUploadPath") - if defaultUploadPath == "" { - defaultUploadPath = userConfig.defaultUploadPath - } trzszFilter.SetDefaultUploadPath(defaultUploadPath) - - downloadPath := args.DownloadPath - if downloadPath == "" { - downloadPath = getExOptionConfig(args, "DefaultDownloadPath") - if downloadPath == "" { - downloadPath = userConfig.defaultDownloadPath - } - } - trzszFilter.SetDefaultDownloadPath(downloadPath) - - dragFileUploadCommand := getExOptionConfig(args, "DragFileUploadCommand") - if dragFileUploadCommand == "" { - dragFileUploadCommand = userConfig.dragFileUploadCommand - } + trzszFilter.SetDefaultDownloadPath(defaultDownloadPath) trzszFilter.SetDragFileUploadCommand(dragFileUploadCommand) - trzszFilter.SetProgressColorPair(userConfig.progressColorPair) // setup tunnel connect