feat: add proxy info in script context

miaokobot · Jul 31, 2022 · df62024 · df62024
1 parent be9b1b5
commit df62024
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/engine/engine.go b/engine/engine.go
@@ -30,6 +30,13 @@ func VMNew() *goja.Runtime {
 func VMNewWithVendor(p interfaces.Vendor, network interfaces.RequestOptionsNetwork) *goja.Runtime {
 	vm := VMNew()
 
+	if p != nil {
+		pi := p.ProxyInfo()
+		vm.Set("proxy", vm.ToValue(pi.Map()))
+	} else {
+		vm.Set("proxy", nil)
+	}
+
 	vm.Set("fetch", factory.FetchFactory(vm, p, network))
 	vm.Set("netcat", factory.NetCatFactory(vm, p, network))
 

diff --git a/interfaces/proxy.go b/interfaces/proxy.go
@@ -37,3 +37,11 @@ type ProxyInfo struct {
 	Address string
 	Type    ProxyType
 }
+
+func (pi *ProxyInfo) Map() map[string]string {
+	return map[string]string{
+		"Name":    pi.Name,
+		"Address": pi.Address,
+		"Type":    string(pi.Type),
+	}
+}