new: implemented tcp.proxy (closes #33)

modules/http_proxy_script_builtins.go

@@ -1,78 +0,0 @@
-package modules
-
-import (
-	"io/ioutil"
-
-	"github.com/bettercap/bettercap/log"
-
-	"github.com/robertkrimen/otto"
-)
-
-var nullOtto = otto.Value{}
-
-func errOtto(format string, args ...interface{}) otto.Value {
-	log.Error(format, args...)
-	return nullOtto
-}
-
-// define functions available to proxy scripts
-func (s *ProxyScript) defineBuiltins() error {
-	// used to read a file ... doh
-	s.VM.Set("readFile", func(call otto.FunctionCall) otto.Value {
-		argv := call.ArgumentList
-		argc := len(argv)
-		if argc != 1 {
-			return errOtto("readFile: expected 1 argument, %d given instead.", argc)
-		}
-
-		filename := argv[0].String()
-		raw, err := ioutil.ReadFile(filename)
-		if err != nil {
-			return errOtto("Could not read %s: %s", filename, err)
-		}
-
-		v, err := s.VM.ToValue(string(raw))
-		if err != nil {
-			return errOtto("Could not convert to string: %s", err)
-		}
-		return v
-	})
-
-	// log something
-	s.VM.Set("log", func(call otto.FunctionCall) otto.Value {
-		for _, v := range call.ArgumentList {
-			log.Info("%s", v.String())
-		}
-		return otto.Value{}
-	})
-
-	// read or write environment variable
-	s.VM.Set("env", func(call otto.FunctionCall) otto.Value {
-		argv := call.ArgumentList
-		argc := len(argv)
-
-		if argc == 1 {
-			// get
-			varName := call.Argument(0).String()
-			if found, varValue := s.sess.Env.Get(varName); found == true {
-				v, err := s.VM.ToValue(varValue)
-				if err != nil {
-					return errOtto("Could not convert to string: %s", varValue)
-				}
-				return v
-			}
-
-		} else if argc == 2 {
-			// set
-			varName := call.Argument(0).String()
-			varValue := call.Argument(1).String()
-			s.sess.Env.Set(varName, varValue)
-		} else {
-			return errOtto("env: expected 1 or 2 arguments, %d given instead.", argc)
-		}
-
-		return nullOtto
-	})
-
-	return nil
-}