implemented the Amass Scripting Engine

config/config.go

@@ -9,6 +9,8 @@ import (
 	"io/ioutil"
 	"log"
 	"net"
+	"os"
+	"path/filepath"
 	"regexp"
 	"strings"
 	"sync"
@@ -623,3 +625,38 @@ func (c *Config) loadResolverSettings(cfg *ini.File) error {
 func (c *Config) UpdateConfig(update Updater) error {
 	return update.OverrideConfig(c)
 }
+
+// AcquireScripts returns all the default and user provided scripts for data sources.
+func (c *Config) AcquireScripts() ([]string, error) {
+	scripts := getDefaultScripts()
+
+	dir := OutputDirectory(c.Dir)
+	if dir == "" {
+		return scripts, nil
+	}
+
+	finfo, err := os.Stat(dir)
+	if os.IsNotExist(err) || !finfo.IsDir() {
+		return scripts, errors.New("The output directory does not exist or is not a directory")
+	}
+
+	filepath.Walk(filepath.Join(dir, "scripts"), func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		// Is this file not a script?
+		if info.IsDir() || filepath.Ext(info.Name()) != ".ads" {
+			return nil
+		}
+		// Get the script content
+		data, err := ioutil.ReadFile(path)
+		if err != nil {
+			return err
+		}
+
+		scripts = append(scripts, string(data))
+		return nil
+	})
+
+	return scripts, nil
+}

config/support.go

@@ -192,3 +192,28 @@ func uniqueIntAppend(s []int, e string) []int {
 	}
 	return s
 }
+
+func getDefaultScripts() []string {
+	fsOnce.Do(openTheFS)
+
+	var scripts []string
+	fs.Walk(StatikFS, "/scripts", func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		// Is this file not a script?
+		if info.IsDir() || filepath.Ext(info.Name()) != ".ads" {
+			return nil
+		}
+		// Get the script content
+		data, err := fs.ReadFile(StatikFS, path)
+		if err != nil {
+			return err
+		}
+
+		scripts = append(scripts, string(data))
+		return nil
+	})
+
+	return scripts
+}