From 403d30af7886edd345c70e270b4e137d4146491f Mon Sep 17 00:00:00 2001
From: wsdjeg <wsdjeg@outlook.com>
Date: Sat, 4 Jun 2022 14:02:40 +0800
Subject: [PATCH] feat(ssh): add `:SSHCommand` command

---
 autoload/SpaceVim/layers/ssh.vim | 31 ++++++++++++++++++++++++++++++-
 docs/layers/ssh.md               | 10 ++++++++++
 2 files changed, 40 insertions(+), 1 deletion(-)

diff --git a/autoload/SpaceVim/layers/ssh.vim b/autoload/SpaceVim/layers/ssh.vim
index 56f460d5c..43f3d0557 100644
--- a/autoload/SpaceVim/layers/ssh.vim
+++ b/autoload/SpaceVim/layers/ssh.vim
@@ -28,16 +28,21 @@ if exists('s:ssh')
   finish
 endif
 
+let s:JOB = SpaceVim#api#import('job')
+let s:NOT = SpaceVim#api#import('notify')
+
 let s:ssh = 'ssh'
 let s:user = 'root'
 let s:ip = '127.0.0.1'
 let s:port = '20'
+let s:pass = ''
 
 function! SpaceVim#layers#ssh#config() abort
   let g:_spacevim_mappings_space.S = {'name' : '+SSH'}
   call SpaceVim#mapping#space#langSPC('nmap', ['S','o'],
         \ 'call SpaceVim#layers#ssh#connect()',
         \ 'connect-to-ssh-server', 1)
+  command! -nargs=* SSHCommand call s:run(<q-args>)
 endfunction
 
 function! SpaceVim#layers#ssh#set_variable(opt) abort
@@ -45,6 +50,7 @@ function! SpaceVim#layers#ssh#set_variable(opt) abort
   let s:user = get(a:opt, 'ssh_user', s:user)
   let s:ip = get(a:opt, 'ssh_address', s:ip)
   let s:port = get(a:opt, 'ssh_port', s:port)
+  let s:pass = get(a:opt, 'ssh_password', s:pass)
 endfunction
 
 function! SpaceVim#layers#ssh#connect() abort
@@ -61,7 +67,7 @@ endfunction
 
 function! SpaceVim#layers#ssh#get_options() abort
 
-  return ['ssh_port', 'ssh_user', 'ssh_address', 'ssh_command']
+  return ['ssh_port', 'ssh_user', 'ssh_address', 'ssh_command', 'ssh_password']
 
 endfunction
 
@@ -69,3 +75,26 @@ function! SpaceVim#layers#ssh#health() abort
   call SpaceVim#layers#ssh#config()
   return 1
 endfunction
+
+function! s:run(cmd) abort
+  let s:NOT.notify_max_width = &columns * 0.70
+  let s:NOT.timeout = 5000
+  call s:JOB.start(['plink', s:user . '@' . s:ip, '-P', s:port, '-pw', s:pass, a:cmd . ' 2>&1'], 
+        \ {
+          \ 'on_stdout' : function('s:on_stdout'),
+          \ 'on_stderr' : function('s:on_stderr')
+          \ })
+endfunction
+
+
+function! s:on_stdout(id, data, event) abort
+  for line in a:data
+    call s:NOT.notify(line)
+  endfor
+endfunction
+
+function! s:on_stderr(id, data, event) abort
+  for line in a:data
+    call s:NOT.notify(line, 'WarningMsg')
+  endfor
+endfunction
diff --git a/docs/layers/ssh.md b/docs/layers/ssh.md
index dead66904..56124addc 100644
--- a/docs/layers/ssh.md
+++ b/docs/layers/ssh.md
@@ -10,6 +10,7 @@ description: "This layer provides ssh support in SpaceVim"
 - [Intro](#intro)
 - [Layer options](#layer-options)
 - [Key bindings](#key-bindings)
+- [Commands](#commands)
 
 <!-- vim-markdown-toc -->
 
@@ -46,3 +47,12 @@ Example:
 | Key Binding | Description                 |
 | ----------- | --------------------------- |
 | `SPC S o`   | open ssh connection windows |
+
+## Commands
+
+- `:SSHCommand`: run comamnd in ssh client.
+
+   for example:
+   ```
+   :SSHCommand nvim --version
+   ```