new: session scripts can now include other scripts via require('file')

session/script.go

@@ -1,20 +1,93 @@
 package session
 
 import (
+	"fmt"
+	"github.com/bettercap/bettercap/caplets"
 	_ "github.com/bettercap/bettercap/js"
+	"github.com/evilsocket/islazy/fs"
 	"github.com/evilsocket/islazy/plugin"
+	"github.com/evilsocket/islazy/str"
+	"io/ioutil"
+	"path/filepath"
+	"regexp"
+	"strings"
 )
 
+// require("telegram.js")
+var requireParser = regexp.MustCompile(`(?msi)^\s*require\s*\(\s*["']([^"']+)["']\s*\);?\s*$`)
+
 type Script struct {
 	*plugin.Plugin
 }
 
-func LoadScript(fileName string, ses *Session) (*Script, error) {
-	if p, err := plugin.Load(fileName); err != nil {
+// yo! we're doing c-like preprocessing on a javascript file from go :D
+func preprocess(basePath string, code string, level int) (string, error) {
+	if level >= 255 {
+		return "", fmt.Errorf("too many nested includes")
+	}
+
+	matches := requireParser.FindAllStringSubmatch(code, -1)
+	for _, match := range matches {
+		expr := match[0]
+		fileName := str.Trim(match[1])
+
+		if fileName[0] != '/' {
+			searchPaths := []string{
+				filepath.Join(basePath, fileName),
+				filepath.Join(caplets.InstallBase, fileName),
+			}
+
+			if strings.Contains(fileName, ".js") == false {
+				searchPaths = append(searchPaths, []string{
+					filepath.Join(basePath, fileName) + ".js",
+					filepath.Join(caplets.InstallBase, fileName) + ".js",
+				}...)
+			}
+
+			found := false
+			for _, fName := range searchPaths {
+				if fs.Exists(fName) {
+					fileName = fName
+					found = true
+					break
+				}
+			}
+			if !found {
+				return "", fmt.Errorf("file %s not found in any of %#v", fileName, searchPaths)
+			}
+		}
+
+		raw, err := ioutil.ReadFile(fileName)
+		if err != nil {
+			return "", fmt.Errorf("%s: %v", fileName, err)
+		}
+
+		if includedBody, err := preprocess(filepath.Dir(fileName), string(raw), level+1); err != nil {
+			return "", fmt.Errorf("%s: %v", fileName, err)
+		} else {
+			code = strings.ReplaceAll(code, expr, includedBody)
+		}
+	}
+
+	return code, nil
+}
+
+func LoadScript(fileName string) (*Script, error) {
+	raw, err := ioutil.ReadFile(fileName)
+	if err != nil {
+		return nil, err
+	}
+
+	basePath := filepath.Dir(fileName)
+	if code, err := preprocess(basePath, string(raw), 0); err != nil {
+		return nil, err
+	} else if p, err := plugin.Parse(code); err != nil {
 		return nil, err
 	} else {
+		p.Path = fileName
+		p.Name = strings.Replace(basePath, ".js", "", -1)
 		return &Script{
-			Plugin:  p,
+			Plugin: p,
 		}, nil
 	}
-}
+}