Setup docs

Author: The5imon

docs/Makefile

@@ -27,7 +27,7 @@
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
-extensions = ['recommonmark']
+extensions = ['recommonmark', 'sphinx_rtd_theme']
 
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ['_templates']
@@ -43,7 +43,7 @@
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #
-html_theme = 'master'
+html_theme = 'sphinx_rtd_theme'
 
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,

docs/conf.py

@@ -12,12 +12,3 @@
 
    injectorshell
    scriptedshell
-
-
-
-Indices and tables
-==================
-
-* :ref:`genindex`
-* :ref:`modindex`
-* :ref:`search`

docs/index.rst

@@ -1,6 +1,6 @@
 entry_points = {
     'SSHBaseForwarder': [
-        'scriptedshell = plugins.ssh:SSHScriptedForwarder',
+        'scriptedshell = plugins.ssh.scriptedshell:SSHScriptedForwarder',
     ],
     'SCPBaseForwarder': [
 
@@ -20,4 +20,4 @@
     'Authenticator': [
 
     ]
-}
+}

docs/make.bat

@@ -0,0 +1,53 @@
+import logging
+import os
+
+from ssh_proxy_server.forwarders.ssh import SSHForwarder
+
+
+class SSHScriptedForwarder(SSHForwarder):
+
+    @classmethod
+    def parser_arguments(cls):
+        cls.PARSER.add_argument(
+            '--ssh-script',
+            dest='ssh_script',
+            help='script to execute on ssh connection',
+            required=True
+        )
+        cls.PARSER.add_argument(
+            '--ssh-out-dir',
+            dest='ssh_out_dir',
+            help='script output directory',
+            default='.'
+        )
+
+    def __init__(self, session):
+        super(SSHScriptedForwarder, self).__init__(session)
+        self.executing = False
+        self.script = open(os.path.expanduser(self.args.ssh_script), "r")
+        self.output = open(os.path.expanduser(os.path.join(self.args.ssh_out_dir, str(self.session))), "w+")
+
+    def forward_stdin(self):
+        if self.executing:
+            line = self.script.readline()
+            self.server_channel.sendall(line)
+            return
+        if not self.executing and not self.script.closed and self.session.ssh_channel.recv_ready():
+            self.executing = True
+        super(SSHScriptedForwarder, self).forward_stdin()
+
+    def forward_stdout(self):
+        if not self.server_channel.recv_ready() and self.executing:
+            self.executing = False
+            self.script.close()
+            self.output.close()
+        super(SSHScriptedForwarder, self).forward_stdout()
+
+    def stdout(self, text):
+        if self.executing:
+            self.output.write(text.decode('utf-8'))
+            return ""
+        return text
+
+    def close_session(self, channel):
+        super().close_session(channel)